From e95cf420a2671c0f381724ee4af719a27ce5873d Mon Sep 17 00:00:00 2001
From: Evgeniy Khramtsov <ekhramtsov@process-one.net>
Date: Thu, 3 Mar 2016 14:10:06 +0300
Subject: [PATCH] Enable flexible offline on disco#info as well

---
 src/mod_offline.erl | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/mod_offline.erl b/src/mod_offline.erl
index 0db1dd681..4e00d9235 100644
--- a/src/mod_offline.erl
+++ b/src/mod_offline.erl
@@ -341,9 +341,15 @@ get_sm_items(_Acc, #jid{luser = U, lserver = S, lresource = R} = JID,
 get_sm_items(Acc, _From, _To, _Node, _Lang) ->
     Acc.
 
-get_info(_Acc, #jid{luser = U, lserver = S}, #jid{luser = U, lserver = S},
-	 ?NS_FLEX_OFFLINE, _Lang) ->
+get_info(_Acc, #jid{luser = U, lserver = S, lresource = R},
+	 #jid{luser = U, lserver = S}, ?NS_FLEX_OFFLINE, _Lang) ->
     N = jlib:integer_to_binary(count_offline_messages(U, S)),
+    case ejabberd_sm:get_session_pid(U, S, R) of
+	Pid when is_pid(Pid) ->
+	    Pid ! dont_ask_offline;
+	none ->
+	    ok
+    end,
     [#xmlel{name = <<"x">>,
 	    attrs = [{<<"xmlns">>, ?NS_XDATA},
 		     {<<"type">>, <<"result">>}],