423 lines
13 KiB
Erlang
423 lines
13 KiB
Erlang
%%%----------------------------------------------------------------------
|
|
%%% File : mod_vcard.erl
|
|
%%% Author : Alexey Shchepin <alexey@sevcom.net>
|
|
%%% Purpose :
|
|
%%% Created : 2 Jan 2003 by Alexey Shchepin <alexey@sevcom.net>
|
|
%%% Id : $Id$
|
|
%%%----------------------------------------------------------------------
|
|
|
|
-module(mod_vcard).
|
|
-author('alexey@sevcom.net').
|
|
-vsn('$Revision$ ').
|
|
|
|
-export([start/0, init/0,
|
|
process_local_iq/3,
|
|
process_sm_iq/3]).
|
|
|
|
-include("ejabberd.hrl").
|
|
-include("namespaces.hrl").
|
|
|
|
|
|
-record(vcard_search, {user, fn, family, given, middle, nickname,
|
|
bday, ctry="", locality="", email,
|
|
orgname, orgunit}).
|
|
-record(vcard, {user, vcard}).
|
|
|
|
|
|
start() ->
|
|
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)}]),
|
|
mnesia:add_table_index(vcard_search, fn),
|
|
mnesia:add_table_index(vcard_search, family),
|
|
mnesia:add_table_index(vcard_search, given),
|
|
mnesia:add_table_index(vcard_search, middle),
|
|
mnesia:add_table_index(vcard_search, nickname),
|
|
mnesia:add_table_index(vcard_search, bday),
|
|
mnesia:add_table_index(vcard_search, ctry),
|
|
mnesia:add_table_index(vcard_search, locality),
|
|
mnesia:add_table_index(vcard_search, email),
|
|
mnesia:add_table_index(vcard_search, orgname),
|
|
mnesia:add_table_index(vcard_search, orgunit),
|
|
|
|
|
|
ejabberd_local:register_iq_handler(?NS_VCARD,
|
|
?MODULE, process_local_iq),
|
|
ejabberd_sm:register_iq_handler(?NS_VCARD,
|
|
?MODULE, process_sm_iq),
|
|
spawn(?MODULE, init, []).
|
|
|
|
init() ->
|
|
ejabberd_router:register_local_route("vjud." ++ ?MYNAME),
|
|
loop().
|
|
|
|
loop() ->
|
|
receive
|
|
{route, From, To, Packet} ->
|
|
do_route(From, To, Packet),
|
|
loop();
|
|
_ ->
|
|
loop()
|
|
end.
|
|
|
|
|
|
process_local_iq(From, To, {iq, ID, Type, XMLNS, SubEl}) ->
|
|
case Type of
|
|
set ->
|
|
{User, Server, _} = From,
|
|
LUser = jlib:tolower(User),
|
|
LServer = jlib:tolower(Server),
|
|
case ?MYNAME of
|
|
LServer ->
|
|
set_vcard(LUser, SubEl),
|
|
{iq, ID, result, XMLNS, []};
|
|
_ ->
|
|
{iq, ID, error, XMLNS,
|
|
[SubEl, {xmlelement, "error",
|
|
[{"code", "405"}],
|
|
[{xmlcdata, "Not Allowed"}]}]}
|
|
end;
|
|
get ->
|
|
{iq, ID, result, XMLNS,
|
|
[{xmlelement, "vCard",
|
|
[{"xmlns", ?NS_VCARD}],
|
|
[{xmlelement, "FN", [],
|
|
[{xmlcdata, "ejabberd"}]},
|
|
{xmlelement, "URL", [],
|
|
[{xmlcdata,
|
|
"http://www.jabber.ru/projects/ejabberd/"}]},
|
|
{xmlelement, "DESC", [],
|
|
[{xmlcdata, "Erlang Jabber Server\n"
|
|
"Copyright (c) 2002, 2003 Alexey Shchepin"}]},
|
|
{xmlelement, "BDAY", [],
|
|
[{xmlcdata, "20021116"}]}
|
|
]}]}
|
|
end.
|
|
|
|
|
|
process_sm_iq(From, To, {iq, ID, Type, XMLNS, SubEl}) ->
|
|
case Type of
|
|
set ->
|
|
{iq, ID, error, XMLNS, [SubEl, {xmlelement, "error",
|
|
[{"code", "405"}],
|
|
[{xmlcdata, "Not Allowed"}]}]};
|
|
get ->
|
|
{User, _, _} = To,
|
|
LUser = jlib:tolower(User),
|
|
F = fun() ->
|
|
mnesia:read({vcard, LUser})
|
|
end,
|
|
Els = case mnesia:transaction(F) of
|
|
{atomic, Rs} ->
|
|
lists:map(fun(R) ->
|
|
R#vcard.vcard
|
|
end, Rs);
|
|
{aborted, Reason} ->
|
|
[]
|
|
end,
|
|
{iq, ID, result, XMLNS, Els}
|
|
end.
|
|
|
|
|
|
set_vcard(LUser, 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, "CTRY"}, cdata]),
|
|
%locality = xml:get_path_s(VCARD, [{elem, "FN"}, 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]),
|
|
|
|
LFN = jlib:tolower(FN),
|
|
LFamily = jlib:tolower(Family),
|
|
LGiven = jlib:tolower(Given),
|
|
LMiddle = jlib:tolower(Middle),
|
|
LNickname = jlib:tolower(Nickname),
|
|
LBDay = jlib:tolower(BDay),
|
|
LEMail = jlib:tolower(EMail),
|
|
LOrgName = jlib:tolower(OrgName),
|
|
LOrgUnit = jlib:tolower(OrgUnit),
|
|
|
|
F = fun() ->
|
|
mnesia:write(#vcard{user = LUser, vcard = VCARD}),
|
|
mnesia:write(#vcard_search{user = LUser,
|
|
fn = LFN,
|
|
family = LFamily,
|
|
given = LGiven,
|
|
middle = LMiddle,
|
|
nickname = LNickname,
|
|
bday = LBDay,
|
|
%ctry = LCTRY,
|
|
%locality = LLocality,
|
|
email = LEMail,
|
|
orgname = LOrgName,
|
|
orgunit = LOrgUnit
|
|
})
|
|
end,
|
|
mnesia:transaction(F).
|
|
|
|
|
|
-define(FORM,
|
|
[{xmlelement, "instructions", [],
|
|
[{xmlcdata, "You need a x:data capable client to search"}]},
|
|
{xmlelement, "x", [{"xmlns", ?NS_XDATA}, {"type", "form"}],
|
|
[{xmlelement, "title", [], [{xmlcdata, "Users Search"}]},
|
|
{xmlelement, "instructions", [],
|
|
[{xmlcdata, "Fill in fields to search "
|
|
"for any matching Jabber User"}]},
|
|
{xmlelement, "field", [{"type", "text-single"},
|
|
{"label", "JID"},
|
|
{"var", "jid"}], []},
|
|
{xmlelement, "field", [{"type", "text-single"},
|
|
{"label", "Full Name"},
|
|
{"var", "fn"}], []},
|
|
{xmlelement, "field", [{"type", "text-single"},
|
|
{"label", "Name"},
|
|
{"var", "given"}], []},
|
|
{xmlelement, "field", [{"type", "text-single"},
|
|
{"label", "Middle Name"},
|
|
{"var", "middle"}], []},
|
|
{xmlelement, "field", [{"type", "text-single"},
|
|
{"label", "Family Name"},
|
|
{"var", "family"}], []},
|
|
{xmlelement, "field", [{"type", "text-single"},
|
|
{"label", "Nickname"},
|
|
{"var", "nickname"}], []},
|
|
{xmlelement, "field", [{"type", "text-single"},
|
|
{"label", "Birthday"},
|
|
{"var", "bday"}], []},
|
|
{xmlelement, "field", [{"type", "text-single"},
|
|
{"label", "Country"},
|
|
{"var", "ctry"}], []},
|
|
{xmlelement, "field", [{"type", "text-single"},
|
|
{"label", "City"},
|
|
{"var", "locality"}], []},
|
|
{xmlelement, "field", [{"type", "text-single"},
|
|
{"label", "email"},
|
|
{"var", "email"}], []},
|
|
{xmlelement, "field", [{"type", "text-single"},
|
|
{"label", "Organization Name"},
|
|
{"var", "orgname"}], []},
|
|
{xmlelement, "field", [{"type", "text-single"},
|
|
{"label", "Organization Unit"},
|
|
{"var", "orgunit"}], []}
|
|
]}]).
|
|
|
|
|
|
|
|
|
|
do_route(From, To, Packet) ->
|
|
{User, Server, Resource} = To,
|
|
if
|
|
(User /= "") or (Resource /= "") ->
|
|
Err = jlib:make_error_reply(Packet, "503", "Service Unavailable"),
|
|
ejabberd_router ! {route, To, From, Err};
|
|
true ->
|
|
IQ = jlib:iq_query_info(Packet),
|
|
case IQ of
|
|
{iq, ID, Type, ?NS_SEARCH, SubEl} ->
|
|
case Type of
|
|
set ->
|
|
XDataEl = find_xdata_el(SubEl),
|
|
case XDataEl of
|
|
false ->
|
|
Err = jlib:make_error_reply(
|
|
Packet, "400", "Bad Request"),
|
|
ejabberd_router:route(To, From, Err);
|
|
_ ->
|
|
XData = jlib:parse_xdata_submit(XDataEl),
|
|
case XData of
|
|
invalid ->
|
|
Err = jlib:make_error_reply(
|
|
Packet,
|
|
"400", "Bad Request"),
|
|
ejabberd_router:route(To, From,
|
|
Err);
|
|
_ ->
|
|
ResIQ =
|
|
{iq, ID, result, ?NS_SEARCH,
|
|
[{xmlelement,
|
|
"query",
|
|
[{"xmlns", ?NS_SEARCH}],
|
|
[{xmlelement, "x",
|
|
[{"xmlns", ?NS_XDATA},
|
|
{"type", "result"}],
|
|
search_result(XData)
|
|
}]}]},
|
|
ejabberd_router:route(
|
|
To, From, jlib:iq_to_xml(ResIQ))
|
|
end
|
|
end;
|
|
get ->
|
|
ResIQ = {iq, ID, result, ?NS_SEARCH,
|
|
[{xmlelement,
|
|
"query",
|
|
[{"xmlns", ?NS_SEARCH}],
|
|
?FORM
|
|
}]},
|
|
ejabberd_router:route(To,
|
|
From,
|
|
jlib:iq_to_xml(ResIQ))
|
|
end;
|
|
{iq, ID, Type, ?NS_DISCO_INFO, SubEl} ->
|
|
case Type of
|
|
set ->
|
|
Err = jlib:make_error_reply(
|
|
Packet, "405", "Not Allowed"),
|
|
ejabberd_router:route(To, From, Err);
|
|
get ->
|
|
ResIQ = {iq, ID, result, ?NS_DISCO_INFO,
|
|
[{xmlelement,
|
|
"query",
|
|
[{"xmlns", ?NS_DISCO_INFO}],
|
|
[{xmlelement, "identity",
|
|
[{"category", "directory"},
|
|
{"type", "user"},
|
|
{"name", "vCard User Search"}], []},
|
|
{xmlelement, "feature",
|
|
[{"var", ?NS_SEARCH}], []}
|
|
]
|
|
}]},
|
|
ejabberd_router:route(To,
|
|
From,
|
|
jlib:iq_to_xml(ResIQ))
|
|
end;
|
|
{iq, ID, Type, ?NS_DISCO_ITEMS, SubEl} ->
|
|
case Type of
|
|
set ->
|
|
Err = jlib:make_error_reply(
|
|
Packet, "405", "Not Allowed"),
|
|
ejabberd_router:route(To, From, Err);
|
|
get ->
|
|
ResIQ = {iq, ID, result, ?NS_DISCO_INFO,
|
|
[{xmlelement,
|
|
"query",
|
|
[{"xmlns", ?NS_DISCO_INFO}], []}]},
|
|
ejabberd_router:route(To,
|
|
From,
|
|
jlib:iq_to_xml(ResIQ))
|
|
end;
|
|
_ ->
|
|
Err = jlib:make_error_reply(Packet,
|
|
"503", "Service Unavailable"),
|
|
ejabberd_router:route(To, From, Err)
|
|
end
|
|
end.
|
|
|
|
find_xdata_el({xmlelement, Name, Attrs, SubEls}) ->
|
|
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).
|
|
|
|
search_result(Data) ->
|
|
[{xmlelement, "title", [], [{xmlcdata, "Users Search Results"}]},
|
|
{xmlelement, "reported", [],
|
|
[{xmlelement, "field", [{"label", "JID"}, {"var", "jid"}], []},
|
|
{xmlelement, "field", [{"label", "Full Name"}, {"var", "fn"}], []},
|
|
{xmlelement, "field", [{"label", "Name"}, {"var", "given"}], []},
|
|
{xmlelement, "field", [{"label", "Middle Name"}, {"var", "middle"}], []},
|
|
{xmlelement, "field", [{"label", "Family Name"}, {"var", "family"}], []},
|
|
{xmlelement, "field", [{"label", "Nickname"}, {"var", "nickname"}], []},
|
|
{xmlelement, "field", [{"label", "Birthday"}, {"var", "bday"}], []},
|
|
{xmlelement, "field", [{"label", "Country"}, {"var", "ctry"}], []},
|
|
{xmlelement, "field", [{"label", "City"}, {"var", "locality"}], []},
|
|
{xmlelement, "field", [{"label", "email"}, {"var", "email"}], []},
|
|
{xmlelement, "field", [{"label", "Organization Name"},
|
|
{"var", "orgname"}], []},
|
|
{xmlelement, "field", [{"label", "Organization Unit"},
|
|
{"var", "orgunit"}], []}
|
|
]}] ++ lists:map(fun record_to_item/1, search(Data)).
|
|
|
|
-define(FIELD(Var, Val),
|
|
{xmlelement, "field", [{"var", Var}],
|
|
[{xmlelement, "value", [],
|
|
[{xmlcdata, Val}]}]}).
|
|
|
|
record_to_item(R) ->
|
|
{xmlelement, "item", [],
|
|
[
|
|
?FIELD("jid", R#vcard_search.user ++ "@" ++ ?MYNAME),
|
|
?FIELD("fn", R#vcard_search.fn),
|
|
?FIELD("family", R#vcard_search.family),
|
|
?FIELD("given", R#vcard_search.given),
|
|
?FIELD("middle", R#vcard_search.middle),
|
|
?FIELD("nickname", R#vcard_search.nickname),
|
|
?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)
|
|
]
|
|
}.
|
|
|
|
|
|
search(Data) ->
|
|
MatchSpec = make_matchspec(Data),
|
|
F = fun() ->
|
|
mnesia:match_object(MatchSpec)
|
|
end,
|
|
case mnesia:transaction(F) of
|
|
{atomic, Rs} ->
|
|
Rs;
|
|
_ ->
|
|
[]
|
|
end.
|
|
|
|
|
|
make_matchspec(Data) ->
|
|
GlobMatch = #vcard_search{user = '_',
|
|
fn = '_',
|
|
family = '_',
|
|
given = '_',
|
|
middle = '_',
|
|
nickname = '_',
|
|
bday = '_',
|
|
ctry = '_',
|
|
locality = '_',
|
|
email = '_',
|
|
orgname = '_',
|
|
orgunit = '_'
|
|
},
|
|
Match = filter_fields(Data, GlobMatch),
|
|
Match.
|
|
|
|
filter_fields([], Match) ->
|
|
Match;
|
|
filter_fields([{SVar, [Val]} | Ds], Match)
|
|
when is_list(Val) and (Val /= "") ->
|
|
NewMatch = case SVar of
|
|
"jid" -> Match;
|
|
"fn" -> Match#vcard_search{fn = Val};
|
|
"family" -> Match#vcard_search{family = Val};
|
|
"given" -> Match#vcard_search{given = Val};
|
|
"middle" -> Match#vcard_search{middle = Val};
|
|
"nickname" -> Match#vcard_search{nickname = Val};
|
|
"bday" -> Match#vcard_search{bday = Val};
|
|
"ctry" -> Match#vcard_search{ctry = Val};
|
|
"locality" -> Match#vcard_search{locality = Val};
|
|
"email" -> Match#vcard_search{email = Val};
|
|
"orgname" -> Match#vcard_search{orgname = Val};
|
|
"orgunit" -> Match#vcard_search{orgunit = Val};
|
|
_ -> Match
|
|
end,
|
|
filter_fields(Ds, NewMatch);
|
|
filter_fields([_ | Ds], Match) ->
|
|
filter_fields(Ds, Match).
|