prosody
291ae816d800
plugins/mod_pubsub.lua

Find changesets by keywords by author, files, the commit message, revision number or hash, or revset expression.

Software / code / prosody

Comparison

plugins/mod_pubsub.lua @ 3619:291ae816d800

mod_pubsub: It's aliiiive!
author Matthew Wild <mwild1@gmail.com>
date Sat, 13 Nov 2010 03:16:58 +0000
child 3621:7e1a1dd42088
comparison
equal deleted inserted replaced
3618:321767e78029 3619:291ae816d800
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
90 service = pubsub.new({
91 broadcaster = simple_broadcast
92 });
93
94 module:hook("iq/host/http://jabber.org/protocol/pubsub:pubsub", handle_pubsub_iq);