75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
package threema
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/bdlm/log"
|
|
"gosrc.io/xmpp"
|
|
|
|
"dev.sum7.eu/genofire/golang-lib/database"
|
|
|
|
"dev.sum7.eu/genofire/thrempp/component"
|
|
"dev.sum7.eu/genofire/thrempp/models"
|
|
)
|
|
|
|
type Threema struct {
|
|
component.Component
|
|
out chan xmpp.Packet
|
|
accountJID map[string]*Account
|
|
}
|
|
|
|
func NewThreema(config map[string]interface{}) (component.Component, error) {
|
|
return &Threema{
|
|
out: make(chan xmpp.Packet),
|
|
accountJID: make(map[string]*Account),
|
|
}, nil
|
|
}
|
|
|
|
func (t *Threema) Connect() (chan xmpp.Packet, error) {
|
|
var jids []*models.JID
|
|
database.Read.Find(&jids)
|
|
for _, jid := range jids {
|
|
a := t.getAccount(jid)
|
|
log.WithFields(map[string]interface{}{
|
|
"jid": jid.String(),
|
|
"threema": string(a.TID),
|
|
}).Info("connected")
|
|
}
|
|
return t.out, nil
|
|
}
|
|
func (t *Threema) Send(packet xmpp.Packet) {
|
|
switch p := packet.(type) {
|
|
case xmpp.Message:
|
|
from := models.ParseJID(p.PacketAttrs.From)
|
|
to := models.ParseJID(p.PacketAttrs.To)
|
|
|
|
if to.IsDomain() {
|
|
msg := xmpp.NewMessage("chat", "", from.String(), "", "en")
|
|
msg.Body = t.Bot(from, p.Body)
|
|
t.out <- msg
|
|
return
|
|
}
|
|
|
|
account := t.getAccount(from)
|
|
if account == nil {
|
|
msg := xmpp.NewMessage("chat", "", from.String(), "", "en")
|
|
msg.Body = "It was not possible to send, becouse we have no account for you.\nPlease generate one, by sending `generate` to this gateway"
|
|
t.out <- msg
|
|
return
|
|
}
|
|
|
|
threemaID := strings.ToUpper(to.Local)
|
|
if err := account.Send(threemaID, p); err != nil {
|
|
msg := xmpp.NewMessage("chat", "", from.String(), "", "en")
|
|
msg.Body = err.Error()
|
|
t.out <- msg
|
|
}
|
|
default:
|
|
log.Warnf("unkown package%v", p)
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
component.AddComponent("threema", NewThreema)
|
|
}
|