package bmattermost import ( "errors" "fmt" "github.com/42wim/matterbridge/bridge" "github.com/42wim/matterbridge/bridge/config" "github.com/42wim/matterbridge/bridge/helper" "github.com/42wim/matterbridge/matterclient" "github.com/42wim/matterbridge/matterhook" "github.com/rs/xid" ) type Bmattermost struct { mh *matterhook.Client mc *matterclient.MMClient uuid string TeamID string *bridge.Config avatarMap map[string]string } const mattermostPlugin = "mattermost.plugin" func New(cfg *bridge.Config) bridge.Bridger { b := &Bmattermost{Config: cfg, avatarMap: make(map[string]string)} b.uuid = xid.New().String() return b } func (b *Bmattermost) Command(cmd string) string { return "" } func (b *Bmattermost) Connect() error { if b.Account == mattermostPlugin { return nil } if b.GetString("WebhookBindAddress") != "" { if err := b.doConnectWebhookBind(); err != nil { return err } go b.handleMatter() return nil } switch { case b.GetString("WebhookURL") != "": if err := b.doConnectWebhookURL(); err != nil { return err } go b.handleMatter() return nil case b.GetString("Token") != "": b.Log.Info("Connecting using token (sending and receiving)") err := b.apiLogin() if err != nil { return err } go b.handleMatter() case b.GetString("Login") != "": b.Log.Info("Connecting using login/password (sending and receiving)") err := b.apiLogin() if err != nil { return err } go b.handleMatter() } if b.GetString("WebhookBindAddress") == "" && b.GetString("WebhookURL") == "" && b.GetString("Login") == "" && b.GetString("Token") == "" { return errors.New("no connection method found. See that you have WebhookBindAddress, WebhookURL or Token/Login/Password/Server/Team configured") } return nil } func (b *Bmattermost) Disconnect() error { return nil } func (b *Bmattermost) JoinChannel(channel config.ChannelInfo) error { if b.Account == mattermostPlugin { return nil } // we can only join channels using the API if b.GetString("WebhookURL") == "" && b.GetString("WebhookBindAddress") == "" { id := b.mc.GetChannelId(channel.Name, b.TeamID) if id == "" { return fmt.Errorf("Could not find channel ID for channel %s", channel.Name) } return b.mc.JoinChannel(id) } return nil } func (b *Bmattermost) Send(msg config.Message) (string, error) { if b.Account == mattermostPlugin { return "", nil } b.Log.Debugf("=> Receiving %#v", msg) // Make a action /me of the message if msg.Event == config.EventUserAction { msg.Text = "*" + msg.Text + "*" } // map the file SHA to our user (caches the avatar) if msg.Event == config.EventAvatarDownload { return b.cacheAvatar(&msg) } // Use webhook to send the message if b.GetString("WebhookURL") != "" { return b.sendWebhook(msg) } // Delete message if msg.Event == config.EventMsgDelete { if msg.ID == "" { return "", nil } return msg.ID, b.mc.DeleteMessage(msg.ID) } // Upload a file if it exists if msg.Extra != nil { for _, rmsg := range helper.HandleExtra(&msg, b.General) { if _, err := b.mc.PostMessage(b.mc.GetChannelId(rmsg.Channel, b.TeamID), rmsg.Username+rmsg.Text, msg.ParentID); err != nil { b.Log.Errorf("PostMessage failed: %s", err) } } if len(msg.Extra["file"]) > 0 { return b.handleUploadFile(&msg) } } // Prepend nick if configured if b.GetBool("PrefixMessagesWithNick") { msg.Text = msg.Username + msg.Text } // Edit message if we have an ID if msg.ID != "" { return b.mc.EditMessage(msg.ID, msg.Text) } // Post normal message return b.mc.PostMessage(b.mc.GetChannelId(msg.Channel, b.TeamID), msg.Text, msg.ParentID) }