*** empty log message ***

SVN Revision: 17
This commit is contained in:
Alexey Shchepin 2002-12-14 20:07:26 +00:00
parent 496c041b28
commit 10f47ce78d
3 changed files with 76 additions and 23 deletions

View File

@ -95,11 +95,10 @@ terminate(Reason, State) ->
%%% Internal functions
%%%----------------------------------------------------------------------
% TODO: lowercase user name
check_password(User, Password) ->
LUser = jlib:tolower(User),
F = fun() ->
case mnesia:read({passwd, User}) of
case mnesia:read({passwd, LUser}) of
[E] ->
E#passwd.password
end
@ -113,20 +112,23 @@ check_password(User, Password) ->
set_password(User, Password) ->
LUser = jlib:tolower(User),
F = fun() ->
mnesia:write(#passwd{user = User, password = Password})
mnesia:write(#passwd{user = LUser, password = Password})
end,
mnesia:transaction(F).
try_register(User, Password) ->
LUser = jlib:tolower(User),
F = fun() ->
case mnesia:read({passwd, User}) of
case mnesia:read({passwd, LUser}) of
[] ->
mnesia:write(#passwd{user = User, password = Password}),
mnesia:write(#passwd{user = LUser,
password = Password}),
ok;
[E] ->
exists
end
end,
mnesia:transaction(F).

View File

@ -10,7 +10,8 @@
-author('alexey@sevcom.net').
-vsn('$Revision$ ').
-export([start/0, init/0, open_session/2, close_session/2]).
-export([start/0, init/0, open_session/2, close_session/2,
get_user_resources/1]).
-include_lib("mnemosyne/include/mnemosyne.hrl").
-include("ejabberd.hrl").
@ -240,3 +241,20 @@ do_route(From, To, Packet) ->
false
end.
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
get_user_resources(User) ->
F = fun() ->
mnemosyne:eval(query [X.resource || X <- table(user_resource),
X.user = User]
end)
end,
case mnesia:transaction(F) of
{atomic, Rs} ->
Rs;
{aborted, Reason} ->
?DEBUG("delivery failed: ~p~n", [Reason]),
[]
end.

View File

@ -78,13 +78,16 @@ process_iq(From, To, IQ) ->
process_iq_get(From, To, {iq, ID, Type, XMLNS, SubEl}) ->
{User, _, _} = From,
LUser = jlib:tolower(User),
F = fun() ->
mnesia:read({roster, User})
mnesia:read({roster, LUser})
end,
case mnesia:transaction(F) of
{atomic, Items} ->
XItems = lists:map(fun item_to_xml/1, Items),
{iq, ID, result, XMLNS, XItems};
{iq, ID, result, XMLNS, [{xmlelement, "query",
[{"xmlns", "jabber:iq:roster"}],
XItems}]};
_ ->
{iq, ID, error, XMLNS,
[SubEl, jlib:make_error_element("500",
@ -102,6 +105,8 @@ item_to_xml(Item) ->
Attrs3 = case Item#roster.subscription of
none ->
[{"subscription", "none"} | Attrs2];
remove ->
[{"subscription", "remove"} | Attrs2];
_ ->
% TODO
Attrs2
@ -124,36 +129,45 @@ process_item_set(User, To, XItem) ->
{xmlelement, Name, Attrs, Els} = XItem,
% TODO: load existing item
JID = jlib:string_to_jid(xml:get_attr_s("jid", Attrs)),
LUser = jlib:tolower(User),
case JID of
error ->
ok;
_ ->
F = fun() ->
Res = mnemosyne:eval(query [X || X <- table(roster),
X.user = User,
X.user = LUser,
X.jid = JID]
end),
Item = case Res of
[] ->
#roster{user = User,
#roster{user = LUser,
jid = JID,
groups = [],
xattrs = [],
xs = []};
[I] ->
mnesia:delete_object(I),
I
I#roster{groups = [],
xattrs = [],
xs = []}
end,
Item1 = process_item_attrs(Item, Attrs),
Item2 = process_item_els(Item1, Els),
mnesia:write(Item2),
case Item2#roster.subscription of
remove ->
ok;
_ ->
mnesia:write(Item2)
end,
Item2
end,
case mnesia:transaction(F) of
{atomic, Item} ->
io:format("ROSTER: push for user ~p: ~p~n", [User, Item]),
push_item(User, To, Item),
ok;
_ ->
E ->
?DEBUG("ROSTER: roster item set error: ~p~n", [E]),
ok
end
end.
@ -163,26 +177,26 @@ process_item_attrs(Item, [{Attr, Val} | Attrs]) ->
"jid" ->
case jlib:string_to_jid(Val) of
error ->
process_item_attrs(Item, [Attrs]);
process_item_attrs(Item, Attrs);
JID ->
process_item_attrs(Item#roster{jid = JID}, [Attrs])
process_item_attrs(Item#roster{jid = JID}, Attrs)
end;
"name" ->
process_item_attrs(Item#roster{name = Val}, [Attrs]);
process_item_attrs(Item#roster{name = Val}, Attrs);
"subscription" ->
case Val of
"remove" ->
process_item_attrs(Item#roster{subscription = remove},
[Attrs]);
Attrs);
_ ->
process_item_attrs(Item, [Attrs])
process_item_attrs(Item, Attrs)
end;
"ask" ->
process_item_attrs(Item, [Attrs]);
process_item_attrs(Item, Attrs);
_ ->
XAttrs = Item#roster.xattrs,
process_item_attrs(Item#roster{xattrs = [{Attr, Val} | XAttrs]},
[Attrs])
Attrs)
end;
process_item_attrs(Item, []) ->
Item.
@ -207,3 +221,22 @@ process_item_els(Item, [{xmlcdata, _} | Els]) ->
process_item_els(Item, Els);
process_item_els(Item, []) ->
Item.
push_item(User, From, Item) ->
lists:foreach(fun(Resource) ->
push_item(User, Resource, From, Item)
end, ejabberd_sm:get_user_resources(User)).
% TODO: don't push to those who not load roster
push_item(User, Resource, From, Item) ->
ResIQ = {iq, "", set, "jabber:iq:roster",
[{xmlelement, "query",
[{"xmlns", "jabber:iq:roster"}],
[item_to_xml(Item)]}]},
ejabberd_router ! {route,
From,
{User, ?MYNAME, Resource},
jlib:iq_to_xml(ResIQ)}.