diff options
Diffstat (limited to 'server.go')
-rw-r--r-- | server.go | 259 |
1 files changed, 259 insertions, 0 deletions
diff --git a/server.go b/server.go new file mode 100644 index 0000000..10721be --- /dev/null +++ b/server.go @@ -0,0 +1,259 @@ +package main + +import ( + "encoding/json" + "fmt" + "net/http" + "strings" + + "github.com/gorilla/mux" + log "github.com/sirupsen/logrus" + + "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector" + "git.deuxfleurs.fr/Deuxfleurs/easybridge/mxlib" +) + +type Config struct { + HttpBindAddr string + Server string + DbType string + DbPath string + MatrixDomain string +} + +var mx *mxlib.Client + +func StartAppService() (chan error, error) { + mx = mxlib.NewClient(config.Server, registration.AsToken) + + err := InitDb() + if err != nil { + return nil, err + } + + err = mx.RegisterUser(registration.SenderLocalpart) + if mxe, ok := err.(*mxlib.MxError); !ok || mxe.ErrCode != "M_USER_IN_USE" { + return nil, err + } + if err == nil { + // If Easybridge account was created, update avatar and display name + err = mx.ProfileAvatar(ezbrMxId(), &connector.FileMediaObject{ + Path: "easybridge.jpg", + }) + if err != nil { + return nil, err + } + err = mx.ProfileDisplayname(ezbrMxId(), fmt.Sprintf("Easybridge (%s)", EASYBRIDGE_SYSTEM_PROTOCOL)) + if err != nil { + return nil, err + } + } + + router := mux.NewRouter() + router.HandleFunc("/_matrix/app/v1/transactions/{txnId}", handleTxn) + router.HandleFunc("/transactions/{txnId}", handleTxn) + + errch := make(chan error) + go func() { + log.Printf("Starting HTTP server on %s", config.HttpBindAddr) + err := http.ListenAndServe(config.HttpBindAddr, checkTokenAndLog(router)) + if err != nil { + errch <- err + } + }() + + return errch, nil +} + +func checkTokenAndLog(handler http.Handler) http.Handler { + return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + r.ParseForm() + if strings.Join(r.Form["access_token"], "") != registration.HsToken { + http.Error(w, "Wrong or no token provided", http.StatusUnauthorized) + return + } + + log.Printf("%s %s %s\n", r.RemoteAddr, r.Method, r.URL) + handler.ServeHTTP(w, r) + }) +} + +func handleTxn(w http.ResponseWriter, r *http.Request) { + if r.Method == "PUT" { + var txn mxlib.Transaction + err := json.NewDecoder(r.Body).Decode(&txn) + if err != nil { + http.Error(w, err.Error(), http.StatusBadRequest) + log.Warnf("JSON decode error: %s\n", err) + return + } + + log.Debugf("Got transaction %#v\n", txn) + + for i := range txn.Events { + ev := &txn.Events[i] + if strings.HasPrefix(ev.Sender, "@"+registration.SenderLocalpart) { + // Don't do anything with ezbr events that come back to us + continue + } + err = handleTxnEvent(ev) + if err != nil { + ezbrSystemSend(ev.Sender, fmt.Sprintf("Could not process %s (%s): %s", ev.Type, ev.Sender, err)) + } + } + + fmt.Fprintf(w, "{}\n") + } else { + http.Error(w, "Expected PUT request", http.StatusBadRequest) + } +} + +func handleTxnEvent(e *mxlib.Event) error { + if e.Type == "m.room.message" { + ev := &connector.Event{ + Type: connector.EVENT_MESSAGE, + Text: e.Content["body"].(string), + } + typ := e.Content["msgtype"].(string) + if typ == "m.emote" { + ev.Type = connector.EVENT_MESSAGE + } else if typ == "m.file" || typ == "m.image" { + ev.Text = "" + ev.Attachments = []connector.MediaObject{mx.ParseMediaInfo(e.Content)} + } + + if pm_room := dbIsPmRoom(e.RoomId); pm_room != nil { + if pm_room.Protocol == EASYBRIDGE_SYSTEM_PROTOCOL { + handleSystemMessage(e.Sender, e.Content["body"].(string)) + return nil + } + // If this is a private message room + acct := FindAccount(pm_room.MxUserID, pm_room.AccountName) + if acct == nil { + return fmt.Errorf("Not connected to %s", pm_room.AccountName) + } else if e.Sender == pm_room.MxUserID { + ev.Author = acct.Conn.User() + ev.Recipient = pm_room.UserID + return acct.Conn.Send(ev) + } + } else if room := dbIsPublicRoom(e.RoomId); room != nil { + // If this is a regular room + acct := FindJoinedAccount(e.Sender, room.Protocol, room.RoomID) + if acct != nil { + ev.Author = acct.Conn.User() + ev.Room = room.RoomID + return acct.Conn.Send(ev) + } else { + mx.RoomKick(e.RoomId, e.Sender, fmt.Sprintf("Not present in %s on %s, please talk with Easybridge to rejoin", room.RoomID, room.Protocol)) + return fmt.Errorf("not joined %s on %s", room.RoomID, room.Protocol) + } + } else { + return fmt.Errorf("Room not bridged") + } + } else if e.Type == "m.room.member" { + ms := e.Content["membership"].(string) + if ms == "leave" { + if pm_room := dbIsPmRoom(e.RoomId); pm_room != nil { + // If leaving a PM room, we must delete it + them_mx := userMxId(pm_room.Protocol, pm_room.UserID) + mx.RoomLeaveAs(e.RoomId, them_mx) + db.Delete(pm_room) + return nil + } else if room := dbIsPublicRoom(e.RoomId); room != nil { + // If leaving a public room, leave from server as well + acct := FindJoinedAccount(e.Sender, room.Protocol, room.RoomID) + if acct != nil { + acct.Conn.Leave(room.RoomID) + return nil + // TODO: manage autojoin list, remove this room + } else { + mx.RoomKick(e.RoomId, e.Sender, fmt.Sprintf("Not present in %s on %s, please talk with Easybridge to rejoin", room.RoomID, room.Protocol)) + return fmt.Errorf("not joined %s on %s", room.RoomID, room.Protocol) + } + } else { + return fmt.Errorf("Room not bridged") + } + } + } else if e.Type == "m.room.topic" { + if room := dbIsPublicRoom(e.RoomId); room != nil { + acct := FindJoinedAccount(e.Sender, room.Protocol, room.RoomID) + if acct != nil { + return acct.Conn.SetRoomInfo(room.RoomID, &connector.RoomInfo{ + Topic: e.Content["topic"].(string), + }) + } else { + return fmt.Errorf("Could not find room account for %s %s %s", e.Sender, room.Protocol, room.RoomID) + } + } + } + return nil +} + +func handleSystemMessage(mxid string, msg string) { + cmd := strings.Fields(msg) + switch cmd[0] { + case "help": + ezbrSystemSend(mxid, "Welcome to Easybridge! Here is a list of available commands:") + ezbrSystemSend(mxid, "- help: request help") + ezbrSystemSend(mxid, "- list: list accounts") + ezbrSystemSend(mxid, "- accounts: list accounts") + ezbrSystemSend(mxid, "- join <protocol or account> <room id>: join public chat room") + ezbrSystemSend(mxid, "- talk <protocol or account> <user id>: open private conversation to contact") + case "list", "account", "accounts": + one := false + if accts, ok := registeredAccounts[mxid]; ok { + for name, acct := range accts { + one = true + ezbrSystemSendf(mxid, "- %s (%s)", name, acct.Protocol) + } + } + if !one { + ezbrSystemSendf(mxid, "No account currently configured") + } + case "join": + account := findAccount(mxid, cmd[1]) + if account != nil { + err := account.Conn.Join(connector.RoomID(cmd[2])) + if err != nil { + ezbrSystemSendf(mxid, "%s", err) + } + } else { + ezbrSystemSendf(mxid, "No account with name or using protocol %s", cmd[1]) + } + case "talk": + account := findAccount(mxid, cmd[1]) + if account != nil { + quser := connector.UserID(cmd[2]) + err := account.Conn.Invite(quser, connector.RoomID("")) + if err != nil { + ezbrSystemSendf(mxid, "%s", err) + return + } + + quser_mxid, err := dbGetMxUser(account.Protocol, quser) + if err != nil { + ezbrSystemSendf(mxid, "%s", err) + return + } + _, err = dbGetMxPmRoom(account.Protocol, quser, quser_mxid, mxid, account.AccountName) + if err != nil { + ezbrSystemSendf(mxid, "%s", err) + } + } else { + ezbrSystemSendf(mxid, "No account with name or using protocol %s", cmd[1]) + } + default: + ezbrSystemSend(mxid, "Unrecognized command. Type `help` if you need some help!") + } +} + +func findAccount(mxid string, q string) *Account { + if accts, ok := registeredAccounts[mxid]; ok { + for name, acct := range accts { + if name == q || acct.Protocol == q { + return acct + } + } + } + return nil +} |