thrempp/component/threema/main.go

package threema

import (
	"errors"
	"strings"

	"github.com/bdlm/log"
	"gosrc.io/xmpp/stanza"

	"dev.sum7.eu/genofire/golang-lib/database"

	"dev.sum7.eu/sum7/thrempp/component"
	"dev.sum7.eu/sum7/thrempp/models"
)

type Threema struct {
	component.Component
	out            chan stanza.Packet
	accountJID     map[string]*Account
	bot            map[string]*Bot
	httpUploadPath string
	httpUploadURL  string
}

func NewThreema(config map[string]interface{}) (component.Component, error) {
	t := &Threema{
		out:        make(chan stanza.Packet),
		accountJID: make(map[string]*Account),
		bot:        make(map[string]*Bot),
	}
	if pathI, ok := config["http_upload_path"]; ok {
		if path, ok := pathI.(string); ok {
			t.httpUploadPath = path
		} else {
			return nil, errors.New("wrong format of http_upload_path")
		}
	}
	if urlI, ok := config["http_upload_url"]; ok {
		if url, ok := urlI.(string); ok {
			t.httpUploadURL = url
		} else {
			return nil, errors.New("wrong format of http_upload_url")
		}
	}
	return t, nil
}

func (t *Threema) Connect() (chan stanza.Packet, error) {
	var jids []*models.JID
	database.Read.Find(&jids)
	for _, jid := range jids {
		logger := log.WithField("jid", jid.String())
		a, err := t.getAccount(jid)
		if err != nil {
			logger.Warnf("unable to connect%s", err)
			continue
		}
		logger = logger.WithField("threema", string(a.TID))
		logger.Info("connected")
	}
	return t.out, nil
}
func (t *Threema) Send(packet stanza.Packet) {
	if p := t.send(packet); p != nil {
		t.out <- p
	}
}
func (t *Threema) send(packet stanza.Packet) stanza.Packet {
	switch p := packet.(type) {
	case stanza.Message:
		from := models.ParseJID(p.Attrs.From)
		to := models.ParseJID(p.Attrs.To)

		if to.IsDomain() {
			if from == nil {
				log.Warn("receive message without sender")
				return nil
			}
			msg := stanza.NewMessage(stanza.Attrs{Type: stanza.MessageTypeChat, To: from.String()})
			msg.Body = t.getBot(from).Handle(p.Body)
			return msg
		}

		account, err := t.getAccount(from)
		if err != nil {
			msg := stanza.NewMessage(stanza.Attrs{Type: stanza.MessageTypeChat, To: from.String()})
			msg.Body = "It was not possible to send, because we have no account for you.\nPlease generate one, by sending `generate` to this gateway"
			return msg
		}

		threemaID := strings.ToUpper(to.Local)
		if err := account.Send(threemaID, p); err != nil {
			msg := stanza.NewMessage(stanza.Attrs{Type: stanza.MessageTypeChat, To: from.String()})
			msg.Body = err.Error()
			return msg
		}
	default:
		log.Warnf("unknown package: %v", p)
	}
	return nil
}

func init() {
	component.AddComponent("threema", NewThreema)
}