diff options
Diffstat (limited to 'bridge/slack/slack.go')
-rw-r--r-- | bridge/slack/slack.go | 180 |
1 files changed, 180 insertions, 0 deletions
diff --git a/bridge/slack/slack.go b/bridge/slack/slack.go new file mode 100644 index 00000000..804eaea4 --- /dev/null +++ b/bridge/slack/slack.go @@ -0,0 +1,180 @@ +package bslack + +import ( + "github.com/42wim/matterbridge/bridge/config" + "github.com/42wim/matterbridge/matterhook" + log "github.com/Sirupsen/logrus" + "github.com/nlopes/slack" + "strings" + "time" +) + +type MMMessage struct { + Text string + Channel string + Username string +} + +type bslack struct { + mh *matterhook.Client + sc *slack.Client + // MMapi + *config.Config + rtm *slack.RTM + Plus bool + Remote chan config.Message + channels []slack.Channel +} + +var flog *log.Entry + +func init() { + flog = log.WithFields(log.Fields{"module": "slack"}) +} + +func New(cfg *config.Config, c chan config.Message) *bslack { + b := &bslack{} + b.Config = cfg + b.Remote = c + b.Plus = cfg.Slack.UseAPI + return b +} + +func (b *bslack) Command(cmd string) string { + return "" +} + +func (b *bslack) Connect() error { + if !b.Plus { + b.mh = matterhook.New(b.Config.Slack.URL, + matterhook.Config{BindAddress: b.Config.Slack.BindAddress}) + } else { + b.sc = slack.New(b.Config.Slack.Token) + flog.Infof("Trying login on slack with Token") + /* + if err != nil { + return err + } + */ + flog.Info("Login ok") + } + b.rtm = b.sc.NewRTM() + go b.rtm.ManageConnection() + go b.handleSlack() + return nil +} + +func (b *bslack) Name() string { + return "slack" +} + +func (b *bslack) Send(msg config.Message) error { + flog.Infof("slack send %#v", msg) + if msg.Origin != "slack" { + return b.SendType(msg.Username, msg.Text, msg.Channel, "") + } + return nil +} + +func (b *bslack) SendType(nick string, message string, channel string, mtype string) error { + if b.Config.Slack.PrefixMessagesWithNick { + message = nick + " " + message + } + if !b.Plus { + matterMessage := matterhook.OMessage{IconURL: b.Config.Slack.IconURL} + matterMessage.Channel = channel + matterMessage.UserName = nick + matterMessage.Type = mtype + matterMessage.Text = message + err := b.mh.Send(matterMessage) + if err != nil { + flog.Info(err) + return err + } + flog.Debug("->slack channel: ", channel, " ", message) + return nil + } + flog.Debugf("sent to slack channel API: %s %s", channel, message) + newmsg := b.rtm.NewOutgoingMessage(message, b.getChannelByName(channel).ID) + b.rtm.SendMessage(newmsg) + return nil +} + +func (b *bslack) getChannelByName(name string) *slack.Channel { + if b.channels == nil { + return nil + } + for _, channel := range b.channels { + if channel.Name == name { + return &channel + } + } + return nil +} + +func (b *bslack) handleSlack() { + flog.Infof("Choosing API based slack connection: %t", b.Plus) + mchan := make(chan *MMMessage) + if b.Plus { + go b.handleSlackClient(mchan) + } else { + go b.handleMatterHook(mchan) + } + time.Sleep(time.Second) + flog.Info("Start listening for Slack messages") + for message := range mchan { + texts := strings.Split(message.Text, "\n") + for _, text := range texts { + flog.Debug("Sending message from " + message.Username + " to " + message.Channel) + b.Remote <- config.Message{Text: text, Username: message.Username, Channel: message.Channel, Origin: "slack"} + } + } +} + +func (b *bslack) handleSlackClient(mchan chan *MMMessage) { + for msg := range b.rtm.IncomingEvents { + switch ev := msg.Data.(type) { + case *slack.MessageEvent: + flog.Debugf("%#v", ev) + channel, err := b.rtm.GetChannelInfo(ev.Channel) + if err != nil { + continue + } + user, err := b.rtm.GetUserInfo(ev.User) + if err != nil { + continue + } + m := &MMMessage{} + m.Username = user.Name + m.Channel = channel.Name + m.Text = ev.Text + mchan <- m + case *slack.OutgoingErrorEvent: + flog.Debugf("%#v", ev.Error()) + case *slack.ConnectedEvent: + b.channels = ev.Info.Channels + for _, val := range b.Config.Channel { + channel := b.getChannelByName(val.Slack) + if channel != nil && !channel.IsMember { + flog.Infof("Joining %s", val.Slack) + b.sc.JoinChannel(channel.ID) + } + } + case *slack.InvalidAuthEvent: + flog.Fatalf("Invalid Token %#v", ev) + default: + } + } +} + +func (b *bslack) handleMatterHook(mchan chan *MMMessage) { + for { + message := b.mh.Receive() + flog.Debugf("receiving from slack %#v", message) + m := &MMMessage{} + m.Username = message.UserName + m.Text = message.Text + m.Channel = message.ChannelName + mchan <- m + } +} |