util/events.lua

   1 -- Prosody IM
   2 -- Copyright (C) 2008-2010 Matthew Wild
   3 -- Copyright (C) 2008-2010 Waqas Hussain
   4 --
   5 -- This project is MIT/X11 licensed. Please see the
   6 -- COPYING file in the source package for more information.
   7 --
   8
   9
  10 local pairs = pairs;
  11 local t_insert = table.insert;
  12 local t_sort = table.sort;
  13 local setmetatable = setmetatable;
  14 local next = next;
  15
  16 module "events"
  17
  18 function new()
  19         local handlers = {};
  20         local event_map = {};
  21         local function _rebuild_index(handlers, event)
  22                 local _handlers = event_map[event];
  23                 if not _handlers or next(_handlers) == nil then return; end
  24                 local index = {};
  25                 for handler in pairs(_handlers) do
  26                         t_insert(index, handler);
  27                 end
  28                 t_sort(index, function(a, b) return _handlers[a] > _handlers[b]; end);
  29                 handlers[event] = index;
  30                 return index;
  31         end;
  32         setmetatable(handlers, { __index = _rebuild_index });
  33         local function add_handler(event, handler, priority)
  34                 local map = event_map[event];
  35                 if map then
  36                         map[handler] = priority or 0;
  37                 else
  38                         map = {[handler] = priority or 0};
  39                         event_map[event] = map;
  40                 end
  41                 handlers[event] = nil;
  42         end;
  43         local function remove_handler(event, handler)
  44                 local map = event_map[event];
  45                 if map then
  46                         map[handler] = nil;
  47                         handlers[event] = nil;
  48                 end
  49         end;
  50         local function add_handlers(handlers)
  51                 for event, handler in pairs(handlers) do
  52                         add_handler(event, handler);
  53                 end
  54         end;
  55         local function remove_handlers(handlers)
  56                 for event, handler in pairs(handlers) do
  57                         remove_handler(event, handler);
  58                 end
  59         end;
  60         local function fire_event(event, ...)
  61                 local h = handlers[event];
  62                 if h then
  63                         for i=1,#h do
  64                                 local ret = h[i](...);
  65                                 if ret ~= nil then return ret; end
  66                         end
  67                 end
  68         end;
  69         return {
  70                 add_handler = add_handler;
  71                 remove_handler = remove_handler;
  72                 add_handlers = add_handlers;
  73                 remove_handlers = remove_handlers;
  74                 fire_event = fire_event;
  75                 _handlers = handlers;
  76                 _event_map = event_map;
  77         };
  78 end
  79
  80 return _M;