2 -- Copyright (C) 2008-2009 Matthew Wild
3 -- Copyright (C) 2008-2009 Waqas Hussain
5 -- This project is MIT/X11 licensed. Please see the
6 -- COPYING file in the source package for more information.
11 local st = require "util.stanza"
13 local jid_split = require "util.jid".split;
14 local jid_prep = require "util.jid".prep;
15 local t_concat = table.concat;
16 local tostring = tostring;
18 local rm_remove_from_roster = require "core.rostermanager".remove_from_roster;
19 local rm_add_to_roster = require "core.rostermanager".add_to_roster;
20 local rm_roster_push = require "core.rostermanager".roster_push;
21 local core_post_stanza = core_post_stanza;
23 module:add_feature("jabber:iq:roster");
25 local rosterver_stream_feature = st.stanza("ver", {xmlns="urn:xmpp:features:rosterver"}):tag("optional"):up();
26 module:add_event_hook("stream-features",
27 function (session, features)
28 if session.username then
29 features:add_child(rosterver_stream_feature);
33 module:add_iq_handler("c2s", "jabber:iq:roster",
34 function (session, stanza)
35 if stanza.tags[1].name == "query" then
36 if stanza.attr.type == "get" then
37 local roster = st.reply(stanza);
39 local ver = stanza.tags[1].attr.ver
41 if (not ver) or tonumber(ver) ~= (session.roster[false].version or 1) then
42 roster:query("jabber:iq:roster");
43 -- Client does not support versioning, or has stale roster
44 for jid in pairs(session.roster) do
45 if jid ~= "pending" and jid then
48 subscription = session.roster[jid].subscription,
49 ask = session.roster[jid].ask,
50 name = session.roster[jid].name,
52 for group in pairs(session.roster[jid].groups) do
53 roster:tag("group"):text(group):up();
55 roster:up(); -- move out from item
58 roster.tags[1].attr.ver = tostring(session.roster[false].version or "1");
61 session.interested = true; -- resource is interested in roster updates
63 elseif stanza.attr.type == "set" then
64 local query = stanza.tags[1];
65 if #query.tags == 1 and query.tags[1].name == "item"
66 and query.tags[1].attr.xmlns == "jabber:iq:roster" and query.tags[1].attr.jid
67 -- Protection against overwriting roster.pending, until we move it
68 and query.tags[1].attr.jid ~= "pending" then
69 local item = query.tags[1];
70 local from_node, from_host = jid_split(stanza.attr.from);
71 local from_bare = from_node and (from_node.."@"..from_host) or from_host; -- bare JID
72 local jid = jid_prep(item.attr.jid);
73 local node, host, resource = jid_split(jid);
74 if not resource and host then
75 if jid ~= from_node.."@"..from_host then
76 if item.attr.subscription == "remove" then
77 local r_item = session.roster[jid];
79 local success, err_type, err_cond, err_msg = rm_remove_from_roster(session, jid);
81 session.send(st.reply(stanza));
82 rm_roster_push(from_node, from_host, jid);
83 local to_bare = node and (node.."@"..host) or host; -- bare JID
84 if r_item.subscription == "both" or r_item.subscription == "from" then
85 core_post_stanza(session, st.presence({type="unsubscribed", from=session.full_jid, to=to_bare}));
86 elseif r_item.subscription == "both" or r_item.subscription == "to" then
87 core_post_stanza(session, st.presence({type="unsubscribe", from=session.full_jid, to=to_bare}));
90 session.send(st.error_reply(stanza, err_type, err_cond, err_msg));
93 session.send(st.error_reply(stanza, "modify", "item-not-found"));
96 local r_item = {name = item.attr.name, groups = {}};
97 if r_item.name == "" then r_item.name = nil; end
98 if session.roster[jid] then
99 r_item.subscription = session.roster[jid].subscription;
100 r_item.ask = session.roster[jid].ask;
102 r_item.subscription = "none";
104 for _, child in ipairs(item) do
105 if child.name == "group" then
106 local text = t_concat(child);
107 if text and text ~= "" then
108 r_item.groups[text] = true;
112 local success, err_type, err_cond, err_msg = rm_add_to_roster(session, jid, r_item);
115 session.send(st.reply(stanza));
116 -- and push change to all resources
117 rm_roster_push(from_node, from_host, jid);
119 -- Adding to roster failed
120 session.send(st.error_reply(stanza, err_type, err_cond, err_msg));
124 -- Trying to add self to roster
125 session.send(st.error_reply(stanza, "cancel", "not-allowed"));
128 -- Invalid JID added to roster
129 session.send(st.error_reply(stanza, "modify", "bad-request")); -- FIXME what's the correct error?
132 -- Roster set didn't include a single item, or its name wasn't 'item'
133 session.send(st.error_reply(stanza, "modify", "bad-request"));