\r
\r
-module(euphorik_bd).\r
--export([\r
- % gestion :\r
- create/0,\r
- connect/0,\r
- connect/1,\r
- reset/0,\r
- \r
+-export([ \r
% users :\r
nouveau_user/2,\r
- nouveau_user/3,\r
- set_profile/10,\r
+ nouveau_user/4,\r
+ set_profile/4,\r
update_date_derniere_connexion/1,\r
update_ip/2,\r
update_pseudo_user/2,\r
- print_users/0,\r
- print_users/1,\r
- print_user/1,\r
user_by_cookie/1, \r
user_by_id/1, \r
user_by_login/1,\r
user_by_login_password/2,\r
user_by_mess/1,\r
- toggle_ek_master/1,\r
css_from_user_cookie/1,\r
+ is_ek_master_from_cookie/1,\r
\r
% messages :e\r
nouveau_message/3,\r
-include_lib("stdlib/include/qlc.hrl").\r
\r
\r
-% Instructions pour créer une nouvelle base : \r
-% $erl -sname yaws -mnesia dir '"/projets/euphorik/BD"'\r
-% voir doc/installation.txt\r
-% >l(euphorik_bd).\r
-% >euphorik_bd:create().\r
-create() ->\r
- mnesia:stop(),\r
- mnesia:delete_schema([node()]),\r
- mnesia:create_schema([node()]), % nécessaire pour les tables sur disc\r
- mnesia:start(),\r
- create_tables(),\r
- reset().\r
- \r
-create_tables() ->\r
- mnesia:create_table(counter, [\r
- {attributes, record_info(fields, counter)},\r
- {disc_copies, [node()]}\r
- ]),\r
- mnesia:create_table(proprietes, [\r
- {attributes, record_info(fields, proprietes)},\r
- {disc_copies, [node()]}\r
- ]),\r
- mnesia:create_table(minichat, [\r
- {attributes, record_info(fields, minichat)},\r
- {index, [auteur_id, troll_id]},\r
- {disc_copies, [node()]}\r
- ]),\r
- mnesia:create_table(reponse_minichat, [\r
- {type, bag},\r
- {attributes, record_info(fields, reponse_minichat)},\r
- {index, [cible]},\r
- {disc_copies, [node()]}\r
- ]),\r
- mnesia:create_table(user, [\r
- {attributes, record_info(fields, user)},\r
- {index, [cookie, login]},\r
- {disc_copies, [node()]}\r
- ]),\r
- mnesia:create_table(ip_table, [\r
- {attributes, record_info(fields, ip_table)},\r
- {disc_copies, [node()]}\r
- ]),\r
- mnesia:create_table(troll, [\r
- {attributes, record_info(fields, troll)},\r
- {index, [date_post]},\r
- {disc_copies, [node()]}\r
- ]).\r
- \r
- \r
-% Connexion à la base de données de yaws sur overnux\r
-connect() ->\r
- connect(yaws@flynux).\r
-connect(Node) ->\r
- mnesia:start(),\r
- mnesia:change_config(extra_db_nodes, [Node]).\r
-\r
-\r
-% Efface tous les users, minichat_reponse et minichat.\r
-reset() ->\r
- mnesia:clear_table(counter),\r
- mnesia:clear_table(proprietes),\r
- mnesia:clear_table(user),\r
- mnesia:clear_table(reponse_minichat),\r
- mnesia:clear_table(minichat),\r
- mnesia:clear_table(troll),\r
- mnesia:clear_table(ip_table),\r
- % crée l'utilisateur root\r
- mnesia:transaction(fun() ->\r
- mnesia:write(#proprietes{nom = version, valeur = ?VERSION_BD}),\r
- User = #user{id = 0, pseudo = "Sys", login = "Sys", date_creation = now(), date_derniere_connexion = now(), ek_master = true},\r
- mnesia:write(User),\r
- User\r
- end).\r
-\r
-\r
% Ajoute un nouveau user et le renvoie\r
-nouveau_user(Pseudo, Cookie) ->\r
+nouveau_user(Cookie, Profile) ->\r
F = fun() ->\r
Id = nouvel_id(user),\r
- User = #user{id = Id, cookie = Cookie, pseudo = Pseudo, date_creation = now(), date_derniere_connexion = now()},\r
+ User = #user{id = Id, cookie = Cookie, date_creation = now(), date_derniere_connexion = now(), profile = Profile},\r
mnesia:write(User),\r
User\r
end,\r
\r
\r
% Ajoute un nouveau user et le renvoie\r
-nouveau_user(Login, Password, Cookie) ->\r
+nouveau_user(Login, Password, Cookie, Profile) ->\r
F = fun() ->\r
Id = nouvel_id(user),\r
- User = #user{id = Id, cookie = Cookie, pseudo = Login, login = Login, password = Password, date_creation = now(), date_derniere_connexion = now()},\r
+ User = #user{id = Id, cookie = Cookie, login = Login, password = Password, date_creation = now(), date_derniere_connexion = now(), profile = Profile#profile{pseudo = Login}},\r
mnesia:write(User),\r
User\r
end,\r
resultat_transaction(mnesia:transaction(F)).\r
+ \r
\r
-\r
-% Mise à par Cookie les autres peuvent être undefined ce qui veut dire qu'ils ne seront pas modifié.\r
-set_profile(Cookie, Login, Password, Pseudo, Email, Css, Nick_format, View_times, View_tooltips, Conversations) ->\r
- if Nick_format =:= nick; Nick_format =:= login; Nick_format =:= nick_login ->\r
- resultat_transaction(mnesia:transaction(\r
- fun() ->\r
- case user_by_cookie(Cookie) of\r
- {ok, User} ->\r
- case user_by_login(Login) of\r
- {ok, U} when Login =/= [], U#user.id =/= User#user.id ->\r
- login_deja_pris;\r
- _ ->\r
- User_modifie = User#user{\r
- % TODO : pourquoi ne pas tester avec la valeur "undefined" plutôt qu'avec "is_list" ?\r
- % TODO : validation plus strict des données (pas de page négative dans les conv par exemple)\r
- login = if is_list(Login) -> Login; true -> User#user.login end,\r
- password = if is_list(Password) andalso Password =/= [] -> Password; true -> User#user.password end,\r
- pseudo = if is_list(Pseudo) -> Pseudo; true -> User#user.pseudo end,\r
- email = if is_list(Email) -> Email; true -> User#user.email end,\r
- css = if is_list(Css) -> Css; true -> User#user.css end,\r
- nick_format = Nick_format,\r
- view_times = View_times,\r
- view_tooltips = View_tooltips,\r
- conversations = if is_list(Conversations) -> Conversations; true -> User#user.conversations end\r
- },\r
- mnesia:write(User_modifie),\r
- ok\r
- end;\r
- _ -> erreur\r
- end\r
- end\r
- ));\r
- true ->\r
- erreur\r
- end.\r
+% Définit les données du profile d'une utilisateur.\r
+set_profile(Cookie, Login, Password, Profile) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ case user_by_cookie(Cookie) of\r
+ {ok, User_existant} ->\r
+ case user_by_login(Login) of\r
+ {ok, U} when Login =/= [], U#user.id =/= User_existant#user.id ->\r
+ login_deja_pris;\r
+ _ ->\r
+ mnesia:write(\r
+ User_existant#user{\r
+ login = if User_existant#user.login =:= [] -> Login; true -> User_existant#user.login end, % on ne peut pas changer de login sauf si on en a pas !\r
+ password = if Password =:= [] -> User_existant#user.password; true -> Password end,\r
+ profile = Profile\r
+ }\r
+ ),\r
+ ok\r
+ end;\r
+ _ -> erreur\r
+ end\r
+ end\r
+ )).\r
\r
\r
% Met à jour la date de la dernière connexion d'un utilisateur à maintenant\r
mnesia:abort("update_date_derniere_connexion: User inconnu")\r
end\r
end\r
- ). \r
+ ).\r
\r
\r
% Met à jour l'ip d'un user\r
mnesia:transaction(\r
fun() -> \r
case mnesia:wread({user, UserId}) of\r
- [User] when User#user.pseudo =/= Pseudo ->\r
- mnesia:write(User#user{pseudo = Pseudo});\r
+ [#user{profile = Profile} = User] when Profile#profile.pseudo =/= Pseudo ->\r
+ mnesia:write(User#user{profile = Profile#profile { pseudo = Pseudo } });\r
_ ->\r
mnesia:abort("update_pseudo_user: User inconnu ou pseudo deja à jour")\r
end\r
end\r
).\r
- \r
- \r
-% Affiche N user trié par leur date de dernière connexion.\r
-% Opt est une liste d'option d'affichage :\r
-% * ekmaster : n'affiche que les admins\r
-print_users(N, Opt) ->\r
- AfficheQueLesEkMaster = lists:any(fun(O) -> O =:= ekmaster end, Opt),\r
- resultat_transaction(mnesia:transaction(fun() ->\r
- C = cursor(\r
- qlc:keysort(\r
- #user.date_derniere_connexion, \r
- if AfficheQueLesEkMaster ->\r
- q([E || E <- mnesia:table(user), E#user.ek_master =:= true]);\r
- true ->\r
- q([E || E <- mnesia:table(user)])\r
- end,\r
- [{order, descending}]\r
- ),\r
- [{tmpdir, ?KEY_SORT_TEMP_DIR}]\r
- ),\r
- Users = qlc:next_answers(C, N),\r
- lists:foreach(\r
- fun(U) ->\r
- print_user(U)\r
- end,\r
- Users\r
- ),\r
- qlc:delete_cursor(C)\r
- end)).\r
- \r
- \r
-% Affiche tous les users.\r
-print_users(Opt) ->\r
- print_users(all_remaining, Opt).\r
-\r
-% Affiche tous les users.\r
-print_users() ->\r
- print_users(all_remaining, []).\r
- \r
-print_user(User) when is_record(User, user) ->\r
- #user{id = Id, pseudo = Pseudo, login = Login, ek_master = Ek_master, date_derniere_connexion = Date, last_ip = IP} = User,\r
- {{Annee, Mois, Jour}, {Heure, Min, _}} = calendar:now_to_local_time(Date),\r
- io:format(\r
- % id pseudo (login) IP Jour Mois Année Heure Minute\r
- "~4w : ~10.10..s(~10.10..s) ~s ~2w.~2.2.0w.~w - ~2wh~2.2.0w~n",\r
- [ \r
- Id,\r
- if Ek_master -> "*"; true -> "" end ++ Pseudo,\r
- Login,\r
- euphorik_common:serialize_ip(IP),\r
- Jour, Mois, Annee, Heure, Min\r
- ]\r
- );\r
-% Affichage d'un user en fonction de son login\r
-print_user(Login) when is_list(Login) ->\r
- case user_by_login(Login) of\r
- {ok, User} ->\r
- print_user(User);\r
- _ ->\r
- {erreur, "Login pas trouvé : " ++ Login}\r
- end;\r
-% Affichage d'un user en fonction de son id\r
-print_user(Id) when is_integer(Id) -> \r
- case user_by_id(Id) of \r
- {ok, User} ->\r
- print_user(User);\r
- _ ->\r
- {erreur, "Id pas trouvé : " ++ integer_to_list(Id)}\r
- end.\r
- \r
+ \r
\r
% Est-ce qu'un utilisateur existe en fonction de son cookie ?\r
% Renvoie {ok, User} ou erreur\r
user_by_id(ID) ->\r
resultat_transaction(mnesia:transaction(\r
fun() ->\r
- case e(q([E || E <- mnesia:table(user), E#user.id =:= ID]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
+% case e(q([E || E <- mnesia:table(user), E#user.id =:= ID]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
+% [User] -> {ok, User};\r
+% _ -> erreur\r
+% end\r
+ case mnesia:read({user, ID}) of\r
[User] -> {ok, User};\r
_ -> erreur\r
end\r
end\r
)).\r
\r
- \r
-toggle_ek_master(User_id) ->\r
- resultat_transaction(mnesia:transaction(\r
- fun() ->\r
- Users = e(q([E || E <- mnesia:table(user), E#user.id =:= User_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]),\r
- case Users of\r
- [User] ->\r
- mnesia:write(User#user{ek_master = not User#user.ek_master});\r
- _ -> erreur\r
- end\r
- end\r
- )).\r
- \r
\r
% Renvoie une chaine représentant le cookie ou undefined si pas trouvé.\r
css_from_user_cookie(Cookie) ->\r
case user_by_cookie(Cookie) of \r
- {ok, User} ->\r
- User#user.css;\r
+ {ok, #user{profile = Profile}} ->\r
+ Profile#profile.css;\r
_ ->\r
undefined\r
end.\r
\r
+ \r
+is_ek_master_from_cookie(Cookie) ->\r
+ case user_by_cookie(Cookie) of\r
+ {ok, #user{ek_master = true}} -> true;\r
+ _ -> false\r
+ end.\r
+ \r
\r
user_by_login_password(Login, Password) ->\r
resultat_transaction(mnesia:transaction(\r
_ ->\r
% est-ce que l'auteur existe ?\r
case e(q([E || E <- mnesia:table(user), E#user.id =:= Auteur_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
- [Auteur] ->\r
+ [#user{profile = Profile} = Auteur] ->\r
if length(Repond_a) =/= length(Repond_A_ids) ->\r
{erreur, "Un ou plusieurs messages introuvable"};\r
true ->\r
% est-ce que l'auteur à trop floodé ?\r
if Auteur#user.indice_flood =/= ?INDICE_SPAM_MAX, Auteur_maj#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->\r
mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}),\r
- nouveau_message_sys("''" ++ Auteur#user.pseudo ++ if Auteur#user.login =/= [] -> " (" ++ Auteur#user.login ++ ")"; true -> "" end ++ "'' est bloqué pour " ++ integer_to_list(trunc(?DUREE_BLOCAGE_SPAM / 1000)) ++ " secondes pour cause de flood.");\r
+ nouveau_message_sys("''" ++ Profile#profile.pseudo ++ if Auteur#user.login =/= [] -> " (" ++ Auteur#user.login ++ ")"; true -> "" end ++ "'' est bloqué pour " ++ integer_to_list(trunc(?DUREE_BLOCAGE_SPAM / 1000)) ++ " secondes pour cause de flood.");\r
Auteur#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->\r
{erreur, "Bloqué pour cause de flood"};\r
true -> \r
id = Id,\r
auteur_id = Auteur#user.id,\r
date = now(),\r
- pseudo = Auteur#user.pseudo,\r
+ pseudo = Profile#profile.pseudo,\r
contenu = Mess,\r
racine_id = if Racine_id =:= undefined -> Id; true -> Racine_id end\r
}),\r
\r
% Création d'un message système lié à un troll.\r
nouveau_message_sys(Mess, Troll_id) ->\r
- {ok, Root} = user_by_id(0),\r
+ {ok, #user{profile = Profile}} = user_by_id(0),\r
resultat_transaction(mnesia:transaction(\r
fun() ->\r
Id = nouvel_id(minichat),\r
- mnesia:write(#minichat{id=Id, auteur_id=0, date=now(), pseudo=Root#user.pseudo, contenu=Mess, troll_id=Troll_id, racine_id=Id}),\r
+ mnesia:write(#minichat{id = Id, auteur_id = 0, date = now(), pseudo = Profile#profile.pseudo, contenu = Mess, troll_id = Troll_id, racine_id = Id}),\r
Id\r
end\r
)).\r
resultat_transaction(mnesia:transaction(\r
fun() ->\r
case mnesia:read({minichat, Id}) of\r
- [] -> erreur;\r
- [M] ->\r
- {ok, M#minichat{contenu = contenu_message(M)}}\r
+ [M] -> {ok, M#minichat{contenu = contenu_message(M)}};\r
+ _ -> erreur\r
end\r
end\r
)).\r
\r
\r
-% Renvoie le contenu d'un message donnée, à utiliser à l'intérieur d'une transaction.\r
+% Renvoie le contenu d'un message donnée en fonction du troll associé, à utiliser à l'intérieur d'une transaction.\r
% TODO : Cette fonction pourrait être remplacé par un "outer-join", est-ce possible avec qlc ?\r
contenu_message(E) ->\r
case mnesia:read({troll, E#minichat.troll_id}) of\r
)).\r
\r
\r
-% Renvoie les parents d'un message M (les messages auquels répond M)\r
+% Renvoie les id des parents d'un message M (les messages auquels répond M)\r
% ordrés du plus petit au plus grand..\r
+% On évite d'utiliser qlc pour des raisons de performance\r
% @spec parents_id(integer()) -> [integer()]\r
-parents_id(M) ->\r
+parents_id(M_id) ->\r
resultat_transaction(mnesia:transaction(fun() ->\r
- e(\r
- qlc:sort(\r
- q([E#reponse_minichat.cible || E <- mnesia:table(reponse_minichat), E#reponse_minichat.repondant =:= M]),\r
- [{order, ascending}]\r
- ),\r
- [{tmpdir, ?KEY_SORT_TEMP_DIR}]\r
- )\r
+ case mnesia:read({reponse_minichat, M_id}) of\r
+ Parents when is_list(Parents) ->\r
+ lists:sort(lists:map(\r
+ fun(#reponse_minichat{cible = Cible}) -> Cible end,\r
+ Parents\r
+ ));\r
+ _ -> [] \r
+ end\r
end)).\r
\r
\r
\r
\r
% Est-ce que le message Id_mess est une réponse d'une message de Id_user ?\r
+% On evite d'utiliser qlc (ce qui était fait avant) pour des raisons de performance.\r
est_une_reponse_a_user(Id_user, Id_mess) ->\r
- case mnesia:transaction(\r
- fun() ->\r
- e(q([\r
- M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat),\r
- M#minichat.auteur_id =:= Id_user, M#minichat.id =:= R#reponse_minichat.cible, R#reponse_minichat.repondant =:= Id_mess\r
- ]), [{unique_all, true}, {tmpdir, ?KEY_SORT_TEMP_DIR}])\r
- end\r
- ) of\r
- {atomic, [_]} -> true;\r
- _ -> false\r
- end.\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ case mnesia:read({reponse_minichat, Id_mess}) of\r
+ [] -> false;\r
+ Cibles ->\r
+ lists:any(\r
+ fun(#reponse_minichat{cible = Cible}) ->\r
+ case mnesia:read({minichat, Cible}) of\r
+ [#minichat{auteur_id = Id_user}] -> true;\r
+ _ -> false\r
+ end\r
+ end,\r
+ Cibles\r
+ )\r
+ end\r
+ end\r
+ )).\r
\r
\r
% Est-ce que Id_user à répondu au message Id_mess\r
+% On evite d'utiliser qlc (ce qui était fait avant) pour des raisons de performance.\r
a_repondu_a_message(Id_user, Id_mess) ->\r
- case mnesia:transaction(\r
- fun() ->\r
- e(q([\r
- M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat),\r
- R#reponse_minichat.cible =:= Id_mess, R#reponse_minichat.repondant =:= M#minichat.id, M#minichat.auteur_id =:= Id_user\r
- ]), [{unique_all, true}, {tmpdir, ?KEY_SORT_TEMP_DIR}])\r
- end\r
- ) of\r
- {atomic, [_]} -> true;\r
- _ -> false\r
- end.\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ case mnesia:match_object({reponse_minichat, '_', Id_mess}) of\r
+ [] -> false;\r
+ Reponses ->\r
+ lists:any(\r
+ fun(#reponse_minichat{repondant = Reponse}) ->\r
+ case mnesia:read({minichat, Reponse}) of\r
+ [#minichat{auteur_id = Id_user}] -> true;\r
+ _ -> false\r
+ end\r
+ end,\r
+ Reponses\r
+ )\r
+ end\r
+ end\r
+ )).\r
\r
\r
% Est-ce que Id_user possède Id_mess ?\r
{\r
IP#ip_table.ip,\r
delta_date_minute(date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration), Now),\r
- e(q([{U#user.pseudo, U#user.login} || U <- mnesia:table(user), U#user.last_ip =:= IP#ip_table.ip]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])\r
+ e(q([{Profile#profile.pseudo, U#user.login} || #user{profile = Profile} = U <- mnesia:table(user), U#user.last_ip =:= IP#ip_table.ip]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])\r
} ||\r
IP <- mnesia:table(ip_table),\r
if IP#ip_table.ban =:= undefined -> false; true -> date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration) > Now end\r
projects / euphorik.git / blobdiff