X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=modules%2Ferl%2Feuphorik_bd.erl;h=d354c21e983e8b86613f2f38de14e1c4a5bffea3;hb=8bec0dac79e750d0040de8a009c6ae864479642e;hp=9ad6da2caf10c0736d404b2941458601455fc90e;hpb=2233e8f8760b118c3cacf541ef32ea456a8e1527;p=euphorik.git diff --git a/modules/erl/euphorik_bd.erl b/modules/erl/euphorik_bd.erl index 9ad6da2..d354c21 100755 --- a/modules/erl/euphorik_bd.erl +++ b/modules/erl/euphorik_bd.erl @@ -4,7 +4,7 @@ % Ce module utilise une base mnesia. % @author G.Burri --module(euphorik_minichat). +-module(euphorik_bd). -export([ % gestion : create/0, @@ -17,6 +17,7 @@ nouveau_user/3, set_profile/9, update_date_derniere_connexion/1, + update_ip/2, update_pseudo_user/2, users/0, user_by_cookie/1, @@ -24,9 +25,11 @@ user_by_login/1, user_by_login_password/2, user_by_mess/1, + toggle_ek_master/1, % messages : nouveau_message/3, + nouveau_message_sys/1, messages/1, messages/2, messages/3, @@ -39,10 +42,32 @@ a_repondu_a_message/2, possede_message/2, - % utiles : + % ip : + ip_table/0, + ban/2, + deban/1, + est_banni/1, can_register/1, + + % trolls : + trolls/0, + trolls/1, + trolls_attente/1, + put_troll/2, + mod_troll/2, + del_troll/1, + troll_by_id/1, + current_troll/0, + elire_troll/0, + + % versions : + update_version/1, + + % utiles : resultat_transaction/1 -]). +]). + +-import(qlc, [e/1, e/2, q/1, cursor/1]). -include("../include/euphorik_bd.hrl"). -include("../include/euphorik_defines.hrl"). @@ -88,6 +113,7 @@ create_tables() -> ]), mnesia:create_table(troll, [ {attributes, record_info(fields, troll)}, + {index, [date_post]}, {disc_copies, [node()]} ]). @@ -105,10 +131,12 @@ reset() -> mnesia:clear_table(counter), mnesia:clear_table(user), mnesia:clear_table(reponse_minichat), - mnesia:clear_table(minichat), + mnesia:clear_table(minichat), + mnesia:clear_table(troll), + mnesia:clear_table(ip_table), % crée l'utilisateur root mnesia:transaction(fun() -> - User = #user{id = 0, pseudo = "Sys", login = "Sys", date_creation = now(), date_derniere_connexion = now()}, + User = #user{id = 0, pseudo = "Sys", login = "Sys", date_creation = now(), date_derniere_connexion = now(), ek_master = true}, mnesia:write(User), User end). @@ -144,14 +172,14 @@ set_profile(Cookie, Login, Password, Pseudo, Email, Css, Nick_format, Page_princ case user_by_cookie(Cookie) of {ok, User} -> case user_by_login(Login) of - {ok, U} when User#user.login =/= [], U#user.id =/= User#user.id -> + {ok, U} when Login =/= [], U#user.id =/= User#user.id -> login_deja_pris; - _ -> + _ -> User_modifie = User#user{ % TODO : pourquoi ne pas tester avec la valeur "undefined" plutôt qu'avec "is_list" ? % TODO : validation plus strict des données (pas de page négative dans les conv par exemple) login = if is_list(Login) -> Login; true -> User#user.login end, - password = if is_list(Password) -> Password; true -> User#user.password end, + password = if is_list(Password) andalso Password =/= [] -> Password; true -> User#user.password end, pseudo = if is_list(Pseudo) -> Pseudo; true -> User#user.pseudo end, email = if is_list(Email) -> Email; true -> User#user.email end, css = if is_list(Css) -> Css; true -> User#user.css end, @@ -172,10 +200,10 @@ set_profile(Cookie, Login, Password, Pseudo, Email, Css, Nick_format, Page_princ % Met à jour la date de la dernière connexion d'un utilisateur à maintenant -update_date_derniere_connexion(UserId) -> +update_date_derniere_connexion(User_id) -> mnesia:transaction( fun() -> - case mnesia:wread({user, UserId}) of + case mnesia:wread({user, User_id}) of [User] -> mnesia:write(User#user{date_derniere_connexion = now()}); _ -> @@ -183,7 +211,21 @@ update_date_derniere_connexion(UserId) -> end end ). - + + +% Met à jour l'ip d'un user +update_ip(User_id, IP) -> + mnesia:transaction( + fun() -> + case mnesia:wread({user, User_id}) of + [User] -> + mnesia:write(User#user{last_ip = IP}); + _ -> + mnesia:abort("update_ip: User inconnu") + end + end + ). + % Met à jour le pseudo du user update_pseudo_user(UserId, Pseudo) -> @@ -202,7 +244,7 @@ update_pseudo_user(UserId, Pseudo) -> % Renvoie tous les users. users() -> resultat_transaction(mnesia:transaction(fun() -> - qlc:e(qlc:q([E || E <- mnesia:table(user)])) + e(q([E || E <- mnesia:table(user)])) end)). @@ -211,8 +253,7 @@ users() -> user_by_cookie(Cookie) -> resultat_transaction(mnesia:transaction( fun() -> - Users = qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.cookie =:= Cookie])), - case Users of + case e(q([E || E <- mnesia:table(user), E#user.cookie =:= Cookie])) of [User] -> {ok, User}; _ -> erreur end @@ -223,8 +264,7 @@ user_by_cookie(Cookie) -> user_by_id(ID) -> resultat_transaction(mnesia:transaction( fun() -> - Users = qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.id =:= ID])), - case Users of + case e(q([E || E <- mnesia:table(user), E#user.id =:= ID])) of [User] -> {ok, User}; _ -> erreur end @@ -235,7 +275,7 @@ user_by_id(ID) -> user_by_login(Login) -> resultat_transaction(mnesia:transaction( fun() -> - Users = qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.login =:= Login])), + Users = e(q([E || E <- mnesia:table(user), E#user.login =:= Login])), case Users of [User] -> {ok, User}; _ -> erreur @@ -244,12 +284,24 @@ user_by_login(Login) -> )). -user_by_login_password(Login, Password) -> +toggle_ek_master(User_id) -> resultat_transaction(mnesia:transaction( fun() -> - Users = qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.login =:= Login, E#user.password =:= Password])), + Users = e(q([E || E <- mnesia:table(user), E#user.id =:= User_id])), case Users of - [User] -> {ok, User}; + [User] -> + mnesia:write(User#user{ek_master = not User#user.ek_master}); + _ -> erreur + end + end + )). + + +user_by_login_password(Login, Password) -> + resultat_transaction(mnesia:transaction( + fun() -> + case e(q([E || E <- mnesia:table(user), E#user.login =:= Login, E#user.password =:= Password])) of + [User | _] -> {ok, User}; _ -> erreur end end @@ -260,35 +312,28 @@ user_by_login_password(Login, Password) -> user_by_mess(Id) -> resultat_transaction(mnesia:transaction( fun() -> - case qlc:e(qlc:q([U || U <- mnesia:table(user), M <- mnesia:table(minichat), M#minichat.id =:= Id, M#minichat.auteur_id =:= U#user.id])) of - [User] -> {ok, User}; + case e(q([U || U <- mnesia:table(user), M <- mnesia:table(minichat), M#minichat.id =:= Id, M#minichat.auteur_id =:= U#user.id])) of + [User | _] -> {ok, User}; _ -> erreur end end )). - -% Renvoie l'utilisateur root -root() -> - {ok, User} = user_by_id(0), - User. - % Ajoute un message. Repond_A est une liste d'id auquel le message répond % retourne soit l'id du message soit erreur. nouveau_message(Mess, Auteur_id, Repond_A) -> % regarde si les id 'Repond_A' existent F = fun() -> - Nb_id_trouve = length(qlc:e(qlc:q([E#minichat.id || E <- mnesia:table(minichat), lists:member(E#minichat.id, Repond_A)]))), + Nb_id_trouve = length(e(q([E#minichat.id || E <- mnesia:table(minichat), lists:member(E#minichat.id, Repond_A)]))), % est-ce que l'auteur existe ? - Auteur = case qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.id =:= Auteur_id])) of + Auteur = case e(q([E || E <- mnesia:table(user), E#user.id =:= Auteur_id])) of [A] -> A; _ -> throw("L'auteur du message est introuvable") end, if Nb_id_trouve =/= length(Repond_A) -> throw("Un ou plusieurs messages introuvable"); true -> ok end, - Id = nouvel_id(minichat), % compare les dernière Delta = delta_date_ms(Auteur#user.date_derniere_connexion, now()), Nouvel_indice_flood = Auteur#user.indice_flood + if Delta =< ?DUREE_SPAM -> 2; true -> -1 end, @@ -298,14 +343,13 @@ nouveau_message(Mess, Auteur_id, Repond_A) -> }, % est-ce que l'auteur à trop floodé ? if Auteur#user.indice_flood =/= ?INDICE_SPAM_MAX, Auteur_maj#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM -> - Root = root(), mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}), - mnesia:write(#minichat{id=Id, auteur_id=Root#user.id, date=now(), pseudo=Root#user.pseudo, contenu=Auteur#user.pseudo ++ "(" ++ Auteur#user.login ++ ") est bloqué pour " ++ integer_to_list(trunc(?DUREE_BLOCAGE_SPAM / 1000)) ++ " secondes pour cause de flood.."}), - Id; + 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."); Auteur#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM -> erreur; true -> mnesia:write(Auteur_maj), + Id = nouvel_id(minichat), inserer_reponses(Id, Repond_A), mnesia:write(#minichat{id=Id, auteur_id=Auteur#user.id, date=now(), pseudo=Auteur#user.pseudo, contenu=Mess}), Id @@ -320,6 +364,18 @@ inserer_reponses(Id_repondant, [Id_mess | Reste]) -> inserer_reponses(_, []) -> ok. + +% Permet de créer un message système. +% Renvoie l'id du message système +nouveau_message_sys(Mess) -> + {ok, Root} = user_by_id(0), + mnesia:transaction( + fun() -> + Id = nouvel_id(minichat), + mnesia:write(#minichat{id=Id, auteur_id=0, date=now(), pseudo=Root#user.pseudo, contenu=Mess}) + end + ). + % Renvoie N messages se trouvant sur la première page messages(N) -> @@ -329,7 +385,7 @@ messages(N) -> % Renvoie N messages se trouvant sur la page P messages(N, P) -> F = fun() -> - C = qlc:cursor(qlc:q([E || E <- qlc:keysort(2, mnesia:table(minichat), [{order, descending}])])), + C = cursor(q([E || E <- qlc:keysort(2, mnesia:table(minichat), [{order, descending}])])), if P > 1 -> qlc:next_answers(C, N * (P - 1)); true -> ok end, @@ -350,7 +406,7 @@ messages(Id, N, P) -> message_by_id(Id) -> case resultat_transaction(mnesia:transaction( fun() -> - qlc:e(qlc:q([E || E <- qlc:keysort(2, mnesia:table(minichat), [{order, ascending}]), Id =:= E#minichat.id])) + e(q([E || E <- qlc:keysort(2, mnesia:table(minichat), [{order, ascending}]), Id =:= E#minichat.id])) end )) of [M] -> {ok, M}; @@ -363,9 +419,8 @@ messages_by_ids(Ids) -> resultat_transaction(mnesia:transaction( fun() -> % TODO : optimisations ? serait-ce du O(n) ? - Query = qlc:q([E || E <- qlc:keysort(2, mnesia:table(minichat), [{order, ascending}]), lists:any(fun(Id) -> Id =:= E#minichat.id end, Ids)]), - %io:format("~s~n", [qlc:info(Query)]), - qlc:e(Query) + Query = q([E || E <- qlc:keysort(2, mnesia:table(minichat), [{order, ascending}]), lists:any(fun(Id) -> Id =:= E#minichat.id end, Ids)]), + e(Query) end )). @@ -374,14 +429,14 @@ messages_by_ids(Ids) -> % TODO : ya pas plus simple ? message_existe(Id) -> resultat_transaction(mnesia:transaction(fun() -> - length(qlc:e(qlc:q([E#minichat.id || E <- mnesia:table(minichat), E#minichat.id =:= Id]))) =:= 1 + length(e(q([E#minichat.id || E <- mnesia:table(minichat), E#minichat.id =:= Id]))) =:= 1 end)). % Renvoie les reponses (utilisé normalement uniquement pendant le debug). reponses() -> F = fun() -> - qlc:e(qlc:q([E || E <- mnesia:table(reponse_minichat)])) + e(q([E || E <- mnesia:table(reponse_minichat)])) end, resultat_transaction(mnesia:transaction(F)). @@ -390,7 +445,7 @@ reponses() -> repond_a(M_id) -> resultat_transaction(mnesia:transaction( fun() -> - qlc:e(qlc:q( + e(q( [M || E <- mnesia:table(reponse_minichat), M <- mnesia:table(minichat), E#reponse_minichat.repondant =:= M_id, @@ -403,7 +458,7 @@ repond_a(M_id) -> est_une_reponse_a_user(Id_user, Id_mess) -> case mnesia:transaction( fun() -> - qlc:e(qlc:q([ + e(q([ M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat), M#minichat.auteur_id =:= Id_user, M#minichat.id =:= R#reponse_minichat.cible, R#reponse_minichat.repondant =:= Id_mess ]), [{unique_all, true}]) @@ -418,7 +473,7 @@ est_une_reponse_a_user(Id_user, Id_mess) -> a_repondu_a_message(Id_user, Id_mess) -> case mnesia:transaction( fun() -> - qlc:e(qlc:q([ + e(q([ M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat), R#reponse_minichat.cible =:= Id_mess, R#reponse_minichat.repondant =:= M#minichat.id, M#minichat.auteur_id =:= Id_user ]), [{unique_all, true}]) @@ -433,21 +488,84 @@ a_repondu_a_message(Id_user, Id_mess) -> possede_message(Id_user, Id_mess) -> case mnesia:transaction( fun() -> - qlc:e(qlc:q([E#minichat.auteur_id || E <- mnesia:table(minichat), E#minichat.id =:= Id_mess])) + e(q([E#minichat.auteur_id || E <- mnesia:table(minichat), E#minichat.id =:= Id_mess])) end ) of {atomic, [Id_user | []]} -> true; _ -> false end. + +ip_table() -> + resultat_transaction(mnesia:transaction( + fun() -> + e(q([IP || IP <- mnesia:table(ip_table)])) + end + )). + + +% Bannie une ip pour un certain temps (en minute). +ban(IP, Duration) -> + mnesia:transaction( + fun() -> + case mnesia:wread({ip_table, IP}) of + [IP_tuple] -> + mnesia:write(IP_tuple#ip_table{ban = now(), ban_duration = Duration}); + _ -> + mnesia:write(#ip_table{ip = IP, ban = now(), ban_duration = Duration}) + end + end + ). + + +% Débanni une ip +deban(IP) -> + ban(IP, 0). + + +% Renvoie soit {true, Temps} où Temps est le temps en minutes pendant lequel le user est encore banni +% ou false. +est_banni(User_id) -> + resultat_transaction(mnesia:transaction( + fun() -> + case e(q([ + {IP#ip_table.ban, IP#ip_table.ban_duration} || + U <- mnesia:table(user), + U#user.id =:= User_id, + IP <- mnesia:table(ip_table), + IP#ip_table.ip =:= U#user.last_ip + ])) of + [{Ban, Ban_duration}] -> + Echeance = date_plus_minutes(Ban, Ban_duration), + Now = now(), + if Echeance < Now -> % l'échéance est passée + false; + true -> + {true, trunc(delta_date_ms(Echeance, Now) / 1000 / 60)} + end; + _ -> + false + end + end + )). + +% Ban est une date tel que retourner par now(). +% Ban_duration est un temps en minutes. +% retourne une date. +date_plus_minutes(Ban, Ban_duration) -> + Duration_sec = Ban_duration * 60, + {MegaSec, Sec, MicroSec} = Ban, + {MegaSec + if Sec + Duration_sec >= 1000000 -> 1; true -> 0 end,(Sec + Duration_sec) rem 1000000, MicroSec}. + + % Si deux enregistrements consequtifs de la même IP sont fait en moins d'une seconde alors % ip_table.nb_try_register est incrémenté de 1 sinon il est décrémenté de 1 (jusqu'a 0). -% Si ip_table.nb_try_register vaut 5 alors l'ip ne peux plus s'enregistrer pour une heure. +% Si ip_table.nb_try_register vaut 5 alors l'ip ne peux plus s'enregistrer pour une heure. can_register(IP) -> resultat_transaction(mnesia:transaction( fun() -> - case qlc:e(qlc:q([I || I <- mnesia:table(ip_table), I#ip_table.ip =:= IP])) of + case e(q([I || I <- mnesia:table(ip_table), I#ip_table.ip =:= IP])) of [] -> mnesia:write(#ip_table{ip = IP, date_last_try_register = now()}), true; @@ -467,6 +585,193 @@ can_register(IP) -> end )). + +% Renvoie tous les trolls. +trolls() -> + resultat_transaction(mnesia:transaction( + fun() -> + e(q([T || T <- qlc:keysort(2, mnesia:table(troll))])) + end + )). + + +% Renvoie les trolls manquants posté après Last_id. +trolls(Last_id) -> + resultat_transaction(mnesia:transaction( + e(q([T || T <- qlc:keysort(2, mnesia:table(troll)), T#troll.id > Last_id, T#troll.date_post =:= undefined])) + )). + + +% Renvoie les trolls manquants posté après Last_id. +% Si pas de trolls alors attend un événement tel qu'un ajout, une modification ou une suppression. +% renvoie : +% {mod, Troll} +% ou {add, [Trolls]} +% ou {del, Troll_id} +% ou timeout +trolls_attente(Last_id) -> + case mnesia:subscribe({table, troll, detailed}) of + {error, E} = E -> + E; + _ -> + R = case resultat_transaction(mnesia:transaction( + fun() -> + e(q([T || T <- qlc:keysort(2, mnesia:table(troll)), T#troll.id > Last_id, T#troll.date_post =:= undefined])) + end + )) of + [] -> % pas de trolls + attend_evenement_troll(); + Trolls -> + {add, Trolls} + end, + mnesia:unsubscribe({table, troll, detailed}), + R + end. + +attend_evenement_troll() -> + % s'il n'y a pas de trolls que l'utilisateur n'a pas connaissance alors on attend un événement + receive + % cas où un troll est choisit comme courant + {mnesia_table_event, {write, troll, Troll, [Old_troll | _], _}} when Old_troll#troll.date_post =:= undefined, Troll#troll.date_post =/= undefined -> + {del, Troll#troll.id}; + {mnesia_table_event, {write, troll, Troll, [_Old_troll | _], _}} -> + {mod, Troll}; + {mnesia_table_event, {write, troll, Troll, [], _}} -> + {add, [Troll]}; + {mnesia_table_event, {delete, troll, {troll, Id}, _, _}} -> + {del, Id}; + _ -> + attend_evenement_troll() + after 1000 * 60 * 60 -> + timeout + end. + + + % Renvoie l'id du nouveau troll + % ou max_troll_reached_per_user si le nombre de troll posté par l'utilisateur max a été atteind + % ou max_troll_reached si le nombre de troll posté max a été atteind + % ou user_unknown +put_troll(User_id, Content) -> + resultat_transaction(mnesia:transaction( + fun() -> + % control le nombre de troll déjà posté + Nb_troll_poste_par_user = length(e(q( + [ + E#troll.id || E <- mnesia:table(troll), + E#troll.id_user =:= User_id, + E#troll.date_post =:= undefined + ] + ))), + Nb_troll_poste_total = length(e(q( + [ + E#troll.id || E <- mnesia:table(troll), + E#troll.date_post =:= undefined + ] + ))), + User = user_by_id(User_id), + case User of + {ok, _} -> + if Nb_troll_poste_par_user >= ?NB_MAX_TROLL_WAITING_BY_USER -> + max_troll_reached_per_user; + Nb_troll_poste_total >= ?NB_MAX_TROLL_WAITING -> + max_troll_reached; + true -> + Id = nouvel_id(minichat), + mnesia:write(#troll{id = Id, id_user = User_id, date_create = now(), content = Content}), + Id + end; + _ -> + user_unknown + end + end + )). + + +% renvoie ok | erreur +mod_troll(Troll_id, Content) -> + mnesia:transaction( + fun() -> + case mnesia:wread({troll, Troll_id}) of + [Troll = #troll{date_post = undefined}] -> + mnesia:write(Troll#troll{content = Content}); + _ -> + mnesia:abort("mod_troll: Troll inconnu ou déjà posté") + end + end + ). + + +del_troll(Troll_id) -> + mnesia:transaction( + fun() -> + case mnesia:wread({troll, Troll_id}) of + [#troll{date_post = undefined}] -> + mnesia:delete({troll, Troll_id}); + _ -> + mnesia:abort("mod_troll: Troll inconnu ou déjà posté") + end + end + ). + + +troll_by_id(Troll_id) -> + resultat_transaction(mnesia:transaction( + fun() -> + case e(q([T || T <- mnesia:table(troll), T#troll.id =:= Troll_id])) of + [T] -> {ok, T}; + _ -> + erreur + end + end + )). + + +% Renvoie le troll actuel qui se trouve sur la page principale. +% Renvois aucun si pas de troll courant. +current_troll() -> + resultat_transaction(mnesia:transaction( + fun() -> + % TODO : ya pas moyen de désigner le champs plutot qu'avec un nombre pour le tri ? + C = cursor(q([T || T <- qlc:keysort(5, mnesia:table(troll), [{order, descending}]), T#troll.date_post =/= undefined])), + R = case qlc:next_answers(C, 1) of + [T] -> T; + _ -> aucun + end, + qlc:delete_cursor(C), + R + end + )). + + +% Elit un troll au hasard parmis les trolls en attente (leur date_post =:= undefined) +% met à jour sa date de post. +% renvoie plus_de_trolls si il n'y a aucun troll en attente. +elire_troll() -> + {A1,A2,A3} = now(), + random:seed(A1, A2, A3), + mnesia:transaction( + fun() -> + case e(q([T || T <- mnesia:table(troll), T#troll.date_post =:= undefined])) of + [] -> + plus_de_trolls; + Trolls -> + Troll = lists:nth(random:uniform(length(Trolls)), Trolls), + Troll2 = Troll#troll{date_post = now()}, + mnesia:write(Troll2) + end + end + ). + + + +update_version(1) -> + mnesia:transform_table( + ip_table, + fun() -> null end, + record_info(fields, ip_table), + ip_table + ). + % Renvoie le résultat d'une transaction (en décomposant le tuple fournit) resultat_transaction({_, T}) -> @@ -479,6 +784,7 @@ delta_date_ms(D1, D2) -> % Bizarre, cette fonction n'existe pas dans la stdlib. +% Pas utilisé mais bon ca me fait de la peine de l'enlever. ceiling(X) -> T = trunc(X), case (X - T) of