From b30e85836e575974bac1b2ea91e9b6c2dd39fe44 Mon Sep 17 00:00:00 2001 From: Wim Date: Mon, 5 Sep 2016 16:34:37 +0200 Subject: Add Slack support --- bridge/bridge.go | 9 +++ bridge/config/config.go | 15 +++- bridge/slack/slack.go | 180 ++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 203 insertions(+), 1 deletion(-) create mode 100644 bridge/slack/slack.go (limited to 'bridge') diff --git a/bridge/bridge.go b/bridge/bridge.go index 047370be..45a0160a 100644 --- a/bridge/bridge.go +++ b/bridge/bridge.go @@ -6,6 +6,7 @@ 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" @@ -42,6 +43,9 @@ 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)) } @@ -72,6 +76,7 @@ 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 @@ -83,6 +88,7 @@ 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 } @@ -105,6 +111,7 @@ 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) } } @@ -138,5 +145,7 @@ 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 cd353fcb..48574e28 100644 --- a/bridge/config/config.go +++ b/bridge/config/config.go @@ -35,7 +35,6 @@ type Config struct { RemoteNickFormat string Token string } - Mattermost struct { URL string ShowJoinPart bool @@ -55,6 +54,19 @@ 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 @@ -70,6 +82,7 @@ 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 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 + } +} -- cgit v1.2.3