summaryrefslogtreecommitdiffstats
path: root/bridge/slack/slack.go
diff options
context:
space:
mode:
Diffstat (limited to 'bridge/slack/slack.go')
-rw-r--r--bridge/slack/slack.go180
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
+ }
+}