core/componentmanager.lua

   1 -- Prosody IM v0.2
   2 -- Copyright (C) 2008 Matthew Wild
   3 -- Copyright (C) 2008 Waqas Hussain
   4 --
   5 -- This program is free software; you can redistribute it and/or
   6 -- modify it under the terms of the GNU General Public License
   7 -- as published by the Free Software Foundation; either version 2
   8 -- of the License, or (at your option) any later version.
   9 --
  10 -- This program is distributed in the hope that it will be useful,
  11 -- but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 -- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 -- GNU General Public License for more details.
  14 --
  15 -- You should have received a copy of the GNU General Public License
  16 -- along with this program; if not, write to the Free Software
  17 -- Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  18 --
  19
  20
  21 \r
  22 \r
  23 local log = require "util.logger".init("componentmanager");
  24 local module_load = require "core.modulemanager".load;
  25 local module_unload = require "core.modulemanager".unload;\r
  26 local jid_split = require "util.jid".split;\r
  27 local hosts = hosts;\r
  28 \r
  29 local components = {};\r
  30 \r
  31 module "componentmanager"\r
  32 \r
  33 function handle_stanza(origin, stanza)\r
  34         local node, host = jid_split(stanza.attr.to);\r
  35         local component = nil;
  36         if not component then component = components[stanza.attr.to]; end -- hack to allow hooking node@server/resource and server/resource
  37         if not component then component = components[node.."@"..host]; end -- hack to allow hooking node@server\r
  38         if not component then component = components[host]; end
  39         if component then\r
  40                 log("debug", "stanza being handled by component: "..host);\r
  41                 component(origin, stanza, hosts[host]);\r
  42         else\r
  43                 log("error", "Component manager recieved a stanza for a non-existing component: " .. stanza.attr.to);\r
  44         end\r
  45 end\r
  46 \r
  47 function register_component(host, component)\r
  48         if not hosts[host] then\r
  49                 -- TODO check for host well-formedness\r
  50                 components[host] = component;\r
  51                 hosts[host] = { type = "component", host = host, connected = true, s2sout = {} };
  52                 -- FIXME only load for a.b.c if b.c has dialback, and/or check in config
  53                 module_load(host, "dialback");\r
  54                 log("debug", "component added: "..host);\r
  55                 return hosts[host];\r
  56         else\r
  57                 log("error", "Attempt to set component for existing host: "..host);\r
  58         end\r
  59 end\r
  60
  61 function deregister_component(host, component)
  62         if components[host] then
  63                 module_unload(host, "dialback");
  64                 components[host] = nil;
  65                 hosts[host] = nil;
  66                 log("debug", "component removed: "..host);
  67                 return true;
  68         else
  69                 log("error", "Attempt to remove component for non-existing host: "..host);
  70         end
  71 end
  72 \r
  73 return _M;\r