Merge with Florob

[prosody.git] / util / helpers.lua

diff --git a/util/helpers.lua b/util/helpers.lua
index 80f72b3b267e49cb53e112a84196fce28cc09c1d..08b86a7c0907ee462af619f11e982dc73afae6ef 100644 (file)
--- a/util/helpers.lua
+++ b/util/helpers.lua
@@ -1,3 +1,12 @@
+-- Prosody IM
+-- Copyright (C) 2008-2010 Matthew Wild
+-- Copyright (C) 2008-2010 Waqas Hussain
+-- 
+-- This project is MIT/X11 licensed. Please see the
+-- COPYING file in the source package for more information.
+--
+
+local debug = require "util.debug";
 
 module("helpers", package.seeall);
 
@@ -5,6 +14,14 @@ module("helpers", package.seeall);
 
 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
@@ -14,13 +31,52 @@ function log_events(events, name, logger)
        name = name or tostring(events);
        function events.fire_event(event, ...)
                logger("debug", "%s firing event: %s", name, event);
+               return f(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; -- :)
+       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
+               i = i + 1;
+               name, value = debug.getupvalue(f, i);
+       until name == get_name or name == nil;
+       return value;
 end
 
 return _M;