-
Notifications
You must be signed in to change notification settings - Fork 5
/
notify.go
79 lines (69 loc) · 1.82 KB
/
notify.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
package server
import (
"fmt"
"time"
"github.com/doomsday-project/doomsday/client/doomsday"
"github.com/doomsday-project/doomsday/server/logger"
"github.com/doomsday-project/doomsday/server/notify"
"github.com/doomsday-project/doomsday/server/notify/backend"
"github.com/doomsday-project/doomsday/server/notify/schedule"
)
type notifier struct {
s schedule.Schedule
b backend.Backend
}
func NotifyFrom(conf notify.Config, m *SourceManager, l *logger.Logger) error {
var n notifier
var err error
if conf.DoomsdayURL == "" {
return fmt.Errorf("Please provide doomsday_url")
}
n.s, err = schedule.New(conf.Schedule.Type, conf.Schedule.Properties)
if err != nil {
return fmt.Errorf("Error creating schedule: %s", err)
}
uni := backend.BackendUniversalConfig{
DoomsdayURL: conf.DoomsdayURL,
Logger: l,
}
n.b, err = backend.New(conf.Backend, uni)
if err != nil {
return fmt.Errorf("Error creating backend: %s", err)
}
n.s.Start()
go func() {
for range n.s.Channel() {
l.WriteF("Triggering notification check")
const (
StateOK = iota
StateExpired
StateSoon
)
d := m.Data()
state := StateOK
expiredThreshold := time.Duration(0)
expiringSoonThreshold := time.Hour * 24 * 7 * 4
if len(d.Filter(doomsday.CacheItemFilter{Within: &expiredThreshold})) > 0 {
state = StateExpired
} else if len(d.Filter(doomsday.CacheItemFilter{Within: &expiringSoonThreshold})) > 0 {
state = StateSoon
}
var sendErr error
switch state {
case StateOK:
l.WriteF("No expiring certs")
sendErr = n.b.OK()
case StateSoon:
l.WriteF("Certs expiring soon")
sendErr = n.b.Soon()
case StateExpired:
l.WriteF("Certs expired")
sendErr = n.b.Expired()
}
if sendErr != nil {
l.WriteF("Could not send notification: %s", sendErr)
}
}
}()
return nil
}