plugins/mod_pubsub.lua

   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 module:hook("iq/host/http://jabber.org/protocol/pubsub:pubsub", handle_pubsub_iq);
  91
  92 service = pubsub.new({
  93         broadcaster = simple_broadcast
  94 });
  95 _M.service = service;
  96