Support for external processes; stub FB messenger bridge

This commit is contained in:
Alex 2020-02-29 18:30:43 +01:00
parent 2649e41c85
commit 7431469632
12 changed files with 729 additions and 28 deletions

View file

@ -2,10 +2,12 @@ BIN=easybridge
DOCKER=lxpz/easybridge_amd64
SRC= mxlib/registration.go mxlib/api.go mxlib/mediaobject.go mxlib/client.go \
connector/connector.go connector/config.go connector/mediaobject.go \
connector/connector.go connector/config.go \
connector/mediaobject.go connector/marshal.go \
connector/irc/irc.go connector/irc/config.go \
connector/xmpp/config.go connector/xmpp/xmpp.go \
connector/mattermost/mattermost.go connector/mattermost/config.go \
connector/external/external.go connector/external/config.go \
web.go account.go main.go server.go db.go util.go
all: $(BIN)

View file

@ -11,6 +11,7 @@ import (
. "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector"
// Necessary for them to register their protocols
_ "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector/external"
_ "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector/irc"
_ "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector/mattermost"
_ "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector/xmpp"
@ -292,7 +293,7 @@ func (a *Account) userInfoUpdatedInternal(user UserID, info *UserInfo) error {
}
}
if info.Avatar != nil {
if MediaObject(info.Avatar) != nil {
cache_key := fmt.Sprintf("%s/user_avatar/%s", a.Protocol, user)
cache_val := info.Avatar.Filename()
if cache_val == "" || dbKvTestAndSet(cache_key, cache_val) {
@ -346,7 +347,7 @@ func (a *Account) roomInfoUpdatedInternal(roomId RoomID, author UserID, info *Ro
}
}
if info.Picture != nil {
if MediaObject(info.Picture) != nil {
cache_key := fmt.Sprintf("%s/room_picture/%s", a.Protocol, roomId)
cache_val := info.Picture.Filename()
if cache_val == "" || dbKvTestAndSet(cache_key, cache_val) {

View file

@ -109,48 +109,48 @@ const (
)
type Event struct {
Type EventType
Type EventType `json:"type"`
// If non-empty, the event Id is used to deduplicate events in a channel
// This is usefull for backends that provide a backlog of channel messages
// when (re-)joining a room
Id string
Id string `json:"id"`
// UserID of the user that sent the event
// If this is a direct message event, this event can only have been authored
// by the user we are talking to (and not by ourself)
Author UserID
Author UserID `json:"author"`
// UserID of the targetted user in the case of a direct message,
// empty if targetting a room
Recipient UserID
Recipient UserID `json:"recipient"`
// RoomID of the room where the event happenned or of the targetted room,
// or empty string if it happenned by direct message
Room RoomID
Room RoomID `json:"room"`
// Message text or action text
Text string
Text string `json:"text`
// Attached files such as images
Attachments []MediaObject
Attachments []SMediaObject `json:"attachments"`
}
type UserInfo struct {
DisplayName string
DisplayName string `json:"display_name"`
// If non-empty, the Filename of the avatar object will be used by Easybridge
// to deduplicate the update events and prevent needless reuploads.
// Example strategy that works for the mattermost backend: use the update timestamp as fictious file name
Avatar MediaObject
Avatar SMediaObject `json:"avatar"`
}
type RoomInfo struct {
Name string
Topic string
Name string `json:"name"`
Topic string `json:"topic"`
// Same deduplication comment as for UserInfo.Avatar
Picture MediaObject
Picture SMediaObject `json:"picture"`
}
type MediaObject interface {
@ -169,7 +169,11 @@ type MediaObject interface {
URL() string
}
type ImageSize struct {
Width int
Height int
type SMediaObject struct {
MediaObject
}
type ImageSize struct {
Width int `json:"width"`
Height int `json:"height"`
}

50
connector/external/config.go vendored Normal file
View file

@ -0,0 +1,50 @@
package external
import (
. "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector"
)
const DUMMYEXT_PROTOCOL = "DummyExt"
const MESSENGER_PROTOCOL = "Messenger"
func init() {
Register(DUMMYEXT_PROTOCOL, Protocol{
NewConnector: func() Connector {
return &External{
protocol: DUMMYEXT_PROTOCOL,
command: "./external/dummy.py",
debug: true,
}
},
Schema: ConfigSchema{
&ConfigEntry{
Name: "user",
Description: "Username",
Required: true,
},
},
})
Register(MESSENGER_PROTOCOL, Protocol{
NewConnector: func() Connector {
return &External{
protocol: MESSENGER_PROTOCOL,
command: "./external/messenger.py",
debug: true,
}
},
Schema: ConfigSchema{
&ConfigEntry{
Name: "email",
Description: "Email address",
Required: true,
},
&ConfigEntry{
Name: "password",
Description: "Password",
IsPassword: true,
Required: true,
},
},
})
}

387
connector/external/external.go vendored Normal file
View file

@ -0,0 +1,387 @@
package external
import (
"encoding/json"
"fmt"
"io"
"os"
"os/exec"
"strings"
"sync"
"sync/atomic"
"time"
log "github.com/sirupsen/logrus"
. "git.deuxfleurs.fr/Deuxfleurs/easybridge/connector"
)
// Serialization protocol
type extMessage struct {
// Header: message type and identifier
MsgType string `json:"_type"`
MsgId uint64 `json:"_id"`
// Message fields
Key string `json:"key"`
Value string `json:"value"`
EventId string `json:"event_id"`
Error string `json:"error"`
Room RoomID `json:"room"`
User UserID `json:"user"`
}
type extMessageWithData struct {
extMessage
Data interface{} `json:"data"`
}
// Possible values for MsgType
const (
// ezbr -> external
CONFIGURE = "configure"
GET_USER = "get_user"
SET_USER_INFO = "set_user_info"
SET_ROOM_INFO = "set_room_info"
JOIN = "join"
INVITE = "invite"
LEAVE = "leave"
SEND = "send"
CLOSE = "close"
// external -> ezbr
JOINED = "joined"
LEFT = "left"
USER_INFO_UPDATED = "user_info_updated"
ROOM_INFO_UPDATED = "room_info_updated"
EVENT = "event"
CACHE_PUT = "cache_put"
CACHE_GET = "cache_get"
// reply messages
// ezbr -> external: all must wait for a reply!
// external -> ezbr: only CACHE_GET produces a reply
REP_OK = "rep_ok"
REP_ERROR = "rep_error"
)
// ----
type External struct {
handler Handler
protocol string
command string
debug bool
config Configuration
recv io.Reader
send io.Writer
sendJson *json.Encoder
generation int
proc *exec.Cmd
counter uint64
inflightRequests map[uint64]chan *extMessageWithData
lock sync.Mutex
}
func (ext *External) SetHandler(h Handler) {
ext.handler = h
}
func (ext *External) Protocol() string {
return ext.protocol
}
func (ext *External) Configure(c Configuration) error {
var err error
if ext.proc != nil {
ext.Close()
}
ext.inflightRequests = map[uint64]chan *extMessageWithData{}
ext.generation += 1
err = ext.setupProc()
if err != nil {
return err
}
go ext.restartLoop(ext.generation)
_, err = ext.cmd(extMessage{
MsgType: CONFIGURE,
}, c)
if err != nil {
return err
}
return nil
}
// ---- Process management and communication logic
func (ext *External) setupProc() error {
var err error
ext.proc = exec.Command(ext.command)
ext.recv, err = ext.proc.StdoutPipe()
if err != nil {
return err
}
ext.send, err = ext.proc.StdinPipe()
if err != nil {
return err
}
if ext.debug {
ext.recv = io.TeeReader(ext.recv, os.Stderr)
ext.send = io.MultiWriter(ext.send, os.Stderr)
}
ext.sendJson = json.NewEncoder(ext.send)
ext.proc.Stderr = os.Stderr
err = ext.proc.Start()
if err != nil {
return err
}
go ext.recvLoop()
return nil
}
func (ext *External) restartLoop(generation int) {
for {
if ext.proc == nil {
break
}
ext.proc.Wait()
if ext.generation != generation {
break
}
log.Printf("Process %s stopped, restarting.", ext.command)
err := ext.setupProc()
if err != nil {
ext.proc = nil
log.Warnf("Unable to restart %s: %s", ext.command, err)
break
}
}
}
func (m *extMessageWithData) DecodeJSON(jj []byte) error {
var c extMessage
err := json.Unmarshal(jj, &c)
if err != nil {
return err
}
*m = extMessageWithData{extMessage: c}
switch c.MsgType {
case USER_INFO_UPDATED:
var ui UserInfo
err := json.Unmarshal(jj, &ui)
if err != nil {
return err
}
m.Data = &ui
case ROOM_INFO_UPDATED:
var ri RoomInfo
err := json.Unmarshal(jj, &ri)
if err != nil {
return err
}
m.Data = &ri
case EVENT:
var ev Event
err := json.Unmarshal(jj, &ev)
if err != nil {
return err
}
m.Data = &ev
}
return nil
}
func (ext *External) recvLoop() {
reader := json.NewDecoder(ext.recv)
for {
var msg extMessageWithData
err := reader.Decode(&msg)
if err != nil {
log.Warnf("Failed to decode from %s: %s. Stopping reading.", ext.command, err)
break
}
if strings.HasPrefix(msg.MsgType, "rep_") {
func() {
ext.lock.Lock()
defer ext.lock.Unlock()
if ch, ok := ext.inflightRequests[msg.MsgId]; ok {
ch <- &msg
delete(ext.inflightRequests, msg.MsgId)
}
}()
} else {
ext.handleCmd(&msg)
}
}
}
func (ext *External) cmd(msg extMessage, data interface{}) (*extMessageWithData, error) {
msg_id := atomic.AddUint64(&ext.counter, 1)
msg.MsgId = msg_id
fullMsg := extMessageWithData{
extMessage: msg,
Data: data,
}
ch := make(chan *extMessageWithData)
func() {
ext.lock.Lock()
defer ext.lock.Unlock()
ext.inflightRequests[msg_id] = ch
}()
defer func() {
ext.lock.Lock()
defer ext.lock.Unlock()
delete(ext.inflightRequests, msg_id)
}()
err := ext.sendJson.Encode(&fullMsg)
if err != nil {
return nil, err
}
select {
case rep := <-ch:
if rep.MsgType == REP_ERROR {
return nil, fmt.Errorf("%s: %s", msg.MsgType, rep.Error)
} else {
return rep, nil
}
case <-time.After(5 * time.Second):
return nil, fmt.Errorf("(%s) timeout", msg.MsgType)
}
}
func (ext *External) Close() {
ext.sendJson.Encode(&extMessage{
MsgType: CLOSE,
})
ext.generation += 1
proc := ext.proc
ext.proc = nil
ext.recv = nil
ext.send = nil
ext.sendJson = nil
go func() {
time.Sleep(10 * time.Second)
proc.Process.Kill()
}()
}
// ---- Actual message handling :)
func (ext *External) handleCmd(msg *extMessageWithData) {
switch msg.MsgType {
case JOINED:
ext.handler.Joined(msg.Room)
case LEFT:
ext.handler.Left(msg.Room)
case USER_INFO_UPDATED:
ext.handler.UserInfoUpdated(msg.User, msg.Data.(*UserInfo))
case ROOM_INFO_UPDATED:
ext.handler.RoomInfoUpdated(msg.Room, msg.User, msg.Data.(*RoomInfo))
case EVENT:
ext.handler.Event(msg.Data.(*Event))
case CACHE_PUT:
ext.handler.CachePut(msg.Key, msg.Value)
case CACHE_GET:
value := ext.handler.CacheGet(msg.Key)
ext.sendJson.Encode(&extMessage{
MsgType: REP_OK,
MsgId: msg.MsgId,
Value: value,
})
}
}
func (ext *External) User() UserID {
rep, err := ext.cmd(extMessage{
MsgType: GET_USER,
}, nil)
if err != nil {
log.Warnf("Unable to get user!")
return ""
}
return rep.User
}
func (ext *External) SetUserInfo(info *UserInfo) error {
_, err := ext.cmd(extMessage{
MsgType: SET_USER_INFO,
}, info)
return err
}
func (ext *External) SetRoomInfo(room RoomID, info *RoomInfo) error {
_, err := ext.cmd(extMessage{
MsgType: SET_ROOM_INFO,
Room: room,
}, info)
return err
}
func (ext *External) Join(room RoomID) error {
_, err := ext.cmd(extMessage{
MsgType: JOIN,
Room: room,
}, nil)
return err
}
func (ext *External) Invite(user UserID, room RoomID) error {
_, err := ext.cmd(extMessage{
MsgType: LEAVE,
User: user,
Room: room,
}, nil)
return err
}
func (ext *External) Leave(room RoomID) {
_, err := ext.cmd(extMessage{
MsgType: LEAVE,
Room: room,
}, nil)
if err != nil {
log.Warnf("Could not leave %s: %s", room, err.Error())
}
}
func (ext *External) Send(event *Event) (string, error) {
rep, err := ext.cmd(extMessage{
MsgType: SEND,
}, event)
if err != nil {
return "", err
}
return rep.EventId, nil
}

View file

@ -142,7 +142,7 @@ func (irc *IRC) SetRoomInfo(roomId RoomID, info *RoomInfo) error {
if info.Name != "" && info.Name != ch {
return fmt.Errorf("May not change IRC room name to other than %s", ch)
}
if info.Picture != nil {
if MediaObject(info.Picture) != nil {
return fmt.Errorf("Room picture not supported on IRC")
}
return nil

99
connector/marshal.go Normal file
View file

@ -0,0 +1,99 @@
package connector
import (
"bytes"
"encoding/base64"
"encoding/json"
"fmt"
"io"
)
func (t EventType) MarshalText() ([]byte, error) {
switch t {
case EVENT_JOIN:
return []byte("join"), nil
case EVENT_LEAVE:
return []byte("leave"), nil
case EVENT_MESSAGE:
return []byte("message"), nil
case EVENT_ACTION:
return []byte("action"), nil
default:
return nil, fmt.Errorf("Invalid event type: %d", t)
}
}
func (t *EventType) UnmarshalText(text []byte) error {
switch string(text) {
case "join":
*t = EVENT_JOIN
return nil
case "leave":
*t = EVENT_LEAVE
return nil
case "message":
*t = EVENT_MESSAGE
return nil
case "action":
*t = EVENT_ACTION
return nil
default:
return fmt.Errorf("Invalid event type: %s", string(text))
}
}
// ----
type MediaObjectJSON struct {
Filename string `json:"filename"`
Mimetype string `json:"mime_type"`
ImageSize *ImageSize `json:"image_size"`
Data string `json:"data"`
}
func (mo SMediaObject) MarshalJSON() ([]byte, error) {
if MediaObject(mo) == nil {
return []byte("null"), nil
}
mod := MediaObjectJSON{
Filename: mo.Filename(),
Mimetype: mo.Mimetype(),
ImageSize: mo.ImageSize(),
}
rd, err := mo.Read()
if err != nil {
return nil, err
}
defer rd.Close()
buf := bytes.NewBuffer([]byte{})
_, err = io.Copy(buf, rd)
if err != nil {
return nil, err
}
mod.Data = base64.StdEncoding.EncodeToString(buf.Bytes())
return json.Marshal(&mod)
}
func (mo *SMediaObject) UnmarshalJSON(jdata []byte) error {
if string(jdata) == "null" {
return nil
}
var d MediaObjectJSON
err := json.Unmarshal(jdata, &d)
if err != nil {
return err
}
bytes, err := base64.StdEncoding.DecodeString(d.Data)
if err != nil {
return err
}
*mo = SMediaObject{&BlobMediaObject{
ObjectFilename: d.Filename,
ObjectMimetype: d.Mimetype,
ObjectImageSize: d.ImageSize,
ObjectData: bytes,
}}
return nil
}

View file

@ -210,7 +210,7 @@ func (mm *Mattermost) SetRoomInfo(roomId RoomID, info *RoomInfo) error {
mm.conn.UpdateChannelHeader(ch, info.Topic)
}
if info.Picture != nil {
if MediaObject(info.Picture) != nil {
err = fmt.Errorf("Not supported: channel picture on mattermost")
}
@ -374,7 +374,7 @@ func (mm *Mattermost) initSyncChannel(ch *model.Channel) {
room_info.Name = t.Team.Name + " / " + room_info.Name
}
if t.Team.LastTeamIconUpdate > 0 {
room_info.Picture = &LazyBlobMediaObject{
room_info.Picture = SMediaObject{&LazyBlobMediaObject{
ObjectFilename: fmt.Sprintf("%s-%d",
t.Team.Name,
t.Team.LastTeamIconUpdate),
@ -388,7 +388,7 @@ func (mm *Mattermost) initSyncChannel(ch *model.Channel) {
o.ObjectMimetype = http.DetectContentType(team_img)
return nil
},
}
}}
}
break
}
@ -487,7 +487,7 @@ func (mm *Mattermost) updateUserInfo(user *model.User) {
DisplayName: userDisp,
}
if user.LastPictureUpdate > 0 {
ui.Avatar = &LazyBlobMediaObject{
ui.Avatar = SMediaObject{&LazyBlobMediaObject{
ObjectFilename: fmt.Sprintf("%s-%d",
user.Username,
user.LastPictureUpdate),
@ -501,7 +501,7 @@ func (mm *Mattermost) updateUserInfo(user *model.User) {
o.ObjectMimetype = http.DetectContentType(img)
return nil
},
}
}}
}
mm.handler.UserInfoUpdated(userId, ui)
mm.caches.displayname[userId] = userDisp
@ -567,7 +567,7 @@ func (mm *Mattermost) handlePost(channel_name string, post *model.Post, only_mes
// Handle files
if post.FileIds != nil && len(post.FileIds) > 0 {
msg_ev.Attachments = []MediaObject{}
msg_ev.Attachments = []SMediaObject{}
for _, file := range post.Metadata.Files {
media_object := &LazyBlobMediaObject{
ObjectFilename: file.Name,
@ -587,7 +587,7 @@ func (mm *Mattermost) handlePost(channel_name string, post *model.Post, only_mes
Height: file.Height,
}
}
msg_ev.Attachments = append(msg_ev.Attachments, media_object)
msg_ev.Attachments = append(msg_ev.Attachments, SMediaObject{media_object})
}
}

View file

@ -272,7 +272,7 @@ func (xm *XMPP) SetRoomInfo(roomId RoomID, info *RoomInfo) error {
}
}
if info.Picture != nil {
if MediaObject(info.Picture) != nil {
// TODO
return fmt.Errorf("Room picture change not implemented on xmpp")
}

38
external/dummy.py vendored Executable file
View file

@ -0,0 +1,38 @@
#!/usr/bin/env python3
import sys
import json
sys.stderr.write("(python) dummy starting")
sys.stderr.flush()
if __name__ == "__main__":
username = ""
while True:
line = sys.stdin.readline()
sys.stderr.write("(python) got: {}\n".format(line))
sys.stderr.flush()
cmd = json.loads(line)
reply = {
"_type": "rep_ok",
"_id": cmd["_id"],
}
if cmd["_type"] == "configure":
username = cmd["data"]["user"]
if cmd["_type"] == "get_user":
reply["user"] = username
repline = json.dumps(reply)
sys.stderr.write("(python) sending: {}\n".format(repline))
sys.stderr.flush()
sys.stdout.write(repline + "\n")
sys.stdout.flush()
if cmd["_type"] == "close":
break
sys.stderr.write("(python) dummy stopping")
sys.stderr.flush()

118
external/messenger.py vendored Executable file
View file

@ -0,0 +1,118 @@
#!/usr/bin/env python3
import sys
import json
import hashlib
import gevent
import fbchat
# ---- MESSAGE TYPES ----
# ezbr -> external
CONFIGURE = "configure"
GET_USER = "get_user"
SET_USER_INFO = "set_user_info"
SET_ROOM_INFO = "set_room_info"
JOIN = "join"
INVITE = "invite"
LEAVE = "leave"
SEND = "send"
CLOSE = "close"
# external -> ezbr
JOINED = "joined"
LEFT = "left"
USER_INFO_UPDATED = "user_info_updated"
ROOM_INFO_UPDATED = "room_info_updated"
EVENT = "event"
CACHE_PUT = "cache_put"
CACHE_GET = "cache_get"
# reply messages
# ezbr -> external: all must wait for a reply!
# external -> ezbr: only CACHE_GET produces a reply
REP_OK = "rep_ok"
REP_ERROR = "rep_error"
# ---- MESSENGER CLIENT CLASS THAT HANDLES EVENTS ----
class MessengerBridgeClient(fbchat.Client):
def __init__(self, bridge, *args, **kwargs):
self.bridge = bridge
super(MessengerBridgeClient, self).__init__(*args, **kwargs)
# TODO: handle events
# ---- MAIN LOOP THAT HANDLES REQUESTS FROM BRIDGE ----
class MessengerBridge:
def run(self):
self.client = None
self.keep_running = True
while self.keep_running:
line = sys.stdin.readline()
sys.stderr.write("(python) reading {}\n".format(line.strip()))
cmd = json.loads(line)
try:
rep = self.handle_cmd(cmd)
if rep is None:
rep = {}
if "_type" not in rep:
rep["_type"] = REP_OK
except Exception as e:
rep = {
"_type": REP_ERROR,
"error": "{}".format(e)
}
rep["_id"] = cmd["_id"]
self.write(rep)
def write(self, msg):
msgstr = json.dumps(msg)
sys.stderr.write("(python) writing {}\n".format(msgstr))
sys.stdout.write(msgstr + "\n")
sys.stdout.flush()
def handle_cmd(self, cmd):
ty = cmd["_type"]
if ty == CONFIGURE:
cookies_file = "/tmp/cookies_" + hashlib.sha224(cmd["data"]["email"].encode("utf-8")).hexdigest()
try:
f = open(cookies_file, "r")
cookies = json.load(f)
f.close()
sys.stderr.write("(python messenger) using previous cookies: {}\n".format(cookies))
except:
cookies = None
self.client = MessengerBridgeClient(self, cmd["data"]["email"], cmd["data"]["password"], session_cookies=cookies)
if self.client.isLoggedIn():
cookies = self.client.getSession()
try:
f = open(cookies_file, "w")
json.dump(cookies, f)
f.close()
except:
pass
elif ty == CLOSE:
self.client.logout()
self.keep_running = False
else:
return {"_type": REP_ERROR, "error": "Not implemented"}
if __name__ == "__main__":
bridge = MessengerBridge()
bridge.run()

View file

@ -130,7 +130,9 @@ func handleTxnEvent(e *mxlib.Event) error {
ev.Type = connector.EVENT_MESSAGE
} else if typ == "m.file" || typ == "m.image" {
ev.Text = ""
ev.Attachments = []connector.MediaObject{mx.ParseMediaInfo(e.Content)}
ev.Attachments = []connector.SMediaObject{
connector.SMediaObject{mx.ParseMediaInfo(e.Content)},
}
}
if pm_room := dbIsPmRoom(e.RoomId); pm_room != nil {