forked from danryan/hal
/
rtm.go
87 lines (78 loc) · 2.22 KB
/
rtm.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
package slack
import (
"github.com/danryan/hal"
"github.com/nlopes/slack"
)
func (a *adapter) startConnection() {
api := slack.New(a.token)
// api.SetDebug(true)
users, err := api.GetUsers()
if err != nil {
hal.Logger.Debugf("%s\n", err)
}
for _, user := range users {
// retrieve the name and mention name of our bot from the server
// if user.Id == api.Id {
// a.name = user.Name
// // skip adding the bot to the users map
// continue
// }
// Initialize a newUser object in case we need it.
newUser := hal.User{
ID: user.ID,
Name: user.Name,
}
// Prepopulate our users map because we can easily do so.
// If a user doesn't exist, set it.
u, err := a.Robot.Users.Get(user.ID)
if err != nil {
a.Robot.Users.Set(user.ID, newUser)
}
// If the user doesn't match completely (say, if someone changes their name),
// then adjust what we have stored.
if u.Name != user.Name {
a.Robot.Users.Set(user.ID, newUser)
}
}
hal.Logger.Debugf("Stored users: %s\n", a.Robot.Users.All())
a.rtm = api.NewRTM()
go a.rtm.ManageConnection()
for {
select {
case msg := <-a.rtm.IncomingEvents:
hal.Logger.Debug("Event Received: ")
switch msg.Data.(type) {
case slack.HelloEvent:
// Ignore hello
case *slack.MessageEvent:
m := msg.Data.(*slack.MessageEvent)
hal.Logger.Debugf("Message: %v\n", m)
msg := a.newMessage(m)
a.Receive(msg)
case *slack.PresenceChangeEvent:
m := msg.Data.(*slack.PresenceChangeEvent)
hal.Logger.Debugf("Presence Change: %v\n", m)
case slack.LatencyReport:
m := msg.Data.(slack.LatencyReport)
hal.Logger.Debugf("Current latency: %v\n", m.Value)
case slack.TeamJoinEvent:
m := msg.Data.(slack.TeamJoinEvent)
hal.Logger.Debugf("New member joined the team: %v\n", m.User)
// Add the new member to the user list
if _, err := a.Robot.Users.Get(m.User.ID); err != nil {
a.Robot.Users.Set(m.User.ID, hal.User{ID: m.User.ID, Name: m.User.Name})
}
default:
hal.Logger.Debugf("Unexpected: %v\n", msg.Data)
}
}
}
}
func (a *adapter) newMessage(msg *slack.MessageEvent) *hal.Message {
user, _ := a.Robot.Users.Get(msg.Msg.User)
return &hal.Message{
User: user,
Room: msg.Msg.Channel,
Text: msg.Text,
}
}