summaryrefslogtreecommitdiffstats
path: root/model/compliance.go
blob: 5546b7830bef1a0f4b82c3391a1df712ef1bacbf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

package model

import (
	"encoding/json"
	"io"
	"net/http"
	"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 {
	b, _ := json.Marshal(o)
	return string(b)
}

func (me *Compliance) PreSave() {
	if me.Id == "" {
		me.Id = NewId()
	}

	if me.Status == "" {
		me.Status = COMPLIANCE_STATUS_CREATED
	}

	me.Count = 0
	me.Emails = NormalizeEmail(me.Emails)
	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 {
		return NewAppError("Compliance.IsValid", "model.compliance.is_valid.id.app_error", nil, "", http.StatusBadRequest)
	}

	if me.CreateAt == 0 {
		return NewAppError("Compliance.IsValid", "model.compliance.is_valid.create_at.app_error", nil, "", http.StatusBadRequest)
	}

	if len(me.Desc) > 512 || len(me.Desc) == 0 {
		return NewAppError("Compliance.IsValid", "model.compliance.is_valid.desc.app_error", nil, "", http.StatusBadRequest)
	}

	if me.StartAt == 0 {
		return NewAppError("Compliance.IsValid", "model.compliance.is_valid.start_at.app_error", nil, "", http.StatusBadRequest)
	}

	if me.EndAt == 0 {
		return NewAppError("Compliance.IsValid", "model.compliance.is_valid.end_at.app_error", nil, "", http.StatusBadRequest)
	}

	if me.EndAt <= me.StartAt {
		return NewAppError("Compliance.IsValid", "model.compliance.is_valid.start_end_at.app_error", nil, "", http.StatusBadRequest)
	}

	return nil
}

func ComplianceFromJson(data io.Reader) *Compliance {
	var o *Compliance
	json.NewDecoder(data).Decode(&o)
	return o
}

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
	json.NewDecoder(data).Decode(&o)
	return o
}