package appservice import ( "fmt" "strings" log "github.com/sirupsen/logrus" . "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector" ) type Account struct { MatrixUser string AccountName string Protocol string Conn Connector JoinedRooms map[RoomID]bool } var registeredAccounts = map[string]map[string]*Account{} func AddAccount(a *Account) { if _, ok := registeredAccounts[a.MatrixUser]; !ok { registeredAccounts[a.MatrixUser] = make(map[string]*Account) } registeredAccounts[a.MatrixUser][a.AccountName] = a ezbrSystemSendf(a.MatrixUser, "Connecting to account %s (%s)", a.AccountName, a.Protocol) } func FindAccount(mxUser string, name string) *Account { if u, ok := registeredAccounts[mxUser]; ok { if a, ok := u[name]; ok { return a } } return nil } func FindJoinedAccount(mxUser string, protocol string, room RoomID) *Account { if u, ok := registeredAccounts[mxUser]; ok { for _, acct := range u { if acct.Protocol == protocol { if j, ok := acct.JoinedRooms[room]; ok && j { return acct } } } } return nil } func RemoveAccount(mxUser string, name string) { if u, ok := registeredAccounts[mxUser]; ok { delete(u, name) } } func (a *Account) ezbrMessagef(format string, args ...interface{}) { msg := fmt.Sprintf(format, args...) msg = fmt.Sprintf("%s: %s", a.Protocol, msg) ezbrSystemSend(a.MatrixUser, msg) } // ---- Begin event handlers ---- func (a *Account) Joined(roomId RoomID) { err := a.joinedInternal(roomId) if err != nil { a.ezbrMessagef("Dropping Account.Joined %s: %s", roomId, err) } } func (a *Account) joinedInternal(roomId RoomID) error { a.JoinedRooms[roomId] = true mx_room_id, err := dbGetMxRoom(a.Protocol, roomId) if err != nil { return err } log.Debugf("Joined %s (%s)\n", roomId, a.MatrixUser) err = mxRoomInvite(mx_room_id, a.MatrixUser) if err != nil && strings.Contains(err.Error(), "already in the room") { err = nil } return err } // ---- func (a *Account) Left(roomId RoomID) { err := a.leftInternal(roomId) if err != nil { a.ezbrMessagef("Dropping Account.Left %s: %s", roomId, err) } } func (a *Account) leftInternal(roomId RoomID) error { delete(a.JoinedRooms, roomId) mx_room_id, err := dbGetMxRoom(a.Protocol, roomId) if err != nil { return err } log.Printf("Left %s (%s)\n", roomId, a.MatrixUser) err = mxRoomKick(mx_room_id, a.MatrixUser, fmt.Sprintf("got leave room event on %s", a.Protocol)) if err != nil && strings.Contains(err.Error(), "not in the room") { err = nil } return err } // ---- func (a *Account) UserInfoUpdated(user UserID, info *UserInfo) { err := a.userInfoUpdatedInternal(user, info) if err != nil { a.ezbrMessagef("Dropping Account.UserInfoUpdated %s: %s", user, err) } } func (a *Account) userInfoUpdatedInternal(user UserID, info *UserInfo) error { mx_user_id, err := dbGetMxUser(a.Protocol, user) if err != nil { return err } if info.DisplayName != "" { err2 := mxProfileDisplayname(mx_user_id, fmt.Sprintf("%s (%s)", info.DisplayName, a.Protocol)) if err2 != nil { err = err2 } } if info.Avatar != nil { err = fmt.Errorf("Avatar: not implemented") } return err } // ---- func (a *Account) RoomInfoUpdated(roomId RoomID, author UserID, info *RoomInfo) { err := a.roomInfoUpdatedInternal(roomId, author, info) if err != nil { a.ezbrMessagef("Dropping Account.RoomInfoUpdated %s: %s", roomId, err) } } func (a *Account) roomInfoUpdatedInternal(roomId RoomID, author UserID, info *RoomInfo) error { mx_room_id, err := dbGetMxRoom(a.Protocol, roomId) if err != nil { return err } as_mxid := ezbrMxId() if len(author) > 0 { mx_user_id, err2 := dbGetMxUser(a.Protocol, author) if err2 == nil { as_mxid = mx_user_id } } if info.Topic != "" { err2 := mxRoomTopicAs(mx_room_id, info.Topic, as_mxid) if err2 != nil { err = err2 } } if info.Name != "" { name := fmt.Sprintf("%s (%s)", info.Name, a.Protocol) err2 := mxRoomNameAs(mx_room_id, name, as_mxid) if err2 != nil { err = err2 } } if info.Picture != nil { // TODO err = fmt.Errorf("Picture: not implemented") } return err } // ---- func (a *Account) Event(event *Event) { err := a.eventInternal(event) if err != nil { a.ezbrMessagef("Dropping Account.Event %s %s %s: %s", event.Author, event.Recipient, event.Room, err) } } func (a *Account) eventInternal(event *Event) error { mx_user_id, err := dbGetMxUser(a.Protocol, event.Author) if err != nil { return err } if event.Type == EVENT_JOIN { log.Printf("%s join %s %s", a.Protocol, event.Author, event.Room) mx_room_id, err := dbGetMxRoom(a.Protocol, event.Room) if err != nil { return err } err = mxRoomInvite(mx_room_id, mx_user_id) if err != nil { if strings.Contains(err.Error(), "already in the room") { err = nil } return err } return mxRoomJoinAs(mx_room_id, mx_user_id) } else if event.Type == EVENT_LEAVE { log.Printf("%s join %s %s", a.Protocol, event.Author, event.Room) mx_room_id, err := dbGetMxRoom(a.Protocol, event.Room) if err != nil { return err } return mxRoomLeaveAs(mx_room_id, mx_user_id) } else { log.Printf("%s msg %s %s", a.Protocol, event.Author, event.Room) mx_room_id := "" if len(event.Room) > 0 { mx_room_id, err = dbGetMxRoom(a.Protocol, event.Room) if err != nil { return err } } else { mx_room_id, err = dbGetMxPmRoom(a.Protocol, event.Author, mx_user_id, a.MatrixUser, a.AccountName) if err != nil { return err } } typ := "m.text" if event.Type == EVENT_ACTION { typ = "m.emote" } return mxSendMessageAs(mx_room_id, typ, event.Text, mx_user_id) } }