X-Git-Url: https://git.enpas.org/?a=blobdiff_plain;f=util%2Ffilters.lua;h=d143666bf8d8f212225d475cdc8312317e746ad4;hb=464a04b551dca8f5c6c11ea706704cd3b9ef489c;hp=f6a3e77ae36c1bfb6c94c0fd2fbead13908026ab;hpb=21150443be50adadd65a04396626e8a4dd7a5093;p=prosody.git diff --git a/util/filters.lua b/util/filters.lua index f6a3e77a..d143666b 100644 --- a/util/filters.lua +++ b/util/filters.lua @@ -10,20 +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 data == nil then break; end + 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) @@ -62,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;