package main import ( "encoding/xml" "fmt" "log" "gosrc.io/xmpp" "gosrc.io/xmpp/stanza" ) func main() { opts := xmpp.ComponentOptions{ TransportConfiguration: xmpp.TransportConfiguration{ Address: "localhost:9999", Domain: "service.localhost", }, Domain: "service.localhost", Secret: "mypass", // TODO: Move that part to a component discovery handler Name: "Test Component", Category: "gateway", Type: "service", } router := xmpp.NewRouter() router.HandleFunc("message", handleMessage) router.NewRoute(). IQNamespaces(stanza.NSDiscoInfo). HandlerFunc(func(s xmpp.Sender, p stanza.Packet) { discoInfo(s, p, opts) }) router.NewRoute(). IQNamespaces("urn:xmpp:delegation:1"). HandlerFunc(handleDelegation) component, err := xmpp.NewComponent(opts, router) if err != nil { log.Fatalf("%+v", err) } // If you pass the component to a stream manager, it will handle the reconnect policy // for you automatically. // TODO: Post Connect could be a feature of the router or the client. Move it somewhere else. cm := xmpp.NewStreamManager(component, nil) log.Fatal(cm.Run()) } func handleMessage(_ xmpp.Sender, p stanza.Packet) { msg, ok := p.(stanza.Message) if !ok { return } var msgProcessed bool for _, ext := range msg.Extensions { delegation, ok := ext.(*stanza.Delegation) if ok { msgProcessed = true fmt.Printf("Delegation confirmed for namespace %s\n", delegation.Delegated.Namespace) } } // TODO: Decode privilege message // if !msgProcessed { fmt.Printf("Ignored received message, not related to delegation: %v\n", msg) } } const ( pubsubNode = "urn:xmpp:delegation:1::http://jabber.org/protocol/pubsub" pepNode = "urn:xmpp:delegation:1:bare:http://jabber.org/protocol/pubsub" ) // TODO: replace xmpp.Sender by ctx xmpp.Context ? // ctx.Stream.Send / SendRaw // ctx.Opts func discoInfo(c xmpp.Sender, p stanza.Packet, opts xmpp.ComponentOptions) { // Type conversion & sanity checks iq, ok := p.(stanza.IQ) if !ok { return } info, ok := iq.Payload.(*stanza.DiscoInfo) if !ok { return } iqResp := stanza.NewIQ(stanza.Attrs{Type: "result", From: iq.To, To: iq.From, Id: iq.Id}) switch info.Node { case "": discoInfoRoot(&iqResp, opts) case pubsubNode: discoInfoPubSub(&iqResp) case pepNode: discoInfoPEP(&iqResp) } _ = c.Send(iqResp) } func discoInfoRoot(iqResp *stanza.IQ, opts xmpp.ComponentOptions) { disco := iqResp.DiscoInfo() disco.AddIdentity(opts.Name, opts.Category, opts.Type) disco.AddFeatures(stanza.NSDiscoInfo, stanza.NSDiscoItems, "jabber:iq:version", "urn:xmpp:delegation:1") } func discoInfoPubSub(iqResp *stanza.IQ) { payload := stanza.DiscoInfo{ XMLName: xml.Name{ Space: stanza.NSDiscoInfo, Local: "query", }, Node: pubsubNode, Features: []stanza.Feature{ {Var: "http://jabber.org/protocol/pubsub"}, {Var: "http://jabber.org/protocol/pubsub#publish"}, {Var: "http://jabber.org/protocol/pubsub#subscribe"}, {Var: "http://jabber.org/protocol/pubsub#publish-options"}, }, } iqResp.Payload = &payload } func discoInfoPEP(iqResp *stanza.IQ) { identity := stanza.Identity{ Category: "pubsub", Type: "pep", } payload := stanza.DiscoInfo{ XMLName: xml.Name{ Space: stanza.NSDiscoInfo, Local: "query", }, Identity: []stanza.Identity{identity}, Node: pepNode, Features: []stanza.Feature{ {Var: "http://jabber.org/protocol/pubsub#access-presence"}, {Var: "http://jabber.org/protocol/pubsub#auto-create"}, {Var: "http://jabber.org/protocol/pubsub#auto-subscribe"}, {Var: "http://jabber.org/protocol/pubsub#config-node"}, {Var: "http://jabber.org/protocol/pubsub#create-and-configure"}, {Var: "http://jabber.org/protocol/pubsub#create-nodes"}, {Var: "http://jabber.org/protocol/pubsub#filtered-notifications"}, {Var: "http://jabber.org/protocol/pubsub#persistent-items"}, {Var: "http://jabber.org/protocol/pubsub#publish"}, {Var: "http://jabber.org/protocol/pubsub#retrieve-items"}, {Var: "http://jabber.org/protocol/pubsub#subscribe"}, }, } iqResp.Payload = &payload } func handleDelegation(s xmpp.Sender, p stanza.Packet) { // Type conversion & sanity checks iq, ok := p.(stanza.IQ) if !ok { return } delegation, ok := iq.Payload.(*stanza.Delegation) if !ok { return } forwardedPacket := delegation.Forwarded.Stanza fmt.Println(forwardedPacket) forwardedIQ, ok := forwardedPacket.(stanza.IQ) if !ok { return } pubsub, ok := forwardedIQ.Payload.(*stanza.PubSubGeneric) if !ok { // We only support pubsub delegation return } if pubsub.Publish.XMLName.Local == "publish" { // Prepare pubsub IQ reply iqResp := stanza.NewIQ(stanza.Attrs{Type: "result", From: forwardedIQ.To, To: forwardedIQ.From, Id: forwardedIQ.Id}) payload := stanza.PubSubGeneric{ XMLName: xml.Name{ Space: "http://jabber.org/protocol/pubsub", Local: "pubsub", }, } iqResp.Payload = &payload // Wrap the reply in delegation 'forward' iqForward := stanza.NewIQ(stanza.Attrs{Type: "result", From: iq.To, To: iq.From, Id: iq.Id}) delegPayload := stanza.Delegation{ XMLName: xml.Name{ Space: "urn:xmpp:delegation:1", Local: "delegation", }, Forwarded: &stanza.Forwarded{ XMLName: xml.Name{ Space: "urn:xmpp:forward:0", Local: "forward", }, Stanza: iqResp, }, } iqForward.Payload = &delegPayload _ = s.Send(iqForward) // TODO: The component should actually broadcast the mood to subscribers } }