/
main.go
109 lines (91 loc) · 2.36 KB
/
main.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
package main
import (
"log"
"regexp"
"strings"
"github.com/bfontaine/stargazer/Godeps/_workspace/src/github.com/nlopes/slack"
"github.com/bfontaine/stargazer/Godeps/_workspace/src/github.com/vrischmann/envconfig"
)
var conf struct {
Token string
}
var api *slack.Client
var rtm *slack.RTM
var whitelist = make(map[string]struct{})
func main() {
if err := envconfig.Init(&conf); err != nil {
log.Fatal(err)
}
api = slack.New(conf.Token)
rtm = api.NewRTM()
go rtm.ManageConnection()
for {
select {
case msg := <-rtm.IncomingEvents:
switch ev := msg.Data.(type) {
case *slack.StarAddedEvent:
notify(newStar(ev.User, true, ev.Item))
case *slack.StarRemovedEvent:
notify(newStar(ev.User, false, ev.Item))
case *slack.DisconnectedEvent:
log.Println("Disconnection.")
return
case *slack.ConnectingEvent:
log.Println("Connecting....")
case *slack.ConnectedEvent:
log.Println("Connected.")
slackInfo = rtm.GetInfo()
case *slack.MessageEvent:
if strings.HasPrefix(ev.Channel, "D") {
resp := gotDM(ev.Channel, ev.User, ev.Text)
if resp != "" {
api.PostMessage(ev.Channel, resp, slackParams)
}
}
default:
// skip
}
}
}
}
func notify(s Star) {
if _, ok := whitelist[s.Message.User]; !ok {
log.Printf("%s is not whitelisted, skipping.", s.Message.User)
return
}
_, _, ch, err := api.OpenIMChannel(s.Message.User)
if err != nil {
log.Println(err.Error())
return
}
msg, err := s.notification()
if err != nil {
log.Printf("Error: %v\n", err)
return
}
api.PostMessage(ch, msg, slackParams)
}
var reEnableMsg = regexp.MustCompile(`\b(enable|yes|oui|want|please)\b`)
var reDisableMsg = regexp.MustCompile(`\b(disable|no|non|stop|fuck you|go away)\b`)
const (
enablingMessage = "Got it! Happy stalking!"
enabledMessage = "I already know that ;)"
disablingMessage = "OK got it. I won't disturb you anymore."
disabledMessage = "I didn’t want to disturb anyway."
)
func gotDM(channel, user, text string) string {
if reEnableMsg.MatchString(strings.ToLower(strings.TrimSpace(text))) {
if _, ok := whitelist[user]; ok {
return enabledMessage
}
whitelist[user] = struct{}{}
return enablingMessage
} else if reDisableMsg.MatchString(text) {
if _, ok := whitelist[user]; !ok {
return disabledMessage
}
delete(whitelist, user)
return disablingMessage
}
return ""
}