plugins/mod_roster.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2010 Matthew Wild
   3 -- Copyright (C) 2008-2010 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9
  10 local st = require "util.stanza"
  11
  12 local jid_split = require "util.jid".split;
  13 local jid_prep = require "util.jid".prep;
  14 local t_concat = table.concat;
  15 local tonumber = tonumber;
  16 local pairs, ipairs = pairs, ipairs;
  17
  18 local rm_load_roster = require "core.rostermanager".load_roster;
  19 local rm_remove_from_roster = require "core.rostermanager".remove_from_roster;
  20 local rm_add_to_roster = require "core.rostermanager".add_to_roster;
  21 local rm_roster_push = require "core.rostermanager".roster_push;
  22 local core_post_stanza = prosody.core_post_stanza;
  23
  24 module:add_feature("jabber:iq:roster");
  25
  26 local rosterver_stream_feature = st.stanza("ver", {xmlns="urn:xmpp:features:rosterver"});
  27 module:hook("stream-features", function(event)
  28         local origin, features = event.origin, event.features;
  29         if origin.username then
  30                 features:add_child(rosterver_stream_feature);
  31         end
  32 end);
  33
  34 module:hook("iq/self/jabber:iq:roster:query", function(event)
  35         local session, stanza = event.origin, event.stanza;
  36
  37         if stanza.attr.type == "get" then
  38                 local roster = st.reply(stanza);
  39
  40                 local client_ver = tonumber(stanza.tags[1].attr.ver);
  41                 local server_ver = tonumber(session.roster[false].version or 1);
  42
  43                 if not (client_ver and server_ver) or client_ver ~= server_ver then
  44                         roster:query("jabber:iq:roster");
  45                         -- Client does not support versioning, or has stale roster
  46                         for jid, item in pairs(session.roster) do
  47                                 if jid then
  48                                         roster:tag("item", {
  49                                                 jid = jid,
  50                                                 subscription = item.subscription,
  51                                                 ask = item.ask,
  52                                                 name = item.name,
  53                                         });
  54                                         for group in pairs(item.groups) do
  55                                                 roster:tag("group"):text(group):up();
  56                                         end
  57                                         roster:up(); -- move out from item
  58                                 end
  59                         end
  60                         roster.tags[1].attr.ver = server_ver;
  61                 end
  62                 session.send(roster);
  63                 session.interested = true; -- resource is interested in roster updates
  64         else -- stanza.attr.type == "set"
  65                 local query = stanza.tags[1];
  66                 if #query.tags == 1 and query.tags[1].name == "item"
  67                                 and query.tags[1].attr.xmlns == "jabber:iq:roster" and query.tags[1].attr.jid then
  68                         local item = query.tags[1];
  69                         local from_node, from_host = jid_split(stanza.attr.from);
  70                         local jid = jid_prep(item.attr.jid);
  71                         local node, host, resource = jid_split(jid);
  72                         if not resource and host then
  73                                 if jid ~= from_node.."@"..from_host then
  74                                         if item.attr.subscription == "remove" then
  75                                                 local roster = session.roster;
  76                                                 local r_item = roster[jid];
  77                                                 if r_item then
  78                                                         module:fire_event("roster-item-removed", {
  79                                                                 username = node, jid = jid, item = r_item, origin = session, roster = roster,
  80                                                         });
  81                                                         local success, err_type, err_cond, err_msg = rm_remove_from_roster(session, jid);
  82                                                         if success then
  83                                                                 session.send(st.reply(stanza));
  84                                                                 rm_roster_push(from_node, from_host, jid);
  85                                                         else
  86                                                                 session.send(st.error_reply(stanza, err_type, err_cond, err_msg));
  87                                                         end
  88                                                 else
  89                                                         session.send(st.error_reply(stanza, "modify", "item-not-found"));
  90                                                 end
  91                                         else
  92                                                 local r_item = {name = item.attr.name, groups = {}};
  93                                                 if r_item.name == "" then r_item.name = nil; end
  94                                                 if session.roster[jid] then
  95                                                         r_item.subscription = session.roster[jid].subscription;
  96                                                         r_item.ask = session.roster[jid].ask;
  97                                                 else
  98                                                         r_item.subscription = "none";
  99                                                 end
 100                                                 for _, child in ipairs(item) do
 101                                                         if child.name == "group" then
 102                                                                 local text = t_concat(child);
 103                                                                 if text and text ~= "" then
 104                                                                         r_item.groups[text] = true;
 105                                                                 end
 106                                                         end
 107                                                 end
 108                                                 local success, err_type, err_cond, err_msg = rm_add_to_roster(session, jid, r_item);
 109                                                 if success then
 110                                                         -- Ok, send success
 111                                                         session.send(st.reply(stanza));
 112                                                         -- and push change to all resources
 113                                                         rm_roster_push(from_node, from_host, jid);
 114                                                 else
 115                                                         -- Adding to roster failed
 116                                                         session.send(st.error_reply(stanza, err_type, err_cond, err_msg));
 117                                                 end
 118                                         end
 119                                 else
 120                                         -- Trying to add self to roster
 121                                         session.send(st.error_reply(stanza, "cancel", "not-allowed"));
 122                                 end
 123                         else
 124                                 -- Invalid JID added to roster
 125                                 session.send(st.error_reply(stanza, "modify", "bad-request")); -- FIXME what's the correct error?
 126                         end
 127                 else
 128                         -- Roster set didn't include a single item, or its name wasn't  'item'
 129                         session.send(st.error_reply(stanza, "modify", "bad-request"));
 130                 end
 131         end
 132         return true;
 133 end);
 134
 135 module:hook_global("user-deleted", function(event)
 136         local username, host = event.username, event.host;
 137         local origin = event.origin or prosody.hosts[host];
 138         if host ~= module.host then return end
 139         local bare = username .. "@" .. host;
 140         local roster = rm_load_roster(username, host);
 141         for jid, item in pairs(roster) do
 142                 if jid then
 143                         module:fire_event("roster-item-removed", {
 144                                 username = username, jid = jid, item = item, roster = roster, origin = origin,
 145                         });
 146                 else
 147                         for jid in pairs(item.pending) do
 148                                 module:fire_event("roster-item-removed", {
 149                                         username = username, jid = jid, roster = roster, origin = origin,
 150                                 });
 151                         end
 152                 end
 153         end
 154 end, 300);