Software / code / prosody
Annotate
plugins/mod_pubsub.lua @ 3622:418354197a02
mod_pubsub: Use module.environment to reference the module's environment
| author | Matthew Wild <mwild1@gmail.com> |
|---|---|
| date | Sat, 13 Nov 2010 04:12:03 +0000 |
| parent | 3621:7e1a1dd42088 |
| child | 3641:3603aeb325de |
| rev | line source |
|---|---|
| 3619 | 1 local pubsub = require "util.pubsub"; |
| 2 local st = require "util.stanza"; | |
| 3 local jid_bare = require "util.jid".bare; | |
| 4 local uuid_generate = require "util.uuid".generate; | |
| 5 | |
| 6 require "core.modulemanager".load(module.host, "iq"); | |
| 7 | |
| 8 local xmlns_pubsub = "http://jabber.org/protocol/pubsub"; | |
| 9 local xmlns_pubsub_errors = "http://jabber.org/protocol/pubsub#errors"; | |
| 10 local xmlns_pubsub_event = "http://jabber.org/protocol/pubsub#event"; | |
| 11 | |
| 12 local service; | |
| 13 | |
| 14 local handlers = {}; | |
| 15 | |
| 16 function handle_pubsub_iq(event) | |
| 17 local origin, stanza = event.origin, event.stanza; | |
| 18 local pubsub = stanza.tags[1]; | |
| 19 local action = pubsub.tags[1]; | |
| 20 local handler = handlers[stanza.attr.type.."_"..action.name]; | |
| 21 if handler then | |
| 22 handler(origin, stanza, action); | |
| 23 return true; | |
| 24 end | |
| 25 end | |
| 26 | |
| 27 local pubsub_errors = { | |
| 28 ["invalid-jid"] = { "modify", "bad-request", nil, "invalid-jid" }; | |
| 29 ["item-not-found"] = { "cancel", "item-not-found" }; | |
| 30 }; | |
| 31 function pubsub_error_reply(stanza, error) | |
| 32 local e = pubsub_errors[error]; | |
| 33 local reply = st.error_reply(stanza, unpack(e, 1, 3)); | |
| 34 if e[4] then | |
| 35 reply:tag(e[4], { xmlns = xmlns_pubsub_errors }):up(); | |
| 36 end | |
| 37 return reply; | |
| 38 end | |
| 39 | |
| 40 function handlers.set_subscribe(origin, stanza, subscribe) | |
| 41 local node, jid = subscribe.attr.node, subscribe.attr.jid; | |
| 42 if jid_bare(jid) ~= jid_bare(stanza.attr.from) then | |
| 43 return origin.send(pubsub_error_reply(stanza, "invalid-jid")); | |
| 44 end | |
| 45 local ok, ret = service:add_subscription(node, stanza.attr.from, jid); | |
| 46 local reply; | |
| 47 if ok then | |
| 48 reply = st.reply(stanza) | |
| 49 :tag("pubsub", { xmlns = xmlns_pubsub }) | |
| 50 :tag("subscription", { | |
| 51 node = node, | |
| 52 jid = jid, | |
| 53 subscription = "subscribed" | |
| 54 }); | |
| 55 else | |
| 56 reply = pubsub_error_reply(stanza, ret); | |
| 57 end | |
| 58 return origin.send(reply); | |
| 59 end | |
| 60 | |
| 61 function handlers.set_publish(origin, stanza, publish) | |
| 62 local node = publish.attr.node; | |
| 63 local item = publish:get_child("item"); | |
| 64 local id = (item and item.attr.id) or uuid_generate(); | |
| 65 local ok, ret = service:publish(node, stanza.attr.from, id, item); | |
| 66 local reply; | |
| 67 if ok then | |
| 68 reply = st.reply(stanza) | |
| 69 :tag("pubsub", { xmlns = xmlns_pubsub }) | |
| 70 :tag("publish", { node = node }) | |
| 71 :tag("item", { id = id }); | |
| 72 else | |
| 73 reply = pubsub_error_reply(stanza, ret); | |
| 74 end | |
| 75 return origin.send(reply); | |
| 76 end | |
| 77 | |
| 78 function simple_broadcast(node, jids, item) | |
| 79 local message = st.message({ from = module.host, type = "headline" }) | |
| 80 :tag("event", { xmlns = xmlns_pubsub_event }) | |
| 81 :tag("items", { node = node }) | |
| 82 :add_child(item); | |
| 83 for jid in pairs(jids) do | |
| 84 module:log("debug", "Sending notification to %s", jid); | |
| 85 message.attr.to = jid; | |
| 86 core_post_stanza(hosts[module.host], message); | |
| 87 end | |
| 88 end | |
| 89 | |
|
3621
7e1a1dd42088
mod_pubsub: Expose 'service'
Matthew Wild <mwild1@gmail.com>
parents:
3619
diff
changeset
|
90 module:hook("iq/host/http://jabber.org/protocol/pubsub:pubsub", handle_pubsub_iq); |
|
7e1a1dd42088
mod_pubsub: Expose 'service'
Matthew Wild <mwild1@gmail.com>
parents:
3619
diff
changeset
|
91 |
| 3619 | 92 service = pubsub.new({ |
| 93 broadcaster = simple_broadcast | |
| 94 }); | |
|
3622
418354197a02
mod_pubsub: Use module.environment to reference the module's environment
Matthew Wild <mwild1@gmail.com>
parents:
3621
diff
changeset
|
95 module.environment.service = service; |
| 3619 | 96 |
