Stanza router: Message to bare JID fixes
[prosody.git] / core / presencemanager.lua
index c6619fea62553fca8c67554b54b800c0b461b449..8fdf3612c1d6e3a0666ea2a8c21970d837fea626 100644 (file)
@@ -1,8 +1,20 @@
+-- Prosody IM v0.3
+-- Copyright (C) 2008-2009 Matthew Wild
+-- Copyright (C) 2008-2009 Waqas Hussain
+-- 
+-- This project is MIT/X11 licensed. Please see the
+-- COPYING file in the source package for more information.
+--
+
+
 \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 +22,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 pairs(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