diff options
Diffstat (limited to 'src/mod_last_odbc.erl')
| -rw-r--r-- | src/mod_last_odbc.erl | 27 |
1 files changed, 14 insertions, 13 deletions
diff --git a/src/mod_last_odbc.erl b/src/mod_last_odbc.erl index e7577ed40..ae63d9a8f 100644 --- a/src/mod_last_odbc.erl +++ b/src/mod_last_odbc.erl @@ -12,8 +12,8 @@ -behaviour(gen_mod). --export([start/1, - stop/0, +-export([start/2, + stop/1, process_local_iq/3, process_sm_iq/3, on_presence_update/4, @@ -24,24 +24,24 @@ -include("jlib.hrl"). -start(Opts) -> +start(Host, Opts) -> IQDisc = gen_mod:get_opt(iqdisc, Opts, one_queue), - gen_iq_handler:add_iq_handler(ejabberd_local, ?NS_LAST, + gen_iq_handler:add_iq_handler(ejabberd_local, Host, ?NS_LAST, ?MODULE, process_local_iq, IQDisc), - gen_iq_handler:add_iq_handler(ejabberd_sm, ?NS_LAST, + gen_iq_handler:add_iq_handler(ejabberd_sm, Host, ?NS_LAST, ?MODULE, process_sm_iq, IQDisc), - ejabberd_hooks:add(remove_user, + ejabberd_hooks:add(remove_user, Host, ?MODULE, remove_user, 50), - ejabberd_hooks:add(unset_presence_hook, + ejabberd_hooks:add(unset_presence_hook, Host, ?MODULE, on_presence_update, 50). -stop() -> - ejabberd_hooks:delete(remove_user, +stop(Host) -> + ejabberd_hooks:delete(remove_user, Host, ?MODULE, remove_user, 50), - ejabberd_hooks:delete(unset_presence_hook, + ejabberd_hooks:delete(unset_presence_hook, Host, ?MODULE, on_presence_update, 50), - gen_iq_handler:remove_iq_handler(ejabberd_local, ?NS_LAST), - gen_iq_handler:remove_iq_handler(ejabberd_sm, ?NS_LAST). + gen_iq_handler:remove_iq_handler(ejabberd_local, Host, ?NS_LAST), + gen_iq_handler:remove_iq_handler(ejabberd_sm, Host, ?NS_LAST). process_local_iq(_From, _To, #iq{type = Type, sub_el = SubEl} = IQ) -> case Type of @@ -66,7 +66,8 @@ process_sm_iq(From, To, #iq{type = Type, sub_el = SubEl} = IQ) -> Server = To#jid.lserver, {Subscription, _Groups} = ejabberd_hooks:run_fold( - roster_get_jid_info, {none, []}, [User, From]), + roster_get_jid_info, Server, + {none, []}, [User, From]), if (Subscription == both) or (Subscription == from) -> case catch mod_privacy:get_user_list(User, Server) of |