package bmatrix import ( "bytes" "fmt" "mime" "regexp" "strings" "sync" "time" "github.com/42wim/matterbridge/bridge" "github.com/42wim/matterbridge/bridge/config" "github.com/42wim/matterbridge/bridge/helper" matrix "github.com/matterbridge/gomatrix" ) var ( htmlTag = regexp.MustCompile("") htmlReplacementTag = regexp.MustCompile("<[^>]*>") ) type NicknameCacheEntry struct { displayName string lastUpdated time.Time } type Bmatrix struct { mc *matrix.Client UserID string NicknameMap map[string]NicknameCacheEntry RoomMap map[string]string rateMutex sync.RWMutex sync.RWMutex *bridge.Config } type httpError struct { Errcode string `json:"errcode"` Err string `json:"error"` RetryAfterMs int `json:"retry_after_ms"` } type matrixUsername struct { plain string formatted string } // SubTextMessage represents the new content of the message in edit messages. type SubTextMessage struct { MsgType string `json:"msgtype"` Body string `json:"body"` FormattedBody string `json:"formatted_body,omitempty"` Format string `json:"format,omitempty"` } // MessageRelation explains how the current message relates to a previous message. // Notably used for message edits. type MessageRelation struct { EventID string `json:"event_id"` Type string `json:"rel_type"` } type EditedMessage struct { NewContent SubTextMessage `json:"m.new_content"` RelatedTo MessageRelation `json:"m.relates_to"` matrix.TextMessage } type InReplyToRelationContent struct { EventID string `json:"event_id"` } type InReplyToRelation struct { InReplyTo InReplyToRelationContent `json:"m.in_reply_to"` } type ReplyMessage struct { RelatedTo InReplyToRelation `json:"m.relates_to"` matrix.TextMessage } func New(cfg *bridge.Config) bridge.Bridger { b := &Bmatrix{Config: cfg} b.RoomMap = make(map[string]string) b.NicknameMap = make(map[string]NicknameCacheEntry) return b } func (b *Bmatrix) Connect() error { var err error b.Log.Infof("Connecting %s", b.GetString("Server")) if b.GetString("MxID") != "" && b.GetString("Token") != "" { b.mc, err = matrix.NewClient( b.GetString("Server"), b.GetString("MxID"), b.GetString("Token"), ) if err != nil { return err } b.UserID = b.GetString("MxID") b.Log.Info("Using existing Matrix credentials") } else { b.mc, err = matrix.NewClient(b.GetString("Server"), "", "") if err != nil { return err } resp, err := b.mc.Login(&matrix.ReqLogin{ Type: "m.login.password", User: b.GetString("Login"), Password: b.GetString("Password"), Identifier: matrix.NewUserIdentifier(b.GetString("Login")), }) if err != nil { return err } b.mc.SetCredentials(resp.UserID, resp.AccessToken) b.UserID = resp.UserID b.Log.Info("Connection succeeded") } go b.handlematrix() return nil } func (b *Bmatrix) Disconnect() error { return nil } func (b *Bmatrix) JoinChannel(channel config.ChannelInfo) error { return b.retry(func() error { resp, err := b.mc.JoinRoom(channel.Name, "", nil) if err != nil { return err } b.Lock() b.RoomMap[resp.RoomID] = channel.Name b.Unlock() return nil }) } func (b *Bmatrix) Send(msg config.Message) (string, error) { b.Log.Debugf("=> Receiving %#v", msg) channel := b.getRoomID(msg.Channel) b.Log.Debugf("Channel %s maps to channel id %s", msg.Channel, channel) username := newMatrixUsername(msg.Username) // Make a action /me of the message if msg.Event == config.EventUserAction { m := matrix.TextMessage{ MsgType: "m.emote", Body: username.plain + msg.Text, FormattedBody: username.formatted + helper.ParseMarkdown(msg.Text), Format: "org.matrix.custom.html", } if b.GetBool("HTMLDisable") { m.Format = "" m.FormattedBody = "" } msgID := "" err := b.retry(func() error { resp, err := b.mc.SendMessageEvent(channel, "m.room.message", m) if err != nil { return err } msgID = resp.EventID return err }) return msgID, err } // Delete message if msg.Event == config.EventMsgDelete { if msg.ID == "" { return "", nil } msgID := "" err := b.retry(func() error { resp, err := b.mc.RedactEvent(channel, msg.ID, &matrix.ReqRedact{}) if err != nil { return err } msgID = resp.EventID return err }) return msgID, err } // Upload a file if it exists if msg.Extra != nil { for _, rmsg := range helper.HandleExtra(&msg, b.General) { rmsg := rmsg err := b.retry(func() error { _, err := b.mc.SendText(channel, rmsg.Username+rmsg.Text) return err }) if err != nil { b.Log.Errorf("sendText failed: %s", err) } } // check if we have files to upload (from slack, telegram or mattermost) if len(msg.Extra["file"]) > 0 { return b.handleUploadFiles(&msg, channel) } } // Edit message if we have an ID if msg.ID != "" { rmsg := EditedMessage{ TextMessage: matrix.TextMessage{ Body: username.plain + msg.Text, MsgType: "m.text", Format: "org.matrix.custom.html", FormattedBody: username.formatted + helper.ParseMarkdown(msg.Text), }, } rmsg.NewContent = SubTextMessage{ Body: rmsg.TextMessage.Body, FormattedBody: rmsg.TextMessage.FormattedBody, Format: rmsg.TextMessage.Format, MsgType: "m.text", } if b.GetBool("HTMLDisable") { rmsg.TextMessage.Format = "" rmsg.TextMessage.FormattedBody = "" rmsg.NewContent.Format = "" rmsg.NewContent.FormattedBody = "" } rmsg.RelatedTo = MessageRelation{ EventID: msg.ID, Type: "m.replace", } err := b.retry(func() error { _, err := b.mc.SendMessageEvent(channel, "m.room.message", rmsg) return err }) if err != nil { return "", err } return msg.ID, nil } // Use notices to send join/leave events if msg.Event == config.EventJoinLeave { m := matrix.TextMessage{ MsgType: "m.notice", Body: username.plain + msg.Text, FormattedBody: username.formatted + msg.Text, Format: "org.matrix.custom.html", } if b.GetBool("HTMLDisable") { m.Format = "" m.FormattedBody = "" } var ( resp *matrix.RespSendEvent err error ) err = b.retry(func() error { resp, err = b.mc.SendMessageEvent(channel, "m.room.message", m) return err }) if err != nil { return "", err } return resp.EventID, err } if msg.ParentValid() { m := ReplyMessage{ TextMessage: matrix.TextMessage{ MsgType: "m.text", Body: username.plain + msg.Text, FormattedBody: username.formatted + helper.ParseMarkdown(msg.Text), Format: "org.matrix.custom.html", }, } if b.GetBool("HTMLDisable") { m.TextMessage.Format = "" m.TextMessage.FormattedBody = "" } m.RelatedTo = InReplyToRelation{ InReplyTo: InReplyToRelationContent{ EventID: msg.ParentID, }, } var ( resp *matrix.RespSendEvent err error ) err = b.retry(func() error { resp, err = b.mc.SendMessageEvent(channel, "m.room.message", m) return err }) if err != nil { return "", err } return resp.EventID, err } if b.GetBool("HTMLDisable") { var ( resp *matrix.RespSendEvent err error ) err = b.retry(func() error { resp, err = b.mc.SendText(channel, username.plain+msg.Text) return err }) if err != nil { return "", err } return resp.EventID, err } // Post normal message with HTML support (eg riot.im) var ( resp *matrix.RespSendEvent err error ) err = b.retry(func() error { resp, err = b.mc.SendFormattedText(channel, username.plain+msg.Text, username.formatted+helper.ParseMarkdown(msg.Text)) return err }) if err != nil { return "", err } return resp.EventID, err } func (b *Bmatrix) handlematrix() { syncer := b.mc.Syncer.(*matrix.DefaultSyncer) syncer.OnEventType("m.room.redaction", b.handleEvent) syncer.OnEventType("m.room.message", b.handleEvent) syncer.OnEventType("m.room.member", b.handleMemberChange) go func() { for { if b == nil { return } if err := b.mc.Sync(); err != nil { b.Log.Println("Sync() returned ", err) } } }() } func (b *Bmatrix) handleEdit(ev *matrix.Event, rmsg config.Message) bool { relationInterface, present := ev.Content["m.relates_to"] newContentInterface, present2 := ev.Content["m.new_content"] if !(present && present2) { return false } var relation MessageRelation if err := interface2Struct(relationInterface, &relation); err != nil { b.Log.Warnf("Couldn't parse 'm.relates_to' object with value %#v", relationInterface) return false } var newContent SubTextMessage if err := interface2Struct(newContentInterface, &newContent); err != nil { b.Log.Warnf("Couldn't parse 'm.new_content' object with value %#v", newContentInterface) return false } if relation.Type != "m.replace" { return false } rmsg.ID = relation.EventID rmsg.Text = newContent.Body b.Remote <- rmsg return true } func (b *Bmatrix) handleReply(ev *matrix.Event, rmsg config.Message) bool { relationInterface, present := ev.Content["m.relates_to"] if !present { return false } var relation InReplyToRelation if err := interface2Struct(relationInterface, &relation); err != nil { // probably fine return false } body := rmsg.Text if !b.GetBool("keepquotedreply") { for strings.HasPrefix(body, "> ") { lineIdx := strings.IndexRune(body, '\n') if lineIdx == -1 { body = "" } else { body = body[(lineIdx + 1):] } } } rmsg.Text = body rmsg.ParentID = relation.InReplyTo.EventID b.Remote <- rmsg return true } func (b *Bmatrix) handleMemberChange(ev *matrix.Event) { // Update the displayname on join messages, according to https://matrix.org/docs/spec/client_server/r0.6.1#events-on-change-of-profile-information if ev.Content["membership"] == "join" { if dn, ok := ev.Content["displayname"].(string); ok { b.cacheDisplayName(ev.Sender, dn) } } } func (b *Bmatrix) handleEvent(ev *matrix.Event) { b.Log.Debugf("== Receiving event: %#v", ev) if ev.Sender != b.UserID { b.RLock() channel, ok := b.RoomMap[ev.RoomID] b.RUnlock() if !ok { b.Log.Debugf("Unknown room %s", ev.RoomID) return } // Create our message rmsg := config.Message{ Username: b.getDisplayName(ev.Sender), Channel: channel, Account: b.Account, UserID: ev.Sender, ID: ev.ID, Avatar: b.getAvatarURL(ev.Sender), } // Remove homeserver suffix if configured if b.GetBool("NoHomeServerSuffix") { re := regexp.MustCompile("(.*?):.*") rmsg.Username = re.ReplaceAllString(rmsg.Username, `$1`) } // Delete event if ev.Type == "m.room.redaction" { rmsg.Event = config.EventMsgDelete rmsg.ID = ev.Redacts rmsg.Text = config.EventMsgDelete b.Remote <- rmsg return } // Text must be a string if rmsg.Text, ok = ev.Content["body"].(string); !ok { b.Log.Errorf("Content[body] is not a string: %T\n%#v", ev.Content["body"], ev.Content) return } // Do we have a /me action if ev.Content["msgtype"].(string) == "m.emote" { rmsg.Event = config.EventUserAction } // Is it an edit? if b.handleEdit(ev, rmsg) { return } // Is it a reply? if b.handleReply(ev, rmsg) { return } // Do we have attachments if b.containsAttachment(ev.Content) { err := b.handleDownloadFile(&rmsg, ev.Content) if err != nil { b.Log.Errorf("download failed: %#v", err) } } b.Log.Debugf("<= Sending message from %s on %s to gateway", ev.Sender, b.Account) b.Remote <- rmsg // not crucial, so no ratelimit check here if err := b.mc.MarkRead(ev.RoomID, ev.ID); err != nil { b.Log.Errorf("couldn't mark message as read %s", err.Error()) } } } // handleDownloadFile handles file download func (b *Bmatrix) handleDownloadFile(rmsg *config.Message, content map[string]interface{}) error { var ( ok bool url, name, msgtype, mtype string info map[string]interface{} size float64 ) rmsg.Extra = make(map[string][]interface{}) if url, ok = content["url"].(string); !ok { return fmt.Errorf("url isn't a %T", url) } url = strings.Replace(url, "mxc://", b.GetString("Server")+"/_matrix/media/v1/download/", -1) if info, ok = content["info"].(map[string]interface{}); !ok { return fmt.Errorf("info isn't a %T", info) } if size, ok = info["size"].(float64); !ok { return fmt.Errorf("size isn't a %T", size) } if name, ok = content["body"].(string); !ok { return fmt.Errorf("name isn't a %T", name) } if msgtype, ok = content["msgtype"].(string); !ok { return fmt.Errorf("msgtype isn't a %T", msgtype) } if mtype, ok = info["mimetype"].(string); !ok { return fmt.Errorf("mtype isn't a %T", mtype) } // check if we have an image uploaded without extension if !strings.Contains(name, ".") { if msgtype == "m.image" { mext, _ := mime.ExtensionsByType(mtype) if len(mext) > 0 { name += mext[0] } } else { // just a default .png extension if we don't have mime info name += ".png" } } // check if the size is ok err := helper.HandleDownloadSize(b.Log, rmsg, name, int64(size), b.General) if err != nil { return err } // actually download the file data, err := helper.DownloadFile(url) if err != nil { return fmt.Errorf("download %s failed %#v", url, err) } // add the downloaded data to the message helper.HandleDownloadData(b.Log, rmsg, name, "", url, data, b.General) return nil } // handleUploadFiles handles native upload of files. func (b *Bmatrix) handleUploadFiles(msg *config.Message, channel string) (string, error) { for _, f := range msg.Extra["file"] { if fi, ok := f.(config.FileInfo); ok { b.handleUploadFile(msg, channel, &fi) } } return "", nil } // handleUploadFile handles native upload of a file. func (b *Bmatrix) handleUploadFile(msg *config.Message, channel string, fi *config.FileInfo) { username := newMatrixUsername(msg.Username) content := bytes.NewReader(*fi.Data) sp := strings.Split(fi.Name, ".") mtype := mime.TypeByExtension("." + sp[len(sp)-1]) // image and video uploads send no username, we have to do this ourself here #715 err := b.retry(func() error { _, err := b.mc.SendFormattedText(channel, username.plain+fi.Comment, username.formatted+fi.Comment) return err }) if err != nil { b.Log.Errorf("file comment failed: %#v", err) } b.Log.Debugf("uploading file: %s %s", fi.Name, mtype) var res *matrix.RespMediaUpload err = b.retry(func() error { res, err = b.mc.UploadToContentRepo(content, mtype, int64(len(*fi.Data))) return err }) if err != nil { b.Log.Errorf("file upload failed: %#v", err) return } switch { case strings.Contains(mtype, "video"): b.Log.Debugf("sendVideo %s", res.ContentURI) err = b.retry(func() error { _, err = b.mc.SendVideo(channel, fi.Name, res.ContentURI) return err }) if err != nil { b.Log.Errorf("sendVideo failed: %#v", err) } case strings.Contains(mtype, "image"): b.Log.Debugf("sendImage %s", res.ContentURI) err = b.retry(func() error { _, err = b.mc.SendImage(channel, fi.Name, res.ContentURI) return err }) if err != nil { b.Log.Errorf("sendImage failed: %#v", err) } case strings.Contains(mtype, "audio"): b.Log.Debugf("sendAudio %s", res.ContentURI) err = b.retry(func() error { _, err = b.mc.SendMessageEvent(channel, "m.room.message", matrix.AudioMessage{ MsgType: "m.audio", Body: fi.Name, URL: res.ContentURI, Info: matrix.AudioInfo{ Mimetype: mtype, Size: uint(len(*fi.Data)), }, }) return err }) if err != nil { b.Log.Errorf("sendAudio failed: %#v", err) } default: b.Log.Debugf("sendFile %s", res.ContentURI) err = b.retry(func() error { _, err = b.mc.SendMessageEvent(channel, "m.room.message", matrix.FileMessage{ MsgType: "m.file", Body: fi.Name, URL: res.ContentURI, Info: matrix.FileInfo{ Mimetype: mtype, Size: uint(len(*fi.Data)), }, }) return err }) if err != nil { b.Log.Errorf("sendFile failed: %#v", err) } } b.Log.Debugf("result: %#v", res) }