mod_component: Deregister component on disconnect
[prosody.git] / plugins / mod_roster.lua
index 83e9835d4f05372a49384bfa3caecf76b3d5cc03..e30bc1f9dd60607f45f832fbc2b19e9b574022e0 100644 (file)
@@ -1,20 +1,9 @@
--- Prosody IM v0.2
--- Copyright (C) 2008 Matthew Wild
--- Copyright (C) 2008 Waqas Hussain
+-- Prosody IM v0.4
+-- Copyright (C) 2008-2009 Matthew Wild
+-- Copyright (C) 2008-2009 Waqas Hussain
 -- 
--- This program is free software; you can redistribute it and/or
--- modify it under the terms of the GNU General Public License
--- as published by the Free Software Foundation; either version 2
--- of the License, or (at your option) any later version.
--- 
--- This program is distributed in the hope that it will be useful,
--- but WITHOUT ANY WARRANTY; without even the implied warranty of
--- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
--- GNU General Public License for more details.
--- 
--- You should have received a copy of the GNU General Public License
--- along with this program; if not, write to the Free Software
--- Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+-- This project is MIT/X11 licensed. Please see the
+-- COPYING file in the source package for more information.
 --
 
 
@@ -22,7 +11,9 @@
 local st = require "util.stanza"
 
 local jid_split = require "util.jid".split;
+local jid_prep = require "util.jid".prep;
 local t_concat = table.concat;
+local tostring = tostring;
 
 local handle_presence = require "core.presencemanager".handle_presence;
 local rm_remove_from_roster = require "core.rostermanager".remove_from_roster;
@@ -38,19 +29,26 @@ module:add_iq_handler("c2s", "jabber:iq:roster",
                                if stanza.attr.type == "get" then
                                        local roster = st.reply(stanza)
                                                                :query("jabber:iq:roster");
-                                       for jid in pairs(session.roster) do
-                                               if jid ~= "pending" then
-                                                       roster:tag("item", {
-                                                               jid = jid,
-                                                               subscription = session.roster[jid].subscription,
-                                                               ask = session.roster[jid].ask,
-                                                               name = session.roster[jid].name,
-                                                       });
-                                                       for group in pairs(session.roster[jid].groups) do
-                                                               roster:tag("group"):text(group):up();
+                                       
+                                       local ver = stanza.tags[1].attr.ver
+                                       
+                                       if (not ver) or tonumber(ver) ~= (session.roster[false].version or 1) then
+                                               -- Client does not support versioning, or has stale roster
+                                               for jid in pairs(session.roster) do
+                                                       if jid ~= "pending" and jid then
+                                                               roster:tag("item", {
+                                                                       jid = jid,
+                                                                       subscription = session.roster[jid].subscription,
+                                                                       ask = session.roster[jid].ask,
+                                                                       name = session.roster[jid].name,
+                                                               });
+                                                               for group in pairs(session.roster[jid].groups) do
+                                                                       roster:tag("group"):text(group):up();
+                                                               end
+                                                               roster:up(); -- move out from item
                                                        end
-                                                       roster:up(); -- move out from item
                                                end
+                                               roster.tags[1].attr.ver = tostring(session.roster[false].version or "1");
                                        end
                                        session.send(roster);
                                        session.interested = true; -- resource is interested in roster updates
@@ -58,22 +56,24 @@ module:add_iq_handler("c2s", "jabber:iq:roster",
                                elseif stanza.attr.type == "set" then
                                        local query = stanza.tags[1];
                                        if #query.tags == 1 and query.tags[1].name == "item"
-                                                       and query.tags[1].attr.xmlns == "jabber:iq:roster" and query.tags[1].attr.jid
+                                                       and query.tags[1].attr.xmlns == "jabber:iq:roster" and query.tags[1].attr.jid 
+                                                       -- Protection against overwriting roster.pending, until we move it
                                                        and query.tags[1].attr.jid ~= "pending" then
                                                local item = query.tags[1];
                                                local from_node, from_host = jid_split(stanza.attr.from);
                                                local from_bare = from_node and (from_node.."@"..from_host) or from_host; -- bare JID
-                                               local node, host, resource = jid_split(item.attr.jid);
-                                               local to_bare = node and (node.."@"..host) or host; -- bare JID
+                                               local jid = jid_prep(item.attr.jid);
+                                               local node, host, resource = jid_split(jid);
                                                if not resource and host then
-                                                       if item.attr.jid ~= from_node.."@"..from_host then
+                                                       if jid ~= from_node.."@"..from_host then
                                                                if item.attr.subscription == "remove" then
-                                                                       local r_item = session.roster[item.attr.jid];
+                                                                       local r_item = session.roster[jid];
                                                                        if r_item then
-                                                                               local success, err_type, err_cond, err_msg = rm_remove_from_roster(session, item.attr.jid);
+                                                                               local success, err_type, err_cond, err_msg = rm_remove_from_roster(session, jid);
                                                                                if success then
                                                                                        session.send(st.reply(stanza));
-                                                                                       rm_roster_push(from_node, from_host, item.attr.jid);
+                                                                                       rm_roster_push(from_node, from_host, jid);
+                                                                                       local to_bare = node and (node.."@"..host) or host; -- bare JID
                                                                                        if r_item.subscription == "both" or r_item.subscription == "from" then
                                                                                                handle_presence(session, st.presence({type="unsubscribed"}), from_bare, to_bare,
                                                                                                        core_route_stanza, false);
@@ -90,9 +90,9 @@ module:add_iq_handler("c2s", "jabber:iq:roster",
                                                                else
                                                                        local r_item = {name = item.attr.name, groups = {}};
                                                                        if r_item.name == "" then r_item.name = nil; end
-                                                                       if session.roster[item.attr.jid] then
-                                                                               r_item.subscription = session.roster[item.attr.jid].subscription;
-                                                                               r_item.ask = session.roster[item.attr.jid].ask;
+                                                                       if session.roster[jid] then
+                                                                               r_item.subscription = session.roster[jid].subscription;
+                                                                               r_item.ask = session.roster[jid].ask;
                                                                        else
                                                                                r_item.subscription = "none";
                                                                        end
@@ -104,10 +104,10 @@ module:add_iq_handler("c2s", "jabber:iq:roster",
                                                                                        end
                                                                                end
                                                                        end
-                                                                       local success, err_type, err_cond, err_msg = rm_add_to_roster(session, item.attr.jid, r_item);
+                                                                       local success, err_type, err_cond, err_msg = rm_add_to_roster(session, jid, r_item);
                                                                        if success then
                                                                                session.send(st.reply(stanza));
-                                                                               rm_roster_push(from_node, from_host, item.attr.jid);
+                                                                               rm_roster_push(from_node, from_host, jid);
                                                                        else
                                                                                session.send(st.error_reply(stanza, err_type, err_cond, err_msg));
                                                                        end