42wim/matterbridge

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
}