Add Slack support

3 files changed, 203 insertions, 1 deletions

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
+	}
+}