Thumbnail

rani/matterbridge.git

Clone URL: https://git.buni.party/rani/matterbridge.git

commit 1e016277effba83762ba5d79c672c1010594401d Author: Wim <wim@42.be> Date: Mon Sep 05 16:34:37 2016 +0000 Add Slack support diff --git a/bridge/bridge.go b/bridge/bridge.go index 047370b..45a0160 100644 --- a/bridge/bridge.go +++ b/bridge/bridge.go @@ -66 +67 @@ import (   "github.com/42wim/matterbridge/bridge/gitter"   "github.com/42wim/matterbridge/bridge/irc"   "github.com/42wim/matterbridge/bridge/mattermost" + "github.com/42wim/matterbridge/bridge/slack"   "github.com/42wim/matterbridge/bridge/xmpp"   log "github.com/Sirupsen/logrus"   "strings" @@ -426 +439 @@ func NewBridge(cfg *config.Config) error {   if cfg.Gitter.Enable {   b.Bridges = append(b.Bridges, bgitter.New(cfg, c))   } + if cfg.Slack.Enable { + b.Bridges = append(b.Bridges, bslack.New(cfg, c)) + }   if len(b.Bridges) < 2 {   log.Fatalf("only %d sections enabled. Need at least 2 sections enabled (eg [IRC] and [mattermost]", len(b.Bridges))   } @@ -726 +767 @@ func (b *Bridge) mapChannels() error {   m["mattermost"] = val.Mattermost   m["xmpp"] = val.Xmpp   m["gitter"] = val.Gitter + m["slack"] = val.Slack   b.Channels = append(b.Channels, m)   }   return nil @@ -836 +887 @@ func (b *Bridge) mapIgnores() {   m["mattermost"] = strings.Fields(b.Config.Mattermost.IgnoreNicks)   m["xmpp"] = strings.Fields(b.Config.Xmpp.IgnoreNicks)   m["gitter"] = strings.Fields(b.Config.Gitter.IgnoreNicks) + m["slack"] = strings.Fields(b.Config.Slack.IgnoreNicks)   b.ignoreNicks = m  }   @@ -1056 +1117 @@ func (b *Bridge) handleMessage(msg config.Message, dest Bridger) {   return   }   b.modifyMessage(&msg, dest.Name()) + log.Debugf("sending %#v from %s to %s", msg, msg.Origin, dest.Name())   dest.Send(msg)   }  } @@ -1385 +1457 @@ func (b *Bridge) modifyMessage(msg *config.Message, dest string) {   setNickFormat(msg, b.Config.Xmpp.RemoteNickFormat)   case "mattermost":   setNickFormat(msg, b.Config.Mattermost.RemoteNickFormat) + case "slack": + setNickFormat(msg, b.Config.Slack.RemoteNickFormat)   }  } diff --git a/bridge/config/config.go b/bridge/config/config.go index cd353fc..48574e2 100644 --- a/bridge/config/config.go +++ b/bridge/config/config.go @@ -357 +356 @@ type Config struct {   RemoteNickFormat string   Token string   } -   Mattermost struct {   URL string   ShowJoinPart bool @@ -556 +5419 @@ type Config struct {   NoTLS bool   Enable bool   } + Slack struct { + BindAddress string + Enable bool + IconURL string + IgnoreNicks string + NickFormatter string + NicksPerRow int + PrefixMessagesWithNick bool + RemoteNickFormat string + Token string + URL string + UseAPI bool + }   Xmpp struct {   IgnoreNicks string   Jid string @@ -706 +827 @@ type Config struct {   Mattermost string   Xmpp string   Gitter string + Slack string   }   General struct {   GiphyAPIKey string diff --git a/bridge/slack/slack.go b/bridge/slack/slack.go new file mode 100644 index 0000000..804eaea --- /dev/null +++ b/bridge/slack/slack.go @@ -00 +1180 @@ +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 + } +} diff --git a/matterhook/matterhook.go b/matterhook/matterhook.go index fc003cc..0880884 100644 --- a/matterhook/matterhook.go +++ b/matterhook/matterhook.go @@ -276 +278 @@ type OMessage struct {    // IMessage for mattermost outgoing webhook. (received from mattermost)  type IMessage struct { + BotID string `schema:"bot_id"` + BotName string `schema:"bot_name"`   Token string `schema:"token"`   TeamID string `schema:"team_id"`   TeamDomain string `schema:"team_domain"` @@ -366 +388 @@ type IMessage struct {   UserID string `schema:"user_id"`   UserName string `schema:"user_name"`   PostId string `schema:"post_id"` + RawText string `schema:"raw_text"` + ServiceId string `schema:"service_id"`   Text string `schema:"text"`   TriggerWord string `schema:"trigger_word"`  }