summaryrefslogtreecommitdiffstats
path: root/bridge/mattermost
diff options
context:
space:
mode:
authorWim <wim@42.be>2018-12-01 00:49:08 +0100
committerGitHub <noreply@github.com>2018-12-01 00:49:08 +0100
commitf94c2b40a3e8eb45054bda1b71eb6fd3f9bc9e6d (patch)
tree5c2ba8a3b443b4d390ab6386b4871e0d235ed3ac /bridge/mattermost
parent47d29ecf63be89fff5c2ecb9e354dc6cca70c5f0 (diff)
downloadmatterbridge-msglm-f94c2b40a3e8eb45054bda1b71eb6fd3f9bc9e6d.tar.gz
matterbridge-msglm-f94c2b40a3e8eb45054bda1b71eb6fd3f9bc9e6d.tar.bz2
matterbridge-msglm-f94c2b40a3e8eb45054bda1b71eb6fd3f9bc9e6d.zip
Refactor mattermost bridge (#622)
* Split up in different files * Decrease complexity * Fix linting issues
Diffstat (limited to 'bridge/mattermost')
-rw-r--r--bridge/mattermost/handlers.go194
-rw-r--r--bridge/mattermost/helpers.go218
-rw-r--r--bridge/mattermost/mattermost.go354
3 files changed, 422 insertions, 344 deletions
diff --git a/bridge/mattermost/handlers.go b/bridge/mattermost/handlers.go
new file mode 100644
index 00000000..d75e8d40
--- /dev/null
+++ b/bridge/mattermost/handlers.go
@@ -0,0 +1,194 @@
+package bmattermost
+
+import (
+ "github.com/42wim/matterbridge/bridge/config"
+ "github.com/42wim/matterbridge/bridge/helper"
+ "github.com/42wim/matterbridge/matterclient"
+ "github.com/mattermost/mattermost-server/model"
+)
+
+// handleDownloadAvatar downloads the avatar of userid from channel
+// sends a EVENT_AVATAR_DOWNLOAD message to the gateway if successful.
+// logs an error message if it fails
+func (b *Bmattermost) handleDownloadAvatar(userid string, channel string) {
+ rmsg := config.Message{
+ Username: "system",
+ Text: "avatar",
+ Channel: channel,
+ Account: b.Account,
+ UserID: userid,
+ Event: config.EventAvatarDownload,
+ Extra: make(map[string][]interface{}),
+ }
+ if _, ok := b.avatarMap[userid]; !ok {
+ data, resp := b.mc.Client.GetProfileImage(userid, "")
+ if resp.Error != nil {
+ b.Log.Errorf("ProfileImage download failed for %#v %s", userid, resp.Error)
+ return
+ }
+ err := helper.HandleDownloadSize(b.Log, &rmsg, userid+".png", int64(len(data)), b.General)
+ if err != nil {
+ b.Log.Error(err)
+ return
+ }
+ helper.HandleDownloadData(b.Log, &rmsg, userid+".png", rmsg.Text, "", &data, b.General)
+ b.Remote <- rmsg
+ }
+}
+
+// handleDownloadFile handles file download
+func (b *Bmattermost) handleDownloadFile(rmsg *config.Message, id string) error {
+ url, _ := b.mc.Client.GetFileLink(id)
+ finfo, resp := b.mc.Client.GetFileInfo(id)
+ if resp.Error != nil {
+ return resp.Error
+ }
+ err := helper.HandleDownloadSize(b.Log, rmsg, finfo.Name, finfo.Size, b.General)
+ if err != nil {
+ return err
+ }
+ data, resp := b.mc.Client.DownloadFile(id, true)
+ if resp.Error != nil {
+ return resp.Error
+ }
+ helper.HandleDownloadData(b.Log, rmsg, finfo.Name, rmsg.Text, url, &data, b.General)
+ return nil
+}
+
+func (b *Bmattermost) handleMatter() {
+ messages := make(chan *config.Message)
+ if b.GetString("WebhookBindAddress") != "" {
+ b.Log.Debugf("Choosing webhooks based receiving")
+ go b.handleMatterHook(messages)
+ } else {
+ if b.GetString("Token") != "" {
+ b.Log.Debugf("Choosing token based receiving")
+ } else {
+ b.Log.Debugf("Choosing login/password based receiving")
+ }
+ go b.handleMatterClient(messages)
+ }
+ var ok bool
+ for message := range messages {
+ message.Avatar = helper.GetAvatar(b.avatarMap, message.UserID, b.General)
+ message.Account = b.Account
+ message.Text, ok = b.replaceAction(message.Text)
+ if ok {
+ message.Event = config.EventUserAction
+ }
+ b.Log.Debugf("<= Sending message from %s on %s to gateway", message.Username, b.Account)
+ b.Log.Debugf("<= Message is %#v", message)
+ b.Remote <- *message
+ }
+}
+
+func (b *Bmattermost) handleMatterClient(messages chan *config.Message) {
+ for message := range b.mc.MessageChan {
+ b.Log.Debugf("%#v", message.Raw.Data)
+
+ if b.skipMessage(message) {
+ b.Log.Debugf("Skipped message: %#v", message)
+ continue
+ }
+
+ // only download avatars if we have a place to upload them (configured mediaserver)
+ if b.General.MediaServerUpload != "" || b.General.MediaDownloadPath != "" {
+ b.handleDownloadAvatar(message.UserID, message.Channel)
+ }
+
+ b.Log.Debugf("== Receiving event %#v", message)
+
+ rmsg := &config.Message{
+ Username: message.Username,
+ UserID: message.UserID,
+ Channel: message.Channel,
+ Text: message.Text,
+ ID: message.Post.Id,
+ Extra: make(map[string][]interface{}),
+ }
+
+ // handle mattermost post properties (override username and attachments)
+ b.handleProps(rmsg, message)
+
+ // create a text for bridges that don't support native editing
+ if message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED && !b.GetBool("EditDisable") {
+ rmsg.Text = message.Text + b.GetString("EditSuffix")
+ }
+
+ if message.Raw.Event == model.WEBSOCKET_EVENT_POST_DELETED {
+ rmsg.Event = config.EventMsgDelete
+ }
+
+ for _, id := range message.Post.FileIds {
+ err := b.handleDownloadFile(rmsg, id)
+ if err != nil {
+ b.Log.Errorf("download failed: %s", err)
+ }
+ }
+
+ // Use nickname instead of username if defined
+ if nick := b.mc.GetNickName(rmsg.UserID); nick != "" {
+ rmsg.Username = nick
+ }
+
+ messages <- rmsg
+ }
+}
+
+func (b *Bmattermost) handleMatterHook(messages chan *config.Message) {
+ for {
+ message := b.mh.Receive()
+ b.Log.Debugf("Receiving from matterhook %#v", message)
+ messages <- &config.Message{
+ UserID: message.UserID,
+ Username: message.UserName,
+ Text: message.Text,
+ Channel: message.ChannelName,
+ }
+ }
+}
+
+// handleUploadFile handles native upload of files
+func (b *Bmattermost) handleUploadFile(msg *config.Message) (string, error) {
+ var err error
+ var res, id string
+ channelID := b.mc.GetChannelId(msg.Channel, b.TeamID)
+ for _, f := range msg.Extra["file"] {
+ fi := f.(config.FileInfo)
+ id, err = b.mc.UploadFile(*fi.Data, channelID, fi.Name)
+ if err != nil {
+ return "", err
+ }
+ msg.Text = fi.Comment
+ if b.GetBool("PrefixMessagesWithNick") {
+ msg.Text = msg.Username + msg.Text
+ }
+ res, err = b.mc.PostMessageWithFiles(channelID, msg.Text, []string{id})
+ }
+ return res, err
+}
+
+func (b *Bmattermost) handleProps(rmsg *config.Message, message *matterclient.Message) {
+ props := message.Post.Props
+ if props == nil {
+ return
+ }
+ if _, ok := props["override_username"].(string); ok {
+ rmsg.Username = props["override_username"].(string)
+ }
+ if _, ok := props["attachments"].([]interface{}); ok {
+ rmsg.Extra["attachments"] = props["attachments"].([]interface{})
+ if rmsg.Text == "" {
+ for _, attachment := range rmsg.Extra["attachments"] {
+ attach := attachment.(map[string]interface{})
+ if attach["text"].(string) != "" {
+ rmsg.Text += attach["text"].(string)
+ continue
+ }
+ if attach["fallback"].(string) != "" {
+ rmsg.Text += attach["fallback"].(string)
+ }
+ }
+ }
+ }
+}
diff --git a/bridge/mattermost/helpers.go b/bridge/mattermost/helpers.go
new file mode 100644
index 00000000..c6117624
--- /dev/null
+++ b/bridge/mattermost/helpers.go
@@ -0,0 +1,218 @@
+package bmattermost
+
+import (
+ "strings"
+
+ "github.com/42wim/matterbridge/bridge/config"
+ "github.com/42wim/matterbridge/bridge/helper"
+ "github.com/42wim/matterbridge/matterclient"
+ "github.com/42wim/matterbridge/matterhook"
+ "github.com/mattermost/mattermost-server/model"
+)
+
+func (b *Bmattermost) doConnectWebhookBind() error {
+ switch {
+ case b.GetString("WebhookURL") != "":
+ b.Log.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)")
+ b.mh = matterhook.New(b.GetString("WebhookURL"),
+ matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
+ BindAddress: b.GetString("WebhookBindAddress")})
+ case b.GetString("Token") != "":
+ b.Log.Info("Connecting using token (sending)")
+ err := b.apiLogin()
+ if err != nil {
+ return err
+ }
+ case b.GetString("Login") != "":
+ b.Log.Info("Connecting using login/password (sending)")
+ err := b.apiLogin()
+ if err != nil {
+ return err
+ }
+ default:
+ b.Log.Info("Connecting using webhookbindaddress (receiving)")
+ b.mh = matterhook.New(b.GetString("WebhookURL"),
+ matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
+ BindAddress: b.GetString("WebhookBindAddress")})
+ }
+ return nil
+}
+
+func (b *Bmattermost) doConnectWebhookURL() error {
+ b.Log.Info("Connecting using webhookurl (sending)")
+ b.mh = matterhook.New(b.GetString("WebhookURL"),
+ matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
+ DisableServer: true})
+ if b.GetString("Token") != "" {
+ b.Log.Info("Connecting using token (receiving)")
+ err := b.apiLogin()
+ if err != nil {
+ return err
+ }
+ } else if b.GetString("Login") != "" {
+ b.Log.Info("Connecting using login/password (receiving)")
+ err := b.apiLogin()
+ if err != nil {
+ return err
+ }
+ }
+ return nil
+}
+
+func (b *Bmattermost) apiLogin() error {
+ password := b.GetString("Password")
+ if b.GetString("Token") != "" {
+ password = "token=" + b.GetString("Token")
+ }
+
+ b.mc = matterclient.New(b.GetString("Login"), password, b.GetString("Team"), b.GetString("Server"))
+ if b.GetBool("debug") {
+ b.mc.SetLogLevel("debug")
+ }
+ b.mc.SkipTLSVerify = b.GetBool("SkipTLSVerify")
+ b.mc.NoTLS = b.GetBool("NoTLS")
+ b.Log.Infof("Connecting %s (team: %s) on %s", b.GetString("Login"), b.GetString("Team"), b.GetString("Server"))
+ err := b.mc.Login()
+ if err != nil {
+ return err
+ }
+ b.Log.Info("Connection succeeded")
+ b.TeamID = b.mc.GetTeamId()
+ go b.mc.WsReceiver()
+ go b.mc.StatusLoop()
+ return nil
+}
+
+// replaceAction replace the message with the correct action (/me) code
+func (b *Bmattermost) replaceAction(text string) (string, bool) {
+ if strings.HasPrefix(text, "*") && strings.HasSuffix(text, "*") {
+ return strings.Replace(text, "*", "", -1), true
+ }
+ return text, false
+}
+
+func (b *Bmattermost) cacheAvatar(msg *config.Message) (string, error) {
+ fi := msg.Extra["file"][0].(config.FileInfo)
+ /* if we have a sha we have successfully uploaded the file to the media server,
+ so we can now cache the sha */
+ if fi.SHA != "" {
+ b.Log.Debugf("Added %s to %s in avatarMap", fi.SHA, msg.UserID)
+ b.avatarMap[msg.UserID] = fi.SHA
+ }
+ return "", nil
+}
+
+// sendWebhook uses the configured WebhookURL to send the message
+func (b *Bmattermost) sendWebhook(msg config.Message) (string, error) {
+ // skip events
+ if msg.Event != "" {
+ return "", nil
+ }
+
+ if b.GetBool("PrefixMessagesWithNick") {
+ msg.Text = msg.Username + msg.Text
+ }
+ if msg.Extra != nil {
+ // this sends a message only if we received a config.EVENT_FILE_FAILURE_SIZE
+ for _, rmsg := range helper.HandleExtra(&msg, b.General) {
+ rmsg := rmsg // scopelint
+ iconURL := config.GetIconURL(&rmsg, b.GetString("iconurl"))
+ matterMessage := matterhook.OMessage{
+ IconURL: iconURL,
+ Channel: rmsg.Channel,
+ UserName: rmsg.Username,
+ Text: rmsg.Text,
+ Props: make(map[string]interface{}),
+ }
+ matterMessage.Props["matterbridge_"+b.uuid] = true
+ if err := b.mh.Send(matterMessage); err != nil {
+ b.Log.Errorf("sendWebhook failed: %s ", err)
+ }
+ }
+
+ // webhook doesn't support file uploads, so we add the url manually
+ if len(msg.Extra["file"]) > 0 {
+ for _, f := range msg.Extra["file"] {
+ fi := f.(config.FileInfo)
+ if fi.URL != "" {
+ msg.Text += fi.URL
+ }
+ }
+ }
+ }
+
+ iconURL := config.GetIconURL(&msg, b.GetString("iconurl"))
+ matterMessage := matterhook.OMessage{
+ IconURL: iconURL,
+ Channel: msg.Channel,
+ UserName: msg.Username,
+ Text: msg.Text,
+ Props: make(map[string]interface{}),
+ }
+ if msg.Avatar != "" {
+ matterMessage.IconURL = msg.Avatar
+ }
+ matterMessage.Props["matterbridge_"+b.uuid] = true
+ err := b.mh.Send(matterMessage)
+ if err != nil {
+ b.Log.Info(err)
+ return "", err
+ }
+ return "", nil
+}
+
+// skipMessages returns true if this message should not be handled
+func (b *Bmattermost) skipMessage(message *matterclient.Message) bool {
+ // Handle join/leave
+ if message.Type == "system_join_leave" ||
+ message.Type == "system_join_channel" ||
+ message.Type == "system_leave_channel" {
+ if b.GetBool("nosendjoinpart") {
+ return true
+ }
+ b.Log.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account)
+ b.Remote <- config.Message{
+ Username: "system",
+ Text: message.Text,
+ Channel: message.Channel,
+ Account: b.Account,
+ Event: config.EventJoinLeave,
+ }
+ return true
+ }
+
+ // Handle edited messages
+ if (message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED) && b.GetBool("EditDisable") {
+ return true
+ }
+
+ // Ignore messages sent from matterbridge
+ if message.Post.Props != nil {
+ if _, ok := message.Post.Props["matterbridge_"+b.uuid].(bool); ok {
+ b.Log.Debugf("sent by matterbridge, ignoring")
+ return true
+ }
+ }
+
+ // Ignore messages sent from a user logged in as the bot
+ if b.mc.User.Username == message.Username {
+ return true
+ }
+
+ // if the message has reactions don't repost it (for now, until we can correlate reaction with message)
+ if message.Post.HasReactions {
+ return true
+ }
+
+ // ignore messages from other teams than ours
+ if message.Raw.Data["team_id"].(string) != b.TeamID {
+ return true
+ }
+
+ // only handle posted, edited or deleted events
+ if !(message.Raw.Event == "posted" || message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED ||
+ message.Raw.Event == model.WEBSOCKET_EVENT_POST_DELETED) {
+ return true
+ }
+ return false
+}
diff --git a/bridge/mattermost/mattermost.go b/bridge/mattermost/mattermost.go
index 4854753d..b0dca0ba 100644
--- a/bridge/mattermost/mattermost.go
+++ b/bridge/mattermost/mattermost.go
@@ -3,14 +3,12 @@ package bmattermost
import (
"errors"
"fmt"
- "strings"
"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/mattermost/mattermost-server/model"
"github.com/rs/xid"
)
@@ -40,54 +38,18 @@ func (b *Bmattermost) Connect() error {
return nil
}
if b.GetString("WebhookBindAddress") != "" {
- switch {
- case b.GetString("WebhookURL") != "":
- b.Log.Info("Connecting using webhookurl (sending) and webhookbindaddress (receiving)")
- b.mh = matterhook.New(b.GetString("WebhookURL"),
- matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
- BindAddress: b.GetString("WebhookBindAddress")})
- case b.GetString("Token") != "":
- b.Log.Info("Connecting using token (sending)")
- err := b.apiLogin()
- if err != nil {
- return err
- }
- case b.GetString("Login") != "":
- b.Log.Info("Connecting using login/password (sending)")
- err := b.apiLogin()
- if err != nil {
- return err
- }
- default:
- b.Log.Info("Connecting using webhookbindaddress (receiving)")
- b.mh = matterhook.New(b.GetString("WebhookURL"),
- matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
- BindAddress: b.GetString("WebhookBindAddress")})
+ if err := b.doConnectWebhookBind(); err != nil {
+ return err
}
go b.handleMatter()
return nil
}
switch {
case b.GetString("WebhookURL") != "":
- b.Log.Info("Connecting using webhookurl (sending)")
- b.mh = matterhook.New(b.GetString("WebhookURL"),
- matterhook.Config{InsecureSkipVerify: b.GetBool("SkipTLSVerify"),
- DisableServer: true})
- if b.GetString("Token") != "" {
- b.Log.Info("Connecting using token (receiving)")
- err := b.apiLogin()
- if err != nil {
- return err
- }
- go b.handleMatter()
- } else if b.GetString("Login") != "" {
- b.Log.Info("Connecting using login/password (receiving)")
- err := b.apiLogin()
- if err != nil {
- return err
- }
- go b.handleMatter()
+ 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)")
@@ -104,7 +66,8 @@ func (b *Bmattermost) Connect() error {
}
go b.handleMatter()
}
- if b.GetString("WebhookBindAddress") == "" && b.GetString("WebhookURL") == "" && b.GetString("Login") == "" && b.GetString("Token") == "" {
+ 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
@@ -161,7 +124,9 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) {
// Upload a file if it exists
if msg.Extra != nil {
for _, rmsg := range helper.HandleExtra(&msg, b.General) {
- b.mc.PostMessage(b.mc.GetChannelId(rmsg.Channel, b.TeamID), rmsg.Username+rmsg.Text)
+ if _, err := b.mc.PostMessage(b.mc.GetChannelId(rmsg.Channel, b.TeamID), rmsg.Username+rmsg.Text); err != nil {
+ b.Log.Errorf("PostMessage failed: %s", err)
+ }
}
if len(msg.Extra["file"]) > 0 {
return b.handleUploadFile(&msg)
@@ -181,302 +146,3 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) {
// Post normal message
return b.mc.PostMessage(b.mc.GetChannelId(msg.Channel, b.TeamID), msg.Text)
}
-
-func (b *Bmattermost) handleMatter() {
- messages := make(chan *config.Message)
- if b.GetString("WebhookBindAddress") != "" {
- b.Log.Debugf("Choosing webhooks based receiving")
- go b.handleMatterHook(messages)
- } else {
- if b.GetString("Token") != "" {
- b.Log.Debugf("Choosing token based receiving")
- } else {
- b.Log.Debugf("Choosing login/password based receiving")
- }
- go b.handleMatterClient(messages)
- }
- var ok bool
- for message := range messages {
- message.Avatar = helper.GetAvatar(b.avatarMap, message.UserID, b.General)
- message.Account = b.Account
- message.Text, ok = b.replaceAction(message.Text)
- if ok {
- message.Event = config.EventUserAction
- }
- b.Log.Debugf("<= Sending message from %s on %s to gateway", message.Username, b.Account)
- b.Log.Debugf("<= Message is %#v", message)
- b.Remote <- *message
- }
-}
-
-func (b *Bmattermost) handleMatterClient(messages chan *config.Message) {
- for message := range b.mc.MessageChan {
- b.Log.Debugf("%#v", message.Raw.Data)
-
- if b.skipMessage(message) {
- b.Log.Debugf("Skipped message: %#v", message)
- continue
- }
-
- // only download avatars if we have a place to upload them (configured mediaserver)
- if b.General.MediaServerUpload != "" || b.General.MediaDownloadPath != "" {
- b.handleDownloadAvatar(message.UserID, message.Channel)
- }
-
- b.Log.Debugf("== Receiving event %#v", message)
-
- rmsg := &config.Message{Username: message.Username, UserID: message.UserID, Channel: message.Channel, Text: message.Text, ID: message.Post.Id, Extra: make(map[string][]interface{})}
-
- // handle mattermost post properties (override username and attachments)
- props := message.Post.Props
- if props != nil {
- if _, ok := props["override_username"].(string); ok {
- rmsg.Username = props["override_username"].(string)
- }
- if _, ok := props["attachments"].([]interface{}); ok {
- rmsg.Extra["attachments"] = props["attachments"].([]interface{})
- if rmsg.Text == "" {
- for _, attachment := range rmsg.Extra["attachments"] {
- attach := attachment.(map[string]interface{})
- if attach["text"].(string) != "" {
- rmsg.Text += attach["text"].(string)
- continue
- }
- if attach["fallback"].(string) != "" {
- rmsg.Text += attach["fallback"].(string)
- }
- }
- }
- }
- }
-
- // create a text for bridges that don't support native editing
- if message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED && !b.GetBool("EditDisable") {
- rmsg.Text = message.Text + b.GetString("EditSuffix")
- }
-
- if message.Raw.Event == model.WEBSOCKET_EVENT_POST_DELETED {
- rmsg.Event = config.EventMsgDelete
- }
-
- if len(message.Post.FileIds) > 0 {
- for _, id := range message.Post.FileIds {
- err := b.handleDownloadFile(rmsg, id)
- if err != nil {
- b.Log.Errorf("download failed: %s", err)
- }
- }
- }
- // Use nickname instead of username if defined
- if nick := b.mc.GetNickName(rmsg.UserID); nick != "" {
- rmsg.Username = nick
- }
-
- messages <- rmsg
- }
-}
-
-func (b *Bmattermost) handleMatterHook(messages chan *config.Message) {
- for {
- message := b.mh.Receive()
- b.Log.Debugf("Receiving from matterhook %#v", message)
- messages <- &config.Message{UserID: message.UserID, Username: message.UserName, Text: message.Text, Channel: message.ChannelName}
- }
-}
-
-func (b *Bmattermost) apiLogin() error {
- password := b.GetString("Password")
- if b.GetString("Token") != "" {
- password = "token=" + b.GetString("Token")
- }
-
- b.mc = matterclient.New(b.GetString("Login"), password, b.GetString("Team"), b.GetString("Server"))
- if b.GetBool("debug") {
- b.mc.SetLogLevel("debug")
- }
- b.mc.SkipTLSVerify = b.GetBool("SkipTLSVerify")
- b.mc.NoTLS = b.GetBool("NoTLS")
- b.Log.Infof("Connecting %s (team: %s) on %s", b.GetString("Login"), b.GetString("Team"), b.GetString("Server"))
- err := b.mc.Login()
- if err != nil {
- return err
- }
- b.Log.Info("Connection succeeded")
- b.TeamID = b.mc.GetTeamId()
- go b.mc.WsReceiver()
- go b.mc.StatusLoop()
- return nil
-}
-
-// replaceAction replace the message with the correct action (/me) code
-func (b *Bmattermost) replaceAction(text string) (string, bool) {
- if strings.HasPrefix(text, "*") && strings.HasSuffix(text, "*") {
- return strings.Replace(text, "*", "", -1), true
- }
- return text, false
-}
-
-func (b *Bmattermost) cacheAvatar(msg *config.Message) (string, error) {
- fi := msg.Extra["file"][0].(config.FileInfo)
- /* if we have a sha we have successfully uploaded the file to the media server,
- so we can now cache the sha */
- if fi.SHA != "" {
- b.Log.Debugf("Added %s to %s in avatarMap", fi.SHA, msg.UserID)
- b.avatarMap[msg.UserID] = fi.SHA
- }
- return "", nil
-}
-
-// handleDownloadAvatar downloads the avatar of userid from channel
-// sends a EVENT_AVATAR_DOWNLOAD message to the gateway if successful.
-// logs an error message if it fails
-func (b *Bmattermost) handleDownloadAvatar(userid string, channel string) {
- rmsg := config.Message{Username: "system", Text: "avatar", Channel: channel, Account: b.Account, UserID: userid, Event: config.EventAvatarDownload, Extra: make(map[string][]interface{})}
- if _, ok := b.avatarMap[userid]; !ok {
- data, resp := b.mc.Client.GetProfileImage(userid, "")
- if resp.Error != nil {
- b.Log.Errorf("ProfileImage download failed for %#v %s", userid, resp.Error)
- return
- }
- err := helper.HandleDownloadSize(b.Log, &rmsg, userid+".png", int64(len(data)), b.General)
- if err != nil {
- b.Log.Error(err)
- return
- }
- helper.HandleDownloadData(b.Log, &rmsg, userid+".png", rmsg.Text, "", &data, b.General)
- b.Remote <- rmsg
- }
-}
-
-// handleDownloadFile handles file download
-func (b *Bmattermost) handleDownloadFile(rmsg *config.Message, id string) error {
- url, _ := b.mc.Client.GetFileLink(id)
- finfo, resp := b.mc.Client.GetFileInfo(id)
- if resp.Error != nil {
- return resp.Error
- }
- err := helper.HandleDownloadSize(b.Log, rmsg, finfo.Name, finfo.Size, b.General)
- if err != nil {
- return err
- }
- data, resp := b.mc.Client.DownloadFile(id, true)
- if resp.Error != nil {
- return resp.Error
- }
- helper.HandleDownloadData(b.Log, rmsg, finfo.Name, rmsg.Text, url, &data, b.General)
- return nil
-}
-
-// handleUploadFile handles native upload of files
-func (b *Bmattermost) handleUploadFile(msg *config.Message) (string, error) {
- var err error
- var res, id string
- channelID := b.mc.GetChannelId(msg.Channel, b.TeamID)
- for _, f := range msg.Extra["file"] {
- fi := f.(config.FileInfo)
- id, err = b.mc.UploadFile(*fi.Data, channelID, fi.Name)
- if err != nil {
- return "", err
- }
- msg.Text = fi.Comment
- if b.GetBool("PrefixMessagesWithNick") {
- msg.Text = msg.Username + msg.Text
- }
- res, err = b.mc.PostMessageWithFiles(channelID, msg.Text, []string{id})
- }
- return res, err
-}
-
-// sendWebhook uses the configured WebhookURL to send the message
-func (b *Bmattermost) sendWebhook(msg config.Message) (string, error) {
- // skip events
- if msg.Event != "" {
- return "", nil
- }
-
- if b.GetBool("PrefixMessagesWithNick") {
- msg.Text = msg.Username + msg.Text
- }
- if msg.Extra != nil {
- // this sends a message only if we received a config.EVENT_FILE_FAILURE_SIZE
- for _, rmsg := range helper.HandleExtra(&msg, b.General) {
- rmsg := rmsg // scopelint
- iconURL := config.GetIconURL(&rmsg, b.GetString("iconurl"))
- matterMessage := matterhook.OMessage{IconURL: iconURL, Channel: rmsg.Channel, UserName: rmsg.Username, Text: rmsg.Text, Props: make(map[string]interface{})}
- matterMessage.Props["matterbridge_"+b.uuid] = true
- b.mh.Send(matterMessage)
- }
-
- // webhook doesn't support file uploads, so we add the url manually
- if len(msg.Extra["file"]) > 0 {
- for _, f := range msg.Extra["file"] {
- fi := f.(config.FileInfo)
- if fi.URL != "" {
- msg.Text += fi.URL
- }
- }
- }
- }
-
- iconURL := config.GetIconURL(&msg, b.GetString("iconurl"))
- matterMessage := matterhook.OMessage{IconURL: iconURL, Channel: msg.Channel, UserName: msg.Username, Text: msg.Text, Props: make(map[string]interface{})}
- if msg.Avatar != "" {
- matterMessage.IconURL = msg.Avatar
- }
- matterMessage.Props["matterbridge_"+b.uuid] = true
- err := b.mh.Send(matterMessage)
- if err != nil {
- b.Log.Info(err)
- return "", err
- }
- return "", nil
-}
-
-// skipMessages returns true if this message should not be handled
-func (b *Bmattermost) skipMessage(message *matterclient.Message) bool {
- // Handle join/leave
- if message.Type == "system_join_leave" ||
- message.Type == "system_join_channel" ||
- message.Type == "system_leave_channel" {
- if b.GetBool("nosendjoinpart") {
- return true
- }
- b.Log.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account)
- b.Remote <- config.Message{Username: "system", Text: message.Text, Channel: message.Channel, Account: b.Account, Event: config.EventJoinLeave}
- return true
- }
-
- // Handle edited messages
- if (message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED) && b.GetBool("EditDisable") {
- return true
- }
-
- // Ignore messages sent from matterbridge
- if message.Post.Props != nil {
- if _, ok := message.Post.Props["matterbridge_"+b.uuid].(bool); ok {
- b.Log.Debugf("sent by matterbridge, ignoring")
- return true
- }
- }
-
- // Ignore messages sent from a user logged in as the bot
- if b.mc.User.Username == message.Username {
- return true
- }
-
- // if the message has reactions don't repost it (for now, until we can correlate reaction with message)
- if message.Post.HasReactions {
- return true
- }
-
- // ignore messages from other teams than ours
- if message.Raw.Data["team_id"].(string) != b.TeamID {
- return true
- }
-
- // only handle posted, edited or deleted events
- if !(message.Raw.Event == "posted" || message.Raw.Event == model.WEBSOCKET_EVENT_POST_EDITED || message.Raw.Event == model.WEBSOCKET_EVENT_POST_DELETED) {
- return true
- }
- return false
-}