Thumbnail

rani/matterbridge.git

Clone URL: https://git.buni.party/rani/matterbridge.git

commit 1a8743b3abcfb8a1a5ed7500fb4a14c041b5e51c Author: Wim <wim@42.be> Date: Sat Jan 28 21:57:06 2023 +0000 Remove mattermost 5 support (#1936) diff --git a/bridge/mattermost/handlers.go b/bridge/mattermost/handlers.go index c7b5124..fccb31b 100644 --- a/bridge/mattermost/handlers.go +++ b/bridge/mattermost/handlers.go @@ -310 +38 @@ package bmattermost  import (   "github.com/42wim/matterbridge/bridge/config"   "github.com/42wim/matterbridge/bridge/helper" - "github.com/42wim/matterbridge/matterclient" - matterclient6 "github.com/matterbridge/matterclient" - "github.com/mattermost/mattermost-server/v5/model" - model6 "github.com/mattermost/mattermost-server/v6/model" + "github.com/matterbridge/matterclient" + "github.com/mattermost/mattermost-server/v6/model"  )    // handleDownloadAvatar downloads the avatar of userid from channel @@ -2620 +2411 @@ func (b *Bmattermost) handleDownloadAvatar(userid string, channel string) {   var (   data []byte   err error - resp *model.Response   ) - if b.mc6 != nil { - data, _, err = b.mc6.Client.GetProfileImage(userid, "") - if err != nil { - b.Log.Errorf("ProfileImage download failed for %#v %s", userid, err) - return - } - } else { - data, resp = b.mc.Client.GetProfileImage(userid, "") - if resp.Error != nil { - b.Log.Errorf("ProfileImage download failed for %#v %s", userid, resp.Error) - return - } + data, _, err = b.mc.Client.GetProfileImage(userid, "") + if err != nil { + b.Log.Errorf("ProfileImage download failed for %#v %s", userid, err) + return   }     err = helper.HandleDownloadSize(b.Log, &rmsg, userid+".png", int64(len(data)), b.General) @@ -5233 +4110 @@ func (b *Bmattermost) handleDownloadAvatar(userid string, channel string) {   }  }   -// handleDownloadFile handles file download -func (b *Bmattermost) handleDownloadFile(rmsg *config.Message, id string) error { - if b.mc6 != nil { - return b.handleDownloadFile6(rmsg, id) - } - - 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 -} -  // nolint:wrapcheck -func (b *Bmattermost) handleDownloadFile6(rmsg *config.Message, id string) error { - url, _, _ := b.mc6.Client.GetFileLink(id) - finfo, _, err := b.mc6.Client.GetFileInfo(id) +func (b *Bmattermost) handleDownloadFile(rmsg *config.Message, id string) error { + url, _, _ := b.mc.Client.GetFileLink(id) + finfo, _, err := b.mc.Client.GetFileInfo(id)   if err != nil {   return err   } @@ -867 +527 @@ func (b *Bmattermost) handleDownloadFile6(rmsg *config.Message, id string) error   if err != nil {   return err   } - data, _, err := b.mc6.Client.DownloadFile(id, true) + data, _, err := b.mc.Client.DownloadFile(id, true)   if err != nil {   return err   } @@ -12515 +9110 @@ func (b *Bmattermost) handleMatter() {   }  }   +// nolint:cyclop  func (b *Bmattermost) handleMatterClient(messages chan *config.Message) { - if b.mc6 != nil { - b.Log.Debug("starting matterclient6") - b.handleMatterClient6(messages) - return - } -   for message := range b.mc.MessageChan { - b.Log.Debugf("%#v", message.Raw.Data) + b.Log.Debugf("%#v %#v", message.Raw.GetData(), message.Raw.EventType())     if b.skipMessage(message) {   b.Log.Debugf("Skipped message: %#v", message) @@ -16611 +12711 @@ func (b *Bmattermost) handleMatterClient(messages chan *config.Message) {   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") { + if message.Raw.EventType() == model.WebsocketEventPostEdited && !b.GetBool("EditDisable") {   rmsg.Text = message.Text + b.GetString("EditSuffix")   }   - if message.Raw.Event == model.WEBSOCKET_EVENT_POST_DELETED { + if message.Raw.EventType() == model.WebsocketEventPostDeleted {   rmsg.Event = config.EventMsgDelete   }   @@ -19268 +1536 @@ func (b *Bmattermost) handleMatterClient(messages chan *config.Message) {   }  }   -// nolint:cyclop -func (b *Bmattermost) handleMatterClient6(messages chan *config.Message) { - for message := range b.mc6.MessageChan { - b.Log.Debugf("%#v %#v", message.Raw.GetData(), message.Raw.EventType()) - - if b.skipMessage6(message) { - b.Log.Debugf("Skipped message: %#v", message) - continue - } - - channelName := b.getChannelName(message.Post.ChannelId) - if channelName == "" { - channelName = message.Channel - } - - // only download avatars if we have a place to upload them (configured mediaserver) - if b.General.MediaServerUpload != "" || b.General.MediaDownloadPath != "" { - b.handleDownloadAvatar(message.UserID, channelName) - } - - b.Log.Debugf("== Receiving event %#v", message) - - rmsg := &config.Message{ - Username: message.Username, - UserID: message.UserID, - Channel: channelName, - Text: message.Text, - ID: message.Post.Id, - ParentID: message.Post.RootId, // ParentID is obsolete with mattermost - Extra: make(map[string][]interface{}), - } - - // handle mattermost post properties (override username and attachments) - b.handleProps6(rmsg, message) - - // create a text for bridges that don't support native editing - if message.Raw.EventType() == model6.WebsocketEventPostEdited && !b.GetBool("EditDisable") { - rmsg.Text = message.Text + b.GetString("EditSuffix") - } - - if message.Raw.EventType() == model6.WebsocketEventPostDeleted { - 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 !b.GetBool("useusername") { - if nick := b.mc6.GetNickName(rmsg.UserID); nick != "" { - rmsg.Username = nick - } - } - - messages <- rmsg - } -} -  func (b *Bmattermost) handleMatterHook(messages chan *config.Message) {   for {   message := b.mh.Receive() @@ -26812 +1677 @@ func (b *Bmattermost) handleMatterHook(messages chan *config.Message) {   }  }   -// handleUploadFile handles native upload of files  func (b *Bmattermost) handleUploadFile(msg *config.Message) (string, error) { - if b.mc6 != nil { - return b.handleUploadFile6(msg) - } -   var err error   var res, id string   channelID := b.getChannelID(msg.Channel) @@ -29253 +1868 @@ func (b *Bmattermost) handleUploadFile(msg *config.Message) (string, error) {   return res, err  }   -// nolint:forcetypeassert,wrapcheck -func (b *Bmattermost) handleUploadFile6(msg *config.Message) (string, error) { - var err error - var res, id string - channelID := b.getChannelID(msg.Channel) - for _, f := range msg.Extra["file"] { - fi := f.(config.FileInfo) - id, err = b.mc6.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.mc6.PostMessageWithFiles(channelID, msg.Text, msg.ParentID, []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) - } - } - } - } -} -  // nolint:forcetypeassert -func (b *Bmattermost) handleProps6(rmsg *config.Message, message *matterclient6.Message) { +func (b *Bmattermost) handleProps(rmsg *config.Message, message *matterclient.Message) {   props := message.Post.Props   if props == nil {   return diff --git a/bridge/mattermost/helpers.go b/bridge/mattermost/helpers.go index c69d94f..3e1262c 100644 --- a/bridge/mattermost/helpers.go +++ b/bridge/mattermost/helpers.go @@ -611 +69 @@ import (     "github.com/42wim/matterbridge/bridge/config"   "github.com/42wim/matterbridge/bridge/helper" - "github.com/42wim/matterbridge/matterclient"   "github.com/42wim/matterbridge/matterhook" - matterclient6 "github.com/matterbridge/matterclient" - "github.com/mattermost/mattermost-server/v5/model" - model6 "github.com/mattermost/mattermost-server/v6/model" + "github.com/matterbridge/matterclient" + "github.com/mattermost/mattermost-server/v6/model"  )    func (b *Bmattermost) doConnectWebhookBind() error { @@ -2433 +2215 @@ func (b *Bmattermost) doConnectWebhookBind() error {   })   case b.GetString("Token") != "":   b.Log.Info("Connecting using token (sending)") - b.Log.Infof("Using mattermost v6 methods: %t", b.v6) - - if b.v6 { - err := b.apiLogin6() - if err != nil { - return err - } - } else { - err := b.apiLogin() - if err != nil { - return err - } + err := b.apiLogin() + if err != nil { + return err   }   case b.GetString("Login") != "":   b.Log.Info("Connecting using login/password (sending)") - b.Log.Infof("Using mattermost v6 methods: %t", b.v6) - - if b.v6 { - err := b.apiLogin6() - if err != nil { - return err - } - } else { - err := b.apiLogin() - if err != nil { - return err - } + err := b.apiLogin() + if err != nil { + return err   }   default:   b.Log.Info("Connecting using webhookbindaddress (receiving)") @@ -7245 +5228 @@ func (b *Bmattermost) doConnectWebhookURL() error {   })   if b.GetString("Token") != "" {   b.Log.Info("Connecting using token (receiving)") - b.Log.Infof("Using mattermost v6 methods: %t", b.v6) - - if b.v6 { - err := b.apiLogin6() - if err != nil { - return err - } - } else { - err := b.apiLogin() - if err != nil { - return err - } + err := b.apiLogin() + if err != nil { + return err   }   } else if b.GetString("Login") != "" {   b.Log.Info("Connecting using login/password (receiving)") - b.Log.Infof("Using mattermost v6 methods: %t", b.v6) - - if b.v6 { - err := b.apiLogin6() - if err != nil { - return err - } - } else { - err := b.apiLogin() - if err != nil { - return err - } + err := b.apiLogin() + if err != nil { + return err   }   }   return nil  }   +// nolint:wrapcheck  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")) + b.mc = matterclient.New(b.GetString("Login"), password, b.GetString("Team"), b.GetString("Server"), "")   if b.GetBool("debug") {   b.mc.SetLogLevel("debug")   } @@ -11839 +8113 @@ func (b *Bmattermost) apiLogin() error {   b.mc.SkipVersionCheck = b.GetBool("SkipVersionCheck")   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 -} - -// nolint:wrapcheck -func (b *Bmattermost) apiLogin6() error { - password := b.GetString("Password") - if b.GetString("Token") != "" { - password = "token=" + b.GetString("Token") - } - - b.mc6 = matterclient6.New(b.GetString("Login"), password, b.GetString("Team"), b.GetString("Server"), "") - if b.GetBool("debug") { - b.mc6.SetLogLevel("debug") - } - b.mc6.SkipTLSVerify = b.GetBool("SkipTLSVerify") - b.mc6.SkipVersionCheck = b.GetBool("SkipVersionCheck") - b.mc6.NoTLS = b.GetBool("NoTLS") - b.Log.Infof("Connecting %s (team: %s) on %s", b.GetString("Login"), b.GetString("Team"), b.GetString("Server"))   - if err := b.mc6.Login(); err != nil { + if err := b.mc.Login(); err != nil {   return err   }     b.Log.Info("Connection succeeded") - b.TeamID = b.mc6.GetTeamID() + b.TeamID = b.mc.GetTeamID()   return nil  }   @@ -23377 +1709 @@ func (b *Bmattermost) sendWebhook(msg config.Message) (string, error) {   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 - } - - channelName := b.getChannelName(message.Post.ChannelId) - if channelName == "" { - channelName = message.Channel - } - - b.Log.Debugf("Sending JOIN_LEAVE event from %s to gateway", b.Account) - b.Remote <- config.Message{ - Username: "system", - Text: message.Text, - Channel: channelName, - 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 non-post messages - if message.Post == nil { - b.Log.Debugf("ignoring nil message.Post: %#v", message) - 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 -} -  // skipMessages returns true if this message should not be handled  // nolint:gocyclo,cyclop -func (b *Bmattermost) skipMessage6(message *matterclient6.Message) bool { +func (b *Bmattermost) skipMessage(message *matterclient.Message) bool {   // Handle join/leave   if message.Type == "system_join_leave" ||   message.Type == "system_join_channel" || @@ -3297 +1987 @@ func (b *Bmattermost) skipMessage6(message *matterclient6.Message) bool {   }     // Handle edited messages - if (message.Raw.EventType() == model6.WebsocketEventPostEdited) && b.GetBool("EditDisable") { + if (message.Raw.EventType() == model.WebsocketEventPostEdited) && b.GetBool("EditDisable") {   return true   }   @@ -3487 +2177 @@ func (b *Bmattermost) skipMessage6(message *matterclient6.Message) bool {   }     // Ignore messages sent from a user logged in as the bot - if b.mc6.User.Username == message.Username { + if b.mc.User.Username == message.Username {   b.Log.Debug("message from same user as bot, ignoring")   return true   } @@ -3658 +2348 @@ func (b *Bmattermost) skipMessage6(message *matterclient6.Message) bool {   }     // only handle posted, edited or deleted events - if !(message.Raw.EventType() == "posted" || message.Raw.EventType() == model6.WebsocketEventPostEdited || - message.Raw.EventType() == model6.WebsocketEventPostDeleted) { + if !(message.Raw.EventType() == "posted" || message.Raw.EventType() == model.WebsocketEventPostEdited || + message.Raw.EventType() == model.WebsocketEventPostDeleted) {   return true   }   return false @@ -39611 +2657 @@ func (b *Bmattermost) getChannelID(name string) string {   return idcheck[1]   }   - if b.mc6 != nil { - return b.mc6.GetChannelID(name, b.TeamID) - } - - return b.mc.GetChannelId(name, b.TeamID) + return b.mc.GetChannelID(name, b.TeamID)  }    func (b *Bmattermost) getChannelName(id string) string { diff --git a/bridge/mattermost/mattermost.go b/bridge/mattermost/mattermost.go index 6151dbd..2b44dcf 100644 --- a/bridge/mattermost/mattermost.go +++ b/bridge/mattermost/mattermost.go @@ -916 +914 @@ import (   "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" - matterclient6 "github.com/matterbridge/matterclient" + "github.com/matterbridge/matterclient"   "github.com/rs/xid"  )    type Bmattermost struct {   mh *matterhook.Client - mc *matterclient.MMClient - mc6 *matterclient6.Client + mc *matterclient.Client   v6 bool   uuid string   TeamID string @@ -7434 +7217 @@ func (b *Bmattermost) Connect() error {   return nil   case b.GetString("Token") != "":   b.Log.Info("Connecting using token (sending and receiving)") - b.Log.Infof("Using mattermost v6 methods: %t", b.v6) - - if b.v6 { - err := b.apiLogin6() - if err != nil { - return err - } - } else { - err := b.apiLogin() - if err != nil { - return err - } + 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)")   b.Log.Infof("Using mattermost v6 methods: %t", b.v6) - - if b.v6 { - err := b.apiLogin6() - if err != nil { - return err - } - } else { - err := b.apiLogin() - if err != nil { - return err - } + err := b.apiLogin() + if err != nil { + return err   }   go b.handleMatter()   } @@ -13210 +1136 @@ func (b *Bmattermost) JoinChannel(channel config.ChannelInfo) error {   return fmt.Errorf("Could not find channel ID for channel %s", channel.Name)   }   - if b.mc6 != nil { - return b.mc6.JoinChannel(id) // nolint:wrapcheck - } -   return b.mc.JoinChannel(id)   }   @@ -1689 +1456 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) {   if msg.ID == "" {   return "", nil   } - if b.mc6 != nil { - return msg.ID, b.mc6.DeleteMessage(msg.ID) // nolint:wrapcheck - }     return msg.ID, b.mc.DeleteMessage(msg.ID)   } @@ -18336 +15720 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) {     // we only can reply to the root of the thread, not to a specific ID (like discord for example does)   if msg.ParentID != "" { - if b.mc6 != nil { - post, _, err := b.mc6.Client.GetPost(msg.ParentID, "") - if err != nil { - b.Log.Errorf("getting post %s failed: %s", msg.ParentID, err) - } - if post.RootId != "" { - msg.ParentID = post.RootId - } - } else { - post, res := b.mc.Client.GetPost(msg.ParentID, "") - if res.Error != nil { - b.Log.Errorf("getting post %s failed: %s", msg.ParentID, res.Error.DetailedError) - } - if post.RootId != "" { - msg.ParentID = post.RootId - } + post, _, err := b.mc.Client.GetPost(msg.ParentID, "") + if err != nil { + b.Log.Errorf("getting post %s failed: %s", msg.ParentID, err) + } + if post.RootId != "" { + msg.ParentID = post.RootId   }   }     // Upload a file if it exists   if msg.Extra != nil {   for _, rmsg := range helper.HandleExtra(&msg, b.General) { - if b.mc6 != nil { - if _, err := b.mc6.PostMessage(b.getChannelID(rmsg.Channel), rmsg.Username+rmsg.Text, msg.ParentID); err != nil { - b.Log.Errorf("PostMessage failed: %s", err) - } - } else { - if _, err := b.mc.PostMessage(b.getChannelID(rmsg.Channel), rmsg.Username+rmsg.Text, msg.ParentID); err != nil { - b.Log.Errorf("PostMessage failed: %s", err) - } + if _, err := b.mc.PostMessage(b.getChannelID(rmsg.Channel), rmsg.Username+rmsg.Text, msg.ParentID); err != nil { + b.Log.Errorf("PostMessage failed: %s", err)   }   }   if len(msg.Extra["file"]) > 0 { @@ -22717 +1859 @@ func (b *Bmattermost) Send(msg config.Message) (string, error) {     // Edit message if we have an ID   if msg.ID != "" { - if b.mc6 != nil { - return b.mc6.EditMessage(msg.ID, msg.Text) // nolint:wrapcheck - } -   return b.mc.EditMessage(msg.ID, msg.Text)   }     // Post normal message - if b.mc6 != nil { - return b.mc6.PostMessage(b.getChannelID(msg.Channel), msg.Text, msg.ParentID) // nolint:wrapcheck - } -   return b.mc.PostMessage(b.getChannelID(msg.Channel), msg.Text, msg.ParentID)  }