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
  14 module "events"
  15
  16 function new()
  17         local handlers = {};
  18         local event_map = {};
  19         local function _rebuild_index(event) -- TODO optimize index rebuilding
  20                 local _handlers = event_map[event];
  21                 local index = {};
  22                 for handler in pairs(_handlers) do
  23                         t_insert(index, handler);
  24                 end
  25                 t_sort(index, function(a, b) return _handlers[a] > _handlers[b]; end);
  26                 handlers[event] = index;
  27         end;
  28         local function add_handler(event, handler, priority)
  29                 local map = event_map[event];
  30                 if map then
  31                         map[handler] = priority or 0;
  32                 else
  33                         map = {[handler] = priority or 0};
  34                         event_map[event] = map;
  35                 end
  36                 _rebuild_index(event);
  37         end;
  38         local function remove_handler(event, handler)
  39                 local map = event_map[event];
  40                 if map then
  41                         map[handler] = nil;
  42                         _rebuild_index(event);
  43                 end
  44         end;
  45         local function add_handlers(handlers)
  46                 for event, handler in pairs(handlers) do
  47                         add_handler(event, handler);
  48                 end
  49         end;
  50         local function remove_handlers(handlers)
  51                 for event, handler in pairs(handlers) do
  52                         remove_handler(event, handler);
  53                 end
  54         end;
  55         local function fire_event(event, ...)
  56                 local h = handlers[event];
  57                 if h then
  58                         for i=1,#h do
  59                                 local ret = h[i](...);
  60                                 if ret ~= nil then return ret; end
  61                         end
  62                 end
  63         end;
  64         return {
  65                 add_handler = add_handler;
  66                 remove_handler = remove_handler;
  67                 add_handlers = add_handlers;
  68                 remove_handlers = remove_handlers;
  69                 fire_event = fire_event;
  70                 _handlers = handlers;
  71                 _event_map = event_map;
  72         };
  73 end
  74
  75 return _M;