mirror of https://github.com/bjc/prosody
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
2.7 KiB
107 lines
2.7 KiB
-- 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 ipairs = ipairs;
|
|
local pairs = pairs;
|
|
local t_insert = table.insert;
|
|
local t_sort = table.sort;
|
|
local select = select;
|
|
|
|
module "events"
|
|
|
|
function new()
|
|
local dispatchers = {};
|
|
local handlers = {};
|
|
local event_map = {};
|
|
local function _rebuild_index(event) -- TODO optimize index rebuilding
|
|
local _handlers = event_map[event];
|
|
local index = handlers[event];
|
|
if index then
|
|
for i=#index,1,-1 do index[i] = nil; end
|
|
else index = {}; handlers[event] = index; end
|
|
for handler in pairs(_handlers) do
|
|
t_insert(index, handler);
|
|
end
|
|
t_sort(index, function(a, b) return _handlers[a] > _handlers[b]; end);
|
|
end;
|
|
local function add_handler(event, handler, priority)
|
|
local map = event_map[event];
|
|
if map then
|
|
map[handler] = priority or 0;
|
|
else
|
|
map = {[handler] = priority or 0};
|
|
event_map[event] = map;
|
|
end
|
|
_rebuild_index(event);
|
|
end;
|
|
local function remove_handler(event, handler)
|
|
local map = event_map[event];
|
|
if map then
|
|
map[handler] = nil;
|
|
_rebuild_index(event);
|
|
end
|
|
end;
|
|
local function add_handlers(handlers)
|
|
for event, handler in pairs(handlers) do
|
|
add_handler(event, handler);
|
|
end
|
|
end;
|
|
local function remove_handlers(handlers)
|
|
for event, handler in pairs(handlers) do
|
|
remove_handler(event, handler);
|
|
end
|
|
end;
|
|
local function _create_dispatcher(event) -- FIXME duplicate code in fire_event
|
|
local h = handlers[event];
|
|
if not h then h = {}; handlers[event] = h; end
|
|
local dispatcher = function(...)
|
|
for i=1,#h do
|
|
local ret = h[i](...);
|
|
if ret ~= nil then return ret; end
|
|
end
|
|
end;
|
|
dispatchers[event] = dispatcher;
|
|
return dispatcher;
|
|
end;
|
|
local function get_dispatcher(event)
|
|
return dispatchers[event] or _create_dispatcher(event);
|
|
end;
|
|
local function fire_event(event, ...) -- FIXME duplicates dispatcher code
|
|
local h = handlers[event];
|
|
if h then
|
|
for i=1,#h do
|
|
local ret = h[i](...);
|
|
if ret ~= nil then return ret; end
|
|
end
|
|
end
|
|
end;
|
|
local function get_named_arg_dispatcher(event, ...)
|
|
local dispatcher = get_dispatcher(event);
|
|
local keys = {...};
|
|
local data = {};
|
|
return function(...)
|
|
for i, key in ipairs(keys) do data[key] = select(i, ...); end
|
|
dispatcher(data);
|
|
end;
|
|
end;
|
|
return {
|
|
add_handler = add_handler;
|
|
remove_handler = remove_handler;
|
|
add_plugin = add_plugin;
|
|
remove_plugin = remove_plugin;
|
|
get_dispatcher = get_dispatcher;
|
|
fire_event = fire_event;
|
|
get_named_arg_dispatcher = get_named_arg_dispatcher;
|
|
_dispatchers = dispatchers;
|
|
_handlers = handlers;
|
|
_event_map = event_map;
|
|
};
|
|
end
|
|
|
|
return _M;
|
|
|