When getting list of subscribed rooms, check all including temporary ones (#1242)

This commit is contained in:
Badlop 2016-09-19 13:46:01 +02:00
parent 9fa92092bf
commit 5bcfcf4c5e
2 changed files with 7 additions and 5 deletions

View File

@ -718,13 +718,12 @@ get_vh_rooms(Host, #rsm_in{max=M, direction=Direction, id=I, index=Index})->
index = NewIndex}}
end.
get_subscribed_rooms(ServerHost, Host, From) ->
Rooms = get_rooms(ServerHost, Host),
get_subscribed_rooms(_ServerHost, Host1, From) ->
Rooms = get_vh_rooms(Host1),
BareFrom = jid:remove_resource(From),
lists:flatmap(
fun(#muc_room{name_host = {Name, _}, opts = Opts}) ->
Subscribers = proplists:get_value(subscribers, Opts, []),
case lists:keymember(BareFrom, 1, Subscribers) of
fun(#muc_online_room{name_host = {Name, Host}, pid = Pid}) ->
case gen_fsm:sync_send_all_state_event(Pid, {is_subscribed, BareFrom}) of
true -> [jid:make(Name, Host, <<>>)];
false -> []
end;

View File

@ -800,6 +800,9 @@ handle_sync_event({muc_unsubscribe, From}, _From, StateName, StateData) ->
{error, Err} ->
{reply, {error, get_error_text(Err)}, StateName, StateData}
end;
handle_sync_event({is_subscribed, From}, _From, StateName, StateData) ->
IsSubs = ?DICT:is_key(jid:split(From), StateData#state.subscribers),
{reply, IsSubs, StateName, StateData};
handle_sync_event(_Event, _From, StateName,
StateData) ->
Reply = ok, {reply, Reply, StateName, StateData}.