-
Notifications
You must be signed in to change notification settings - Fork 6
/
client.go
114 lines (100 loc) · 2.7 KB
/
client.go
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
/*
Copyright 2021 Adevinta
*/
package reports
import (
"crypto/tls"
"encoding/json"
"net/http"
"github.com/aws/aws-sdk-go/aws"
"github.com/aws/aws-sdk-go/aws/arn"
"github.com/aws/aws-sdk-go/aws/session"
"github.com/aws/aws-sdk-go/service/sns"
"github.com/aws/aws-sdk-go/service/sns/snsiface"
)
const (
livereportType = "livereport"
)
type Client struct {
cfg Config
httpClient *http.Client
snsAPI snsiface.SNSAPI
}
// NewClient builds a new reports client.
func NewClient(cfg Config) (*Client, error) {
arn, err := arn.Parse(cfg.SNSARN)
if err != nil {
return nil, err
}
sess, err := session.NewSession()
if err != nil {
return nil, err
}
awsCfg := aws.NewConfig()
if arn.Region != "" {
awsCfg = awsCfg.WithRegion(arn.Region)
}
if cfg.SNSEndpoint != "" {
awsCfg = awsCfg.WithEndpoint(cfg.SNSEndpoint)
}
// Build http client.
httpClient := &http.Client{
Transport: &http.Transport{
TLSClientConfig: &tls.Config{
InsecureSkipVerify: cfg.InsecureTLS, // nolint
},
},
}
return &Client{
cfg: cfg,
httpClient: httpClient,
snsAPI: sns.New(sess, awsCfg),
}, nil
}
// GenerateDigestReport pushes an SNS event to trigger the digest report
// generation for the specified teamID.
func (c *Client) GenerateDigestReport(teamID, teamName, dateFrom, dateTo, liveReportURL string,
recipients []string, severitiesStats map[string]int, autoSend bool) error {
event := genDigestReportEvent{
Typ: livereportType,
TeamInfo: teamInfo{
ID: teamID,
Name: teamName,
Recipients: recipients,
},
Data: digestReportData{
TeamID: teamID,
DateFrom: dateFrom,
DateTo: dateTo,
LiveReportURL: liveReportURL,
Info: severitiesStats["info"],
InfoDiff: severitiesStats["infoDiff"],
InfoFixed: severitiesStats["infoFixed"],
Low: severitiesStats["low"],
LowDiff: severitiesStats["lowDiff"],
LowFixed: severitiesStats["lowFixed"],
Medium: severitiesStats["medium"],
MediumDiff: severitiesStats["mediumDiff"],
MediumFixed: severitiesStats["mediumFixed"],
High: severitiesStats["high"],
HighDiff: severitiesStats["highDiff"],
HighFixed: severitiesStats["highFixed"],
Critical: severitiesStats["critical"],
CriticalDiff: severitiesStats["criticalDiff"],
CriticalFixed: severitiesStats["criticalFixed"],
},
AutoSend: autoSend,
}
return c.publish(event)
}
func (c *Client) publish(event interface{}) error {
eventPayload, err := json.Marshal(event)
if err != nil {
return err
}
_, err = c.snsAPI.Publish(&sns.PublishInput{
Message: aws.String(string(eventPayload)),
TopicArn: aws.String(c.cfg.SNSARN),
})
return err
}