Merge with 0.5
authorMatthew Wild <mwild1@gmail.com>
Wed, 22 Jul 2009 20:35:58 +0000 (21:35 +0100)
committerMatthew Wild <mwild1@gmail.com>
Wed, 22 Jul 2009 20:35:58 +0000 (21:35 +0100)
net/xmppclient_listener.lua
plugins/mod_console.lua
plugins/mod_muc.lua
prosody
tools/ejabberd2prosody.lua
tools/erlparse.lua
util/broadcast.lua [new file with mode: 0644]
util/helpers.lua [new file with mode: 0644]
util/pubsub.lua [deleted file]

index 35838692ec6f76fc5af49210b32170c57551cf16..cc65ff75d930504699be8e7a3faa6f7c2c03d9ba 100644 (file)
@@ -95,7 +95,7 @@ local function session_close(session, reason)
                end
                session.send("</stream:stream>");
                session.conn.close();
-               xmppclient.disconnect(session.conn, (reason and reason.condition) or reason or "session closed");
+               xmppclient.disconnect(session.conn, (reason and (reason.text or reason.condition)) or reason or "session closed");
        end
 end
 
index 3248ca8c557764bf108d714c383298b5caad37fd..28e12a626a90c5abf01a22620678c1815b4544a5 100644 (file)
@@ -160,7 +160,8 @@ end
 -- Anything in def_env will be accessible within the session as a global variable
 
 def_env.server = {};
-function def_env.server:reload()
+
+function def_env.server:insane_reload()
        prosody.unlock_globals();
        dofile "prosody"
        prosody = _G.prosody;
@@ -185,6 +186,11 @@ function def_env.server:uptime()
                minutes, (minutes ~= 1 and "s") or "", os.date("%c", prosody.start_time));
 end
 
+function def_env.server:shutdown(reason)
+       prosody.shutdown(reason);
+       return true, "Shutdown initiated";
+end
+
 def_env.module = {};
 
 local function get_hosts_set(hosts, module)
@@ -288,6 +294,11 @@ function def_env.config:get(host, section, key)
        return true, tostring(config_get(host, section, key));
 end
 
+function def_env.config:reload()
+       local ok, err = prosody.reload_config();
+       return ok, (ok and "Config reloaded (you may need to reload modules to take effect)") or tostring(err);
+end
+
 def_env.hosts = {};
 function def_env.hosts:list()
        for host, host_session in pairs(hosts) do
index e99ef83c0adab6f39a69d05b9e7036027750b5d3..b38468ea3151dabe3058800fc93bd057b7f84667 100644 (file)
@@ -76,6 +76,8 @@ component = register_component(muc_host, function(origin, stanza)
        handle_to_domain(origin, stanza);
 end);
 
+prosody.hosts[module:get_host()].muc = { rooms = rooms };
+
 module.unload = function()
        deregister_component(muc_host);
 end
@@ -84,4 +86,5 @@ module.save = function()
 end
 module.restore = function(data)
        rooms = data.rooms or {};
+       prosody.hosts[module:get_host()].muc = { rooms = rooms };
 end
diff --git a/prosody b/prosody
index 404dc3eab0ada98b0c902db3105c6054fb988d7d..e0d9f7681c8f7a8cd62bff3a56738341463f9941 100755 (executable)
--- a/prosody
+++ b/prosody
@@ -36,7 +36,7 @@ pcall(require, "luarocks.require")
 
 config = require "core.configmanager"
 
-do
+function read_config()
        -- TODO: Check for other formats when we add support for them
        -- Use lfs? Make a new conf/ dir?
        local ok, level, err = config.load((CFG_CONFIGDIR or ".").."/prosody.cfg.lua");
@@ -62,238 +62,271 @@ do
        end
 end
 
---- Initialize logging
-require "core.loggingmanager"
-
---- Check runtime dependencies
-require "util.dependencies"
-
---- Load socket framework
-local server = require "net.server"
-
-bare_sessions = {};
-full_sessions = {};
-hosts = {};
-
--- Global 'prosody' object
-prosody = {};
-local prosody = prosody;
-
-prosody.bare_sessions = bare_sessions;
-prosody.full_sessions = full_sessions;
-prosody.hosts = hosts;
-
-prosody.paths = { source = CFG_SOURCEDIR, config = CFG_CONFIGDIR, 
-                  plugins = CFG_PLUGINDIR, data = CFG_DATADIR };
-
-prosody.arg = arg;
-
-prosody.events = require "util.events".new();
+function load_libraries()
+       --- Initialize logging
+       require "core.loggingmanager"
+       
+       --- Check runtime dependencies
+       require "util.dependencies"
+       
+       --- Load socket framework
+       server = require "net.server"
+end    
+
+function init_global_state()
+       bare_sessions = {};
+       full_sessions = {};
+       hosts = {};
+
+       -- Global 'prosody' object
+       prosody = {};
+       local prosody = prosody;
+       
+       prosody.bare_sessions = bare_sessions;
+       prosody.full_sessions = full_sessions;
+       prosody.hosts = hosts;
+       
+       prosody.paths = { source = CFG_SOURCEDIR, config = CFG_CONFIGDIR, 
+                         plugins = CFG_PLUGINDIR, data = CFG_DATADIR };
+       
+       prosody.arg = _G.arg;
 
--- Try to determine version
-local version_file = io.open((CFG_SOURCEDIR or ".").."/prosody.version");
-if version_file then
-       prosody.version = version_file:read("*a"):gsub("%s*$", "");
-       version_file:close();
-       if #prosody.version == 12 and prosody.version:match("^[a-f0-9]+$") then
-               prosody.version = "hg:"..prosody.version;
+       prosody.events = require "util.events".new();
+       
+       
+       -- Function to reload the config file
+       function prosody.reload_config()
+               log("info", "Reloading configuration file");
+               prosody.events.fire_event("reloading-config");
+               local ok, level, err = config.load((rawget(_G, "CFG_CONFIGDIR") or ".").."/prosody.cfg.lua");
+               if not ok then
+                       if level == "parser" then
+                               log("error", "There was an error parsing the configuration file: %s", tostring(err));
+                       elseif level == "file" then
+                               log("error", "Couldn't read the config file when trying to reload: %s", tostring(err));
+                       end
+               end
+               return ok, (err and tostring(level)..": "..tostring(err)) or nil;
        end
-else
-       prosody.version = "unknown";
-end
 
-log("info", "Hello and welcome to Prosody version %s", prosody.version);
-
---- Load and initialise core modules
-require "util.import"
-require "core.xmlhandlers"
-require "core.rostermanager"
-require "core.eventmanager"
-require "core.hostmanager"
-require "core.modulemanager"
-require "core.usermanager"
-require "core.sessionmanager"
-require "core.stanza_router"
-
-require "util.array"
-require "util.iterators"
-require "util.timer"
-
--- Commented to protect us from 
--- the second kind of people
---[[ 
-pcall(require, "remdebug.engine");
-if remdebug then remdebug.engine.start() end
-]]
-
-local cl = require "net.connlisteners";
-
-require "util.stanza"
-require "util.jid"
-
-------------------------------------------------------------------------
-
-
-------------- Begin code without a home ---------------------
-
-local data_path = config.get("*", "core", "data_path") or CFG_DATADIR or "data";
-require "util.datamanager".set_data_path(data_path);
-require "util.datamanager".add_callback(function(username, host, datastore, data)
-       if config.get(host, "core", "anonymous_login") then
-               return false;
+       -- Function to reopen logfiles
+       function prosody.reopen_logfiles()
+               log("info", "Re-opening log files");
+               eventmanager.fire_event("reopen-log-files"); -- Handled by appropriate log sinks
+               prosody.events.fire_event("reopen-log-files");
        end
-       return username, host, datastore, data;
-end);
 
------------ End of out-of-place code --------------
+       -- Function to initiate prosody shutdown
+       function prosody.shutdown(reason)
+               log("info", "Shutting down: %s", reason or "unknown reason");
+               prosody.shutdown_reason = reason;
+               prosody.events.fire_event("server-stopping", {reason = reason});
+               server.setquitting(true);
+       end
+end
 
--- Function to reload the config file
-function prosody.reload_config()
-       log("info", "Reloading configuration file");
-       prosody.events.fire_event("reloading-config");
-       local ok, level, err = config.load((rawget(_G, "CFG_CONFIGDIR") or ".").."/prosody.cfg.lua");
-       if not ok then
-               if level == "parser" then
-                       log("error", "There was an error parsing the configuration file: %s", tostring(err));
-               elseif level == "file" then
-                       log("error", "Couldn't read the config file when trying to reload: %s", tostring(err));
+function read_version()
+       -- Try to determine version
+       local version_file = io.open((CFG_SOURCEDIR or ".").."/prosody.version");
+       if version_file then
+               prosody.version = version_file:read("*a"):gsub("%s*$", "");
+               version_file:close();
+               if #prosody.version == 12 and prosody.version:match("^[a-f0-9]+$") then
+                       prosody.version = "hg:"..prosody.version;
                end
+       else
+               prosody.version = "unknown";
        end
 end
 
--- Function to reopen logfiles
-function prosody.reopen_logfiles()
-       log("info", "Re-opening log files");
-       eventmanager.fire_event("reopen-log-files"); -- Handled by appropriate log sinks
-       prosody.events.fire_event("reopen-log-files");
+function load_secondary_libraries()
+       --- Load and initialise core modules
+       require "util.import"
+       require "core.xmlhandlers"
+       require "core.rostermanager"
+       require "core.eventmanager"
+       require "core.hostmanager"
+       require "core.modulemanager"
+       require "core.usermanager"
+       require "core.sessionmanager"
+       require "core.stanza_router"
+
+       require "util.array"
+       require "util.iterators"
+       require "util.timer"
+       require "util.helpers"
+       
+       -- Commented to protect us from 
+       -- the second kind of people
+       --[[ 
+       pcall(require, "remdebug.engine");
+       if remdebug then remdebug.engine.start() end
+       ]]
+
+       require "net.connlisteners";
+       
+       require "util.stanza"
+       require "util.jid"
 end
 
--- Function to initiate prosody shutdown
-function prosody.shutdown(reason)
-       log("info", "Shutting down: %s", reason or "unknown reason");
-       prosody.events.fire_event("server-stopping", {reason = reason});
-       server.setquitting(true);
+function init_data_store()
+       local data_path = config.get("*", "core", "data_path") or CFG_DATADIR or "data";
+       require "util.datamanager".set_data_path(data_path);
+       require "util.datamanager".add_callback(function(username, host, datastore, data)
+               if config.get(host, "core", "anonymous_login") then
+                       return false;
+               end
+               return username, host, datastore, data;
+       end);
 end
 
--- Signal to modules that we are ready to start
-eventmanager.fire_event("server-starting");
-prosody.events.fire_event("server-starting");
+function prepare_to_start()
+       -- Signal to modules that we are ready to start
+       eventmanager.fire_event("server-starting");
+       prosody.events.fire_event("server-starting");
 
--- Load SSL settings from config, and create a ctx table
-local global_ssl_ctx = ssl and config.get("*", "core", "ssl");
-if global_ssl_ctx then
-       local default_ssl_ctx = { mode = "server", protocol = "sslv23", capath = "/etc/ssl/certs", verify = "none"; };
-       setmetatable(global_ssl_ctx, { __index = default_ssl_ctx });
-end
+       -- Load SSL settings from config, and create a ctx table
+       local global_ssl_ctx = ssl and config.get("*", "core", "ssl");
+       if global_ssl_ctx then
+               local default_ssl_ctx = { mode = "server", protocol = "sslv23", capath = "/etc/ssl/certs", verify = "none"; };
+               setmetatable(global_ssl_ctx, { __index = default_ssl_ctx });
+       end
 
--- start listening on sockets
-function net_activate_ports(option, listener, default, conntype)
-       local ports = config.get("*", "core", option.."_ports") or default;
-       if type(ports) == "number" then ports = {ports} end;
-       
-       if type(ports) ~= "table" then
-               log("error", "core."..option.." is not a table");
-       else
-               for _, port in ipairs(ports) do
-                       if type(port) ~= "number" then
-                               log("error", "Non-numeric "..option.."_ports: "..tostring(port));
-                       else
-                               cl.start(listener, { 
-                                       ssl = conntype ~= "tcp" and global_ssl_ctx,
-                                       port = port,
-                                       interface = config.get("*", "core", option.."_interface") 
-                                               or cl.get(listener).default_interface 
-                                               or config.get("*", "core", "interface"),
-                                       type = conntype
-                               });
+       local cl = require "net.connlisteners";
+       -- start listening on sockets
+       function net_activate_ports(option, listener, default, conntype)
+               local ports = config.get("*", "core", option.."_ports") or default;
+               if type(ports) == "number" then ports = {ports} end;
+               
+               if type(ports) ~= "table" then
+                       log("error", "core."..option.." is not a table");
+               else
+                       for _, port in ipairs(ports) do
+                               if type(port) ~= "number" then
+                                       log("error", "Non-numeric "..option.."_ports: "..tostring(port));
+                               else
+                                       cl.start(listener, { 
+                                               ssl = conntype ~= "tcp" and global_ssl_ctx,
+                                               port = port,
+                                               interface = config.get("*", "core", option.."_interface") 
+                                                       or cl.get(listener).default_interface 
+                                                       or config.get("*", "core", "interface"),
+                                               type = conntype
+                                       });
+                               end
                        end
                end
        end
-end
-
-net_activate_ports("c2s", "xmppclient", {5222}, (global_ssl_ctx and "tls") or "tcp");
-net_activate_ports("s2s", "xmppserver", {5269}, "tcp");
-net_activate_ports("component", "xmppcomponent", {}, "tcp");
-net_activate_ports("legacy_ssl", "xmppclient", {}, "ssl");
 
-if cl.get("console") then
-       cl.start("console", { interface = config.get("*", "core", "console_interface") or "127.0.0.1" })
-end
+       net_activate_ports("c2s", "xmppclient", {5222}, (global_ssl_ctx and "tls") or "tcp");
+       net_activate_ports("s2s", "xmppserver", {5269}, "tcp");
+       net_activate_ports("component", "xmppcomponent", {}, "tcp");
+       net_activate_ports("legacy_ssl", "xmppclient", {}, "ssl");
+       
+       if cl.get("console") then
+               cl.start("console", { interface = config.get("*", "core", "console_interface") or "127.0.0.1" })
+       end
 
--- Catch global accesses --
-local locked_globals_mt = { __index = function (t, k) error("Attempt to read a non-existent global '"..k.."'", 2); end, __newindex = function (t, k, v) error("Attempt to set a global: "..tostring(k).." = "..tostring(v), 2); end }
+       prosody.start_time = os.time();
+end    
 
-function prosody.unlock_globals()
-       setmetatable(_G, nil);
-end
+function init_global_protection()
+       -- Catch global accesses --
+       local locked_globals_mt = { __index = function (t, k) error("Attempt to read a non-existent global '"..k.."'", 2); end, __newindex = function (t, k, v) error("Attempt to set a global: "..tostring(k).." = "..tostring(v), 2); end }
+               
+       function prosody.unlock_globals()
+               setmetatable(_G, nil);
+       end
+       
+       function prosody.lock_globals()
+               setmetatable(_G, locked_globals_mt);
+       end
 
-function prosody.lock_globals()
-       setmetatable(_G, locked_globals_mt);
+       -- And lock now...
+       prosody.lock_globals();
 end
 
--- And lock now...
-prosody.lock_globals();
-
-prosody.start_time = os.time();
-
-eventmanager.fire_event("server-started");
-prosody.events.fire_event("server-started");
-
--- Error handler for errors that make it this far
-local function catch_uncaught_error(err)
-       if err:match("%d*: interrupted!$") then
-               return "quitting";
+function loop()
+       -- Error handler for errors that make it this far
+       local function catch_uncaught_error(err)
+               if err:match("%d*: interrupted!$") then
+                       return "quitting";
+               end
+               
+               log("error", "Top-level error, please report:\n%s", tostring(err));
+               local traceback = debug.traceback("", 2);
+               if traceback then
+                       log("error", "%s", traceback);
+               end
+               
+               prosody.events.fire_event("very-bad-error", {error = err, traceback = traceback});
        end
        
-       log("error", "Top-level error, please report:\n%s", tostring(err));
-       local traceback = debug.traceback("", 2);
-       if traceback then
-               log("error", "%s", traceback);
+       while select(2, xpcall(server.loop, catch_uncaught_error)) ~= "quitting" do
+               socket.sleep(0.2);
        end
-       
-       prosody.events.fire_event("very-bad-error", {error = err, traceback = traceback});
 end
 
-while select(2, xpcall(server.loop, catch_uncaught_error)) ~= "quitting" do
-       socket.sleep(0.2);
-end
-
-log("info", "Shutdown status: Cleaning up");
-prosody.events.fire_event("server-cleanup");
-
--- Ok, we're quitting I know, but we
--- need to do some tidying before we go :)
-server.setquitting(false);
-
-log("info", "Shutdown status: Closing all active sessions");
-for hostname, host in pairs(hosts) do
-       log("debug", "Shutdown status: Closing client connections for %s", hostname)
-       if host.sessions then
-               for username, user in pairs(host.sessions) do
-                       for resource, session in pairs(user.sessions) do
-                               log("debug", "Closing connection for %s@%s/%s", username, hostname, resource);
-                               session:close("system-shutdown");
+function cleanup()
+       log("info", "Shutdown status: Cleaning up");
+       prosody.events.fire_event("server-cleanup");
+       
+       -- Ok, we're quitting I know, but we
+       -- need to do some tidying before we go :)
+       server.setquitting(false);
+       
+       log("info", "Shutdown status: Closing all active sessions");
+       for hostname, host in pairs(hosts) do
+               log("debug", "Shutdown status: Closing client connections for %s", hostname)
+               if host.sessions then
+                       local reason = { condition = "system-shutdown", text = "Server is shutting down" };
+                       if prosody.shutdown_reason then
+                               reason.text = reason.text..": "..prosody.shutdown_reason;
+                       end
+                       for username, user in pairs(host.sessions) do
+                               for resource, session in pairs(user.sessions) do
+                                       log("debug", "Closing connection for %s@%s/%s", username, hostname, resource);
+                                       session:close(reason);
+                               end
                        end
                end
-       end
        
-       log("debug", "Shutdown status: Closing outgoing s2s connections from %s", hostname);
-       if host.s2sout then
-               for remotehost, session in pairs(host.s2sout) do
-                       if session.close then
-                               session:close("system-shutdown");
-                       else
-                               log("warn", "Unable to close outgoing s2s session to %s, no session:close()?!", remotehost);
+               log("debug", "Shutdown status: Closing outgoing s2s connections from %s", hostname);
+               if host.s2sout then
+                       for remotehost, session in pairs(host.s2sout) do
+                               if session.close then
+                                       session:close("system-shutdown");
+                               else
+                                       log("warn", "Unable to close outgoing s2s session to %s, no session:close()?!", remotehost);
+                               end
                        end
                end
        end
+
+       log("info", "Shutdown status: Closing all server connections");
+       server.closeall();
+       
+       server.setquitting(true);
 end
 
-log("info", "Shutdown status: Closing all server connections");
-server.closeall();
+read_config();
+load_libraries();
+init_global_state();
+read_version();
+log("info", "Hello and welcome to Prosody version %s", prosody.version);
+load_secondary_libraries();
+init_data_store();
+prepare_to_start();
+init_global_protection();
 
-server.setquitting(true);
+eventmanager.fire_event("server-started");
+prosody.events.fire_event("server-started");
 
+loop();
+
+log("info", "Shutting down...");
+cleanup();
 eventmanager.fire_event("server-stopped");
 prosody.events.fire_event("server-stopped");
-log("info", "Shutdown status: Complete!");
+log("info", "Shutdown complete");
+
index 14a4d1d770594108eee5656933007ec9ff3109e2..4fef3f3abbf434e4eac798d8ee6650cb6f7202b1 100755 (executable)
@@ -8,44 +8,44 @@
 --
 
 
-\r
-require "erlparse";\r
-\r
-package.path = package.path ..";../?.lua";\r
+
+require "erlparse";
+
+package.path = package.path ..";../?.lua";
 local serialize = require "util.serialization".serialize;
-local st = require "util.stanza";\r
-package.loaded["util.logger"] = {init = function() return function() end; end}\r
-local dm = require "util.datamanager"\r
-dm.set_data_path("data");\r
-\r
-function build_stanza(tuple, stanza)\r
-       if tuple[1] == "xmlelement" then\r
-               local name = tuple[2];\r
-               local attr = {};\r
-               for _, a in ipairs(tuple[3]) do attr[a[1]] = a[2]; end\r
-               local up;\r
-               if stanza then stanza:tag(name, attr); up = true; else stanza = st.stanza(name, attr); end\r
-               for _, a in ipairs(tuple[4]) do build_stanza(a, stanza); end\r
-               if up then stanza:up(); else return stanza end\r
-       elseif tuple[1] == "xmlcdata" then\r
-               stanza:text(tuple[2]);\r
-       else\r
-               error("unknown element type: "..serialize(tuple));\r
-       end\r
-end\r
-function build_time(tuple)\r
-       local Megaseconds,Seconds,Microseconds = unpack(tuple);\r
-       return Megaseconds * 1000000 + Seconds;\r
-end\r
-\r
-function vcard(node, host, stanza)\r
-       local ret, err = dm.store(node, host, "vcard", st.preserialize(stanza));\r
-       print("["..(err or "success").."] vCard: "..node.."@"..host);\r
-end\r
-function password(node, host, password)\r
-       local ret, err = dm.store(node, host, "accounts", {password = password});\r
-       print("["..(err or "success").."] accounts: "..node.."@"..host.." = "..password);\r
-end\r
+local st = require "util.stanza";
+package.loaded["util.logger"] = {init = function() return function() end; end}
+local dm = require "util.datamanager"
+dm.set_data_path("data");
+
+function build_stanza(tuple, stanza)
+       if tuple[1] == "xmlelement" then
+               local name = tuple[2];
+               local attr = {};
+               for _, a in ipairs(tuple[3]) do attr[a[1]] = a[2]; end
+               local up;
+               if stanza then stanza:tag(name, attr); up = true; else stanza = st.stanza(name, attr); end
+               for _, a in ipairs(tuple[4]) do build_stanza(a, stanza); end
+               if up then stanza:up(); else return stanza end
+       elseif tuple[1] == "xmlcdata" then
+               stanza:text(tuple[2]);
+       else
+               error("unknown element type: "..serialize(tuple));
+       end
+end
+function build_time(tuple)
+       local Megaseconds,Seconds,Microseconds = unpack(tuple);
+       return Megaseconds * 1000000 + Seconds;
+end
+
+function vcard(node, host, stanza)
+       local ret, err = dm.store(node, host, "vcard", st.preserialize(stanza));
+       print("["..(err or "success").."] vCard: "..node.."@"..host);
+end
+function password(node, host, password)
+       local ret, err = dm.store(node, host, "accounts", {password = password});
+       print("["..(err or "success").."] accounts: "..node.."@"..host.." = "..password);
+end
 function roster(node, host, jid, item)
        local roster = dm.load(node, host, "roster") or {};
        roster[jid] = item;
@@ -59,99 +59,99 @@ function roster_pending(node, host, jid)
        local ret, err = dm.store(node, host, "roster", roster);
        print("["..(err or "success").."] roster: " ..node.."@"..host.." - "..jid);
 end
-function private_storage(node, host, xmlns, stanza)\r
-       local private = dm.load(node, host, "private") or {};\r
-       private[stanza.name..":"..xmlns] = st.preserialize(stanza);\r
-       local ret, err = dm.store(node, host, "private", private);\r
-       print("["..(err or "success").."] private: " ..node.."@"..host.." - "..xmlns);\r
-end\r
-function offline_msg(node, host, t, stanza)\r
-       stanza.attr.stamp = os.date("!%Y-%m-%dT%H:%M:%SZ", t);\r
-       stanza.attr.stamp_legacy = os.date("!%Y%m%dT%H:%M:%S", t);\r
-       local ret, err = dm.list_append(node, host, "offline", st.preserialize(stanza));\r
-       print("["..(err or "success").."] offline: " ..node.."@"..host.." - "..os.date("!%Y-%m-%dT%H:%M:%SZ", t));\r
-end\r
-\r
-\r
-local filters = {\r
-       passwd = function(tuple)\r
-               password(tuple[2][1], tuple[2][2], tuple[3]);\r
-       end;\r
-       vcard = function(tuple)\r
-               vcard(tuple[2][1], tuple[2][2], build_stanza(tuple[3]));\r
-       end;\r
-       roster = function(tuple)\r
-               local node = tuple[3][1]; local host = tuple[3][2];\r
-               local contact = (type(tuple[4][1]) == "table") and tuple[4][2] or tuple[4][1].."@"..tuple[4][2];\r
-               local name = tuple[5]; local subscription = tuple[6];\r
-               local ask = tuple[7]; local groups = tuple[8];\r
-               if type(name) ~= type("") then name = nil; end\r
+function private_storage(node, host, xmlns, stanza)
+       local private = dm.load(node, host, "private") or {};
+       private[stanza.name..":"..xmlns] = st.preserialize(stanza);
+       local ret, err = dm.store(node, host, "private", private);
+       print("["..(err or "success").."] private: " ..node.."@"..host.." - "..xmlns);
+end
+function offline_msg(node, host, t, stanza)
+       stanza.attr.stamp = os.date("!%Y-%m-%dT%H:%M:%SZ", t);
+       stanza.attr.stamp_legacy = os.date("!%Y%m%dT%H:%M:%S", t);
+       local ret, err = dm.list_append(node, host, "offline", st.preserialize(stanza));
+       print("["..(err or "success").."] offline: " ..node.."@"..host.." - "..os.date("!%Y-%m-%dT%H:%M:%SZ", t));
+end
+
+
+local filters = {
+       passwd = function(tuple)
+               password(tuple[2][1], tuple[2][2], tuple[3]);
+       end;
+       vcard = function(tuple)
+               vcard(tuple[2][1], tuple[2][2], build_stanza(tuple[3]));
+       end;
+       roster = function(tuple)
+               local node = tuple[3][1]; local host = tuple[3][2];
+               local contact = (type(tuple[4][1]) == "table") and tuple[4][2] or tuple[4][1].."@"..tuple[4][2];
+               local name = tuple[5]; local subscription = tuple[6];
+               local ask = tuple[7]; local groups = tuple[8];
+               if type(name) ~= type("") then name = nil; end
                if ask == "none" then ask = nil; elseif ask == "out" then ask = "subscribe" elseif ask == "in" then
                        roster_pending(node, host, contact);
                        return;
-               else error(ask) end\r
-               if subscription ~= "both" and subscription ~= "from" and subscription ~= "to" and subscription ~= "none" then error(subscription) end\r
-               local item = {name = name, ask = ask, subscription = subscription, groups = {}};\r
-               for _, g in ipairs(groups) do item.groups[g] = true; end\r
-               roster(node, host, contact, item);\r
-       end;\r
-       private_storage = function(tuple)\r
-               private_storage(tuple[2][1], tuple[2][2], tuple[2][3], build_stanza(tuple[3]));\r
-       end;\r
-       offline_msg = function(tuple)\r
-               offline_msg(tuple[2][1], tuple[2][2], build_time(tuple[3]), build_stanza(tuple[7]));\r
-       end;\r
-       config = function(tuple)\r
-               if tuple[2] == "hosts" then\r
-                       local output = io.output(); io.output("prosody.cfg.lua");\r
-                       io.write("-- Configuration imported from ejabberd --\n");\r
-                       io.write([[Host "*"\r
-       modules_enabled = {\r
-               "saslauth"; -- Authentication for clients and servers. Recommended if you want to log in.\r
-               "legacyauth"; -- Legacy authentication. Only used by some old clients and bots.\r
-               "roster"; -- Allow users to have a roster. Recommended ;)\r
-               "register"; -- Allow users to register on this server using a client\r
-               "tls"; -- Add support for secure TLS on c2s/s2s connections\r
-               "vcard"; -- Allow users to set vCards\r
-               "private"; -- Private XML storage (for room bookmarks, etc.)\r
-               "version"; -- Replies to server version requests\r
-               "dialback"; -- s2s dialback support\r
-               "uptime";\r
-               "disco";\r
-               "time";\r
-               "ping";\r
-               --"selftests";\r
-       };\r
-]]);\r
-                       for _, h in ipairs(tuple[3]) do\r
-                               io.write("Host \"" .. h .. "\"\n");\r
-                       end\r
-                       io.output(output);\r
-                       print("prosody.cfg.lua created");\r
-               end\r
-       end;\r
-};\r
-\r
-local arg = ...;\r
-local help = "/? -? ? /h -h /help -help --help";\r
-if not arg or help:find(arg, 1, true) then\r
-       print([[ejabberd db dump importer for Prosody\r
-\r
-  Usage: ejabberd2prosody.lua filename.txt\r
-\r
-The file can be generated from ejabberd using:\r
-  sudo ./bin/ejabberdctl dump filename.txt\r
-\r
-Note: The path of ejabberdctl depends on your ejabberd installation, and ejabberd needs to be running for ejabberdctl to work.]]);\r
-       os.exit(1);\r
-end\r
-local count = 0;\r
-local t = {};\r
-for item in erlparse.parseFile(arg) do\r
-       count = count + 1;\r
-       local name = item[1];\r
-       t[name] = (t[name] or 0) + 1;\r
-       --print(count, serialize(item));\r
-       if filters[name] then filters[name](item); end\r
-end\r
---print(serialize(t));\r
+               else error(ask) end
+               if subscription ~= "both" and subscription ~= "from" and subscription ~= "to" and subscription ~= "none" then error(subscription) end
+               local item = {name = name, ask = ask, subscription = subscription, groups = {}};
+               for _, g in ipairs(groups) do item.groups[g] = true; end
+               roster(node, host, contact, item);
+       end;
+       private_storage = function(tuple)
+               private_storage(tuple[2][1], tuple[2][2], tuple[2][3], build_stanza(tuple[3]));
+       end;
+       offline_msg = function(tuple)
+               offline_msg(tuple[2][1], tuple[2][2], build_time(tuple[3]), build_stanza(tuple[7]));
+       end;
+       config = function(tuple)
+               if tuple[2] == "hosts" then
+                       local output = io.output(); io.output("prosody.cfg.lua");
+                       io.write("-- Configuration imported from ejabberd --\n");
+                       io.write([[Host "*"
+       modules_enabled = {
+               "saslauth"; -- Authentication for clients and servers. Recommended if you want to log in.
+               "legacyauth"; -- Legacy authentication. Only used by some old clients and bots.
+               "roster"; -- Allow users to have a roster. Recommended ;)
+               "register"; -- Allow users to register on this server using a client
+               "tls"; -- Add support for secure TLS on c2s/s2s connections
+               "vcard"; -- Allow users to set vCards
+               "private"; -- Private XML storage (for room bookmarks, etc.)
+               "version"; -- Replies to server version requests
+               "dialback"; -- s2s dialback support
+               "uptime";
+               "disco";
+               "time";
+               "ping";
+               --"selftests";
+       };
+]]);
+                       for _, h in ipairs(tuple[3]) do
+                               io.write("Host \"" .. h .. "\"\n");
+                       end
+                       io.output(output);
+                       print("prosody.cfg.lua created");
+               end
+       end;
+};
+
+local arg = ...;
+local help = "/? -? ? /h -h /help -help --help";
+if not arg or help:find(arg, 1, true) then
+       print([[ejabberd db dump importer for Prosody
+
+  Usage: ejabberd2prosody.lua filename.txt
+
+The file can be generated from ejabberd using:
+  sudo ./bin/ejabberdctl dump filename.txt
+
+Note: The path of ejabberdctl depends on your ejabberd installation, and ejabberd needs to be running for ejabberdctl to work.]]);
+       os.exit(1);
+end
+local count = 0;
+local t = {};
+for item in erlparse.parseFile(arg) do
+       count = count + 1;
+       local name = item[1];
+       t[name] = (t[name] or 0) + 1;
+       --print(count, serialize(item));
+       if filters[name] then filters[name](item); end
+end
+--print(serialize(t));
index 2e14f41518ca6ac912a2a43ec782f7e210e8c989..0bd4b3b84a61988c03ca383b61509ac24363c3f1 100644 (file)
 --
 
 
-\r
-local file = nil;\r
-local last = nil;\r
-local function read(expected)\r
-       local ch;\r
-       if last then\r
-               ch = last; last = nil;\r
-       else ch = file:read(1); end\r
-       if expected and ch ~= expected then error("expected: "..expected.."; got: "..(ch or "nil")); end\r
-       return ch;\r
-end\r
-local function pushback(ch)\r
-       if last then error(); end\r
-       last = ch;\r
-end\r
-local function peek()\r
-       if not last then last = read(); end\r
-       return last;\r
-end\r
-\r
-local _A, _a, _Z, _z, _0, _9, __, _space = string.byte("AaZz09_ ", 1, 8);\r
-local function isAlpha(ch)\r
-       ch = string.byte(ch) or 0;\r
-       return (ch >= _A and ch <= _Z) or (ch >= _a and ch <= _z);\r
-end\r
-local function isNumeric(ch)\r
-       ch = string.byte(ch) or 0;\r
-       return (ch >= _0 and ch <= _9);\r
-end\r
-local function isVar(ch)\r
-       ch = string.byte(ch) or 0;\r
-       return (ch >= _A and ch <= _Z) or (ch >= _a and ch <= _z) or (ch >= _0 and ch <= _9) or ch == __;\r
-end\r
-local function isSpace(ch)\r
-       ch = string.byte(ch) or "x";\r
-       return ch <= _space;\r
-end\r
-\r
-local function readString()\r
-       read("\""); -- skip quote\r
-       local slash = nil;\r
-       local str = "";\r
-       while true do\r
-               local ch = read();\r
-               if ch == "\"" and not slash then break; end\r
-               str = str..ch;\r
-       end\r
-       str = str:gsub("\\.", {["\\b"]="\b", ["\\d"]="\d", ["\\e"]="\e", ["\\f"]="\f", ["\\n"]="\n", ["\\r"]="\r", ["\\s"]="\s", ["\\t"]="\t", ["\\v"]="\v", ["\\\""]="\"", ["\\'"]="'", ["\\\\"]="\\"});\r
-       return str;\r
-end\r
-local function readSpecialString()\r
-       read("<"); read("<"); -- read <<\r
-       local str = "";\r
-       if peek() == "\"" then\r
-               str = readString();\r
-       elseif peek() ~= ">" then\r
-               error();\r
-       end\r
-       read(">"); read(">"); -- read >>\r
-       return str;\r
-end\r
-local function readVar()\r
-       local var = read();\r
-       while isVar(peek()) do\r
-               var = var..read();\r
-       end\r
-       return var;\r
-end\r
-local function readNumber()\r
-       local num = read();\r
-       while isNumeric(peek()) do\r
-               num = num..read();\r
-       end\r
-       return tonumber(num);\r
-end\r
-local readItem = nil;\r
-local function readTuple()\r
-       local t = {};\r
-       read(); -- read { or [\r
-       while true do\r
-               local item = readItem();\r
-               if not item then break; end\r
-               table.insert(t, item);\r
-       end\r
-       read(); -- read } or ]\r
-       return t;\r
-end\r
-readItem = function()\r
-       local ch = peek();\r
-       if ch == nil then return nil end\r
-       if ch == "{" or ch == "[" then\r
-               return readTuple();\r
-       elseif isAlpha(ch) then\r
-               return readVar();\r
-       elseif isNumeric(ch) then\r
-               return readNumber();\r
-       elseif ch == "\"" then\r
-               return readString();\r
-       elseif ch == "<" then\r
-               return readSpecialString();\r
-       elseif isSpace(ch) or ch == "," or ch == "|" then\r
-               read();\r
-               return readItem();\r
-       else\r
-               --print("Unknown char: "..ch);\r
-               return nil;\r
-       end\r
-end\r
-local function readChunk()\r
-       local x = readItem();\r
-       if x then read("."); end\r
-       return x;\r
-end\r
-local function readFile(filename)\r
-       file = io.open(filename);\r
-       if not file then error("File not found: "..filename); os.exit(0); end\r
-       return function()\r
-               local x = readChunk();\r
-               if not x and peek() then error("Invalid char: "..peek()); end\r
-               return x;\r
-       end;\r
-end\r
-\r
-module "erlparse"\r
-\r
-function parseFile(file)\r
-       return readFile(file);\r
-end\r
-\r
-return _M;\r
+
+local file = nil;
+local last = nil;
+local function read(expected)
+       local ch;
+       if last then
+               ch = last; last = nil;
+       else ch = file:read(1); end
+       if expected and ch ~= expected then error("expected: "..expected.."; got: "..(ch or "nil")); end
+       return ch;
+end
+local function pushback(ch)
+       if last then error(); end
+       last = ch;
+end
+local function peek()
+       if not last then last = read(); end
+       return last;
+end
+
+local _A, _a, _Z, _z, _0, _9, __, _space = string.byte("AaZz09_ ", 1, 8);
+local function isAlpha(ch)
+       ch = string.byte(ch) or 0;
+       return (ch >= _A and ch <= _Z) or (ch >= _a and ch <= _z);
+end
+local function isNumeric(ch)
+       ch = string.byte(ch) or 0;
+       return (ch >= _0 and ch <= _9);
+end
+local function isVar(ch)
+       ch = string.byte(ch) or 0;
+       return (ch >= _A and ch <= _Z) or (ch >= _a and ch <= _z) or (ch >= _0 and ch <= _9) or ch == __;
+end
+local function isSpace(ch)
+       ch = string.byte(ch) or "x";
+       return ch <= _space;
+end
+
+local function readString()
+       read("\""); -- skip quote
+       local slash = nil;
+       local str = "";
+       while true do
+               local ch = read();
+               if ch == "\"" and not slash then break; end
+               str = str..ch;
+       end
+       str = str:gsub("\\.", {["\\b"]="\b", ["\\d"]="\d", ["\\e"]="\e", ["\\f"]="\f", ["\\n"]="\n", ["\\r"]="\r", ["\\s"]="\s", ["\\t"]="\t", ["\\v"]="\v", ["\\\""]="\"", ["\\'"]="'", ["\\\\"]="\\"});
+       return str;
+end
+local function readSpecialString()
+       read("<"); read("<"); -- read <<
+       local str = "";
+       if peek() == "\"" then
+               str = readString();
+       elseif peek() ~= ">" then
+               error();
+       end
+       read(">"); read(">"); -- read >>
+       return str;
+end
+local function readVar()
+       local var = read();
+       while isVar(peek()) do
+               var = var..read();
+       end
+       return var;
+end
+local function readNumber()
+       local num = read();
+       while isNumeric(peek()) do
+               num = num..read();
+       end
+       return tonumber(num);
+end
+local readItem = nil;
+local function readTuple()
+       local t = {};
+       read(); -- read { or [
+       while true do
+               local item = readItem();
+               if not item then break; end
+               table.insert(t, item);
+       end
+       read(); -- read } or ]
+       return t;
+end
+readItem = function()
+       local ch = peek();
+       if ch == nil then return nil end
+       if ch == "{" or ch == "[" then
+               return readTuple();
+       elseif isAlpha(ch) then
+               return readVar();
+       elseif isNumeric(ch) then
+               return readNumber();
+       elseif ch == "\"" then
+               return readString();
+       elseif ch == "<" then
+               return readSpecialString();
+       elseif isSpace(ch) or ch == "," or ch == "|" then
+               read();
+               return readItem();
+       else
+               --print("Unknown char: "..ch);
+               return nil;
+       end
+end
+local function readChunk()
+       local x = readItem();
+       if x then read("."); end
+       return x;
+end
+local function readFile(filename)
+       file = io.open(filename);
+       if not file then error("File not found: "..filename); os.exit(0); end
+       return function()
+               local x = readChunk();
+               if not x and peek() then error("Invalid char: "..peek()); end
+               return x;
+       end;
+end
+
+module "erlparse"
+
+function parseFile(file)
+       return readFile(file);
+end
+
+return _M;
diff --git a/util/broadcast.lua b/util/broadcast.lua
new file mode 100644 (file)
index 0000000..8f6af2f
--- /dev/null
@@ -0,0 +1,68 @@
+-- Prosody IM
+-- 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.
+--
+
+
+local ipairs, pairs, setmetatable, type = 
+        ipairs, pairs, setmetatable, type;
+
+module "pubsub"
+
+local pubsub_node_mt = { __index = _M };
+
+function new_node(name)
+       return setmetatable({ name = name, subscribers = {} }, pubsub_node_mt);
+end
+
+function set_subscribers(node, subscribers_list, list_type)
+       local subscribers = node.subscribers;
+       
+       if list_type == "array" then
+               for _, jid in ipairs(subscribers_list) do
+                       if not subscribers[jid] then
+                               node:add_subscriber(jid);
+                       end
+               end
+       elseif (not list_type) or list_type == "set" then
+               for jid in pairs(subscribers_list) do
+                       if type(jid) == "string" then
+                               node:add_subscriber(jid);
+                       end
+               end
+       end
+end
+
+function get_subscribers(node)
+       return node.subscribers;
+end
+
+function publish(node, item, dispatcher, data)
+       local subscribers = node.subscribers;
+       for i = 1,#subscribers do
+               item.attr.to = subscribers[i];
+               dispatcher(data, item);
+       end
+end
+
+function add_subscriber(node, jid)
+       local subscribers = node.subscribers;
+       if not subscribers[jid] then
+               local space = #subscribers;
+               subscribers[space] = jid;
+               subscribers[jid] = space;
+       end
+end
+
+function remove_subscriber(node, jid)
+       local subscribers = node.subscribers;
+       if subscribers[jid] then
+               subscribers[subscribers[jid]] = nil;
+               subscribers[jid] = nil;
+       end
+end
+
+return _M;
diff --git a/util/helpers.lua b/util/helpers.lua
new file mode 100644 (file)
index 0000000..80f72b3
--- /dev/null
@@ -0,0 +1,26 @@
+
+module("helpers", package.seeall);
+
+-- Helper functions for debugging
+
+local log = require "util.logger".init("util.debug");
+
+function log_events(events, name, logger)
+       local f = events.fire_event;
+       if not f then
+               error("Object does not appear to be a util.events object");
+       end
+       logger = logger or log;
+       name = name or tostring(events);
+       function events.fire_event(event, ...)
+               logger("debug", "%s firing event: %s", name, event);
+       end
+       events[events.fire_event] = f;
+       return events;
+end
+
+function revert_log_events(events)
+       events.fire_event, events[events.fire_event] = events[events.fire_event], nil; -- :)
+end
+
+return _M;
diff --git a/util/pubsub.lua b/util/pubsub.lua
deleted file mode 100644 (file)
index 8f6af2f..0000000
+++ /dev/null
@@ -1,68 +0,0 @@
--- Prosody IM
--- 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.
---
-
-
-local ipairs, pairs, setmetatable, type = 
-        ipairs, pairs, setmetatable, type;
-
-module "pubsub"
-
-local pubsub_node_mt = { __index = _M };
-
-function new_node(name)
-       return setmetatable({ name = name, subscribers = {} }, pubsub_node_mt);
-end
-
-function set_subscribers(node, subscribers_list, list_type)
-       local subscribers = node.subscribers;
-       
-       if list_type == "array" then
-               for _, jid in ipairs(subscribers_list) do
-                       if not subscribers[jid] then
-                               node:add_subscriber(jid);
-                       end
-               end
-       elseif (not list_type) or list_type == "set" then
-               for jid in pairs(subscribers_list) do
-                       if type(jid) == "string" then
-                               node:add_subscriber(jid);
-                       end
-               end
-       end
-end
-
-function get_subscribers(node)
-       return node.subscribers;
-end
-
-function publish(node, item, dispatcher, data)
-       local subscribers = node.subscribers;
-       for i = 1,#subscribers do
-               item.attr.to = subscribers[i];
-               dispatcher(data, item);
-       end
-end
-
-function add_subscriber(node, jid)
-       local subscribers = node.subscribers;
-       if not subscribers[jid] then
-               local space = #subscribers;
-               subscribers[space] = jid;
-               subscribers[jid] = space;
-       end
-end
-
-function remove_subscriber(node, jid)
-       local subscribers = node.subscribers;
-       if subscribers[jid] then
-               subscribers[subscribers[jid]] = nil;
-               subscribers[jid] = nil;
-       end
-end
-
-return _M;