From 2e194bd997132f4fde5959b54e6bfe269d4de433 Mon Sep 17 00:00:00 2001
From: Badlop <badlop@process-one.net>
Date: Wed, 22 Apr 2015 13:27:25 +0200
Subject: [PATCH] Add private and roster local handlers to get listed in
 disco#info (EJAB-682)

---
 src/mod_private.erl | 3 +++
 src/mod_roster.erl  | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/src/mod_private.erl b/src/mod_private.erl
index f09c6100e..e127c2029 100644
--- 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 605e8e367..2f5d771ce 100644
--- 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).