-
Notifications
You must be signed in to change notification settings - Fork 0
/
service.go
74 lines (65 loc) · 2.16 KB
/
service.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
package slack
import (
"context"
"time"
log "github.com/cantara/bragi/sbragi"
"github.com/cantara/gober/consensus"
scheduletasks "github.com/cantara/gober/scheduletasks"
"github.com/cantara/gober/stream"
"github.com/cantara/wamper/screenshot"
"github.com/cantara/wamper/sites"
)
type Task struct {
Site sites.Site `json:"site"`
Time time.Time `json:"time"`
Interval time.Duration `json:"interval"`
SlackToken log.RedactedString `json:"slack_token"`
SlackChannel string `json:"slack_channel"`
}
type Service interface {
Set(task Task) error
Tasks() []scheduletasks.TaskMetadata
}
type service struct {
schedule scheduletasks.Tasks[Task]
screenshots screenshot.Store
}
func Init(s stream.Stream, consBuild consensus.ConsBuilderFunc, scr screenshot.Store, cryptoKey log.RedactedString, ctx context.Context) (out Service, err error) {
ser := service{
screenshots: scr,
}
//t, err := tasks.Init[Task](s, "slack_task", "", cryptKeyProvider(cryptoKey), ctx)
tas, err := scheduletasks.Init(s, consBuild, "slack_task", "1.0.0", stream.StaticProvider(cryptoKey), ser.executeTask, time.Second*30, true, 10, ctx)
if err != nil {
return
}
ser.schedule = tas
out = &ser
return
}
func (s *service) Set(t Task) error {
return s.schedule.Create(t.Site.Name, t.Time, t.Interval, t)
}
func (s *service) Tasks() []scheduletasks.TaskMetadata {
return s.schedule.Tasks()
}
func (s *service) executeTask(t Task, _ context.Context) bool {
scr, err := s.screenshots.Get(t.Site.Id())
if err != nil {
log.WithError(err).Error("while getting screenshot during scheduled task execution")
return false
}
slack, err := NewClient(t.SlackToken)
if err != nil {
log.WithError(err).Error("while creating slack client during scheduled task execution")
return false
}
r, err := slack.SendFile(t.SlackChannel, "Today's "+t.Site.Name+"!\n"+t.Site.Url.String(), scr.Buf)
if err != nil {
log.WithError(err).Error("while posting slack message during scheduled task execution")
return false
}
log.Debug("slack response", "response", r)
log.Info("posted slack message", "site", t.Site, "channel", t.SlackChannel)
return true
}