github.com/psyb0t/mattermost-server@v4.6.1-0.20180125161845-5503a1351abf+incompatible/app/compliance.go (about)

     1  // Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
     2  // See License.txt for license information.
     3  
     4  package app
     5  
     6  import (
     7  	"io/ioutil"
     8  
     9  	"net/http"
    10  
    11  	"github.com/mattermost/mattermost-server/model"
    12  	"github.com/mattermost/mattermost-server/utils"
    13  )
    14  
    15  func (a *App) GetComplianceReports(page, perPage int) (model.Compliances, *model.AppError) {
    16  	if !*a.Config().ComplianceSettings.Enable || !utils.IsLicensed() || !*utils.License().Features.Compliance {
    17  		return nil, model.NewAppError("GetComplianceReports", "ent.compliance.licence_disable.app_error", nil, "", http.StatusNotImplemented)
    18  	}
    19  
    20  	if result := <-a.Srv.Store.Compliance().GetAll(page*perPage, perPage); result.Err != nil {
    21  		return nil, result.Err
    22  	} else {
    23  		return result.Data.(model.Compliances), nil
    24  	}
    25  }
    26  
    27  func (a *App) SaveComplianceReport(job *model.Compliance) (*model.Compliance, *model.AppError) {
    28  	if !*a.Config().ComplianceSettings.Enable || !utils.IsLicensed() || !*utils.License().Features.Compliance || a.Compliance == nil {
    29  		return nil, model.NewAppError("saveComplianceReport", "ent.compliance.licence_disable.app_error", nil, "", http.StatusNotImplemented)
    30  	}
    31  
    32  	job.Type = model.COMPLIANCE_TYPE_ADHOC
    33  
    34  	if result := <-a.Srv.Store.Compliance().Save(job); result.Err != nil {
    35  		return nil, result.Err
    36  	} else {
    37  		job = result.Data.(*model.Compliance)
    38  		a.Go(func() {
    39  			a.Compliance.RunComplianceJob(job)
    40  		})
    41  	}
    42  
    43  	return job, nil
    44  }
    45  
    46  func (a *App) GetComplianceReport(reportId string) (*model.Compliance, *model.AppError) {
    47  	if !*a.Config().ComplianceSettings.Enable || !utils.IsLicensed() || !*utils.License().Features.Compliance || a.Compliance == nil {
    48  		return nil, model.NewAppError("downloadComplianceReport", "ent.compliance.licence_disable.app_error", nil, "", http.StatusNotImplemented)
    49  	}
    50  
    51  	if result := <-a.Srv.Store.Compliance().Get(reportId); result.Err != nil {
    52  		return nil, result.Err
    53  	} else {
    54  		return result.Data.(*model.Compliance), nil
    55  	}
    56  }
    57  
    58  func (a *App) GetComplianceFile(job *model.Compliance) ([]byte, *model.AppError) {
    59  	if f, err := ioutil.ReadFile(*a.Config().ComplianceSettings.Directory + "compliance/" + job.JobName() + ".zip"); err != nil {
    60  		return nil, model.NewAppError("readFile", "api.file.read_file.reading_local.app_error", nil, err.Error(), http.StatusNotImplemented)
    61  	} else {
    62  		return f, nil
    63  	}
    64  }