mirror of
https://github.com/processone/ejabberd.git
synced 2024-12-22 17:28:25 +01:00
b0cc0cdcf9
presence to: presence is sent (EJAB-255) * src/ejabberd_c2s.erl: Likewise SVN Revision: 1881
741 lines
18 KiB
Erlang
741 lines
18 KiB
Erlang
%%%----------------------------------------------------------------------
|
|
%%% File : mod_privacy.erl
|
|
%%% Author : Alexey Shchepin <alexey@process-one.net>
|
|
%%% Purpose : jabber:iq:privacy support
|
|
%%% Created : 21 Jul 2003 by Alexey Shchepin <alexey@process-one.net>
|
|
%%%
|
|
%%%
|
|
%%% ejabberd, Copyright (C) 2002-2009 ProcessOne
|
|
%%%
|
|
%%% This program is free software; you can redistribute it and/or
|
|
%%% modify it under the terms of the GNU General Public License as
|
|
%%% published by the Free Software Foundation; either version 2 of the
|
|
%%% License, or (at your option) any later version.
|
|
%%%
|
|
%%% This program is distributed in the hope that it will be useful,
|
|
%%% but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
%%% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
%%% General Public License for more details.
|
|
%%%
|
|
%%% You should have received a copy of the GNU General Public License
|
|
%%% along with this program; if not, write to the Free Software
|
|
%%% Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
|
|
%%% 02111-1307 USA
|
|
%%%
|
|
%%%----------------------------------------------------------------------
|
|
|
|
-module(mod_privacy).
|
|
-author('alexey@process-one.net').
|
|
|
|
-behaviour(gen_mod).
|
|
|
|
-export([start/2, stop/1,
|
|
process_iq/3,
|
|
process_iq_set/4,
|
|
process_iq_get/5,
|
|
get_user_list/3,
|
|
check_packet/6,
|
|
remove_user/2,
|
|
updated_list/3]).
|
|
|
|
-include("ejabberd.hrl").
|
|
-include("jlib.hrl").
|
|
-include("mod_privacy.hrl").
|
|
|
|
|
|
start(Host, Opts) ->
|
|
IQDisc = gen_mod:get_opt(iqdisc, Opts, one_queue),
|
|
mnesia:create_table(privacy, [{disc_copies, [node()]},
|
|
{attributes, record_info(fields, privacy)}]),
|
|
update_table(),
|
|
ejabberd_hooks:add(privacy_iq_get, Host,
|
|
?MODULE, process_iq_get, 50),
|
|
ejabberd_hooks:add(privacy_iq_set, Host,
|
|
?MODULE, process_iq_set, 50),
|
|
ejabberd_hooks:add(privacy_get_user_list, Host,
|
|
?MODULE, get_user_list, 50),
|
|
ejabberd_hooks:add(privacy_check_packet, Host,
|
|
?MODULE, check_packet, 50),
|
|
ejabberd_hooks:add(privacy_updated_list, Host,
|
|
?MODULE, updated_list, 50),
|
|
ejabberd_hooks:add(remove_user, Host,
|
|
?MODULE, remove_user, 50),
|
|
gen_iq_handler:add_iq_handler(ejabberd_sm, Host, ?NS_PRIVACY,
|
|
?MODULE, process_iq, IQDisc).
|
|
|
|
stop(Host) ->
|
|
ejabberd_hooks:delete(privacy_iq_get, Host,
|
|
?MODULE, process_iq_get, 50),
|
|
ejabberd_hooks:delete(privacy_iq_set, Host,
|
|
?MODULE, process_iq_set, 50),
|
|
ejabberd_hooks:delete(privacy_get_user_list, Host,
|
|
?MODULE, get_user_list, 50),
|
|
ejabberd_hooks:delete(privacy_check_packet, Host,
|
|
?MODULE, check_packet, 50),
|
|
ejabberd_hooks:delete(privacy_updated_list, Host,
|
|
?MODULE, updated_list, 50),
|
|
ejabberd_hooks:delete(remove_user, Host,
|
|
?MODULE, remove_user, 50),
|
|
gen_iq_handler:remove_iq_handler(ejabberd_sm, Host, ?NS_PRIVACY).
|
|
|
|
process_iq(_From, _To, IQ) ->
|
|
SubEl = IQ#iq.sub_el,
|
|
IQ#iq{type = error, sub_el = [SubEl, ?ERR_NOT_ALLOWED]}.
|
|
|
|
|
|
process_iq_get(_, From, _To, #iq{sub_el = SubEl},
|
|
#userlist{name = Active}) ->
|
|
#jid{luser = LUser, lserver = LServer} = From,
|
|
{xmlelement, _, _, Els} = SubEl,
|
|
case xml:remove_cdata(Els) of
|
|
[] ->
|
|
process_lists_get(LUser, LServer, Active);
|
|
[{xmlelement, Name, Attrs, _SubEls}] ->
|
|
case Name of
|
|
"list" ->
|
|
ListName = xml:get_attr("name", Attrs),
|
|
process_list_get(LUser, LServer, ListName);
|
|
_ ->
|
|
{error, ?ERR_BAD_REQUEST}
|
|
end;
|
|
_ ->
|
|
{error, ?ERR_BAD_REQUEST}
|
|
end.
|
|
|
|
|
|
process_lists_get(LUser, LServer, Active) ->
|
|
case catch mnesia:dirty_read(privacy, {LUser, LServer}) of
|
|
{'EXIT', _Reason} ->
|
|
{error, ?ERR_INTERNAL_SERVER_ERROR};
|
|
[] ->
|
|
{result, [{xmlelement, "query", [{"xmlns", ?NS_PRIVACY}], []}]};
|
|
[#privacy{default = Default, lists = Lists}] ->
|
|
case Lists of
|
|
[] ->
|
|
{result, [{xmlelement, "query",
|
|
[{"xmlns", ?NS_PRIVACY}], []}]};
|
|
_ ->
|
|
LItems = lists:map(
|
|
fun({N, _}) ->
|
|
{xmlelement, "list",
|
|
[{"name", N}], []}
|
|
end, Lists),
|
|
DItems =
|
|
case Default of
|
|
none ->
|
|
LItems;
|
|
_ ->
|
|
[{xmlelement, "default",
|
|
[{"name", Default}], []} | LItems]
|
|
end,
|
|
ADItems =
|
|
case Active of
|
|
none ->
|
|
DItems;
|
|
_ ->
|
|
[{xmlelement, "active",
|
|
[{"name", Active}], []} | DItems]
|
|
end,
|
|
{result,
|
|
[{xmlelement, "query", [{"xmlns", ?NS_PRIVACY}],
|
|
ADItems}]}
|
|
end
|
|
end.
|
|
|
|
process_list_get(LUser, LServer, {value, Name}) ->
|
|
case catch mnesia:dirty_read(privacy, {LUser, LServer}) of
|
|
{'EXIT', _Reason} ->
|
|
{error, ?ERR_INTERNAL_SERVER_ERROR};
|
|
[] ->
|
|
{error, ?ERR_ITEM_NOT_FOUND};
|
|
%{result, [{xmlelement, "query", [{"xmlns", ?NS_PRIVACY}], []}]};
|
|
[#privacy{lists = Lists}] ->
|
|
case lists:keysearch(Name, 1, Lists) of
|
|
{value, {_, List}} ->
|
|
LItems = lists:map(fun item_to_xml/1, List),
|
|
{result,
|
|
[{xmlelement, "query", [{"xmlns", ?NS_PRIVACY}],
|
|
[{xmlelement, "list",
|
|
[{"name", Name}], LItems}]}]};
|
|
_ ->
|
|
{error, ?ERR_ITEM_NOT_FOUND}
|
|
end
|
|
end;
|
|
|
|
process_list_get(_LUser, _LServer, false) ->
|
|
{error, ?ERR_BAD_REQUEST}.
|
|
|
|
item_to_xml(Item) ->
|
|
Attrs1 = [{"action", action_to_list(Item#listitem.action)},
|
|
{"order", order_to_list(Item#listitem.order)}],
|
|
Attrs2 = case Item#listitem.type of
|
|
none ->
|
|
Attrs1;
|
|
Type ->
|
|
[{"type", type_to_list(Item#listitem.type)},
|
|
{"value", value_to_list(Type, Item#listitem.value)} |
|
|
Attrs1]
|
|
end,
|
|
SubEls = case Item#listitem.match_all of
|
|
true ->
|
|
[];
|
|
false ->
|
|
SE1 = case Item#listitem.match_iq of
|
|
true ->
|
|
[{xmlelement, "iq", [], []}];
|
|
false ->
|
|
[]
|
|
end,
|
|
SE2 = case Item#listitem.match_message of
|
|
true ->
|
|
[{xmlelement, "message", [], []} | SE1];
|
|
false ->
|
|
SE1
|
|
end,
|
|
SE3 = case Item#listitem.match_presence_in of
|
|
true ->
|
|
[{xmlelement, "presence-in", [], []} | SE2];
|
|
false ->
|
|
SE2
|
|
end,
|
|
SE4 = case Item#listitem.match_presence_out of
|
|
true ->
|
|
[{xmlelement, "presence-out", [], []} | SE3];
|
|
false ->
|
|
SE3
|
|
end,
|
|
SE4
|
|
end,
|
|
{xmlelement, "item", Attrs2, SubEls}.
|
|
|
|
|
|
action_to_list(Action) ->
|
|
case Action of
|
|
allow -> "allow";
|
|
deny -> "deny"
|
|
end.
|
|
|
|
order_to_list(Order) ->
|
|
integer_to_list(Order).
|
|
|
|
type_to_list(Type) ->
|
|
case Type of
|
|
jid -> "jid";
|
|
group -> "group";
|
|
subscription -> "subscription"
|
|
end.
|
|
|
|
value_to_list(Type, Val) ->
|
|
case Type of
|
|
jid -> jlib:jid_to_string(Val);
|
|
group -> Val;
|
|
subscription ->
|
|
case Val of
|
|
both -> "both";
|
|
to -> "to";
|
|
from -> "from";
|
|
none -> "none"
|
|
end
|
|
end.
|
|
|
|
|
|
|
|
list_to_action(S) ->
|
|
case S of
|
|
"allow" -> allow;
|
|
"deny" -> deny
|
|
end.
|
|
|
|
|
|
|
|
process_iq_set(_, From, _To, #iq{sub_el = SubEl}) ->
|
|
#jid{luser = LUser, lserver = LServer} = From,
|
|
{xmlelement, _, _, Els} = SubEl,
|
|
case xml:remove_cdata(Els) of
|
|
[{xmlelement, Name, Attrs, SubEls}] ->
|
|
ListName = xml:get_attr("name", Attrs),
|
|
case Name of
|
|
"list" ->
|
|
process_list_set(LUser, LServer, ListName,
|
|
xml:remove_cdata(SubEls));
|
|
"active" ->
|
|
process_active_set(LUser, LServer, ListName);
|
|
"default" ->
|
|
process_default_set(LUser, LServer, ListName);
|
|
_ ->
|
|
{error, ?ERR_BAD_REQUEST}
|
|
end;
|
|
_ ->
|
|
{error, ?ERR_BAD_REQUEST}
|
|
end.
|
|
|
|
|
|
process_default_set(LUser, LServer, {value, Name}) ->
|
|
F = fun() ->
|
|
case mnesia:read({privacy, {LUser, LServer}}) of
|
|
[] ->
|
|
{error, ?ERR_ITEM_NOT_FOUND};
|
|
[#privacy{lists = Lists} = P] ->
|
|
case lists:keymember(Name, 1, Lists) of
|
|
true ->
|
|
mnesia:write(P#privacy{default = Name,
|
|
lists = Lists}),
|
|
{result, []};
|
|
false ->
|
|
{error, ?ERR_ITEM_NOT_FOUND}
|
|
end
|
|
end
|
|
end,
|
|
case mnesia:transaction(F) of
|
|
{atomic, {error, _} = Error} ->
|
|
Error;
|
|
{atomic, {result, _} = Res} ->
|
|
Res;
|
|
_ ->
|
|
{error, ?ERR_INTERNAL_SERVER_ERROR}
|
|
end;
|
|
|
|
process_default_set(LUser, LServer, false) ->
|
|
F = fun() ->
|
|
case mnesia:read({privacy, {LUser, LServer}}) of
|
|
[] ->
|
|
{result, []};
|
|
[R] ->
|
|
mnesia:write(R#privacy{default = none}),
|
|
{result, []}
|
|
end
|
|
end,
|
|
case mnesia:transaction(F) of
|
|
{atomic, {error, _} = Error} ->
|
|
Error;
|
|
{atomic, {result, _} = Res} ->
|
|
Res;
|
|
_ ->
|
|
{error, ?ERR_INTERNAL_SERVER_ERROR}
|
|
end.
|
|
|
|
|
|
process_active_set(LUser, LServer, {value, Name}) ->
|
|
case catch mnesia:dirty_read(privacy, {LUser, LServer}) of
|
|
[] ->
|
|
{error, ?ERR_ITEM_NOT_FOUND};
|
|
[#privacy{lists = Lists}] ->
|
|
case lists:keysearch(Name, 1, Lists) of
|
|
{value, {_, List}} ->
|
|
NeedDb = is_list_needdb(List),
|
|
{result, [], #userlist{name = Name, list = List, needdb = NeedDb}};
|
|
false ->
|
|
{error, ?ERR_ITEM_NOT_FOUND}
|
|
end
|
|
end;
|
|
|
|
process_active_set(_LUser, _LServer, false) ->
|
|
{result, [], #userlist{}}.
|
|
|
|
|
|
process_list_set(LUser, LServer, {value, Name}, Els) ->
|
|
case parse_items(Els) of
|
|
false ->
|
|
{error, ?ERR_BAD_REQUEST};
|
|
remove ->
|
|
F =
|
|
fun() ->
|
|
case mnesia:read({privacy, {LUser, LServer}}) of
|
|
[] ->
|
|
{result, []};
|
|
[#privacy{default = Default, lists = Lists} = P] ->
|
|
% TODO: check active
|
|
if
|
|
Name == Default ->
|
|
{error, ?ERR_CONFLICT};
|
|
true ->
|
|
NewLists =
|
|
lists:keydelete(Name, 1, Lists),
|
|
mnesia:write(
|
|
P#privacy{lists = NewLists}),
|
|
{result, []}
|
|
end
|
|
end
|
|
end,
|
|
case mnesia:transaction(F) of
|
|
{atomic, {error, _} = Error} ->
|
|
Error;
|
|
{atomic, {result, _} = Res} ->
|
|
ejabberd_router:route(
|
|
jlib:make_jid(LUser, LServer, ""),
|
|
jlib:make_jid(LUser, LServer, ""),
|
|
{xmlelement, "broadcast", [],
|
|
[{privacy_list,
|
|
#userlist{name = Name, list = []},
|
|
Name}]}),
|
|
Res;
|
|
_ ->
|
|
{error, ?ERR_INTERNAL_SERVER_ERROR}
|
|
end;
|
|
List ->
|
|
F =
|
|
fun() ->
|
|
case mnesia:wread({privacy, {LUser, LServer}}) of
|
|
[] ->
|
|
NewLists = [{Name, List}],
|
|
mnesia:write(#privacy{us = {LUser, LServer},
|
|
lists = NewLists}),
|
|
{result, []};
|
|
[#privacy{lists = Lists} = P] ->
|
|
NewLists1 = lists:keydelete(Name, 1, Lists),
|
|
NewLists = [{Name, List} | NewLists1],
|
|
mnesia:write(P#privacy{lists = NewLists}),
|
|
{result, []}
|
|
end
|
|
end,
|
|
case mnesia:transaction(F) of
|
|
{atomic, {error, _} = Error} ->
|
|
Error;
|
|
{atomic, {result, _} = Res} ->
|
|
ejabberd_router:route(
|
|
jlib:make_jid(LUser, LServer, ""),
|
|
jlib:make_jid(LUser, LServer, ""),
|
|
{xmlelement, "broadcast", [],
|
|
[{privacy_list,
|
|
#userlist{name = Name, list = List},
|
|
Name}]}),
|
|
Res;
|
|
_ ->
|
|
{error, ?ERR_INTERNAL_SERVER_ERROR}
|
|
end
|
|
end;
|
|
|
|
process_list_set(_LUser, _LServer, false, _Els) ->
|
|
{error, ?ERR_BAD_REQUEST}.
|
|
|
|
|
|
parse_items([]) ->
|
|
remove;
|
|
parse_items(Els) ->
|
|
parse_items(Els, []).
|
|
|
|
parse_items([], Res) ->
|
|
%% Sort the items by their 'order' attribute
|
|
lists:keysort(#listitem.order, Res);
|
|
parse_items([{xmlelement, "item", Attrs, SubEls} | Els], Res) ->
|
|
Type = xml:get_attr("type", Attrs),
|
|
Value = xml:get_attr("value", Attrs),
|
|
SAction = xml:get_attr("action", Attrs),
|
|
SOrder = xml:get_attr("order", Attrs),
|
|
Action = case catch list_to_action(element(2, SAction)) of
|
|
{'EXIT', _} -> false;
|
|
Val -> Val
|
|
end,
|
|
Order = case catch list_to_integer(element(2, SOrder)) of
|
|
{'EXIT', _} ->
|
|
false;
|
|
IntVal ->
|
|
if
|
|
IntVal >= 0 ->
|
|
IntVal;
|
|
true ->
|
|
false
|
|
end
|
|
end,
|
|
if
|
|
(Action /= false) and (Order /= false) ->
|
|
I1 = #listitem{action = Action, order = Order},
|
|
I2 = case {Type, Value} of
|
|
{{value, T}, {value, V}} ->
|
|
case T of
|
|
"jid" ->
|
|
case jlib:string_to_jid(V) of
|
|
error ->
|
|
false;
|
|
JID ->
|
|
I1#listitem{
|
|
type = jid,
|
|
value = jlib:jid_tolower(JID)}
|
|
end;
|
|
"group" ->
|
|
I1#listitem{type = group,
|
|
value = V};
|
|
"subscription" ->
|
|
case V of
|
|
"none" ->
|
|
I1#listitem{type = subscription,
|
|
value = none};
|
|
"both" ->
|
|
I1#listitem{type = subscription,
|
|
value = both};
|
|
"from" ->
|
|
I1#listitem{type = subscription,
|
|
value = from};
|
|
"to" ->
|
|
I1#listitem{type = subscription,
|
|
value = to};
|
|
_ ->
|
|
false
|
|
end
|
|
end;
|
|
{{value, _}, false} ->
|
|
false;
|
|
_ ->
|
|
I1
|
|
end,
|
|
case I2 of
|
|
false ->
|
|
false;
|
|
_ ->
|
|
case parse_matches(I2, xml:remove_cdata(SubEls)) of
|
|
false ->
|
|
false;
|
|
I3 ->
|
|
parse_items(Els, [I3 | Res])
|
|
end
|
|
end;
|
|
true ->
|
|
false
|
|
end;
|
|
|
|
parse_items(_, _Res) ->
|
|
false.
|
|
|
|
|
|
parse_matches(Item, []) ->
|
|
Item#listitem{match_all = true};
|
|
parse_matches(Item, Els) ->
|
|
parse_matches1(Item, Els).
|
|
|
|
parse_matches1(Item, []) ->
|
|
Item;
|
|
parse_matches1(Item, [{xmlelement, "message", _, _} | Els]) ->
|
|
parse_matches1(Item#listitem{match_message = true}, Els);
|
|
parse_matches1(Item, [{xmlelement, "iq", _, _} | Els]) ->
|
|
parse_matches1(Item#listitem{match_iq = true}, Els);
|
|
parse_matches1(Item, [{xmlelement, "presence-in", _, _} | Els]) ->
|
|
parse_matches1(Item#listitem{match_presence_in = true}, Els);
|
|
parse_matches1(Item, [{xmlelement, "presence-out", _, _} | Els]) ->
|
|
parse_matches1(Item#listitem{match_presence_out = true}, Els);
|
|
parse_matches1(_Item, [{xmlelement, _, _, _} | _Els]) ->
|
|
false.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
is_list_needdb(Items) ->
|
|
lists:any(
|
|
fun(X) ->
|
|
case X#listitem.type of
|
|
subscription -> true;
|
|
group -> true;
|
|
_ -> false
|
|
end
|
|
end, Items).
|
|
|
|
get_user_list(_, User, Server) ->
|
|
LUser = jlib:nodeprep(User),
|
|
LServer = jlib:nameprep(Server),
|
|
case catch mnesia:dirty_read(privacy, {LUser, LServer}) of
|
|
[] ->
|
|
#userlist{};
|
|
[#privacy{default = Default, lists = Lists}] ->
|
|
case Default of
|
|
none ->
|
|
#userlist{};
|
|
_ ->
|
|
case lists:keysearch(Default, 1, Lists) of
|
|
{value, {_, List}} ->
|
|
NeedDb = is_list_needdb(List),
|
|
#userlist{name = Default, list = List, needdb = NeedDb};
|
|
_ ->
|
|
#userlist{}
|
|
end
|
|
end;
|
|
_ ->
|
|
#userlist{}
|
|
end.
|
|
|
|
|
|
%% From is the sender, To is the destination.
|
|
%% If Dir = out, User@Server is the sender account (From).
|
|
%% If Dir = in, User@Server is the destination account (To).
|
|
check_packet(_, User, Server,
|
|
#userlist{list = List, needdb = NeedDb},
|
|
{From, To, {xmlelement, PName, Attrs, _}},
|
|
Dir) ->
|
|
case List of
|
|
[] ->
|
|
allow;
|
|
_ ->
|
|
PType = case PName of
|
|
"message" -> message;
|
|
"iq" -> iq;
|
|
"presence" ->
|
|
case xml:get_attr_s("type", Attrs) of
|
|
%% notification
|
|
"" -> presence;
|
|
"unavailable" -> presence;
|
|
%% subscribe, subscribed, unsubscribe,
|
|
%% unsubscribed, error, probe, or other
|
|
_ -> other
|
|
end
|
|
end,
|
|
PType2 = case {PType, Dir} of
|
|
{message, in} -> message;
|
|
{iq, in} -> iq;
|
|
{presence, in} -> presence_in;
|
|
{presence, out} -> presence_out;
|
|
{_, _} -> other
|
|
end,
|
|
LJID = case Dir of
|
|
in -> jlib:jid_tolower(From);
|
|
out -> jlib:jid_tolower(To)
|
|
end,
|
|
{Subscription, Groups} =
|
|
case NeedDb of
|
|
true -> ejabberd_hooks:run_fold(roster_get_jid_info,
|
|
jlib:nameprep(Server),
|
|
{none, []},
|
|
[User, Server, LJID]);
|
|
false -> {[], []}
|
|
end,
|
|
check_packet_aux(List, PType2, LJID, Subscription, Groups)
|
|
end.
|
|
|
|
%% Ptype = mesage | iq | presence_in | presence_out | other
|
|
check_packet_aux([], _PType, _JID, _Subscription, _Groups) ->
|
|
allow;
|
|
check_packet_aux([Item | List], PType, JID, Subscription, Groups) ->
|
|
#listitem{type = Type, value = Value, action = Action} = Item,
|
|
case is_ptype_match(Item, PType) of
|
|
true ->
|
|
case Type of
|
|
none ->
|
|
Action;
|
|
_ ->
|
|
case is_type_match(Type, Value,
|
|
JID, Subscription, Groups) of
|
|
true ->
|
|
Action;
|
|
false ->
|
|
check_packet_aux(List, PType,
|
|
JID, Subscription, Groups)
|
|
end
|
|
end;
|
|
false ->
|
|
check_packet_aux(List, PType, JID, Subscription, Groups)
|
|
end.
|
|
|
|
|
|
is_ptype_match(Item, PType) ->
|
|
case Item#listitem.match_all of
|
|
true ->
|
|
true;
|
|
false ->
|
|
case PType of
|
|
message ->
|
|
Item#listitem.match_message;
|
|
iq ->
|
|
Item#listitem.match_iq;
|
|
presence_in ->
|
|
Item#listitem.match_presence_in;
|
|
presence_out ->
|
|
Item#listitem.match_presence_out;
|
|
other ->
|
|
false
|
|
end
|
|
end.
|
|
|
|
|
|
is_type_match(Type, Value, JID, Subscription, Groups) ->
|
|
case Type of
|
|
jid ->
|
|
case Value of
|
|
{"", Server, ""} ->
|
|
case JID of
|
|
{_, Server, _} ->
|
|
true;
|
|
_ ->
|
|
false
|
|
end;
|
|
{User, Server, ""} ->
|
|
case JID of
|
|
{User, Server, _} ->
|
|
true;
|
|
_ ->
|
|
false
|
|
end;
|
|
_ ->
|
|
Value == JID
|
|
end;
|
|
subscription ->
|
|
Value == Subscription;
|
|
group ->
|
|
lists:member(Value, Groups)
|
|
end.
|
|
|
|
|
|
remove_user(User, Server) ->
|
|
LUser = jlib:nodeprep(User),
|
|
LServer = jlib:nameprep(Server),
|
|
F = fun() ->
|
|
mnesia:delete({privacy,
|
|
{LUser, LServer}})
|
|
end,
|
|
mnesia:transaction(F).
|
|
|
|
|
|
updated_list(_,
|
|
#userlist{name = OldName} = Old,
|
|
#userlist{name = NewName} = New) ->
|
|
if
|
|
OldName == NewName ->
|
|
New;
|
|
true ->
|
|
Old
|
|
end.
|
|
|
|
|
|
update_table() ->
|
|
Fields = record_info(fields, privacy),
|
|
case mnesia:table_info(privacy, attributes) of
|
|
Fields ->
|
|
ok;
|
|
[user, default, lists] ->
|
|
?INFO_MSG("Converting privacy table from "
|
|
"{user, default, lists} format", []),
|
|
Host = ?MYNAME,
|
|
{atomic, ok} = mnesia:create_table(
|
|
mod_privacy_tmp_table,
|
|
[{disc_only_copies, [node()]},
|
|
{type, bag},
|
|
{local_content, true},
|
|
{record_name, privacy},
|
|
{attributes, record_info(fields, privacy)}]),
|
|
mnesia:transform_table(privacy, ignore, Fields),
|
|
F1 = fun() ->
|
|
mnesia:write_lock_table(mod_privacy_tmp_table),
|
|
mnesia:foldl(
|
|
fun(#privacy{us = U} = R, _) ->
|
|
mnesia:dirty_write(
|
|
mod_privacy_tmp_table,
|
|
R#privacy{us = {U, Host}})
|
|
end, ok, privacy)
|
|
end,
|
|
mnesia:transaction(F1),
|
|
mnesia:clear_table(privacy),
|
|
F2 = fun() ->
|
|
mnesia:write_lock_table(privacy),
|
|
mnesia:foldl(
|
|
fun(R, _) ->
|
|
mnesia:dirty_write(R)
|
|
end, ok, mod_privacy_tmp_table)
|
|
end,
|
|
mnesia:transaction(F2),
|
|
mnesia:delete_table(mod_privacy_tmp_table);
|
|
_ ->
|
|
?INFO_MSG("Recreating privacy table", []),
|
|
mnesia:transform_table(privacy, ignore, Fields)
|
|
end.
|
|
|
|
|