2017-08-16 14:37:37 -07:00
|
|
|
// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
|
2016-05-21 05:14:08 -07:00
|
|
|
// See License.txt for license information.
|
|
|
|
|
|
|
|
package model
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/json"
|
|
|
|
"io"
|
2018-02-08 15:11:04 -08:00
|
|
|
"net/http"
|
2016-05-21 05:14:08 -07:00
|
|
|
"strings"
|
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
|
|
|
COMPLIANCE_STATUS_CREATED = "created"
|
|
|
|
COMPLIANCE_STATUS_RUNNING = "running"
|
|
|
|
COMPLIANCE_STATUS_FINISHED = "finished"
|
|
|
|
COMPLIANCE_STATUS_FAILED = "failed"
|
|
|
|
COMPLIANCE_STATUS_REMOVED = "removed"
|
|
|
|
|
|
|
|
COMPLIANCE_TYPE_DAILY = "daily"
|
|
|
|
COMPLIANCE_TYPE_ADHOC = "adhoc"
|
|
|
|
)
|
|
|
|
|
|
|
|
type Compliance struct {
|
|
|
|
Id string `json:"id"`
|
|
|
|
CreateAt int64 `json:"create_at"`
|
|
|
|
UserId string `json:"user_id"`
|
|
|
|
Status string `json:"status"`
|
|
|
|
Count int `json:"count"`
|
|
|
|
Desc string `json:"desc"`
|
|
|
|
Type string `json:"type"`
|
|
|
|
StartAt int64 `json:"start_at"`
|
|
|
|
EndAt int64 `json:"end_at"`
|
|
|
|
Keywords string `json:"keywords"`
|
|
|
|
Emails string `json:"emails"`
|
|
|
|
}
|
|
|
|
|
|
|
|
type Compliances []Compliance
|
|
|
|
|
|
|
|
func (o *Compliance) ToJson() string {
|
2018-11-18 09:55:05 -08:00
|
|
|
b, _ := json.Marshal(o)
|
|
|
|
return string(b)
|
2016-05-21 05:14:08 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
func (me *Compliance) PreSave() {
|
|
|
|
if me.Id == "" {
|
|
|
|
me.Id = NewId()
|
|
|
|
}
|
|
|
|
|
|
|
|
if me.Status == "" {
|
|
|
|
me.Status = COMPLIANCE_STATUS_CREATED
|
|
|
|
}
|
|
|
|
|
|
|
|
me.Count = 0
|
2018-11-18 09:55:05 -08:00
|
|
|
me.Emails = NormalizeEmail(me.Emails)
|
2016-05-21 05:14:08 -07:00
|
|
|
me.Keywords = strings.ToLower(me.Keywords)
|
|
|
|
|
|
|
|
me.CreateAt = GetMillis()
|
|
|
|
}
|
|
|
|
|
|
|
|
func (me *Compliance) JobName() string {
|
|
|
|
jobName := me.Type
|
|
|
|
if me.Type == COMPLIANCE_TYPE_DAILY {
|
|
|
|
jobName += "-" + me.Desc
|
|
|
|
}
|
|
|
|
|
|
|
|
jobName += "-" + me.Id
|
|
|
|
|
|
|
|
return jobName
|
|
|
|
}
|
|
|
|
|
|
|
|
func (me *Compliance) IsValid() *AppError {
|
|
|
|
|
|
|
|
if len(me.Id) != 26 {
|
2018-02-08 15:11:04 -08:00
|
|
|
return NewAppError("Compliance.IsValid", "model.compliance.is_valid.id.app_error", nil, "", http.StatusBadRequest)
|
2016-05-21 05:14:08 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if me.CreateAt == 0 {
|
2018-02-08 15:11:04 -08:00
|
|
|
return NewAppError("Compliance.IsValid", "model.compliance.is_valid.create_at.app_error", nil, "", http.StatusBadRequest)
|
2016-05-21 05:14:08 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if len(me.Desc) > 512 || len(me.Desc) == 0 {
|
2018-02-08 15:11:04 -08:00
|
|
|
return NewAppError("Compliance.IsValid", "model.compliance.is_valid.desc.app_error", nil, "", http.StatusBadRequest)
|
2016-05-21 05:14:08 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if me.StartAt == 0 {
|
2018-02-08 15:11:04 -08:00
|
|
|
return NewAppError("Compliance.IsValid", "model.compliance.is_valid.start_at.app_error", nil, "", http.StatusBadRequest)
|
2016-05-21 05:14:08 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if me.EndAt == 0 {
|
2018-02-08 15:11:04 -08:00
|
|
|
return NewAppError("Compliance.IsValid", "model.compliance.is_valid.end_at.app_error", nil, "", http.StatusBadRequest)
|
2016-05-21 05:14:08 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if me.EndAt <= me.StartAt {
|
2018-02-08 15:11:04 -08:00
|
|
|
return NewAppError("Compliance.IsValid", "model.compliance.is_valid.start_end_at.app_error", nil, "", http.StatusBadRequest)
|
2016-05-21 05:14:08 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func ComplianceFromJson(data io.Reader) *Compliance {
|
2018-11-18 09:55:05 -08:00
|
|
|
var o *Compliance
|
|
|
|
json.NewDecoder(data).Decode(&o)
|
|
|
|
return o
|
2016-05-21 05:14:08 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
func (o Compliances) ToJson() string {
|
|
|
|
if b, err := json.Marshal(o); err != nil {
|
|
|
|
return "[]"
|
|
|
|
} else {
|
|
|
|
return string(b)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func CompliancesFromJson(data io.Reader) Compliances {
|
|
|
|
var o Compliances
|
2018-11-18 09:55:05 -08:00
|
|
|
json.NewDecoder(data).Decode(&o)
|
|
|
|
return o
|
2016-05-21 05:14:08 -07:00
|
|
|
}
|