2 -- Copyright (C) 2008-2010 Matthew Wild
3 -- Copyright (C) 2008-2010 Waqas Hussain
5 -- This project is MIT/X11 licensed. Please see the
6 -- COPYING file in the source package for more information.
10 local st = require "util.stanza"
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;
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;
24 module:add_feature("jabber:iq:roster");
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);
34 module:hook("iq/self/jabber:iq:roster:query", function(event)
35 local session, stanza = event.origin, event.stanza;
37 if stanza.attr.type == "get" then
38 local roster = st.reply(stanza);
40 local client_ver = tonumber(stanza.tags[1].attr.ver);
41 local server_ver = tonumber(session.roster[false].version or 1);
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
50 subscription = item.subscription,
54 for group in pairs(item.groups) do
55 roster:tag("group"):text(group):up();
57 roster:up(); -- move out from item
60 roster.tags[1].attr.ver = server_ver;
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];
78 module:fire_event("roster-item-removed", {
79 username = node, jid = jid, item = r_item, origin = session, roster = roster,
81 local success, err_type, err_cond, err_msg = rm_remove_from_roster(session, jid);
83 session.send(st.reply(stanza));
84 rm_roster_push(from_node, from_host, jid);
86 session.send(st.error_reply(stanza, err_type, err_cond, err_msg));
89 session.send(st.error_reply(stanza, "modify", "item-not-found"));
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;
98 r_item.subscription = "none";
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;
108 local success, err_type, err_cond, err_msg = rm_add_to_roster(session, jid, r_item);
111 session.send(st.reply(stanza));
112 -- and push change to all resources
113 rm_roster_push(from_node, from_host, jid);
115 -- Adding to roster failed
116 session.send(st.error_reply(stanza, err_type, err_cond, err_msg));
120 -- Trying to add self to roster
121 session.send(st.error_reply(stanza, "cancel", "not-allowed"));
124 -- Invalid JID added to roster
125 session.send(st.error_reply(stanza, "modify", "bad-request")); -- FIXME what's the correct error?
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"));
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
143 module:fire_event("roster-item-removed", {
144 username = username, jid = jid, item = item, roster = roster, origin = origin,
147 for jid in pairs(item.pending) do
148 module:fire_event("roster-item-removed", {
149 username = username, jid = jid, roster = roster, origin = origin,