add delivery support
This commit is contained in:
parent
0f61e08e1c
commit
e0d2232aeb
|
@ -17,6 +17,7 @@ type Account struct {
|
||||||
Session o3.SessionContext
|
Session o3.SessionContext
|
||||||
send chan<- o3.Message
|
send chan<- o3.Message
|
||||||
recieve <-chan o3.ReceivedMsg
|
recieve <-chan o3.ReceivedMsg
|
||||||
|
deliveredMSG map[uint64]string
|
||||||
}
|
}
|
||||||
|
|
||||||
func (t *Threema) getAccount(jid *models.JID) *Account {
|
func (t *Threema) getAccount(jid *models.JID) *Account {
|
||||||
|
@ -44,6 +45,7 @@ func (t *Threema) getAccount(jid *models.JID) *Account {
|
||||||
a.XMPP = *jid
|
a.XMPP = *jid
|
||||||
a.Session = o3.NewSessionContext(tid)
|
a.Session = o3.NewSessionContext(tid)
|
||||||
a.send, a.recieve, err = a.Session.Run()
|
a.send, a.recieve, err = a.Session.Run()
|
||||||
|
a.deliveredMSG = make(map[uint64]string)
|
||||||
|
|
||||||
// TODO error handling
|
// TODO error handling
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
@ -53,7 +55,6 @@ func (t *Threema) getAccount(jid *models.JID) *Account {
|
||||||
go a.reciever(t.out)
|
go a.reciever(t.out)
|
||||||
|
|
||||||
t.accountJID[jid.String()] = a
|
t.accountJID[jid.String()] = a
|
||||||
t.accountTID[string(a.TID)] = a
|
|
||||||
return a
|
return a
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -71,14 +72,55 @@ func (a *Account) reciever(out chan<- xmpp.Packet) {
|
||||||
}
|
}
|
||||||
xMSG := xmpp.NewMessage("chat", sender, a.XMPP.String(), strconv.FormatUint(msg.ID(), 10), "en")
|
xMSG := xmpp.NewMessage("chat", sender, a.XMPP.String(), strconv.FormatUint(msg.ID(), 10), "en")
|
||||||
xMSG.Body = msg.Text()
|
xMSG.Body = msg.Text()
|
||||||
|
xMSG.Extensions = append(xMSG.Extensions, xmpp.ReceiptRequest{})
|
||||||
out <- xMSG
|
out <- xMSG
|
||||||
case o3.DeliveryReceiptMessage:
|
case o3.DeliveryReceiptMessage:
|
||||||
// msg.MsgID()
|
if id, ok := a.deliveredMSG[msg.MsgID()]; ok {
|
||||||
|
xMSG := xmpp.NewMessage("chat", msg.Sender().String(), a.XMPP.String(), "", "en")
|
||||||
|
log.Warnf("found id %s", id)
|
||||||
|
xMSG.Extensions = append(xMSG.Extensions, xmpp.ReceiptReceived{
|
||||||
|
Id: id,
|
||||||
|
})
|
||||||
|
out <- xMSG
|
||||||
|
delete(a.deliveredMSG, msg.MsgID())
|
||||||
|
} else {
|
||||||
|
log.Warnf("found not id in cache to announce received on xmpp side")
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *Account) Send(to string, msg string) error {
|
func (a *Account) Send(to string, msg xmpp.Message) error {
|
||||||
return a.Session.SendTextMessage(to, msg, a.send)
|
reci := ""
|
||||||
|
for _, el := range msg.Extensions {
|
||||||
|
switch ex := el.(type) {
|
||||||
|
case *xmpp.ReceiptReceived:
|
||||||
|
reci = ex.Id
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if reci != "" {
|
||||||
|
id, err := strconv.ParseUint(reci, 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
drm, err := o3.NewDeliveryReceiptMessage(&a.Session, to, id, o3.MSGDELIVERED)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.send <- drm
|
||||||
|
log.WithFields(map[string]interface{}{
|
||||||
|
"tid": to,
|
||||||
|
"msg_id": id,
|
||||||
|
}).Debug("delivered")
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
msg3, err := o3.NewTextMessage(&a.Session, to, msg.Body)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
a.deliveredMSG[msg3.ID()] = msg.Id
|
||||||
|
a.send <- msg3
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
|
@ -16,14 +16,12 @@ type Threema struct {
|
||||||
component.Component
|
component.Component
|
||||||
out chan xmpp.Packet
|
out chan xmpp.Packet
|
||||||
accountJID map[string]*Account
|
accountJID map[string]*Account
|
||||||
accountTID map[string]*Account
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewThreema(config map[string]interface{}) (component.Component, error) {
|
func NewThreema(config map[string]interface{}) (component.Component, error) {
|
||||||
return &Threema{
|
return &Threema{
|
||||||
out: make(chan xmpp.Packet),
|
out: make(chan xmpp.Packet),
|
||||||
accountJID: make(map[string]*Account),
|
accountJID: make(map[string]*Account),
|
||||||
accountTID: make(map[string]*Account),
|
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -35,7 +33,7 @@ func (t *Threema) Connect() (chan xmpp.Packet, error) {
|
||||||
log.WithFields(map[string]interface{}{
|
log.WithFields(map[string]interface{}{
|
||||||
"jid": jid.String(),
|
"jid": jid.String(),
|
||||||
"threema": string(a.TID),
|
"threema": string(a.TID),
|
||||||
}).Debug("connected")
|
}).Info("connected")
|
||||||
}
|
}
|
||||||
return t.out, nil
|
return t.out, nil
|
||||||
}
|
}
|
||||||
|
@ -45,11 +43,6 @@ func (t *Threema) Send(packet xmpp.Packet) {
|
||||||
from := models.ParseJID(p.PacketAttrs.From)
|
from := models.ParseJID(p.PacketAttrs.From)
|
||||||
to := models.ParseJID(p.PacketAttrs.To)
|
to := models.ParseJID(p.PacketAttrs.To)
|
||||||
|
|
||||||
logger := log.WithFields(map[string]interface{}{
|
|
||||||
"from": from,
|
|
||||||
"to": to,
|
|
||||||
})
|
|
||||||
logger.Debug(p.Body)
|
|
||||||
if to.IsDomain() {
|
if to.IsDomain() {
|
||||||
msg := xmpp.NewMessage("chat", "", from.String(), "", "en")
|
msg := xmpp.NewMessage("chat", "", from.String(), "", "en")
|
||||||
msg.Body = t.Bot(from, p.Body)
|
msg.Body = t.Bot(from, p.Body)
|
||||||
|
@ -66,7 +59,7 @@ func (t *Threema) Send(packet xmpp.Packet) {
|
||||||
}
|
}
|
||||||
|
|
||||||
threemaID := strings.ToUpper(to.Local)
|
threemaID := strings.ToUpper(to.Local)
|
||||||
if err := account.Send(threemaID, p.Body); err != nil {
|
if err := account.Send(threemaID, p); err != nil {
|
||||||
msg := xmpp.NewMessage("chat", "", from.String(), "", "en")
|
msg := xmpp.NewMessage("chat", "", from.String(), "", "en")
|
||||||
msg.Body = err.Error()
|
msg.Body = err.Error()
|
||||||
t.out <- msg
|
t.out <- msg
|
||||||
|
|
|
@ -43,11 +43,11 @@ func (c *Config) recieve(packets chan xmpp.Packet) {
|
||||||
} else {
|
} else {
|
||||||
p.PacketAttrs.From += "@" + c.Host
|
p.PacketAttrs.From += "@" + c.Host
|
||||||
}
|
}
|
||||||
loggerMSG := logger.WithFields(map[string]interface{}{
|
logger.WithFields(map[string]interface{}{
|
||||||
"from": p.PacketAttrs.From,
|
"from": p.PacketAttrs.From,
|
||||||
"to": p.PacketAttrs.To,
|
"to": p.PacketAttrs.To,
|
||||||
})
|
"id": p.PacketAttrs.Id,
|
||||||
loggerMSG.Debug(p.Body)
|
}).Debug(p.XMPPFormat())
|
||||||
c.xmpp.Send(p)
|
c.xmpp.Send(p)
|
||||||
default:
|
default:
|
||||||
log.Warn("ignoring packet:", packet)
|
log.Warn("ignoring packet:", packet)
|
||||||
|
@ -90,6 +90,7 @@ func (c *Config) sender() {
|
||||||
Features: []xmpp.Feature{
|
Features: []xmpp.Feature{
|
||||||
{Var: "http://jabber.org/protocol/disco#info"},
|
{Var: "http://jabber.org/protocol/disco#info"},
|
||||||
{Var: "http://jabber.org/protocol/disco#item"},
|
{Var: "http://jabber.org/protocol/disco#item"},
|
||||||
|
{Var: xmpp.NSSpaceXEP0184Receipt},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
iq.AddPayload(&payload)
|
iq.AddPayload(&payload)
|
||||||
|
@ -124,6 +125,11 @@ func (c *Config) sender() {
|
||||||
}
|
}
|
||||||
|
|
||||||
case xmpp.Message:
|
case xmpp.Message:
|
||||||
|
logger.WithFields(map[string]interface{}{
|
||||||
|
"from": p.PacketAttrs.From,
|
||||||
|
"to": p.PacketAttrs.To,
|
||||||
|
"id": p.PacketAttrs.Id,
|
||||||
|
}).Debug(p.XMPPFormat())
|
||||||
c.comp.Send(packet)
|
c.comp.Send(packet)
|
||||||
|
|
||||||
case xmpp.Presence:
|
case xmpp.Presence:
|
||||||
|
|
Reference in New Issue