X-Git-Url: http://git.euphorik.ch/?p=euphorik.git;a=blobdiff_plain;f=modules%2Ferl%2Feuphorik_bd.erl;h=dff71dbf1fb68aef172c68509d213ac6b3c83c21;hp=eb6f1d4a9a3bc2caae2e267a78abfe3e4ae3d889;hb=654e586482addf692991118ab9a4c5ead082457e;hpb=eae50232e73fca26d6f631c8177cbea649e5036e diff --git a/modules/erl/euphorik_bd.erl b/modules/erl/euphorik_bd.erl index eb6f1d4..dff71db 100755 --- a/modules/erl/euphorik_bd.erl +++ b/modules/erl/euphorik_bd.erl @@ -1,611 +1,814 @@ -% coding: utf-8 -% 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. -% @author G.Burri - +% coding: utf-8 +% Copyright 2008 Grégory Burri +% +% This file is part of Euphorik. +% +% Euphorik is free software: you can redistribute it and/or modify +% it under the terms of the GNU General Public License as published by +% the Free Software Foundation, either version 3 of the License, or +% (at your option) any later version. +% +% Euphorik is distributed in the hope that it will be useful, +% but WITHOUT ANY WARRANTY; without even the implied warranty of +% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +% GNU General Public License for more details. +% +% 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. +% @author G.Burri + + -module(euphorik_bd). --export([ - % gestion : - create/0, - connect/0, - connect/1, - reset/0, - - % users : - nouveau_user/2, - nouveau_user/3, - set_profile/9, - update_date_derniere_connexion/1, - update_ip/2, - update_pseudo_user/2, - users/0, - user_by_cookie/1, - user_by_id/1, - 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, +-export([ + % users : + nouveau_user/2, + nouveau_user/4, + set_profile/4, + update_date_derniere_connexion/1, + update_ip/2, + update_pseudo_user/2, + 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 :e + nouveau_message/3, + nouveau_message_sys/1, + nouveau_message_sys/2, + messages/1, messages/2, - messages/3, - message_by_id/1, - messages_by_ids/1, - message_existe/1, - reponses/0, - repond_a/1, - est_une_reponse_a_user/2, - a_repondu_a_message/2, - possede_message/2, - - % ip : - ip_table/0, - ban/2, - deban/1, - est_banni/1, - can_register/1, - - % versions : - update_version/1, - - % utiles : + messages/3, + message_by_id/1, + messages_by_ids/1, + message_existe/1, + reponses/0, + parents/1, + enfants/1, + parents_id/1, + enfants_id/1, + 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, + message_id_associe/1, + + % utiles : resultat_transaction/1 ]). - --include("../include/euphorik_bd.hrl"). +-import(qlc, [e/2, q/1, cursor/2]). +-include("../include/euphorik_bd.hrl"). -include("../include/euphorik_defines.hrl"). -include_lib("stdlib/include/qlc.hrl"). - - -% Instructions pour créer une nouvelle base : -% $erl -sname yaws -mnesia dir '"/projets/euphorik/BD"' -% voir doc/installation.txt -% >l(euphorik_bd). -% >euphorik_bd:create(). -create() -> - mnesia:stop(), - mnesia:delete_schema([node()]), - mnesia:create_schema([node()]), % nécessaire pour les tables sur disc - mnesia:start(), - create_tables(), - reset(). -create_tables() -> - mnesia:create_table(counter, [ - {attributes, record_info(fields, counter)}, - {disc_copies, [node()]} - ]), - mnesia:create_table(minichat, [ - {attributes, record_info(fields, minichat)}, - {index, [auteur_id]}, - {disc_copies, [node()]} - ]), - mnesia:create_table(reponse_minichat, [ - {type, bag}, - {attributes, record_info(fields, reponse_minichat)}, - {index, [cible]}, - {disc_copies, [node()]} - ]), - mnesia:create_table(user, [ - {attributes, record_info(fields, user)}, - {index, [cookie, login]}, - {disc_copies, [node()]} - ]), - mnesia:create_table(ip_table, [ - {attributes, record_info(fields, ip_table)}, - {disc_copies, [node()]} - ]), - mnesia:create_table(troll, [ - {attributes, record_info(fields, troll)}, - {index, [date_post]}, - {disc_copies, [node()]} - ]). - - -% Connexion à la base de données de yaws sur overnux -connect() -> - connect(yaws@flynux). -connect(Node) -> - mnesia:start(), - mnesia:change_config(extra_db_nodes, [Node]). - - -% Efface tous les users, minichat_reponse et minichat. -reset() -> - mnesia:clear_table(counter), - mnesia:clear_table(user), - mnesia:clear_table(reponse_minichat), - mnesia:clear_table(minichat), - 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(), ek_master = true}, + + +% Ajoute un nouveau user et le renvoie +nouveau_user(Cookie, Profile) -> + F = fun() -> + Id = nouvel_id(user), + User = #user{id = Id, cookie = Cookie, date_creation = now(), date_derniere_connexion = now(), profile = Profile}, + mnesia:write(User), + User + end, + resultat_transaction(mnesia:transaction(F)). + + +% Ajoute un nouveau user et le renvoie +nouveau_user(Login, Password, Cookie, Profile) -> + F = fun() -> + Id = nouvel_id(user), + User = #user{id = Id, cookie = Cookie, login = Login, password = Password, date_creation = now(), date_derniere_connexion = now(), profile = Profile#profile{pseudo = Login}}, mnesia:write(User), User - end). - - -% Ajoute un nouveau user et le renvoie -nouveau_user(Pseudo, Cookie) -> - F = fun() -> - Id = nouvel_id(user), - User = #user{id = Id, cookie = Cookie, pseudo = Pseudo, date_creation = now(), date_derniere_connexion = now()}, - mnesia:write(User), - User - end, - resultat_transaction(mnesia:transaction(F)). - - -% Ajoute un nouveau user et le renvoie -nouveau_user(Login, Password, Cookie) -> - F = fun() -> - Id = nouvel_id(user), - User = #user{id = Id, cookie = Cookie, pseudo = Login, login = Login, password = Password, date_creation = now(), date_derniere_connexion = now()}, - mnesia:write(User), - User - end, - resultat_transaction(mnesia:transaction(F)). - - -% Mise à par Cookie les autres peuvent être undefined ce qui veut dire qu'ils ne seront pas modifié. -set_profile(Cookie, Login, Password, Pseudo, Email, Css, Nick_format, Page_principale, Conversations) -> - if Nick_format =:= nick; Nick_format =:= login; Nick_format =:= nick_login -> - resultat_transaction(mnesia:transaction( - fun() -> - 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 -> - 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) 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, - nick_format = Nick_format, - page_principale = if is_integer(Page_principale), Page_principale > 0 -> Page_principale; true -> User#user.page_principale end, - conversations = if is_list(Conversations) -> Conversations; true -> User#user.conversations end - }, - mnesia:write(User_modifie), - ok - end; - _ -> erreur - end - end - )); - true -> - erreur - end. - - -% Met à jour la date de la dernière connexion d'un utilisateur à maintenant -update_date_derniere_connexion(User_id) -> - mnesia:transaction( - fun() -> - case mnesia:wread({user, User_id}) of - [User] -> - mnesia:write(User#user{date_derniere_connexion = now()}); - _ -> - mnesia:abort("update_date_derniere_connexion: User inconnu") - 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) -> - mnesia:transaction( - fun() -> - case mnesia:wread({user, UserId}) of - [User] when User#user.pseudo =/= Pseudo -> - mnesia:write(User#user{pseudo = Pseudo}); - _ -> - mnesia:abort("update_pseudo_user: User inconnu ou pseudo deja à jour") - end - end - ). - - -% Renvoie tous les users. -users() -> - resultat_transaction(mnesia:transaction(fun() -> - qlc:e(qlc:q([E || E <- mnesia:table(user)])) - end)). - - -% Est-ce qu'un utilisateur existe en fonction de son cookie ? -% Renvoie {ok, User} ou erreur -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 - [User] -> {ok, User}; - _ -> erreur - end - end - )). - - -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 - [User] -> {ok, User}; - _ -> erreur - end - end - )). + end, + resultat_transaction(mnesia:transaction(F)). + + +% Définit les données du profile d'une utilisateur. +set_profile(Cookie, Login, Password, Profile) -> + resultat_transaction(mnesia:transaction( + fun() -> + case user_by_cookie(Cookie) of + {ok, User_existant} -> + case user_by_login(Login) of + {ok, U} when Login =/= [], U#user.id =/= User_existant#user.id -> + login_deja_pris; + _ -> + mnesia:write( + User_existant#user{ + 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 ! + password = if Password =:= [] -> User_existant#user.password; true -> Password end, + profile = Profile + } + ), + ok + end; + _ -> erreur + end + end + )). + + +% Met à jour la date de la dernière connexion d'un utilisateur à maintenant +update_date_derniere_connexion(User_id) -> + mnesia:transaction( + fun() -> + case mnesia:wread({user, User_id}) of + [User] -> + mnesia:write(User#user{date_derniere_connexion = now()}); + _ -> + mnesia:abort("update_date_derniere_connexion: User inconnu") + 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) -> + mnesia:transaction( + 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 } }); + _ -> + mnesia:abort("update_pseudo_user: User inconnu ou pseudo deja à jour") + end + end + ). + + +% Est-ce qu'un utilisateur existe en fonction de son cookie ? +% Renvoie {ok, User} ou erreur +user_by_cookie(Cookie) -> + resultat_transaction(mnesia:transaction( + fun() -> + case e(q([E || E <- mnesia:table(user), E#user.cookie =:= Cookie]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of + [User] -> {ok, User}; + _ -> erreur + end + end + )). + + +user_by_id(ID) -> + resultat_transaction(mnesia:transaction( + fun() -> +% case e(q([E || E <- mnesia:table(user), E#user.id =:= ID]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of +% [User] -> {ok, User}; +% _ -> erreur +% end + case mnesia:read({user, ID}) of + [User] -> {ok, User}; + _ -> erreur + end + end + )). 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]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]), case Users of [User] -> {ok, User}; _ -> erreur end end )). - - -toggle_ek_master(User_id) -> - resultat_transaction(mnesia:transaction( - fun() -> - Users = qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.id =:= User_id])), - case Users of - [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() -> - Users = qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.login =:= Login, E#user.password =:= Password])), - case Users of - [User] -> {ok, User}; - _ -> erreur - end - end - )). - - -% Renvoie {ok, User} où User est un #user possédant le message donné. -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}; - _ -> erreur - 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 + {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( + fun() -> + case e(q([E || E <- mnesia:table(user), E#user.login =:= Login, E#user.password =:= Password]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of + [User | _] -> {ok, User}; + _ -> erreur + end + end + )). + + +% Renvoie {ok, User} où User est un #user possédant le message donné. +user_by_mess(Id) -> + resultat_transaction(mnesia:transaction( + fun() -> + case e(q([U || U <- mnesia:table(user), M <- mnesia:table(minichat), M#minichat.id =:= Id, M#minichat.auteur_id =:= U#user.id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of + [User | _] -> {ok, User}; + _ -> erreur + 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() -> + Repond_a = e( + q([M || M <- mnesia:table(minichat), lists:member(M#minichat.id, Repond_A_ids)]), + [{tmpdir, ?KEY_SORT_TEMP_DIR}] + ), + Racine_id = case Repond_a of + [] -> undefined; + [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 + true -> + Une_racine; + _ -> + {erreur, "Les messages ne font pas partie de la même conversation"} + end + end, + case Racine_id of + {erreur, E} -> {erreur, E}; + _ -> + % est-ce que l'auteur existe ? + case e(q([E || E <- mnesia:table(user), E#user.id =:= Auteur_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of + [#user{profile = Profile} = Auteur] -> + if length(Repond_a) =/= length(Repond_A_ids) -> + {erreur, "Un ou plusieurs messages introuvable"}; + true -> + % comparaison entre la date du dernier poste et maintenant (gestion du flood) + 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, + Auteur_maj = Auteur#user{ + indice_flood = if Nouvel_indice_flood > ?INDICE_SPAM_MAX -> ?INDICE_SPAM_MAX; Nouvel_indice_flood < 0 -> 0; true -> Nouvel_indice_flood end, + date_derniere_connexion = now() + }, + % 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 -> + mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}), + 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 -> + mnesia:write(Auteur_maj), + Id = nouvel_id(minichat), + inserer_reponses(Id, Repond_A_ids), + mnesia:write(#minichat{ + id = Id, + auteur_id = Auteur#user.id, + date = now(), + pseudo = Profile#profile.pseudo, + contenu = Mess, + racine_id = if Racine_id =:= undefined -> Id; true -> Racine_id end + }), + Id + end + end; + _ -> + {erreur, "L'auteur du message est introuvable"} + 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) -> + nouveau_message_sys(Mess, undefined). + + +% Création d'un message système lié à un troll. +nouveau_message_sys(Mess, Troll_id) -> + {ok, #user{profile = Profile}} = user_by_id(0), + resultat_transaction(mnesia:transaction( + fun() -> + Id = nouvel_id(minichat), + mnesia:write(#minichat{id = Id, auteur_id = 0, date = now(), pseudo = Profile#profile.pseudo, contenu = Mess, troll_id = Troll_id, racine_id = Id}), + Id + end + )). + + +% Renvoie N messages se trouvant sur la première page +messages(N) -> + messages(N, 1). + + +% Renvoie N messages se trouvant sur la page P +messages(N, P) -> + F = fun() -> + C = cursor( + qlc:keysort( + #minichat.id, + q([E#minichat{contenu = contenu_message(E)} || E <- mnesia:table(minichat)]), + [{order, descending}] + ), + [{tmpdir, ?KEY_SORT_TEMP_DIR}] + ), + if P > 1 -> qlc:next_answers(C, N * (P - 1)); + true -> ok + end, + R = qlc:next_answers(C, N), + qlc:delete_cursor(C), + R + end, + lists:reverse(resultat_transaction(mnesia:transaction(F))). + + +% 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)}}; + _ -> 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é par un "outer-join", est-ce possible avec qlc ? +contenu_message(E) -> + case mnesia:read({troll, E#minichat.troll_id}) 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) ? +% Bon de toutes façons on s'en fout c'est pas utilisé :) +messages_by_ids(Ids) -> + resultat_transaction(mnesia:transaction( + fun() -> + e(qlc:keysort( + #minichat.id, + q([E || E <- mnesia:table(minichat), lists:any(fun(Id) -> Id =:= E#minichat.id end, Ids)]), + [{order, ascending}] + ),[{tmpdir, ?KEY_SORT_TEMP_DIR}]) + 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) -> - % 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)]))), - % est-ce que l'auteur existe ? - Auteur = case qlc:e(qlc: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, - % 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, - Auteur_maj = Auteur#user{ - indice_flood = if Nouvel_indice_flood > ?INDICE_SPAM_MAX -> ?INDICE_SPAM_MAX; Nouvel_indice_flood < 0 -> 0; true -> Nouvel_indice_flood end, - date_derniere_connexion = now() - }, - % 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 -> - mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}), - nouveau_message_sys(Auteur#user.pseudo ++ "(" ++ Auteur#user.login ++ ") 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 - 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) -> - {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) -> - messages(N, 1). - - -% 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}])])), - if P > 1 -> qlc:next_answers(C, N * (P - 1)); - true -> ok - end, - R = qlc:next_answers(C, N), - qlc:delete_cursor(C), - lists:reverse(R) - end, - resultat_transaction(mnesia:transaction(F)). - - -% 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) -> - case resultat_transaction(mnesia:transaction( - fun() -> - qlc:e(qlc:q([E || E <- qlc:keysort(2, mnesia:table(minichat), [{order, ascending}]), Id =:= E#minichat.id])) - end - )) of - [M] -> {ok, M}; - _ -> erreur - end. - - -% Renvoie une liste de message (voir #minichat de euphorik_bd.hrl) à partir d'une liste d'id (Ids). -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)]), - qlc:e(Query) - end - )). - - -% Est-ce qu'un message existe ? Renvoie un boolean. -% 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 - end)). - - -% Renvoie les reponses (utilisé normalement uniquement pendant le debug). -reponses() -> - F = fun() -> - qlc:e(qlc:q([E || E <- mnesia:table(reponse_minichat)])) - end, - resultat_transaction(mnesia:transaction(F)). - - -% Renvoie les messages auquel M_id répond. -repond_a(M_id) -> - resultat_transaction(mnesia:transaction( - fun() -> - qlc:e(qlc:q( - [M || E <- mnesia:table(reponse_minichat), - M <- mnesia:table(minichat), - E#reponse_minichat.repondant =:= M_id, - M#minichat.id =:= E#reponse_minichat.cible])) - end - )). - - -% Est-ce que le message Id_mess est une réponse d'une message de Id_user ? -est_une_reponse_a_user(Id_user, Id_mess) -> - case mnesia:transaction( - fun() -> - qlc:e(qlc: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}]) - end - ) of - {atomic, [_]} -> true; - _ -> false - end. - - -% Est-ce que Id_user à répondu au message Id_mess -a_repondu_a_message(Id_user, Id_mess) -> - case mnesia:transaction( - fun() -> - qlc:e(qlc: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}]) - end - ) of - {atomic, [_]} -> true; - _ -> false - end. - - -% Est-ce que Id_user possède 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])) - end - ) of - {atomic, [Id_user | []]} -> true; - _ -> false - end. - - -ip_table() -> - resultat_transaction(mnesia:transaction( - fun() -> - qlc:e(qlc: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 qlc:e(qlc: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. -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 - [] -> - mnesia:write(#ip_table{ip = IP, date_last_try_register = now()}), - true; - [T] -> - Delta = delta_date_ms(T#ip_table.date_last_try_register, now()), - if T#ip_table.nb_try_register =:= ?NB_MAX_FLOOD_REGISTER, Delta < ?TEMPS_BAN_FLOOD_REGISTER -> - false; - true -> - mnesia:write(#ip_table{ - ip = IP, - date_last_try_register = now(), - nb_try_register = T#ip_table.nb_try_register + if Delta < ?TEMPS_FLOOD_REGISTER -> 1; T#ip_table.nb_try_register > 0 -> -1; true -> 0 end - }), - true - end - 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}) -> - T. - - -% Retourne la difference entre deux timestamp (erlang:now()) en miliseconde -delta_date_ms(D1, D2) -> - 1000000000 * abs(element(1, D1) - element(1, D2)) + 1000 * abs(element(2, D1) - element(2, D2)) + trunc(abs(element(3, D1) - element(3, D2)) / 1000). - - -% 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 - Neg when Neg < 0 -> T; - Pos when Pos > 0 -> T + 1; - _ -> T - end. - - -% Renvoie un nouvel id pour une table donnée -nouvel_id(Table) -> + +% Est-ce qu'un message existe ? Renvoie un boolean. +% TODO : ya pas plus simple ? +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( + fun() -> + e(q( + [M || R <- mnesia:table(reponse_minichat), + M <- mnesia:table(minichat), + R#reponse_minichat.repondant =:= 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( + fun() -> + e(q( + [M || R <- mnesia:table(reponse_minichat), + M <- mnesia:table(minichat), + R#reponse_minichat.cible =:= 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 +% @spec parents_id(integer()) -> [integer()] +parents_id(M_id) -> + resultat_transaction(mnesia:transaction(fun() -> + case mnesia:read({reponse_minichat, M_id}) of + Parents when is_list(Parents) -> + lists:sort(lists:map( + 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()] +enfants_id(M) -> + resultat_transaction(mnesia:transaction(fun() -> + e( + qlc:sort( + q([E#reponse_minichat.repondant || E <- mnesia:table(reponse_minichat), E#reponse_minichat.cible =:= M]), + [{order, ascending}] + ), + [{tmpdir, ?KEY_SORT_TEMP_DIR}] + ) + 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. +est_une_reponse_a_user(Id_user, Id_mess) -> + resultat_transaction(mnesia:transaction( + fun() -> + case mnesia:read({reponse_minichat, Id_mess}) of + [] -> false; + Cibles -> + lists:any( + fun(#reponse_minichat{cible = Cible}) -> + case mnesia:read({minichat, Cible}) of + [#minichat{auteur_id = Id_user}] -> true; + _ -> false + end + end, + Cibles + ) + end + 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) -> + resultat_transaction(mnesia:transaction( + fun() -> + case mnesia:match_object({reponse_minichat, '_', Id_mess}) of + [] -> false; + Reponses -> + lists:any( + fun(#reponse_minichat{repondant = Reponse}) -> + case mnesia:read({minichat, Reponse}) of + [#minichat{auteur_id = Id_user}] -> true; + _ -> false + end + end, + Reponses + ) + end + end + )). + + +% Est-ce que Id_user possède Id_mess ? +possede_message(Id_user, Id_mess) -> + case mnesia:transaction( + fun() -> + e(q([E#minichat.auteur_id || E <- mnesia:table(minichat), E#minichat.id =:= Id_mess]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) + end + ) of + {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) +list_ban() -> + resultat_transaction(mnesia:transaction( + fun() -> + Now = now(), + e(qlc:keysort(1, q([ + { + IP#ip_table.ip, + delta_date_minute(date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration), Now), + 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}]) + } || + IP <- mnesia:table(ip_table), + if IP#ip_table.ban =:= undefined -> false; true -> date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration) > Now end + ])), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) + 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 + ]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) 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, delta_date_minute(Echeance, Now)} + 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. +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] -> + Delta = delta_date_ms(T#ip_table.date_last_try_register, now()), + if T#ip_table.nb_try_register =:= ?NB_MAX_FLOOD_REGISTER, Delta < ?TEMPS_BAN_FLOOD_REGISTER -> + false; + true -> + mnesia:write(#ip_table{ + ip = IP, + date_last_try_register = now(), + nb_try_register = T#ip_table.nb_try_register + if Delta < ?TEMPS_FLOOD_REGISTER -> 1; T#ip_table.nb_try_register > 0 -> -1; true -> 0 end + }), + true + end + 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(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]), [{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), + Troll2 = Troll#troll{date_post = now()}, + mnesia:write(Troll2), + nouveau_message_sys("Troll de la semaine : ", Troll2#troll.id) + end + end + ). + + +% Renvoie l'id du message associé au troll dont l'id est donnée. +% Renvoie undefined si il n'y en a pas. +message_id_associe(Troll_id) -> + resultat_transaction(mnesia:transaction( + fun() -> + case e(q([M#minichat.id || M <- mnesia:table(minichat), M#minichat.troll_id =:= Troll_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of + [Id] -> Id; + _ -> undefined + end + end + )). + + +% Renvoie le résultat d'une transaction (en décomposant le tuple fournit) +resultat_transaction({_, T}) -> + T. + + +% Retourne la difference entre deux timestamp (erlang:now()) en miliseconde +delta_date_ms(D1, D2) -> + 1000000000 * abs(element(1, D1) - element(1, D2)) + 1000 * abs(element(2, D1) - element(2, D2)) + trunc(abs(element(3, D1) - element(3, D2)) / 1000). + +% Retourne la différence entre deux timestamp (erlang:now) en minutes +delta_date_minute(D1, D2) -> + trunc(delta_date_ms(D1, D2) / 1000 / 60). + + +% 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