X-Git-Url: http://git.euphorik.ch/?p=euphorik.git;a=blobdiff_plain;f=modules%2Ferl%2Feuphorik_bd.erl;h=6c3485dba49ab6cee643de4728fa6dde12f07976;hp=f1d5a601e6cc7db50dcc9cff5aeb2983cdb3e2e8;hb=81393a74d648e4830dc8c7c562fa203f50a105f6;hpb=828daf607fdce322ea17c1786dff84f712aa1632 diff --git a/modules/erl/euphorik_bd.erl b/modules/erl/euphorik_bd.erl index f1d5a60..6c3485d 100755 --- a/modules/erl/euphorik_bd.erl +++ b/modules/erl/euphorik_bd.erl @@ -15,7 +15,7 @@ % % You should have received a copy of the GNU General Public License % along with Euphorik. If not, see . -% +% % Ce module permet de gérer les données persistantes lié au site d'euphorik.ch. % Il permet d'ajouter des message, de demande les messages sur une page donnée, etc.. % Ce module utilise une base mnesia. @@ -23,11 +23,11 @@ -module(euphorik_bd). --export([ +-export([ % texte : get_texte/1, get_texte/2, - + % users : nouveau_user/2, nouveau_user/4, @@ -35,14 +35,14 @@ update_date_derniere_connexion/1, update_ip/2, update_pseudo_user/2, - user_by_cookie/1, - user_by_id/1, + user_by_cookie/1, + user_by_id/1, user_by_login/1, user_by_login_password/2, user_by_mess/1, css_from_user_cookie/1, is_ek_master_from_cookie/1, - + % messages : nouveau_message/3, nouveau_message_sys/1, @@ -60,26 +60,16 @@ est_une_reponse_a_user/2, a_repondu_a_message/2, possede_message/2, - + % ip : list_ban/0, ban/2, deban/1, est_banni/1, can_register/1, - - % trolls : - trolls/0, - trolls/1, - put_troll/2, - mod_troll/2, - del_troll/1, - troll_by_id/1, - current_troll/0, - elire_troll/0, - + % utiles : - resultat_transaction/1, + resultat_transaction/1, get_tuples/3 % must be in a transaction ]). -import(qlc, [e/2, q/1, cursor/2]). @@ -111,8 +101,8 @@ nouveau_user(Cookie, Profile) -> User end, resultat_transaction(mnesia:transaction(F)). - - + + % Ajoute un nouveau user et le renvoie nouveau_user(Login, Password, Cookie, Profile) -> F = fun() -> @@ -122,7 +112,7 @@ nouveau_user(Login, Password, Cookie, Profile) -> User end, resultat_transaction(mnesia:transaction(F)). - + % Définit les données du profile d'une utilisateur. set_profile(Cookie, Login, Password, Profile) -> @@ -174,13 +164,13 @@ update_ip(User_id, IP) -> mnesia:abort("update_ip: User inconnu") end end - ). - - + ). + + % Met à jour le pseudo du user update_pseudo_user(UserId, Pseudo) -> mnesia:transaction( - fun() -> + fun() -> case mnesia:wread({user, UserId}) of [#user{profile = Profile} = User] when Profile#profile.pseudo =/= Pseudo -> mnesia:write(User#user{profile = Profile#profile { pseudo = Pseudo } }); @@ -189,7 +179,7 @@ update_pseudo_user(UserId, Pseudo) -> end end ). - + % Est-ce qu'un utilisateur existe en fonction de son cookie ? % Renvoie {ok, User} ou erreur @@ -202,8 +192,8 @@ user_by_cookie(Cookie) -> end end )). - - + + user_by_id(ID) -> resultat_transaction(mnesia:transaction( fun() -> @@ -217,8 +207,8 @@ user_by_id(ID) -> end end )). - - + + user_by_login(Login) -> resultat_transaction(mnesia:transaction( fun() -> @@ -229,24 +219,24 @@ user_by_login(Login) -> end end )). - + % Renvoie une chaine représentant le cookie ou undefined si pas trouvé. css_from_user_cookie(Cookie) -> - case user_by_cookie(Cookie) of + case user_by_cookie(Cookie) of {ok, #user{profile = Profile}} -> Profile#profile.css; _ -> undefined end. - - + + is_ek_master_from_cookie(Cookie) -> case user_by_cookie(Cookie) of {ok, #user{ek_master = true}} -> true; _ -> false end. - + user_by_login_password(Login, Password) -> resultat_transaction(mnesia:transaction( @@ -257,8 +247,8 @@ user_by_login_password(Login, Password) -> end end )). - - + + % Renvoie {ok, User} où User est un #user possédant le message donné. user_by_mess(Id) -> resultat_transaction(mnesia:transaction( @@ -269,26 +259,26 @@ user_by_mess(Id) -> end end )). - - + + % 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_ids) -> % regarde si les id 'Repond_A' existent - F = fun() -> + F = fun() -> Repond_a = lists:foldr( fun(Repond_a_id, Acc) -> - case mnesia:read({minichat, Repond_a_id}) of + case mnesia:read({minichat, Repond_a_id}) of [M] -> [M | Acc]; - _ -> Acc % le message n'est pas trouvé + _ -> Acc % le message n'est pas trouvé end - end, + end, [], Repond_A_ids ), Racine_id = case Repond_a of [] -> undefined; - [M | _] -> + [M | _] -> Une_racine = M#minichat.racine_id, % vérification que tout les messages de Repond_a possède la même racine (même conversation) case lists:all(fun(R) -> R#minichat.racine_id =:= Une_racine end, Repond_a) of @@ -297,9 +287,9 @@ nouveau_message(Mess, Auteur_id, Repond_A_ids) -> _ -> {erreur, "Les messages ne font pas partie de la même conversation"} end - end, - if length(Repond_a) =/= length(Repond_A_ids) -> - {erreur, "Un ou plusieurs messages introuvable"}; + end, + if length(Repond_a) =/= length(Repond_A_ids) -> + {erreur, "Un ou plusieurs messages introuvable"}; true -> case Racine_id of {erreur, E} -> {erreur, E}; @@ -321,7 +311,7 @@ nouveau_message(Mess, Auteur_id, Repond_A_ids) -> 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."); Auteur#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM -> {erreur, "Bloqué pour cause de flood"}; - true -> + true -> mnesia:write(Auteur_maj), Id = nouvel_id(minichat), inserer_reponses(Id, Repond_A_ids), @@ -338,19 +328,19 @@ nouveau_message(Mess, Auteur_id, Repond_A_ids) -> _ -> {erreur, "L'auteur du message est introuvable"} end - end + end end end, resultat_transaction(mnesia:transaction(F)). - + % Définit Id_repondant comme étant la réponse à Ids. Ids est une liste d'id. inserer_reponses(Id_repondant, [Id_mess | Reste]) -> mnesia:write(#reponse_minichat{repondant = Id_repondant, cible = Id_mess}), inserer_reponses(Id_repondant, Reste); inserer_reponses(_, []) -> ok. - - + + % Permet de créer un message système. % Renvoie l'id du message système nouveau_message_sys(Mess) -> @@ -362,62 +352,51 @@ nouveau_message_sys(Mess) -> Id end )). - - + + % Renvoie N messages se trouvant sur la première page -messages(N) -> +messages(N) -> messages(N, 1). -% Renvoie N messages se trouvant sur la page P +% Renvoie N messages se trouvant sur la page P % TODO FIXME : fonction en O(N * P) ! messages(N, P) -> - F = fun() -> - % % #minichat{contenu = contenu_message(E)} + F = fun() -> get_tuples(minichat, mnesia:table_info(minichat, size) - N * P + 1, N) end, - resultat_transaction(mnesia:transaction(F)). - - -get_tuples(Table, First_id, N) -> - lists:foldr( - fun(Id, Acc) -> - case mnesia:read({Table, Id}) of - [T] -> [T | Acc]; - _ -> Acc - end - end, - [], - lists:seq(First_id, First_id + N - 1) - ). - + resultat_transaction(mnesia:transaction(F)). + + +get_tuples(Table, First_id, N) -> + lists:foldr( + fun(Id, Acc) -> + case mnesia:read({Table, Id}) of + [T] -> [T | Acc]; + _ -> Acc + end + end, + [], + lists:seq(First_id, First_id + N - 1) + ). + % Renvoie les messages manquants pour la page P en sachant qu'il y a N message % par page et que le dernier message que l'on possède est Id messages(Id, N, P) -> lists:filter(fun (M) -> M#minichat.id > Id end, messages(N, P)). - - + + % Renvoie {ok, #minichat} (voir #minichat de euphorik_bd.hrl) à partir de son id. message_by_id(Id) -> resultat_transaction(mnesia:transaction( fun() -> case mnesia:read({minichat, Id}) of - [M] -> {ok, M#minichat{contenu = contenu_message(M)}}; + [M] -> {ok, M}; _ -> erreur end end )). - - -% Renvoie le contenu d'un message donnée en fonction du troll associé, à utiliser à l'intérieur d'une transaction. -% TODO : Cette fonction pourrait être remplacée par un "outer-join", est-ce possible avec qlc ? -contenu_message(E) -> - case mnesia:index_read(troll, E#minichat.id, #troll.id_minichat) of - [] -> E#minichat.contenu; - [T] -> E#minichat.contenu ++ T#troll.content - end. - % Renvoie une liste de message (voir #minichat de euphorik_bd.hrl) à partir d'une liste d'id (Ids). % TODO : optimisations ? serait-ce du O(n) ? @@ -432,7 +411,7 @@ messages_by_ids(Ids) -> ),[{tmpdir, ?KEY_SORT_TEMP_DIR}]) end )). - + % Est-ce qu'un message existe ? Renvoie un boolean. % TODO : ya pas plus simple ? @@ -440,16 +419,16 @@ message_existe(Id) -> resultat_transaction(mnesia:transaction(fun() -> length(e(q([E#minichat.id || E <- mnesia:table(minichat), E#minichat.id =:= Id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])) =:= 1 end)). - - + + % Renvoie les reponses (utilisé normalement uniquement pendant le debug). reponses() -> F = fun() -> e(q([E || E <- mnesia:table(reponse_minichat)]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) end, resultat_transaction(mnesia:transaction(F)). - - + + % Renvoie les messages auquel M_id répond. parents(M_id) -> resultat_transaction(mnesia:transaction( @@ -461,8 +440,8 @@ parents(M_id) -> M#minichat.id =:= R#reponse_minichat.cible]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) end )). - - + + % Renvoie les message qui repondent à M_id enfants(M_id) -> resultat_transaction(mnesia:transaction( @@ -474,8 +453,8 @@ enfants(M_id) -> M#minichat.id =:= R#reponse_minichat.repondant]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) end )). - - + + % Renvoie les id des parents d'un message M (les messages auquels répond M) % ordrés du plus petit au plus grand.. % On évite d'utiliser qlc pour des raisons de performance @@ -488,11 +467,11 @@ parents_id(M_id) -> fun(#reponse_minichat{cible = Cible}) -> Cible end, Parents )); - _ -> [] + _ -> [] end end)). - - + + % Renvoie les id des enfants d'un message M (les messages qui répondent à M) % ordrés du plus petit au plus grand. % @spec enfants_id(integer()) -> [integer()] @@ -504,10 +483,10 @@ enfants_id(M_id) -> fun(#reponse_minichat{repondant = Repondant}) -> Repondant end, Enfants )); - _ -> [] + _ -> [] end end)). - + % Est-ce que le message Id_mess est une réponse d'une message de Id_user ? % On evite d'utiliser qlc (ce qui était fait avant) pour des raisons de performance. @@ -530,7 +509,7 @@ est_une_reponse_a_user(Id_user, Id_mess) -> end )). - + % Est-ce que Id_user à répondu au message Id_mess % On evite d'utiliser qlc (ce qui était fait avant) pour des raisons de performance. a_repondu_a_message(Id_user, Id_mess) -> @@ -551,8 +530,8 @@ a_repondu_a_message(Id_user, Id_mess) -> end end )). - - + + % Est-ce que Id_user possède Id_mess ? possede_message(Id_user, Id_mess) -> case mnesia:transaction( @@ -563,8 +542,8 @@ possede_message(Id_user, Id_mess) -> {atomic, [Id_user | []]} -> true; _ -> false end. - - + + % renvoie la liste des ip bannies % liste de {ip, temps_restant(en minutes), Users} ou Users est une liste de {pseudo, login} % TODO : déterminer la complexité de cette fonction. (Il n'y a pas d'index sur #user.last_ip) @@ -597,13 +576,13 @@ ban(IP, 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) -> @@ -629,8 +608,8 @@ est_banni(User_id) -> end end )). - - + + % Ban est une date tel que retourner par now(). % Ban_duration est un temps en minutes. % retourne une date. @@ -638,16 +617,16 @@ 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 + +% 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. can_register(IP) -> resultat_transaction(mnesia:transaction( fun() -> case e(q([I || I <- mnesia:table(ip_table), I#ip_table.ip =:= IP]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of - [] -> + [] -> mnesia:write(#ip_table{ip = IP, date_last_try_register = now()}), true; [T] -> @@ -665,144 +644,8 @@ can_register(IP) -> end end )). - - -% Renvoie tous les trolls -trolls() -> - resultat_transaction(mnesia:transaction( - fun() -> - e(qlc:keysort(#troll.id, q([T || T <- mnesia:table(troll)])), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) - end - )). - - -% Renvoie les trolls manquants posté après Last_id. -trolls(Last_id) -> - resultat_transaction(mnesia:transaction( - fun() -> - e(qlc:keysort(#troll.id, q([T || T <- mnesia:table(troll), T#troll.id > Last_id, T#troll.date_post =:= undefined])), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) - end - )). - - - % Crée un nouveau troll. - % 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 - ] - ), [{tmpdir, ?KEY_SORT_TEMP_DIR}])), - Nb_troll_poste_total = length(e(q( - [ - E#troll.id || E <- mnesia:table(troll), - E#troll.date_post =:= undefined - ] - ), [{tmpdir, ?KEY_SORT_TEMP_DIR}])), - 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(troll), - 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]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) 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() -> - C = cursor(qlc:keysort(#troll.date_post, q([T || T <- mnesia:table(troll), T#troll.date_post =/= undefined]), [{order, descending}]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]), - 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) -% Un message est posté par 'Sys' et le troll elu est lié à ce message -% 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]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of - [] -> - plus_de_trolls; - Trolls -> - Troll = lists:nth(random:uniform(length(Trolls)), Trolls), - Id_message = nouveau_message_sys("Troll de la semaine : "), - Troll2 = Troll#troll{date_post = now(), id_minichat = Id_message}, - mnesia:write(Troll2) - end - end - ). - - % Renvoie le résultat d'une transaction (en décomposant le tuple fournit) resultat_transaction({_, T}) -> T. @@ -811,4 +654,3 @@ resultat_transaction({_, T}) -> % Renvoie un nouvel id pour une table donnée nouvel_id(Table) -> mnesia:dirty_update_counter(counter, Table, 1). - \ No newline at end of file