2003-01-02 22:01:12 +01:00
|
|
|
%%%----------------------------------------------------------------------
|
|
|
|
%%% File : mod_vcard.erl
|
|
|
|
%%% Author : Alexey Shchepin <alexey@sevcom.net>
|
|
|
|
%%% Purpose :
|
|
|
|
%%% Created : 2 Jan 2003 by Alexey Shchepin <alexey@sevcom.net>
|
|
|
|
%%%----------------------------------------------------------------------
|
|
|
|
|
|
|
|
-module(mod_vcard).
|
|
|
|
-author('alexey@sevcom.net').
|
|
|
|
|
2003-01-24 21:18:33 +01:00
|
|
|
-behaviour(gen_mod).
|
|
|
|
|
2005-08-25 22:48:45 +02:00
|
|
|
-export([start/2, init/3, stop/1,
|
2005-09-04 03:58:47 +02:00
|
|
|
get_sm_features/5,
|
2003-01-02 22:01:12 +01:00
|
|
|
process_local_iq/3,
|
2003-01-09 20:59:16 +01:00
|
|
|
process_sm_iq/3,
|
2003-01-26 21:16:53 +01:00
|
|
|
reindex_vcards/0,
|
2005-04-17 20:08:34 +02:00
|
|
|
remove_user/2]).
|
2003-01-02 22:01:12 +01:00
|
|
|
|
|
|
|
-include("ejabberd.hrl").
|
2003-03-09 21:46:47 +01:00
|
|
|
-include("jlib.hrl").
|
2003-01-02 22:01:12 +01:00
|
|
|
|
|
|
|
|
2004-09-10 22:57:00 +02:00
|
|
|
-define(JUD_MATCHES, 30).
|
2003-10-28 21:26:43 +01:00
|
|
|
|
2005-04-17 20:08:34 +02:00
|
|
|
-record(vcard_search, {us,
|
|
|
|
user, luser,
|
2003-01-09 20:59:16 +01:00
|
|
|
fn, lfn,
|
|
|
|
family, lfamily,
|
|
|
|
given, lgiven,
|
|
|
|
middle, lmiddle,
|
|
|
|
nickname, lnickname,
|
|
|
|
bday, lbday,
|
|
|
|
ctry, lctry,
|
|
|
|
locality, llocality,
|
|
|
|
email, lemail,
|
|
|
|
orgname, lorgname,
|
|
|
|
orgunit, lorgunit
|
|
|
|
}).
|
2005-04-17 20:08:34 +02:00
|
|
|
-record(vcard, {us, vcard}).
|
2003-01-02 22:01:12 +01:00
|
|
|
|
2005-06-20 05:18:13 +02:00
|
|
|
-define(PROCNAME, ejabberd_mod_vcard).
|
2003-01-02 22:01:12 +01:00
|
|
|
|
2005-06-20 05:18:13 +02:00
|
|
|
start(Host, Opts) ->
|
2003-01-02 22:01:12 +01:00
|
|
|
mnesia:create_table(vcard, [{disc_only_copies, [node()]},
|
|
|
|
{attributes, record_info(fields, vcard)}]),
|
|
|
|
mnesia:create_table(vcard_search,
|
|
|
|
[{disc_copies, [node()]},
|
|
|
|
{attributes, record_info(fields, vcard_search)}]),
|
2005-04-17 20:08:34 +02:00
|
|
|
update_tables(),
|
2004-11-21 22:35:51 +01:00
|
|
|
mnesia:add_table_index(vcard_search, luser),
|
2003-01-09 20:59:16 +01:00
|
|
|
mnesia:add_table_index(vcard_search, lfn),
|
|
|
|
mnesia:add_table_index(vcard_search, lfamily),
|
|
|
|
mnesia:add_table_index(vcard_search, lgiven),
|
|
|
|
mnesia:add_table_index(vcard_search, lmiddle),
|
|
|
|
mnesia:add_table_index(vcard_search, lnickname),
|
|
|
|
mnesia:add_table_index(vcard_search, lbday),
|
|
|
|
mnesia:add_table_index(vcard_search, lctry),
|
|
|
|
mnesia:add_table_index(vcard_search, llocality),
|
|
|
|
mnesia:add_table_index(vcard_search, lemail),
|
|
|
|
mnesia:add_table_index(vcard_search, lorgname),
|
|
|
|
mnesia:add_table_index(vcard_search, lorgunit),
|
2003-01-02 22:01:12 +01:00
|
|
|
|
2005-06-20 05:18:13 +02:00
|
|
|
ejabberd_hooks:add(remove_user, Host,
|
2004-12-19 21:47:35 +01:00
|
|
|
?MODULE, remove_user, 50),
|
2003-01-24 21:18:33 +01:00
|
|
|
IQDisc = gen_mod:get_opt(iqdisc, Opts, one_queue),
|
2005-06-20 05:18:13 +02:00
|
|
|
gen_iq_handler:add_iq_handler(ejabberd_local, Host, ?NS_VCARD,
|
2003-01-24 21:18:33 +01:00
|
|
|
?MODULE, process_local_iq, IQDisc),
|
2005-06-20 05:18:13 +02:00
|
|
|
gen_iq_handler:add_iq_handler(ejabberd_sm, Host, ?NS_VCARD,
|
2003-01-24 21:18:33 +01:00
|
|
|
?MODULE, process_sm_iq, IQDisc),
|
2005-09-04 03:58:47 +02:00
|
|
|
ejabberd_hooks:add(disco_sm_features, Host, ?MODULE, get_sm_features, 50),
|
2007-08-25 19:24:00 +02:00
|
|
|
MyHost = gen_mod:get_opt_host(Host, Opts, "vjud.@HOST@"),
|
2004-07-10 00:34:26 +02:00
|
|
|
Search = gen_mod:get_opt(search, Opts, true),
|
2005-06-20 05:18:13 +02:00
|
|
|
register(gen_mod:get_module_proc(Host, ?PROCNAME),
|
2005-08-25 22:48:45 +02:00
|
|
|
spawn(?MODULE, init, [MyHost, Host, Search])).
|
2003-01-02 22:01:12 +01:00
|
|
|
|
2003-01-16 21:24:53 +01:00
|
|
|
|
2005-08-25 22:48:45 +02:00
|
|
|
init(Host, ServerHost, Search) ->
|
2004-07-10 00:34:26 +02:00
|
|
|
case Search of
|
2004-07-11 22:51:54 +02:00
|
|
|
false ->
|
2005-08-25 22:48:45 +02:00
|
|
|
loop(Host, ServerHost);
|
2004-07-10 00:34:26 +02:00
|
|
|
_ ->
|
2005-08-25 22:48:45 +02:00
|
|
|
ejabberd_router:register_route(Host),
|
|
|
|
loop(Host, ServerHost)
|
2004-07-10 00:34:26 +02:00
|
|
|
end.
|
2003-01-02 22:01:12 +01:00
|
|
|
|
2005-08-25 22:48:45 +02:00
|
|
|
loop(Host, ServerHost) ->
|
2003-01-02 22:01:12 +01:00
|
|
|
receive
|
|
|
|
{route, From, To, Packet} ->
|
2005-08-25 22:48:45 +02:00
|
|
|
case catch do_route(ServerHost, From, To, Packet) of
|
2003-10-07 22:31:44 +02:00
|
|
|
{'EXIT', Reason} ->
|
|
|
|
?ERROR_MSG("~p", [Reason]);
|
|
|
|
_ ->
|
|
|
|
ok
|
|
|
|
end,
|
2005-08-25 22:48:45 +02:00
|
|
|
loop(Host, ServerHost);
|
2004-07-10 00:34:26 +02:00
|
|
|
stop ->
|
2005-08-25 22:48:45 +02:00
|
|
|
ejabberd_router:unregister_route(Host),
|
2004-07-10 00:34:26 +02:00
|
|
|
ok;
|
2003-01-02 22:01:12 +01:00
|
|
|
_ ->
|
2005-08-25 22:48:45 +02:00
|
|
|
loop(Host, ServerHost)
|
2003-01-02 22:01:12 +01:00
|
|
|
end.
|
|
|
|
|
2005-06-20 05:18:13 +02:00
|
|
|
stop(Host) ->
|
|
|
|
ejabberd_hooks:delete(remove_user, Host,
|
2004-12-19 21:47:35 +01:00
|
|
|
?MODULE, remove_user, 50),
|
2005-06-20 05:18:13 +02:00
|
|
|
gen_iq_handler:remove_iq_handler(ejabberd_local, Host, ?NS_VCARD),
|
|
|
|
gen_iq_handler:remove_iq_handler(ejabberd_sm, Host, ?NS_VCARD),
|
2005-09-04 03:58:47 +02:00
|
|
|
ejabberd_hooks:delete(disco_sm_features, Host, ?MODULE, get_sm_features, 50),
|
2005-06-20 05:18:13 +02:00
|
|
|
Proc = gen_mod:get_module_proc(Host, ?PROCNAME),
|
|
|
|
Proc ! stop,
|
|
|
|
{wait, Proc}.
|
2003-01-02 22:01:12 +01:00
|
|
|
|
2005-09-04 03:58:47 +02:00
|
|
|
get_sm_features({error, _Error} = Acc, _From, _To, _Node, _Lang) ->
|
|
|
|
Acc;
|
|
|
|
|
|
|
|
get_sm_features(Acc, _From, _To, Node, _Lang) ->
|
|
|
|
case Node of
|
|
|
|
[] ->
|
|
|
|
case Acc of
|
|
|
|
{result, Features} ->
|
|
|
|
{result, [?NS_VCARD | Features]};
|
|
|
|
empty ->
|
|
|
|
{result, [?NS_VCARD]}
|
|
|
|
end;
|
|
|
|
_ ->
|
|
|
|
Acc
|
|
|
|
end.
|
|
|
|
|
2004-02-26 23:00:04 +01:00
|
|
|
process_local_iq(_From, _To, #iq{type = Type, lang = Lang, sub_el = SubEl} = IQ) ->
|
2003-01-02 22:01:12 +01:00
|
|
|
case Type of
|
|
|
|
set ->
|
2003-12-17 21:13:21 +01:00
|
|
|
IQ#iq{type = error, sub_el = [SubEl, ?ERR_NOT_ALLOWED]};
|
2003-01-02 22:01:12 +01:00
|
|
|
get ->
|
2003-12-17 21:13:21 +01:00
|
|
|
IQ#iq{type = result,
|
|
|
|
sub_el = [{xmlelement, "vCard",
|
|
|
|
[{"xmlns", ?NS_VCARD}],
|
|
|
|
[{xmlelement, "FN", [],
|
|
|
|
[{xmlcdata, "ejabberd"}]},
|
|
|
|
{xmlelement, "URL", [],
|
|
|
|
[{xmlcdata,
|
|
|
|
"http://ejabberd.jabberstudio.org/"}]},
|
|
|
|
{xmlelement, "DESC", [],
|
2004-02-26 23:00:04 +01:00
|
|
|
[{xmlcdata,
|
|
|
|
translate:translate(
|
|
|
|
Lang,
|
|
|
|
"Erlang Jabber Server\n"
|
2006-02-06 06:12:54 +01:00
|
|
|
"Copyright (c) 2002-2006 Alexey Shchepin")}]},
|
2003-12-17 21:13:21 +01:00
|
|
|
{xmlelement, "BDAY", [],
|
|
|
|
[{xmlcdata, "2002-11-16"}]}
|
|
|
|
]}]}
|
2003-01-02 22:01:12 +01:00
|
|
|
end.
|
|
|
|
|
|
|
|
|
2003-12-17 21:13:21 +01:00
|
|
|
process_sm_iq(From, To, #iq{type = Type, sub_el = SubEl} = IQ) ->
|
2003-01-02 22:01:12 +01:00
|
|
|
case Type of
|
|
|
|
set ->
|
2005-09-04 03:58:47 +02:00
|
|
|
#jid{user = User, lserver = LServer} = From,
|
2005-04-17 20:08:34 +02:00
|
|
|
case lists:member(LServer, ?MYHOSTS) of
|
|
|
|
true ->
|
|
|
|
set_vcard(User, LServer, SubEl),
|
2003-12-17 21:13:21 +01:00
|
|
|
IQ#iq{type = result, sub_el = []};
|
2005-04-17 20:08:34 +02:00
|
|
|
false ->
|
2003-12-17 21:13:21 +01:00
|
|
|
IQ#iq{type = error, sub_el = [SubEl, ?ERR_NOT_ALLOWED]}
|
2003-07-21 22:01:22 +02:00
|
|
|
end;
|
2003-01-02 22:01:12 +01:00
|
|
|
get ->
|
2005-04-17 20:08:34 +02:00
|
|
|
#jid{luser = LUser, lserver = LServer} = To,
|
|
|
|
US = {LUser, LServer},
|
2003-01-02 22:01:12 +01:00
|
|
|
F = fun() ->
|
2005-04-17 20:08:34 +02:00
|
|
|
mnesia:read({vcard, US})
|
2003-01-02 22:01:12 +01:00
|
|
|
end,
|
|
|
|
Els = case mnesia:transaction(F) of
|
|
|
|
{atomic, Rs} ->
|
|
|
|
lists:map(fun(R) ->
|
|
|
|
R#vcard.vcard
|
|
|
|
end, Rs);
|
2003-10-20 20:23:30 +02:00
|
|
|
{aborted, _Reason} ->
|
2003-01-02 22:01:12 +01:00
|
|
|
[]
|
|
|
|
end,
|
2003-12-17 21:13:21 +01:00
|
|
|
IQ#iq{type = result, sub_el = Els}
|
2003-01-02 22:01:12 +01:00
|
|
|
end.
|
|
|
|
|
2005-04-17 20:08:34 +02:00
|
|
|
set_vcard(User, LServer, VCARD) ->
|
2003-01-02 22:01:12 +01:00
|
|
|
FN = xml:get_path_s(VCARD, [{elem, "FN"}, cdata]),
|
|
|
|
Family = xml:get_path_s(VCARD, [{elem, "N"}, {elem, "FAMILY"}, cdata]),
|
|
|
|
Given = xml:get_path_s(VCARD, [{elem, "N"}, {elem, "GIVEN"}, cdata]),
|
|
|
|
Middle = xml:get_path_s(VCARD, [{elem, "N"}, {elem, "MIDDLE"}, cdata]),
|
|
|
|
Nickname = xml:get_path_s(VCARD, [{elem, "NICKNAME"}, cdata]),
|
|
|
|
BDay = xml:get_path_s(VCARD, [{elem, "BDAY"}, cdata]),
|
2003-01-05 21:24:59 +01:00
|
|
|
CTRY = xml:get_path_s(VCARD, [{elem, "ADR"}, {elem, "CTRY"}, cdata]),
|
|
|
|
Locality = xml:get_path_s(VCARD, [{elem, "ADR"}, {elem, "LOCALITY"},cdata]),
|
2004-03-08 21:20:15 +01:00
|
|
|
EMail1 = xml:get_path_s(VCARD, [{elem, "EMAIL"}, {elem, "USERID"},cdata]),
|
|
|
|
EMail2 = xml:get_path_s(VCARD, [{elem, "EMAIL"}, cdata]),
|
2003-01-02 22:01:12 +01:00
|
|
|
OrgName = xml:get_path_s(VCARD, [{elem, "ORG"}, {elem, "ORGNAME"}, cdata]),
|
|
|
|
OrgUnit = xml:get_path_s(VCARD, [{elem, "ORG"}, {elem, "ORGUNIT"}, cdata]),
|
2004-03-08 21:20:15 +01:00
|
|
|
EMail = case EMail1 of
|
|
|
|
"" ->
|
|
|
|
EMail2;
|
|
|
|
_ ->
|
|
|
|
EMail1
|
|
|
|
end,
|
2003-01-03 20:36:59 +01:00
|
|
|
|
2003-10-07 22:31:44 +02:00
|
|
|
LUser = jlib:nodeprep(User),
|
|
|
|
LFN = stringprep:tolower(FN),
|
|
|
|
LFamily = stringprep:tolower(Family),
|
|
|
|
LGiven = stringprep:tolower(Given),
|
|
|
|
LMiddle = stringprep:tolower(Middle),
|
|
|
|
LNickname = stringprep:tolower(Nickname),
|
|
|
|
LBDay = stringprep:tolower(BDay),
|
|
|
|
LCTRY = stringprep:tolower(CTRY),
|
|
|
|
LLocality = stringprep:tolower(Locality),
|
|
|
|
LEMail = stringprep:tolower(EMail),
|
|
|
|
LOrgName = stringprep:tolower(OrgName),
|
|
|
|
LOrgUnit = stringprep:tolower(OrgUnit),
|
2003-01-03 20:36:59 +01:00
|
|
|
|
2005-04-17 20:08:34 +02:00
|
|
|
US = {LUser, LServer},
|
|
|
|
|
2003-10-20 20:23:30 +02:00
|
|
|
if
|
|
|
|
(LUser == error) or
|
|
|
|
(LFN == error) or
|
|
|
|
(LFamily == error) or
|
|
|
|
(LGiven == error) or
|
|
|
|
(LMiddle == error) or
|
|
|
|
(LNickname == error) or
|
|
|
|
(LBDay == error) or
|
|
|
|
(LCTRY == error) or
|
|
|
|
(LLocality == error) or
|
|
|
|
(LEMail == error) or
|
|
|
|
(LOrgName == error) or
|
|
|
|
(LOrgUnit == error) ->
|
|
|
|
{error, badarg};
|
|
|
|
true ->
|
|
|
|
F = fun() ->
|
2005-04-17 20:08:34 +02:00
|
|
|
mnesia:write(#vcard{us = US, vcard = VCARD}),
|
2003-01-09 20:59:16 +01:00
|
|
|
mnesia:write(
|
2005-04-17 20:08:34 +02:00
|
|
|
#vcard_search{us = US,
|
|
|
|
user = {User, LServer},
|
|
|
|
luser = LUser,
|
2003-01-09 20:59:16 +01:00
|
|
|
fn = FN, lfn = LFN,
|
|
|
|
family = Family, lfamily = LFamily,
|
|
|
|
given = Given, lgiven = LGiven,
|
|
|
|
middle = Middle, lmiddle = LMiddle,
|
|
|
|
nickname = Nickname, lnickname = LNickname,
|
|
|
|
bday = BDay, lbday = LBDay,
|
|
|
|
ctry = CTRY, lctry = LCTRY,
|
|
|
|
locality = Locality, llocality = LLocality,
|
|
|
|
email = EMail, lemail = LEMail,
|
|
|
|
orgname = OrgName, lorgname = LOrgName,
|
|
|
|
orgunit = OrgUnit, lorgunit = LOrgUnit
|
|
|
|
})
|
2003-10-20 20:23:30 +02:00
|
|
|
end,
|
|
|
|
mnesia:transaction(F)
|
|
|
|
end.
|
2003-01-02 22:01:12 +01:00
|
|
|
|
2003-01-05 21:24:59 +01:00
|
|
|
-define(TLFIELD(Type, Label, Var),
|
|
|
|
{xmlelement, "field", [{"type", Type},
|
2003-01-06 20:57:05 +01:00
|
|
|
{"label", translate:translate(Lang, Label)},
|
2003-01-05 21:24:59 +01:00
|
|
|
{"var", Var}], []}).
|
|
|
|
|
2003-01-03 20:36:59 +01:00
|
|
|
|
2004-02-26 23:00:04 +01:00
|
|
|
-define(FORM(JID),
|
2003-01-02 22:01:12 +01:00
|
|
|
[{xmlelement, "instructions", [],
|
2004-02-26 23:00:04 +01:00
|
|
|
[{xmlcdata, translate:translate(Lang, "You need an x:data capable client to search")}]},
|
2003-01-02 22:01:12 +01:00
|
|
|
{xmlelement, "x", [{"xmlns", ?NS_XDATA}, {"type", "form"}],
|
2003-01-06 20:57:05 +01:00
|
|
|
[{xmlelement, "title", [],
|
2004-02-26 23:00:04 +01:00
|
|
|
[{xmlcdata, translate:translate(Lang, "Search users in ") ++
|
|
|
|
jlib:jid_to_string(JID)}]},
|
2003-01-02 22:01:12 +01:00
|
|
|
{xmlelement, "instructions", [],
|
2004-09-10 22:57:00 +02:00
|
|
|
[{xmlcdata, translate:translate(Lang, "Fill in the form to search "
|
|
|
|
"for any matching Jabber User "
|
|
|
|
"(Add * to the end of field to "
|
|
|
|
"match substring)")}]},
|
2003-01-05 21:24:59 +01:00
|
|
|
?TLFIELD("text-single", "User", "user"),
|
|
|
|
?TLFIELD("text-single", "Full Name", "fn"),
|
2007-11-27 15:06:44 +01:00
|
|
|
?TLFIELD("text-single", "Name", "first"),
|
2003-01-05 21:24:59 +01:00
|
|
|
?TLFIELD("text-single", "Middle Name", "middle"),
|
2007-11-27 15:06:44 +01:00
|
|
|
?TLFIELD("text-single", "Family Name", "last"),
|
|
|
|
?TLFIELD("text-single", "Nickname", "nick"),
|
2003-01-05 21:24:59 +01:00
|
|
|
?TLFIELD("text-single", "Birthday", "bday"),
|
|
|
|
?TLFIELD("text-single", "Country", "ctry"),
|
|
|
|
?TLFIELD("text-single", "City", "locality"),
|
2006-02-18 20:56:16 +01:00
|
|
|
?TLFIELD("text-single", "Email", "email"),
|
2003-01-05 21:24:59 +01:00
|
|
|
?TLFIELD("text-single", "Organization Name", "orgname"),
|
|
|
|
?TLFIELD("text-single", "Organization Unit", "orgunit")
|
2003-01-02 22:01:12 +01:00
|
|
|
]}]).
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2005-08-25 22:48:45 +02:00
|
|
|
do_route(ServerHost, From, To, Packet) ->
|
2003-10-07 22:31:44 +02:00
|
|
|
#jid{user = User, resource = Resource} = To,
|
2003-01-02 22:01:12 +01:00
|
|
|
if
|
|
|
|
(User /= "") or (Resource /= "") ->
|
2003-05-09 21:32:59 +02:00
|
|
|
Err = jlib:make_error_reply(Packet, ?ERR_SERVICE_UNAVAILABLE),
|
2003-01-02 22:01:12 +01:00
|
|
|
ejabberd_router ! {route, To, From, Err};
|
|
|
|
true ->
|
|
|
|
IQ = jlib:iq_query_info(Packet),
|
|
|
|
case IQ of
|
2004-02-26 23:00:04 +01:00
|
|
|
#iq{type = Type, xmlns = ?NS_SEARCH, lang = Lang, sub_el = SubEl} ->
|
2003-01-02 22:01:12 +01:00
|
|
|
case Type of
|
|
|
|
set ->
|
2003-01-03 20:36:59 +01:00
|
|
|
XDataEl = find_xdata_el(SubEl),
|
|
|
|
case XDataEl of
|
|
|
|
false ->
|
|
|
|
Err = jlib:make_error_reply(
|
2003-05-09 21:32:59 +02:00
|
|
|
Packet, ?ERR_BAD_REQUEST),
|
2003-01-03 20:36:59 +01:00
|
|
|
ejabberd_router:route(To, From, Err);
|
|
|
|
_ ->
|
|
|
|
XData = jlib:parse_xdata_submit(XDataEl),
|
|
|
|
case XData of
|
|
|
|
invalid ->
|
|
|
|
Err = jlib:make_error_reply(
|
|
|
|
Packet,
|
2003-05-09 21:32:59 +02:00
|
|
|
?ERR_BAD_REQUEST),
|
2003-01-03 20:36:59 +01:00
|
|
|
ejabberd_router:route(To, From,
|
|
|
|
Err);
|
|
|
|
_ ->
|
|
|
|
ResIQ =
|
2003-12-17 21:13:21 +01:00
|
|
|
IQ#iq{
|
|
|
|
type = result,
|
|
|
|
sub_el =
|
|
|
|
[{xmlelement,
|
|
|
|
"query",
|
|
|
|
[{"xmlns", ?NS_SEARCH}],
|
|
|
|
[{xmlelement, "x",
|
|
|
|
[{"xmlns", ?NS_XDATA},
|
|
|
|
{"type", "result"}],
|
2005-08-25 22:48:45 +02:00
|
|
|
search_result(Lang, To, ServerHost, XData)
|
2003-12-17 21:13:21 +01:00
|
|
|
}]}]},
|
2003-01-03 20:36:59 +01:00
|
|
|
ejabberd_router:route(
|
|
|
|
To, From, jlib:iq_to_xml(ResIQ))
|
|
|
|
end
|
|
|
|
end;
|
2003-01-02 22:01:12 +01:00
|
|
|
get ->
|
2003-12-17 21:13:21 +01:00
|
|
|
ResIQ = IQ#iq{type = result,
|
|
|
|
sub_el = [{xmlelement,
|
|
|
|
"query",
|
|
|
|
[{"xmlns", ?NS_SEARCH}],
|
2004-02-26 23:00:04 +01:00
|
|
|
?FORM(To)
|
2003-12-17 21:13:21 +01:00
|
|
|
}]},
|
2003-01-02 22:01:12 +01:00
|
|
|
ejabberd_router:route(To,
|
|
|
|
From,
|
|
|
|
jlib:iq_to_xml(ResIQ))
|
|
|
|
end;
|
2007-05-29 15:14:09 +02:00
|
|
|
#iq{type = Type, xmlns = ?NS_DISCO_INFO, lang = Lang} ->
|
2003-01-02 22:01:12 +01:00
|
|
|
case Type of
|
|
|
|
set ->
|
|
|
|
Err = jlib:make_error_reply(
|
2003-05-09 21:32:59 +02:00
|
|
|
Packet, ?ERR_NOT_ALLOWED),
|
2003-01-02 22:01:12 +01:00
|
|
|
ejabberd_router:route(To, From, Err);
|
|
|
|
get ->
|
2003-12-17 21:13:21 +01:00
|
|
|
ResIQ =
|
|
|
|
IQ#iq{type = result,
|
|
|
|
sub_el = [{xmlelement,
|
|
|
|
"query",
|
|
|
|
[{"xmlns", ?NS_DISCO_INFO}],
|
|
|
|
[{xmlelement, "identity",
|
|
|
|
[{"category", "directory"},
|
|
|
|
{"type", "user"},
|
|
|
|
{"name",
|
2007-05-29 15:14:09 +02:00
|
|
|
translate:translate(Lang, "vCard User Search")}],
|
2003-12-17 21:13:21 +01:00
|
|
|
[]},
|
|
|
|
{xmlelement, "feature",
|
2004-02-26 23:00:04 +01:00
|
|
|
[{"var", ?NS_SEARCH}], []},
|
|
|
|
{xmlelement, "feature",
|
|
|
|
[{"var", ?NS_VCARD}], []}
|
2003-12-17 21:13:21 +01:00
|
|
|
]
|
|
|
|
}]},
|
2003-01-02 22:01:12 +01:00
|
|
|
ejabberd_router:route(To,
|
|
|
|
From,
|
|
|
|
jlib:iq_to_xml(ResIQ))
|
|
|
|
end;
|
2005-09-04 03:58:47 +02:00
|
|
|
#iq{type = Type, xmlns = ?NS_DISCO_ITEMS} ->
|
2003-01-02 22:01:12 +01:00
|
|
|
case Type of
|
|
|
|
set ->
|
|
|
|
Err = jlib:make_error_reply(
|
2003-05-09 21:32:59 +02:00
|
|
|
Packet, ?ERR_NOT_ALLOWED),
|
2003-01-02 22:01:12 +01:00
|
|
|
ejabberd_router:route(To, From, Err);
|
|
|
|
get ->
|
2003-12-17 21:13:21 +01:00
|
|
|
ResIQ =
|
|
|
|
IQ#iq{type = result,
|
|
|
|
sub_el = [{xmlelement,
|
|
|
|
"query",
|
2005-11-19 02:20:05 +01:00
|
|
|
[{"xmlns", ?NS_DISCO_ITEMS}],
|
2003-12-17 21:13:21 +01:00
|
|
|
[]}]},
|
2003-01-02 22:01:12 +01:00
|
|
|
ejabberd_router:route(To,
|
|
|
|
From,
|
|
|
|
jlib:iq_to_xml(ResIQ))
|
|
|
|
end;
|
2004-02-26 23:00:04 +01:00
|
|
|
#iq{type = get, xmlns = ?NS_VCARD, lang = Lang} ->
|
|
|
|
ResIQ =
|
|
|
|
IQ#iq{type = result,
|
|
|
|
sub_el = [{xmlelement,
|
2004-03-02 22:16:55 +01:00
|
|
|
"vCard",
|
2004-02-26 23:00:04 +01:00
|
|
|
[{"xmlns", ?NS_VCARD}],
|
|
|
|
iq_get_vcard(Lang)}]},
|
|
|
|
ejabberd_router:route(To,
|
|
|
|
From,
|
|
|
|
jlib:iq_to_xml(ResIQ));
|
2003-01-02 22:01:12 +01:00
|
|
|
_ ->
|
|
|
|
Err = jlib:make_error_reply(Packet,
|
2003-05-09 21:32:59 +02:00
|
|
|
?ERR_SERVICE_UNAVAILABLE),
|
2003-01-02 22:01:12 +01:00
|
|
|
ejabberd_router:route(To, From, Err)
|
|
|
|
end
|
|
|
|
end.
|
|
|
|
|
2004-02-26 23:00:04 +01:00
|
|
|
iq_get_vcard(Lang) ->
|
|
|
|
[{xmlelement, "FN", [],
|
|
|
|
[{xmlcdata, "ejabberd/mod_vcard"}]},
|
|
|
|
{xmlelement, "URL", [],
|
|
|
|
[{xmlcdata,
|
|
|
|
"http://ejabberd.jabberstudio.org/"}]},
|
|
|
|
{xmlelement, "DESC", [],
|
2004-03-02 22:16:55 +01:00
|
|
|
[{xmlcdata, translate:translate(
|
|
|
|
Lang,
|
|
|
|
"ejabberd vCard module\n"
|
2006-02-06 06:12:54 +01:00
|
|
|
"Copyright (c) 2003-2006 Alexey Shchepin")}]}].
|
2004-02-26 23:00:04 +01:00
|
|
|
|
2003-10-20 20:23:30 +02:00
|
|
|
find_xdata_el({xmlelement, _Name, _Attrs, SubEls}) ->
|
2003-01-03 20:36:59 +01:00
|
|
|
find_xdata_el1(SubEls).
|
|
|
|
|
|
|
|
find_xdata_el1([]) ->
|
|
|
|
false;
|
|
|
|
find_xdata_el1([{xmlelement, Name, Attrs, SubEls} | Els]) ->
|
|
|
|
case xml:get_attr_s("xmlns", Attrs) of
|
|
|
|
?NS_XDATA ->
|
|
|
|
{xmlelement, Name, Attrs, SubEls};
|
|
|
|
_ ->
|
|
|
|
find_xdata_el1(Els)
|
|
|
|
end;
|
|
|
|
find_xdata_el1([_ | Els]) ->
|
|
|
|
find_xdata_el1(Els).
|
|
|
|
|
2003-01-05 21:24:59 +01:00
|
|
|
-define(LFIELD(Label, Var),
|
2003-01-06 20:57:05 +01:00
|
|
|
{xmlelement, "field", [{"label", translate:translate(Lang, Label)},
|
|
|
|
{"var", Var}], []}).
|
2003-01-05 21:24:59 +01:00
|
|
|
|
2005-08-25 22:48:45 +02:00
|
|
|
search_result(Lang, JID, ServerHost, Data) ->
|
2003-01-06 20:57:05 +01:00
|
|
|
[{xmlelement, "title", [],
|
2006-02-18 20:56:16 +01:00
|
|
|
[{xmlcdata, translate:translate(Lang, "Search Results for ") ++
|
2004-02-26 23:00:04 +01:00
|
|
|
jlib:jid_to_string(JID)}]},
|
2003-01-03 20:36:59 +01:00
|
|
|
{xmlelement, "reported", [],
|
2006-02-18 20:56:16 +01:00
|
|
|
[?LFIELD("Jabber ID", "jid"),
|
2003-01-05 21:24:59 +01:00
|
|
|
?LFIELD("Full Name", "fn"),
|
2007-11-27 15:06:44 +01:00
|
|
|
?LFIELD("Name", "first"),
|
2003-01-05 21:24:59 +01:00
|
|
|
?LFIELD("Middle Name", "middle"),
|
2007-11-27 15:06:44 +01:00
|
|
|
?LFIELD("Family Name", "last"),
|
|
|
|
?LFIELD("Nickname", "nick"),
|
2003-01-05 21:24:59 +01:00
|
|
|
?LFIELD("Birthday", "bday"),
|
|
|
|
?LFIELD("Country", "ctry"),
|
|
|
|
?LFIELD("City", "locality"),
|
2006-02-18 20:56:16 +01:00
|
|
|
?LFIELD("Email", "email"),
|
2003-01-05 21:24:59 +01:00
|
|
|
?LFIELD("Organization Name", "orgname"),
|
|
|
|
?LFIELD("Organization Unit", "orgunit")
|
2005-08-25 22:48:45 +02:00
|
|
|
]}] ++ lists:map(fun record_to_item/1, search(ServerHost, Data)).
|
2003-01-03 20:36:59 +01:00
|
|
|
|
|
|
|
-define(FIELD(Var, Val),
|
|
|
|
{xmlelement, "field", [{"var", Var}],
|
|
|
|
[{xmlelement, "value", [],
|
|
|
|
[{xmlcdata, Val}]}]}).
|
|
|
|
|
|
|
|
record_to_item(R) ->
|
2005-04-17 20:08:34 +02:00
|
|
|
{User, Server} = R#vcard_search.user,
|
|
|
|
{xmlelement, "item", [],
|
|
|
|
[
|
|
|
|
?FIELD("jid", User ++ "@" ++ Server),
|
2003-01-03 20:36:59 +01:00
|
|
|
?FIELD("fn", R#vcard_search.fn),
|
2007-11-27 15:06:44 +01:00
|
|
|
?FIELD("last", R#vcard_search.family),
|
|
|
|
?FIELD("first", R#vcard_search.given),
|
2003-01-03 20:36:59 +01:00
|
|
|
?FIELD("middle", R#vcard_search.middle),
|
2007-11-27 15:06:44 +01:00
|
|
|
?FIELD("nick", R#vcard_search.nickname),
|
2003-01-03 20:36:59 +01:00
|
|
|
?FIELD("bday", R#vcard_search.bday),
|
|
|
|
?FIELD("ctry", R#vcard_search.ctry),
|
|
|
|
?FIELD("locality", R#vcard_search.locality),
|
|
|
|
?FIELD("email", R#vcard_search.email),
|
|
|
|
?FIELD("orgname", R#vcard_search.orgname),
|
|
|
|
?FIELD("orgunit", R#vcard_search.orgunit)
|
|
|
|
]
|
|
|
|
}.
|
|
|
|
|
2003-10-28 21:26:43 +01:00
|
|
|
|
2005-04-17 20:08:34 +02:00
|
|
|
search(LServer, Data) ->
|
|
|
|
MatchSpec = make_matchspec(LServer, Data),
|
2005-06-20 05:18:13 +02:00
|
|
|
AllowReturnAll = gen_mod:get_module_opt(LServer, ?MODULE,
|
|
|
|
allow_return_all, false),
|
2003-10-28 21:26:43 +01:00
|
|
|
if
|
2005-04-17 20:08:34 +02:00
|
|
|
(MatchSpec == #vcard_search{_ = '_'}) and (not AllowReturnAll) ->
|
2003-10-28 21:26:43 +01:00
|
|
|
[];
|
|
|
|
true ->
|
|
|
|
case catch mnesia:dirty_select(vcard_search,
|
|
|
|
[{MatchSpec, [], ['$_']}]) of
|
|
|
|
{'EXIT', Reason} ->
|
|
|
|
?ERROR_MSG("~p", [Reason]),
|
|
|
|
[];
|
|
|
|
Rs ->
|
2005-06-20 05:18:13 +02:00
|
|
|
case gen_mod:get_module_opt(LServer, ?MODULE,
|
|
|
|
matches, ?JUD_MATCHES) of
|
2004-09-10 22:57:00 +02:00
|
|
|
infinity ->
|
|
|
|
Rs;
|
|
|
|
Val when is_integer(Val) and (Val > 0) ->
|
|
|
|
lists:sublist(Rs, Val);
|
|
|
|
Val ->
|
|
|
|
?ERROR_MSG("Illegal option value ~p. "
|
|
|
|
"Default value ~p substituted.",
|
|
|
|
[{matches, Val}, ?JUD_MATCHES]),
|
|
|
|
lists:sublist(Rs, ?JUD_MATCHES)
|
|
|
|
end
|
2003-10-28 21:26:43 +01:00
|
|
|
end
|
|
|
|
end.
|
|
|
|
|
|
|
|
|
2005-04-17 20:08:34 +02:00
|
|
|
make_matchspec(LServer, Data) ->
|
2003-01-08 21:42:03 +01:00
|
|
|
GlobMatch = #vcard_search{_ = '_'},
|
2005-04-17 20:08:34 +02:00
|
|
|
Match = filter_fields(Data, GlobMatch, LServer),
|
2003-01-03 20:36:59 +01:00
|
|
|
Match.
|
|
|
|
|
2005-04-17 20:08:34 +02:00
|
|
|
filter_fields([], Match, _LServer) ->
|
2003-01-03 20:36:59 +01:00
|
|
|
Match;
|
2005-04-17 20:08:34 +02:00
|
|
|
filter_fields([{SVar, [Val]} | Ds], Match, LServer)
|
2003-01-03 20:36:59 +01:00
|
|
|
when is_list(Val) and (Val /= "") ->
|
2003-10-07 22:31:44 +02:00
|
|
|
LVal = stringprep:tolower(Val),
|
2003-01-03 20:36:59 +01:00
|
|
|
NewMatch = case SVar of
|
2005-04-17 20:08:34 +02:00
|
|
|
"user" ->
|
2005-06-20 05:18:13 +02:00
|
|
|
case gen_mod:get_module_opt(LServer, ?MODULE,
|
|
|
|
search_all_hosts, true) of
|
2005-04-17 20:08:34 +02:00
|
|
|
true ->
|
|
|
|
Match#vcard_search{luser = make_val(LVal)};
|
|
|
|
false ->
|
|
|
|
Host = find_my_host(LServer),
|
|
|
|
Match#vcard_search{us = {make_val(LVal), Host}}
|
|
|
|
end;
|
2004-09-10 22:57:00 +02:00
|
|
|
"fn" -> Match#vcard_search{lfn = make_val(LVal)};
|
2007-11-27 15:06:44 +01:00
|
|
|
"last" -> Match#vcard_search{lfamily = make_val(LVal)};
|
|
|
|
"first" -> Match#vcard_search{lgiven = make_val(LVal)};
|
2004-09-10 22:57:00 +02:00
|
|
|
"middle" -> Match#vcard_search{lmiddle = make_val(LVal)};
|
2007-11-27 15:06:44 +01:00
|
|
|
"nick" -> Match#vcard_search{lnickname = make_val(LVal)};
|
2004-09-10 22:57:00 +02:00
|
|
|
"bday" -> Match#vcard_search{lbday = make_val(LVal)};
|
|
|
|
"ctry" -> Match#vcard_search{lctry = make_val(LVal)};
|
|
|
|
"locality" -> Match#vcard_search{llocality = make_val(LVal)};
|
|
|
|
"email" -> Match#vcard_search{lemail = make_val(LVal)};
|
|
|
|
"orgname" -> Match#vcard_search{lorgname = make_val(LVal)};
|
|
|
|
"orgunit" -> Match#vcard_search{lorgunit = make_val(LVal)};
|
2003-01-03 20:36:59 +01:00
|
|
|
_ -> Match
|
|
|
|
end,
|
2005-04-17 20:08:34 +02:00
|
|
|
filter_fields(Ds, NewMatch, LServer);
|
|
|
|
filter_fields([_ | Ds], Match, LServer) ->
|
|
|
|
filter_fields(Ds, Match, LServer).
|
2003-01-09 20:59:16 +01:00
|
|
|
|
2004-09-10 22:57:00 +02:00
|
|
|
make_val(Val) ->
|
|
|
|
case lists:suffix("*", Val) of
|
|
|
|
true ->
|
|
|
|
lists:sublist(Val, length(Val) - 1) ++ '_';
|
|
|
|
_ ->
|
|
|
|
Val
|
|
|
|
end.
|
2003-01-09 20:59:16 +01:00
|
|
|
|
2005-04-17 20:08:34 +02:00
|
|
|
find_my_host(LServer) ->
|
|
|
|
Parts = string:tokens(LServer, "."),
|
|
|
|
find_my_host(Parts, ?MYHOSTS).
|
|
|
|
|
|
|
|
find_my_host([], _Hosts) ->
|
|
|
|
?MYNAME;
|
|
|
|
find_my_host([_ | Tail] = Parts, Hosts) ->
|
|
|
|
Domain = parts_to_string(Parts),
|
|
|
|
case lists:member(Domain, Hosts) of
|
|
|
|
true ->
|
|
|
|
Domain;
|
|
|
|
false ->
|
|
|
|
find_my_host(Tail, Hosts)
|
|
|
|
end.
|
|
|
|
|
|
|
|
parts_to_string(Parts) ->
|
|
|
|
string:strip(lists:flatten(lists:map(fun(S) -> [S, $.] end, Parts)),
|
|
|
|
right, $.).
|
|
|
|
|
|
|
|
|
|
|
|
|
2003-01-09 20:59:16 +01:00
|
|
|
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
|
|
|
|
|
|
|
|
set_vcard_t(R, _) ->
|
2005-04-17 20:08:34 +02:00
|
|
|
US = R#vcard.us,
|
|
|
|
User = US,
|
2003-01-09 20:59:16 +01:00
|
|
|
VCARD = R#vcard.vcard,
|
|
|
|
|
|
|
|
FN = xml:get_path_s(VCARD, [{elem, "FN"}, cdata]),
|
|
|
|
Family = xml:get_path_s(VCARD, [{elem, "N"}, {elem, "FAMILY"}, cdata]),
|
|
|
|
Given = xml:get_path_s(VCARD, [{elem, "N"}, {elem, "GIVEN"}, cdata]),
|
|
|
|
Middle = xml:get_path_s(VCARD, [{elem, "N"}, {elem, "MIDDLE"}, cdata]),
|
|
|
|
Nickname = xml:get_path_s(VCARD, [{elem, "NICKNAME"}, cdata]),
|
|
|
|
BDay = xml:get_path_s(VCARD, [{elem, "BDAY"}, cdata]),
|
|
|
|
CTRY = xml:get_path_s(VCARD, [{elem, "ADR"}, {elem, "CTRY"}, cdata]),
|
|
|
|
Locality = xml:get_path_s(VCARD, [{elem, "ADR"}, {elem, "LOCALITY"},cdata]),
|
|
|
|
EMail = xml:get_path_s(VCARD, [{elem, "EMAIL"}, cdata]),
|
|
|
|
OrgName = xml:get_path_s(VCARD, [{elem, "ORG"}, {elem, "ORGNAME"}, cdata]),
|
|
|
|
OrgUnit = xml:get_path_s(VCARD, [{elem, "ORG"}, {elem, "ORGUNIT"}, cdata]),
|
|
|
|
|
2005-04-17 20:08:34 +02:00
|
|
|
{LUser, _LServer} = US,
|
2003-10-07 22:31:44 +02:00
|
|
|
LFN = stringprep:tolower(FN),
|
|
|
|
LFamily = stringprep:tolower(Family),
|
|
|
|
LGiven = stringprep:tolower(Given),
|
|
|
|
LMiddle = stringprep:tolower(Middle),
|
|
|
|
LNickname = stringprep:tolower(Nickname),
|
|
|
|
LBDay = stringprep:tolower(BDay),
|
|
|
|
LCTRY = stringprep:tolower(CTRY),
|
|
|
|
LLocality = stringprep:tolower(Locality),
|
|
|
|
LEMail = stringprep:tolower(EMail),
|
|
|
|
LOrgName = stringprep:tolower(OrgName),
|
|
|
|
LOrgUnit = stringprep:tolower(OrgUnit),
|
2003-01-09 20:59:16 +01:00
|
|
|
|
2003-10-20 20:23:30 +02:00
|
|
|
if
|
|
|
|
(LUser == error) or
|
|
|
|
(LFN == error) or
|
|
|
|
(LFamily == error) or
|
|
|
|
(LGiven == error) or
|
|
|
|
(LMiddle == error) or
|
|
|
|
(LNickname == error) or
|
|
|
|
(LBDay == error) or
|
|
|
|
(LCTRY == error) or
|
|
|
|
(LLocality == error) or
|
|
|
|
(LEMail == error) or
|
|
|
|
(LOrgName == error) or
|
|
|
|
(LOrgUnit == error) ->
|
|
|
|
{error, badarg};
|
|
|
|
true ->
|
|
|
|
mnesia:write(
|
2005-04-17 20:08:34 +02:00
|
|
|
#vcard_search{us = US,
|
|
|
|
user = User, luser = LUser,
|
2003-10-20 20:23:30 +02:00
|
|
|
fn = FN, lfn = LFN,
|
|
|
|
family = Family, lfamily = LFamily,
|
|
|
|
given = Given, lgiven = LGiven,
|
|
|
|
middle = Middle, lmiddle = LMiddle,
|
|
|
|
nickname = Nickname, lnickname = LNickname,
|
|
|
|
bday = BDay, lbday = LBDay,
|
|
|
|
ctry = CTRY, lctry = LCTRY,
|
|
|
|
locality = Locality, llocality = LLocality,
|
|
|
|
email = EMail, lemail = LEMail,
|
|
|
|
orgname = OrgName, lorgname = LOrgName,
|
|
|
|
orgunit = OrgUnit, lorgunit = LOrgUnit
|
|
|
|
})
|
|
|
|
end.
|
2003-01-09 20:59:16 +01:00
|
|
|
|
|
|
|
|
|
|
|
reindex_vcards() ->
|
|
|
|
F = fun() ->
|
|
|
|
mnesia:foldl(fun set_vcard_t/2, [], vcard)
|
|
|
|
end,
|
|