package gateway import ( "encoding/xml" "strings" "sync" "dev.narayana.im/narayana/telegabber/xmpp/extensions" log "github.com/sirupsen/logrus" "github.com/soheilhy/args" "gosrc.io/xmpp" "gosrc.io/xmpp/stanza" ) const NSNick string = "http://jabber.org/protocol/nick" // Queue stores presences to send later var Queue = make(map[string]*stanza.Presence) var QueueLock = sync.Mutex{} // Jid stores the component's JID object var Jid *stanza.Jid // DirtySessions denotes that some Telegram session configurations // were changed and need to be re-flushed to the YamlDB var DirtySessions = false // SendMessage creates and sends a message stanza func SendMessage(to string, from string, body string, component *xmpp.Component) { componentJid := Jid.Full() var logFrom string var messageFrom string if from == "" { logFrom = componentJid messageFrom = componentJid } else { logFrom = from messageFrom = from + "@" + componentJid } log.WithFields(log.Fields{ "from": logFrom, "to": to, }).Warn("Got message") message := stanza.Message{ Attrs: stanza.Attrs{ From: messageFrom, To: to, Type: "chat", }, Body: body, } sendMessage(&message, component) } // SetNickname sets a new nickname for a contact func SetNickname(to string, from string, nickname string, component *xmpp.Component) { componentJid := Jid.Bare() messageFrom := from + "@" + componentJid log.WithFields(log.Fields{ "from": from, "to": to, }).Warn("Set nickname") message := stanza.Message{ Attrs: stanza.Attrs{ From: messageFrom, To: to, Type: "headline", }, Extensions: []stanza.MsgExtension{ stanza.PubSubEvent{ EventElement: stanza.ItemsEvent{ Node: NSNick, Items: []stanza.ItemEvent{ stanza.ItemEvent{ Any: &stanza.Node{ XMLName: xml.Name{Space: NSNick, Local: "nick"}, Content: nickname, }, }, }, }, }, }, } sendMessage(&message, component) } func sendMessage(message *stanza.Message, component *xmpp.Component) { // explicit check, as marshalling is expensive if log.GetLevel() == log.DebugLevel { xmlMessage, err := xml.Marshal(message) if err == nil { log.Debug(string(xmlMessage)) } else { log.Debugf("%#v", message) } } _ = ResumableSend(component, message) } // LogBadPresence verbosely logs a presence func LogBadPresence(presence *stanza.Presence) { log.Errorf("Couldn't send presence: %#v", presence) } // SPFrom is a Telegram user id var SPFrom = args.NewString() // SPType is a presence type var SPType = args.NewString() // SPShow is a availability status var SPShow = args.NewString() // SPStatus is a verbose status var SPStatus = args.NewString() // SPNickname is a XEP-0172 nickname var SPNickname = args.NewString() // SPPhoto is a XEP-0153 hash of avatar in vCard var SPPhoto = args.NewString() // SPResource is an optional resource var SPResource = args.NewString() // SPImmed skips queueing var SPImmed = args.NewBool(args.Default(true)) func newPresence(bareJid string, to string, args ...args.V) stanza.Presence { var presenceFrom string if SPFrom.IsSet(args) { presenceFrom = SPFrom.Get(args) + "@" + bareJid if SPResource.IsSet(args) { resource := SPResource.Get(args) if resource != "" { presenceFrom += "/" + resource } } } else { presenceFrom = bareJid } presence := stanza.Presence{Attrs: stanza.Attrs{ From: presenceFrom, To: to, }} if SPType.IsSet(args) { t := SPType.Get(args) if t != "" { presence.Attrs.Type = stanza.StanzaType(t) } } if SPShow.IsSet(args) { show := SPShow.Get(args) if show != "" { presence.Show = stanza.PresenceShow(show) } } if SPStatus.IsSet(args) { status := SPStatus.Get(args) if status != "" { presence.Status = status } } if SPNickname.IsSet(args) { nickname := SPNickname.Get(args) if nickname != "" { presence.Extensions = append(presence.Extensions, extensions.PresenceNickExtension{ Text: nickname, }) } } if SPPhoto.IsSet(args) { photo := SPPhoto.Get(args) if photo != "" { presence.Extensions = append(presence.Extensions, extensions.PresenceXVCardUpdateExtension{ Photo: extensions.PresenceXVCardUpdatePhoto{ Text: photo, }, }) } } return presence } // SendPresence creates and sends a presence stanza func SendPresence(component *xmpp.Component, to string, args ...args.V) error { var logFrom string bareJid := Jid.Bare() if SPFrom.IsSet(args) { logFrom = SPFrom.Get(args) } else { logFrom = bareJid } log.WithFields(log.Fields{ "type": SPType.Get(args), "from": logFrom, "to": to, }).Info("Got presence") presence := newPresence(bareJid, to, args...) // explicit check, as marshalling is expensive if log.GetLevel() == log.DebugLevel { xmlPresence, err := xml.Marshal(presence) if err == nil { log.Debug(string(xmlPresence)) } else { log.Debugf("%#v", presence) } } immed := SPImmed.Get(args) if immed { err := ResumableSend(component, presence) if err != nil { LogBadPresence(&presence) return err } } else { QueueLock.Lock() Queue[presence.From+presence.To] = &presence QueueLock.Unlock() } return nil } // ResumableSend tries to resume the connection once and sends the packet again func ResumableSend(component *xmpp.Component, packet stanza.Packet) error { err := component.Send(packet) if err != nil && strings.HasPrefix(err.Error(), "cannot send packet") { log.Warn("Packet send failed, trying to resume the connection...") err = component.Connect() if err == nil { err = component.Send(packet) } } if err != nil { log.Error(err.Error()) } return err }