diff options
Diffstat (limited to 'teleirc/matterbridge/bridge/keybase')
| -rw-r--r-- | teleirc/matterbridge/bridge/keybase/handlers.go | 59 | ||||
| -rw-r--r-- | teleirc/matterbridge/bridge/keybase/keybase.go | 106 |
2 files changed, 165 insertions, 0 deletions
diff --git a/teleirc/matterbridge/bridge/keybase/handlers.go b/teleirc/matterbridge/bridge/keybase/handlers.go new file mode 100644 index 0000000..a29208d --- /dev/null +++ b/teleirc/matterbridge/bridge/keybase/handlers.go @@ -0,0 +1,59 @@ +package bkeybase + +import ( + "strconv" + + "github.com/42wim/matterbridge/bridge/config" + "github.com/keybase/go-keybase-chat-bot/kbchat/types/chat1" +) + +func (b *Bkeybase) handleKeybase() { + sub, err := b.kbc.ListenForNewTextMessages() + if err != nil { + b.Log.Errorf("Error listening: %s", err.Error()) + } + + go func() { + for { + msg, err := sub.Read() + if err != nil { + b.Log.Errorf("failed to read message: %s", err.Error()) + } + + if msg.Message.Content.TypeName != "text" { + continue + } + + if msg.Message.Sender.Username == b.kbc.GetUsername() { + continue + } + + b.handleMessage(msg.Message) + + } + }() +} + +func (b *Bkeybase) handleMessage(msg chat1.MsgSummary) { + b.Log.Debugf("== Receiving event: %#v", msg) + if msg.Channel.TopicName != b.channel || msg.Channel.Name != b.team { + return + } + + if msg.Sender.Username != b.kbc.GetUsername() { + + // TODO download avatar + + // Create our message + rmsg := config.Message{Username: msg.Sender.Username, Text: msg.Content.Text.Body, UserID: string(msg.Sender.Uid), Channel: msg.Channel.TopicName, ID: strconv.Itoa(int(msg.Id)), Account: b.Account} + + // Text must be a string + if msg.Content.TypeName != "text" { + b.Log.Errorf("message is not text") + return + } + + b.Log.Debugf("<= Sending message from %s on %s to gateway", msg.Sender.Username, msg.Channel.Name) + b.Remote <- rmsg + } +} diff --git a/teleirc/matterbridge/bridge/keybase/keybase.go b/teleirc/matterbridge/bridge/keybase/keybase.go new file mode 100644 index 0000000..d41e95f --- /dev/null +++ b/teleirc/matterbridge/bridge/keybase/keybase.go @@ -0,0 +1,106 @@ +package bkeybase + +import ( + "io/ioutil" + "os" + "path/filepath" + "strconv" + + "github.com/42wim/matterbridge/bridge" + "github.com/42wim/matterbridge/bridge/config" + "github.com/keybase/go-keybase-chat-bot/kbchat" +) + +// Bkeybase bridge structure +type Bkeybase struct { + kbc *kbchat.API + user string + channel string + team string + *bridge.Config +} + +// New initializes Bkeybase object and sets team +func New(cfg *bridge.Config) bridge.Bridger { + b := &Bkeybase{Config: cfg} + b.team = b.Config.GetString("Team") + return b +} + +// Connect starts keybase API and listener loop +func (b *Bkeybase) Connect() error { + var err error + b.Log.Infof("Connecting %s", b.GetString("Team")) + + // use default keybase location (`keybase`) + b.kbc, err = kbchat.Start(kbchat.RunOptions{}) + if err != nil { + return err + } + b.user = b.kbc.GetUsername() + b.Log.Info("Connection succeeded") + go b.handleKeybase() + return nil +} + +// Disconnect doesn't do anything for now +func (b *Bkeybase) Disconnect() error { + return nil +} + +// JoinChannel sets channel name in struct +func (b *Bkeybase) JoinChannel(channel config.ChannelInfo) error { + if _, err := b.kbc.JoinChannel(b.team, channel.Name); err != nil { + return err + } + b.channel = channel.Name + return nil +} + +// Send receives bridge messages and sends them to Keybase chat room +func (b *Bkeybase) Send(msg config.Message) (string, error) { + b.Log.Debugf("=> Receiving %#v", msg) + + // Handle /me events + if msg.Event == config.EventUserAction { + msg.Text = "_" + msg.Text + "_" + } + + // Delete message if we have an ID + // Delete message not supported by keybase go library yet + + // Edit message if we have an ID + // kbchat lib does not support message editing yet + + if len(msg.Extra["file"]) > 0 { + // Upload a file + dir, err := ioutil.TempDir("", "matterbridge") + if err != nil { + return "", err + } + defer os.RemoveAll(dir) + + for _, f := range msg.Extra["file"] { + fname := f.(config.FileInfo).Name + fdata := *f.(config.FileInfo).Data + fcaption := f.(config.FileInfo).Comment + fpath := filepath.Join(dir, fname) + + if err = ioutil.WriteFile(fpath, fdata, 0600); err != nil { + return "", err + } + + _, _ = b.kbc.SendAttachmentByTeam(b.team, &b.channel, fpath, fcaption) + } + + return "", nil + } + + // Send regular message + text := msg.Username + msg.Text + resp, err := b.kbc.SendMessageByTeamName(b.team, &b.channel, text) + if err != nil { + return "", err + } + return strconv.Itoa(int(*resp.Result.MessageID)), err +} |
