package bmatrix import ( "bytes" "mime" "regexp" "strings" "sync" "github.com/42wim/matterbridge/bridge/config" "github.com/42wim/matterbridge/bridge/helper" log "github.com/Sirupsen/logrus" matrix "github.com/matterbridge/gomatrix" ) type Bmatrix struct { mc *matrix.Client UserID string RoomMap map[string]string sync.RWMutex *config.BridgeConfig } var flog *log.Entry var protocol = "matrix" func init() { flog = log.WithFields(log.Fields{"module": protocol}) } func New(cfg *config.BridgeConfig) *Bmatrix { b := &Bmatrix{BridgeConfig: cfg} b.RoomMap = make(map[string]string) return b } func (b *Bmatrix) Connect() error { var err error flog.Infof("Connecting %s", b.Config.Server) b.mc, err = matrix.NewClient(b.Config.Server, "", "") if err != nil { flog.Debugf("%#v", err) return err } resp, err := b.mc.Login(&matrix.ReqLogin{ Type: "m.login.password", User: b.Config.Login, Password: b.Config.Password, }) if err != nil { flog.Debugf("%#v", err) return err } b.mc.SetCredentials(resp.UserID, resp.AccessToken) b.UserID = resp.UserID flog.Info("Connection succeeded") go b.handlematrix() return nil } func (b *Bmatrix) Disconnect() error { return nil } func (b *Bmatrix) JoinChannel(channel config.ChannelInfo) 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 err } func (b *Bmatrix) Send(msg config.Message) (string, error) { flog.Debugf("Receiving %#v", msg) channel := b.getRoomID(msg.Channel) // ignore delete messages if msg.Event == config.EVENT_MSG_DELETE { if msg.ID == "" { return "", nil } resp, err := b.mc.RedactEvent(channel, msg.ID, &matrix.ReqRedact{}) if err != nil { return "", err } return resp.EventID, err } flog.Debugf("Sending to channel %s", channel) if msg.Event == config.EVENT_USER_ACTION { resp, err := b.mc.SendMessageEvent(channel, "m.room.message", matrix.TextMessage{"m.emote", msg.Username + msg.Text}) if err != nil { return "", err } return resp.EventID, err } if msg.Extra != nil { // check if we have files to upload (from slack, telegram or mattermost) if len(msg.Extra["file"]) > 0 { for _, f := range msg.Extra["file"] { fi := f.(config.FileInfo) content := bytes.NewReader(*fi.Data) sp := strings.Split(fi.Name, ".") mtype := mime.TypeByExtension("." + sp[len(sp)-1]) if strings.Contains(mtype, "image") || strings.Contains(mtype, "video") { if fi.Comment != "" { resp, err := b.mc.SendText(channel, msg.Username+fi.Comment) if err != nil { flog.Errorf("file comment failed: %#v", err) } } flog.Debugf("uploading file: %s %s", fi.Name, mtype) res, err := b.mc.UploadToContentRepo(content, mtype, int64(len(*fi.Data))) if err != nil { flog.Errorf("file upload failed: %#v", err) continue } if strings.Contains(mtype, "video") { flog.Debugf("sendVideo %s", res.ContentURI) _, err = b.mc.SendVideo(channel, fi.Name, res.ContentURI) if err != nil { flog.Errorf("sendVideo failed: %#v", err) } } if strings.Contains(mtype, "image") { flog.Debugf("sendImage %s", res.ContentURI) _, err = b.mc.SendImage(channel, fi.Name, res.ContentURI) if err != nil { flog.Errorf("sendImage failed: %#v", err) } } flog.Debugf("result: %#v", res) } } return "", nil } } resp, err := b.mc.SendText(channel, msg.Username+msg.Text) if err != nil { return "", err } return resp.EventID, err } func (b *Bmatrix) getRoomID(channel string) string { b.RLock() defer b.RUnlock() for ID, name := range b.RoomMap { if name == channel { return ID } } return "" } func (b *Bmatrix) handlematrix() error { syncer := b.mc.Syncer.(*matrix.DefaultSyncer) syncer.OnEventType("m.room.redaction", b.handleEvent) syncer.OnEventType("m.room.message", b.handleEvent) go func() { for { if err := b.mc.Sync(); err != nil { flog.Println("Sync() returned ", err) } } }() return nil } func (b *Bmatrix) handleEvent(ev *matrix.Event) { flog.Debugf("Received: %#v", ev) if ev.Sender != b.UserID { b.RLock() channel, ok := b.RoomMap[ev.RoomID] b.RUnlock() if !ok { flog.Debugf("Unknown room %s", ev.RoomID) return } username := ev.Sender[1:] if b.Config.NoHomeServerSuffix { re := regexp.MustCompile("(.*?):.*") username = re.ReplaceAllString(username, `$1`) } var text string text, _ = ev.Content["body"].(string) rmsg := config.Message{Username: username, Text: text, Channel: channel, Account: b.Account, UserID: ev.Sender} rmsg.ID = ev.ID if ev.Type == "m.room.redaction" { rmsg.Event = config.EVENT_MSG_DELETE rmsg.ID = ev.Redacts rmsg.Text = config.EVENT_MSG_DELETE b.Remote <- rmsg return } if ev.Content["msgtype"].(string) == "m.emote" { rmsg.Event = config.EVENT_USER_ACTION } if ev.Content["msgtype"] != nil && ev.Content["msgtype"].(string) == "m.image" || ev.Content["msgtype"].(string) == "m.video" || ev.Content["msgtype"].(string) == "m.file" { flog.Debugf("ev: %#v", ev) rmsg.Extra = make(map[string][]interface{}) url := ev.Content["url"].(string) url = strings.Replace(url, "mxc://", b.Config.Server+"/_matrix/media/v1/download/", -1) info := ev.Content["info"].(map[string]interface{}) size := info["size"].(float64) name := ev.Content["body"].(string) // check if we have an image uploaded without extension if !strings.Contains(name, ".") { if ev.Content["msgtype"].(string) == "m.image" { if mtype, ok := ev.Content["mimetype"].(string); ok { mext, _ := mime.ExtensionsByType(mtype) if len(mext) > 0 { name = name + mext[0] } } else { // just a default .png extension if we don't have mime info name = name + ".png" } } } flog.Debugf("trying to download %#v with size %#v", name, size) if size <= float64(b.General.MediaDownloadSize) { data, err := helper.DownloadFile(url) if err != nil { flog.Errorf("download %s failed %#v", url, err) } else { flog.Debugf("download OK %#v %#v %#v", name, len(*data), len(url)) rmsg.Extra["file"] = append(rmsg.Extra["file"], config.FileInfo{Name: name, Data: data}) } } rmsg.Text = "" } flog.Debugf("Sending message from %s on %s to gateway", ev.Sender, b.Account) b.Remote <- rmsg } }