159 lines
3.4 KiB
Go
159 lines
3.4 KiB
Go
package appservice
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
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
|
|
}
|
|
|
|
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) Joined(roomId RoomID) {
|
|
a.JoinedRooms[roomId] = true
|
|
|
|
mx_room_id, err := dbGetMxRoom(a.Protocol, roomId)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
log.Printf("Joined %s (%s)\n", roomId, a.MatrixUser)
|
|
|
|
err = mxRoomInvite(mx_room_id, a.MatrixUser)
|
|
if err != nil {
|
|
log.Printf("Could not invite %s to %s", a.MatrixUser, mx_room_id)
|
|
}
|
|
}
|
|
|
|
func (a *Account) Left(roomId RoomID) {
|
|
delete(a.JoinedRooms, roomId)
|
|
|
|
mx_room_id, err := dbGetMxRoom(a.Protocol, roomId)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
log.Printf("Joined %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 {
|
|
log.Printf("Could not invite %s to %s", a.MatrixUser, mx_room_id)
|
|
}
|
|
}
|
|
|
|
func (a *Account) UserInfoUpdated(user UserID, info *UserInfo) {
|
|
// TODO
|
|
}
|
|
|
|
func (a *Account) RoomInfoUpdated(roomId RoomID, info *RoomInfo) {
|
|
// TODO
|
|
}
|
|
|
|
func (a *Account) Event(event *Event) {
|
|
mx_user_id, err := dbGetMxUser(a.Protocol, event.Author)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
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 = mxRoomInvite(mx_room_id, mx_user_id)
|
|
if err != nil {
|
|
log.Printf("Could not invite %s to %s", a.MatrixUser, mx_room_id)
|
|
}
|
|
|
|
err = mxRoomJoinAs(mx_room_id, mx_user_id)
|
|
if err != nil {
|
|
log.Printf("Could not join %s as %s", a.MatrixUser, mx_room_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 = mxRoomLeaveAs(mx_room_id, mx_user_id)
|
|
if err != nil {
|
|
log.Printf("Could not leave %s as %s", a.MatrixUser, mx_room_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
|
|
}
|
|
} else {
|
|
mx_room_id, err = dbGetMxPmRoom(a.Protocol, event.Author, mx_user_id, a.MatrixUser, a.AccountName)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
typ := "m.text"
|
|
if event.Type == EVENT_ACTION {
|
|
typ = "m.emote"
|
|
}
|
|
|
|
err = mxSendMessageAs(mx_room_id, typ, event.Text, mx_user_id)
|
|
if err != nil {
|
|
log.Printf("Could not send %s as %s", event.Text, mx_user_id)
|
|
}
|
|
}
|
|
}
|
|
|