Sat, 13 Nov 2010 04:12:03 +0000
mod_pubsub: Use module.environment to reference the module's environment
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 |