X-Git-Url: https://git.enpas.org/?a=blobdiff_plain;f=util%2Ffilters.lua;h=d143666bf8d8f212225d475cdc8312317e746ad4;hb=492c253d150aeb7edb6687eb9bf085be6c33133a;hp=ee0e6bbaddac2be7c8f7e04889a71fdabc877f72;hpb=7d194ebd7a69e7f19d84aa40493f8d9b1cd2e1ca;p=prosody.git diff --git a/util/filters.lua b/util/filters.lua index ee0e6bba..d143666b 100644 --- a/util/filters.lua +++ b/util/filters.lua @@ -10,19 +10,30 @@ local t_insert, t_remove = table.insert, table.remove; module "filters" +local new_filter_hooks = {}; + function initialize(session) - local filters = {}; - session.filters = filters; - - function session.filter(type, data) - local filter_list = filters[type]; - if filter_list then - for i = 1, #filter_list do - data = filter_list[i](data); + if not session.filters then + local filters = {}; + session.filters = filters; + + function session.filter(type, data) + local filter_list = filters[type]; + if filter_list then + for i = 1, #filter_list do + data = filter_list[i](data, session); + if data == nil then break; end + end end + return data; end - return data; end + + for i=1,#new_filter_hooks do + new_filter_hooks[i](session); + end + + return session.filter; end function add_filter(session, type, callback, priority) @@ -61,4 +72,16 @@ function remove_filter(session, type, callback) end end -return _M; \ No newline at end of file +function add_filter_hook(callback) + t_insert(new_filter_hooks, callback); +end + +function remove_filter_hook(callback) + for i=1,#new_filter_hooks do + if new_filter_hooks[i] == callback then + t_remove(new_filter_hooks, i); + end + end +end + +return _M;