Add private and roster local handlers to get listed in disco#info (EJAB-682)

diff --git a/src/mod_private.erl b/src/mod_private.erl
index f09c6100edeb3a4a0419f89ec0bba1bec1beee2f..e127c20294e4354040a0ac54c24c7d2bec494616 100644 (file)
--- a/src/mod_private.erl
+++ b/src/mod_private.erl
@@ -60,12 +60,15 @@ start(Host, Opts) ->
     end,
     ejabberd_hooks:add(remove_user, Host, ?MODULE,
                       remove_user, 50),
+    gen_iq_handler:add_iq_handler(ejabberd_local, Host,
+                                 ?NS_PRIVATE, ?MODULE, process_sm_iq, IQDisc),
     gen_iq_handler:add_iq_handler(ejabberd_sm, Host,
                                  ?NS_PRIVATE, ?MODULE, process_sm_iq, IQDisc).
 
 stop(Host) ->
     ejabberd_hooks:delete(remove_user, Host, ?MODULE,
                          remove_user, 50),
+    gen_iq_handler:remove_iq_handler(ejabberd_local, Host, ?NS_PRIVATE),
     gen_iq_handler:remove_iq_handler(ejabberd_sm, Host,
                                     ?NS_PRIVATE).
 

diff --git a/src/mod_roster.erl b/src/mod_roster.erl
index 605e8e36702bb170b701ff38fe40d73fe4a47434..2f5d771ce4de26db198fc1ffdd96e384b6c96619 100644 (file)
--- a/src/mod_roster.erl
+++ b/src/mod_roster.erl
@@ -100,6 +100,8 @@ start(Host, Opts) ->
                       webadmin_page, 50),
     ejabberd_hooks:add(webadmin_user, Host, ?MODULE,
                       webadmin_user, 50),
+    gen_iq_handler:add_iq_handler(ejabberd_local, Host,
+                                 ?NS_ROSTER, ?MODULE, process_iq, IQDisc),
     gen_iq_handler:add_iq_handler(ejabberd_sm, Host,
                                  ?NS_ROSTER, ?MODULE, process_iq, IQDisc).
 
@@ -126,6 +128,7 @@ stop(Host) ->
                          webadmin_page, 50),
     ejabberd_hooks:delete(webadmin_user, Host, ?MODULE,
                          webadmin_user, 50),
+    gen_iq_handler:remove_iq_handler(ejabberd_local, Host, ?NS_ROSTER),
     gen_iq_handler:remove_iq_handler(ejabberd_sm, Host,
                                     ?NS_ROSTER).