mod_admin_telnet: Add server:memory() command to view details of Prosody's memory...

[prosody.git] / util / helpers.lua

diff --git a/util/helpers.lua b/util/helpers.lua
index a8c8c61283c389e4d75ab6d9807e51df61f643a9..08b86a7c0907ee462af619f11e982dc73afae6ef 100644 (file)
--- a/util/helpers.lua
+++ b/util/helpers.lua
@@ -6,12 +6,22 @@
 -- COPYING file in the source package for more information.
 --
 
+local debug = require "util.debug";
+
 module("helpers", package.seeall);
 
 -- Helper functions for debugging
 
 local log = require "util.logger".init("util.debug");
 
+function log_host_events(host)
+       return log_events(prosody.hosts[host].events, host);
+end
+
+function revert_log_host_events(host)
+       return revert_log_events(prosody.hosts[host].events);
+end
+
 function log_events(events, name, logger)
        local f = events.fire_event;
        if not f then
@@ -31,6 +41,35 @@ function revert_log_events(events)
        events.fire_event, events[events.fire_event] = events[events.fire_event], nil; -- :))
 end
 
+function show_events(events, specific_event)
+       local event_handlers = events._handlers;
+       local events_array = {};
+       local event_handler_arrays = {};
+       for event in pairs(events._event_map) do
+               local handlers = event_handlers[event];
+               if handlers and (event == specific_event or not specific_event) then
+                       table.insert(events_array, event);
+                       local handler_strings = {};
+                       for i, handler in ipairs(handlers) do
+                               local upvals = debug.string_from_var_table(debug.get_upvalues_table(handler));
+                               handler_strings[i] = "  "..i..": "..tostring(handler)..(upvals and ("\n        "..upvals) or "");
+                       end
+                       event_handler_arrays[event] = handler_strings;
+               end
+       end
+       table.sort(events_array);
+       local i = 1;
+       while i <= #events_array do
+               local handlers = event_handler_arrays[events_array[i]];
+               for j=#handlers, 1, -1 do
+                       table.insert(events_array, i+1, handlers[j]);
+               end
+               if i > 1 then events_array[i] = "\n"..events_array[i]; end
+               i = i + #handlers + 1
+       end
+       return table.concat(events_array, "\n");
+end
+
 function get_upvalue(f, get_name)
        local i, name, value = 0;
        repeat