This repository has been archived on 2020-09-27. You can view files and clone it, but cannot push or open issues or pull requests.
thrempp/component/threema/main.go

106 lines
2.6 KiB
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)
}