github.com/haalcala/mattermost-server-change-repo@v0.0.0-20210713015153-16753fbeee5f/app/compliance.go (about)

     1  // Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
     2  // See LICENSE.txt for license information.
     3  
     4  package app
     5  
     6  import (
     7  	"errors"
     8  	"io/ioutil"
     9  	"net/http"
    10  
    11  	"github.com/mattermost/mattermost-server/v5/model"
    12  	"github.com/mattermost/mattermost-server/v5/store"
    13  )
    14  
    15  func (a *App) GetComplianceReports(page, perPage int) (model.Compliances, *model.AppError) {
    16  	if license := a.Srv().License(); !*a.Config().ComplianceSettings.Enable || license == nil || !*license.Features.Compliance {
    17  		return nil, model.NewAppError("GetComplianceReports", "ent.compliance.licence_disable.app_error", nil, "", http.StatusNotImplemented)
    18  	}
    19  
    20  	compliances, err := a.Srv().Store.Compliance().GetAll(page*perPage, perPage)
    21  	if err != nil {
    22  		return nil, model.NewAppError("GetComplianceReports", "app.compliance.get.finding.app_error", nil, err.Error(), http.StatusInternalServerError)
    23  	}
    24  
    25  	return compliances, nil
    26  }
    27  
    28  func (a *App) SaveComplianceReport(job *model.Compliance) (*model.Compliance, *model.AppError) {
    29  	if license := a.Srv().License(); !*a.Config().ComplianceSettings.Enable || license == nil || !*license.Features.Compliance || a.Compliance() == nil {
    30  		return nil, model.NewAppError("saveComplianceReport", "ent.compliance.licence_disable.app_error", nil, "", http.StatusNotImplemented)
    31  	}
    32  
    33  	job.Type = model.COMPLIANCE_TYPE_ADHOC
    34  
    35  	job, err := a.Srv().Store.Compliance().Save(job)
    36  	if err != nil {
    37  		var appErr *model.AppError
    38  		switch {
    39  		case errors.As(err, &appErr):
    40  			return nil, appErr
    41  		default:
    42  			return nil, model.NewAppError("SaveComplianceReport", "app.compliance.save.saving.app_error", nil, err.Error(), http.StatusInternalServerError)
    43  		}
    44  	}
    45  
    46  	jCopy := job.DeepCopy()
    47  	a.Srv().Go(func() {
    48  		a.Compliance().RunComplianceJob(jCopy)
    49  	})
    50  
    51  	return job, nil
    52  }
    53  
    54  func (a *App) GetComplianceReport(reportId string) (*model.Compliance, *model.AppError) {
    55  	if license := a.Srv().License(); !*a.Config().ComplianceSettings.Enable || license == nil || !*license.Features.Compliance || a.Compliance() == nil {
    56  		return nil, model.NewAppError("downloadComplianceReport", "ent.compliance.licence_disable.app_error", nil, "", http.StatusNotImplemented)
    57  	}
    58  
    59  	compliance, err := a.Srv().Store.Compliance().Get(reportId)
    60  	if err != nil {
    61  		var nfErr *store.ErrNotFound
    62  		switch {
    63  		case errors.As(err, &nfErr):
    64  			return nil, model.NewAppError("GetComplicanceReport", "app.compliance.get.finding.app_error", nil, nfErr.Error(), http.StatusNotFound)
    65  		default:
    66  			return nil, model.NewAppError("GetComplianceReport", "app.compliance.get.finding.app_error", nil, err.Error(), http.StatusInternalServerError)
    67  		}
    68  	}
    69  
    70  	return compliance, nil
    71  }
    72  
    73  func (a *App) GetComplianceFile(job *model.Compliance) ([]byte, *model.AppError) {
    74  	f, err := ioutil.ReadFile(*a.Config().ComplianceSettings.Directory + "compliance/" + job.JobName() + ".zip")
    75  	if err != nil {
    76  		return nil, model.NewAppError("readFile", "api.file.read_file.reading_local.app_error", nil, err.Error(), http.StatusNotImplemented)
    77  	}
    78  	return f, nil
    79  }