Automated merge with http://waqas.ath.cx:8000/
[prosody.git] / core / presencemanager.lua
index 388f08c4e903d1d2011364fe1b96360df00b32ee..0f0808aa94790152c7b84460ae4218a7f64d77a6 100644 (file)
@@ -1,8 +1,31 @@
+-- Prosody IM v0.2
+-- Copyright (C) 2008 Matthew Wild
+-- Copyright (C) 2008 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.
+--
+
+
 \r
 local log = require "util.logger".init("presencemanager")\r
 \r
 local require = require;\r
-local pairs = pairs;\r
+local pairs, ipairs = pairs, ipairs;\r
+local t_concat = table.concat;\r
+local s_find = string.find;\r
+local tonumber = tonumber;\r
 \r
 local st = require "util.stanza";\r
 local jid_split = require "util.jid".split;\r
@@ -10,9 +33,106 @@ local hosts = hosts;
 \r
 local rostermanager = require "core.rostermanager";\r
 local sessionmanager = require "core.sessionmanager";\r
+local offlinemanager = require "core.offlinemanager";\r
 \r
 module "presencemanager"\r
 \r
+function handle_presence(origin, stanza, from_bare, to_bare, core_route_stanza, inbound)\r
+       local type = stanza.attr.type;\r
+       if type and type ~= "unavailable" then\r
+               if inbound then\r
+                       handle_inbound_presence_subscriptions_and_probes(origin, stanza, from_bare, to_bare, core_route_stanza);\r
+               else\r
+                       handle_outbound_presence_subscriptions_and_probes(origin, stanza, from_bare, to_bare, core_route_stanza);\r
+               end\r
+       elseif not inbound and not stanza.attr.to then\r
+               handle_normal_presence(origin, stanza, core_route_stanza);\r
+       else\r
+               core_route_stanza(origin, stanza);\r
+       end\r
+end\r
+\r
+function handle_normal_presence(origin, stanza, core_route_stanza)\r
+       if origin.roster then\r
+               for jid in pairs(origin.roster) do -- broadcast to all interested contacts\r
+                       local subscription = origin.roster[jid].subscription;\r
+                       if subscription == "both" or subscription == "from" then\r
+                               stanza.attr.to = jid;\r
+                               core_route_stanza(origin, stanza);\r
+                       end\r
+               end\r
+               local node, host = jid_split(stanza.attr.from);\r
+               for _, res in pairs(hosts[host].sessions[node].sessions) do -- broadcast to all resources\r
+                       if res ~= origin and res.full_jid then -- to resource. FIXME is res.full_jid the correct check? Maybe it should be res.presence\r
+                               stanza.attr.to = res.full_jid;\r
+                               core_route_stanza(origin, stanza);\r
+                       end\r
+               end\r
+               if stanza.attr.type == nil and not origin.presence then -- initial presence\r
+                       local probe = st.presence({from = origin.full_jid, type = "probe"});\r
+                       for jid in pairs(origin.roster) do -- probe all contacts we are subscribed to\r
+                               local subscription = origin.roster[jid].subscription;\r
+                               if subscription == "both" or subscription == "to" then\r
+                                       probe.attr.to = jid;\r
+                                       core_route_stanza(origin, probe);\r
+                               end\r
+                       end\r
+                       for _, res in pairs(hosts[host].sessions[node].sessions) do -- broadcast from all available resources\r
+                               if res ~= origin and res.presence then\r
+                                       res.presence.attr.to = origin.full_jid;\r
+                                       core_route_stanza(res, res.presence);\r
+                                       res.presence.attr.to = nil;\r
+                               end\r
+                       end\r
+                       if origin.roster.pending then -- resend incoming subscription requests\r
+                               for jid in pairs(origin.roster.pending) do\r
+                                       origin.send(st.presence({type="subscribe", from=jid})); -- TODO add to attribute? Use original?\r
+                               end\r
+                       end\r
+                       local request = st.presence({type="subscribe", from=origin.username.."@"..origin.host});\r
+                       for jid, item in pairs(origin.roster) do -- resend outgoing subscription requests\r
+                               if item.ask then\r
+                                       request.attr.to = jid;\r
+                                       core_route_stanza(origin, request);\r
+                               end\r
+                       end
+                       local offline = offlinemanager.load(node, host);
+                       if offline then\r
+                               for _, msg in ipairs(offline) do\r
+                                       origin.send(msg); -- FIXME do we need to modify to/from in any way?\r
+                               end\r
+                               offlinemanager.deleteAll(node, host);
+                       end\r
+               end\r
+               origin.priority = 0;\r
+               if stanza.attr.type == "unavailable" then\r
+                       origin.presence = nil;
+                       if origin.directed then
+                               for _, jid in ipairs(origin.directed) do
+                                       stanza.attr.to = jid;
+                                       core_route_stanza(origin, stanza);
+                               end
+                               origin.directed = nil;
+                       end\r
+               else\r
+                       origin.presence = stanza;\r
+                       local priority = stanza:child_with_name("priority");\r
+                       if priority and #priority > 0 then\r
+                               priority = t_concat(priority);\r
+                               if s_find(priority, "^[+-]?[0-9]+$") then\r
+                                       priority = tonumber(priority);\r
+                                       if priority < -128 then priority = -128 end\r
+                                       if priority > 127 then priority = 127 end\r
+                                       origin.priority = priority;\r
+                               end\r
+                       end\r
+               end\r
+               stanza.attr.to = nil; -- reset it\r
+       else\r
+               log("error", "presence recieved from client with no roster");\r
+       end\r
+end\r
+\r
 function send_presence_of_available_resources(user, host, jid, recipient_session, core_route_stanza)\r
        local h = hosts[host];\r
        local count = 0;\r
@@ -32,7 +152,7 @@ function send_presence_of_available_resources(user, host, jid, recipient_session
                        end\r
                end\r
        end\r
-       log("info", "broadcasted presence of "..count.." resources from "..user.."@"..host.." to "..to_bare);\r
+       log("info", "broadcasted presence of "..count.." resources from "..user.."@"..host.." to "..jid);\r
        return count;\r
 end\r
 \r
@@ -61,16 +181,16 @@ function handle_outbound_presence_subscriptions_and_probes(origin, stanza, from_
                -- 3. send_presence_of_available_resources\r
                if rostermanager.subscribed(node, host, to_bare) then\r
                        rostermanager.roster_push(node, host, to_bare);\r
-                       core_route_stanza(origin, stanza);\r
-                       send_presence_of_available_resources(node, host, to_bare, origin, core_route_stanza);\r
                end\r
+               core_route_stanza(origin, stanza);\r
+               send_presence_of_available_resources(node, host, to_bare, origin, core_route_stanza);\r
        elseif stanza.attr.type == "unsubscribed" then\r
                -- 1. route stanza\r
                -- 2. roster push (subscription = none or to)\r
                if rostermanager.unsubscribed(node, host, to_bare) then\r
                        rostermanager.roster_push(node, host, to_bare);\r
-                       core_route_stanza(origin, stanza);\r
                end\r
+               core_route_stanza(origin, stanza);\r
        end\r
        stanza.attr.from, stanza.attr.to = st_from, st_to;\r
 end\r
@@ -91,6 +211,10 @@ function handle_inbound_presence_subscriptions_and_probes(origin, stanza, from_b
        elseif stanza.attr.type == "subscribe" then\r
                if rostermanager.is_contact_subscribed(node, host, from_bare) then\r
                        core_route_stanza(origin, st.presence({from=to_bare, to=from_bare, type="subscribed"})); -- already subscribed\r
+                       -- Sending presence is not clearly stated in the RFC, but it seems appropriate\r
+                       if 0 == send_presence_of_available_resources(node, host, from_bare, origin, core_route_stanza) then\r
+                               -- TODO send last recieved unavailable presence (or we MAY do nothing, which is fine too)\r
+                       end\r
                else\r
                        if not rostermanager.is_contact_pending_in(node, host, from_bare) then\r
                                if rostermanager.set_contact_pending_in(node, host, from_bare) then\r
@@ -107,7 +231,7 @@ function handle_inbound_presence_subscriptions_and_probes(origin, stanza, from_b
                        rostermanager.roster_push(node, host, from_bare);\r
                end\r
        elseif stanza.attr.type == "unsubscribed" then\r
-               if rostermanager.process_inbound_subscription_approval(node, host, from_bare) then\r
+               if rostermanager.process_inbound_subscription_cancellation(node, host, from_bare) then\r
                        rostermanager.roster_push(node, host, from_bare);\r
                end\r
        end -- discard any other type\r