-% Module de création de la base de données euphorik.
-% @author G.Burri
-
--module(euphorik_bd).\r
--export([\r
- create/0,\r
- create_tables/0,\r
- vers_version/1,\r
- peupler/0\r
- ]).\r
+% coding: utf-8\r
+% Copyright 2008 Grégory Burri\r
+%\r
+% This file is part of Euphorik.\r
+%\r
+% Euphorik is free software: you can redistribute it and/or modify\r
+% it under the terms of the GNU General Public License as published by\r
+% the Free Software Foundation, either version 3 of the License, or\r
+% (at your option) any later version.\r
+%\r
+% Euphorik is distributed in the hope that it will be useful,\r
+% but WITHOUT ANY WARRANTY; without even the implied warranty of\r
+% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\r
+% GNU General Public License for more details.\r
+%\r
+% You should have received a copy of the GNU General Public License\r
+% along with Euphorik. If not, see <http://www.gnu.org/licenses/>.\r
+% \r
+% Ce module permet de gérer les données persistantes lié au site d'euphorik.ch.\r
+% Il permet d'ajouter des message, de demande les messages sur une page donnée, etc..\r
+% Ce module utilise une base mnesia.\r
+% @author G.Burri\r
+\r
\r
+-module(euphorik_bd).\r
+-export([ \r
+ % users :\r
+ nouveau_user/2,\r
+ nouveau_user/4,\r
+ set_profile/4,\r
+ update_date_derniere_connexion/1,\r
+ update_ip/2,\r
+ update_pseudo_user/2,\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
+ css_from_user_cookie/1,\r
+ is_ek_master_from_cookie/1,\r
+ \r
+ % messages :e\r
+ nouveau_message/3,\r
+ nouveau_message_sys/1,\r
+ nouveau_message_sys/2,\r
+ messages/1,\r
+ messages/2,\r
+ messages/3,\r
+ message_by_id/1,\r
+ messages_by_ids/1,\r
+ message_existe/1,\r
+ reponses/0,\r
+ parents/1,\r
+ enfants/1,\r
+ parents_id/1,\r
+ enfants_id/1,\r
+ est_une_reponse_a_user/2,\r
+ a_repondu_a_message/2,\r
+ possede_message/2,\r
+ \r
+ % ip :\r
+ list_ban/0,\r
+ ban/2,\r
+ deban/1,\r
+ est_banni/1,\r
+ can_register/1,\r
+ \r
+ % trolls :\r
+ trolls/0,\r
+ trolls/1,\r
+ put_troll/2,\r
+ mod_troll/2,\r
+ del_troll/1,\r
+ troll_by_id/1,\r
+ current_troll/0,\r
+ elire_troll/0,\r
+ message_id_associe/1,\r
+ \r
+ % utiles :\r
+ resultat_transaction/1\r
+]).\r
+-import(qlc, [e/2, q/1, cursor/2]).\r
-include("../include/euphorik_bd.hrl").\r
-
-
-% Instructions pour créer une nouvelle base :
-% $erl -sname yaws -mnesia dir '"/projets/euphorik/BD"'
-% >l(euphorik_bd).
-% >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
-\r
-
-create_tables() ->\r
- mnesia:create_table(counter, [\r
- {attributes, record_info(fields, counter)},\r
- {disc_copies, [node()]}\r
- ]),\r
- mnesia:create_table(minichat, [
- {attributes, record_info(fields, minichat)},
- {index, [auteur_id]},
- {disc_copies, [node()]}
- ]),\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)},
- {index, [cookie, login]},\r
- {disc_copies, [node()]}\r
- ]).
-
-
-vers_version(2) ->
- mnesia:transform_table(
- user,
- fun({user, Id, Cookie, Pseudo, Date_creation, Date_derniere_connexion, Css}) ->
- {user, Id, Cookie, Pseudo, "", "", "", Date_creation, Date_derniere_connexion, Css}
- end,
- record_info(fields, user),
- user
- );
-vers_version(3) ->
- mnesia:transform_table(
- user,
- fun({user, Id, Cookie, Pseudo, Login, Password, Email, Date_creation, Date_derniere_connexion, Css}) ->
- {user, Id, Cookie, Pseudo, Login, Password, Email, Date_creation, Date_derniere_connexion, Css, 0}
- end,
- record_info(fields, user),
- user
- );
-vers_version(4) ->
- mnesia:transform_table(
- user,
- fun({user, Id, Cookie, Pseudo, Login, Password, Email, Date_creation, Date_derniere_connexion, Css, Indice_flood}) ->
- {user, Id, Cookie, Pseudo, Login, Password, Email, Date_creation, Date_derniere_connexion, Css, Indice_flood, []}
- end,
- record_info(fields, user),
- user
- ).
-
-% exemple de peuplage de la BD, utilisé pour les tests
-peupler() ->
- mnesia:transaction(
- fun() ->
- mnesia:write({minichat, now(), "Pierre", "Salut tout le monde"}),
- mnesia:write({minichat, now(), "Paul", "Salut à toi !"})
- end
- ).
-
-
+-include("../include/euphorik_defines.hrl").\r
+-include_lib("stdlib/include/qlc.hrl").\r
+\r
+\r
+% Ajoute un nouveau user et le renvoie\r
+nouveau_user(Cookie, Profile) ->\r
+ F = fun() ->\r
+ Id = nouvel_id(user),\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
+ resultat_transaction(mnesia:transaction(F)).\r
+ \r
+ \r
+% Ajoute un nouveau user et le renvoie\r
+nouveau_user(Login, Password, Cookie, Profile) ->\r
+ F = fun() ->\r
+ Id = nouvel_id(user),\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
+% 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
+update_date_derniere_connexion(User_id) ->\r
+ mnesia:transaction(\r
+ fun() ->\r
+ case mnesia:wread({user, User_id}) of\r
+ [User] ->\r
+ mnesia:write(User#user{date_derniere_connexion = now()});\r
+ _ ->\r
+ mnesia:abort("update_date_derniere_connexion: User inconnu")\r
+ end\r
+ end\r
+ ).\r
+\r
+\r
+% Met à jour l'ip d'un user\r
+update_ip(User_id, IP) ->\r
+ mnesia:transaction(\r
+ fun() ->\r
+ case mnesia:wread({user, User_id}) of\r
+ [User] ->\r
+ mnesia:write(User#user{last_ip = IP});\r
+ _ ->\r
+ mnesia:abort("update_ip: User inconnu")\r
+ end\r
+ end\r
+ ). \r
+ \r
+ \r
+% Met à jour le pseudo du user\r
+update_pseudo_user(UserId, Pseudo) ->\r
+ mnesia:transaction(\r
+ fun() -> \r
+ case mnesia:wread({user, UserId}) of\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
+% Est-ce qu'un utilisateur existe en fonction de son cookie ?\r
+% Renvoie {ok, User} ou erreur\r
+user_by_cookie(Cookie) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ case e(q([E || E <- mnesia:table(user), E#user.cookie =:= Cookie]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
+ [User] -> {ok, User};\r
+ _ -> erreur\r
+ end\r
+ end\r
+ )).\r
+ \r
+ \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
+% [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
+user_by_login(Login) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ Users = e(q([E || E <- mnesia:table(user), E#user.login =:= Login]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]),\r
+ case Users of\r
+ [User] -> {ok, User};\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{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
+ fun() ->\r
+ case e(q([E || E <- mnesia:table(user), E#user.login =:= Login, E#user.password =:= Password]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
+ [User | _] -> {ok, User};\r
+ _ -> erreur\r
+ end\r
+ end\r
+ )).\r
+ \r
+ \r
+% Renvoie {ok, User} où User est un #user possédant le message donné.\r
+user_by_mess(Id) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ 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\r
+ [User | _] -> {ok, User};\r
+ _ -> erreur\r
+ end\r
+ end\r
+ )).\r
+ \r
+ \r
+% Ajoute un message. Repond_A est une liste d'id auquel le message répond\r
+% retourne soit l'id du message soit {erreur, <raison>}.\r
+nouveau_message(Mess, Auteur_id, Repond_A_ids) ->\r
+ % regarde si les id 'Repond_A' existent\r
+ F = fun() -> \r
+ Repond_a = e(\r
+ q([M || M <- mnesia:table(minichat), lists:member(M#minichat.id, Repond_A_ids)]),\r
+ [{tmpdir, ?KEY_SORT_TEMP_DIR}]\r
+ ), \r
+ Racine_id = case Repond_a of\r
+ [] -> undefined;\r
+ [M | _ ] -> \r
+ Une_racine = M#minichat.racine_id,\r
+ % vérification que tout les messages de Repond_a possède la même racine (même conversation)\r
+ case lists:all(fun(R) -> R#minichat.racine_id =:= Une_racine end, Repond_a) of\r
+ true ->\r
+ Une_racine;\r
+ _ ->\r
+ {erreur, "Les messages ne font pas partie de la même conversation"}\r
+ end\r
+ end, \r
+ case Racine_id of\r
+ {erreur, E} -> {erreur, E};\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
+ [#user{profile = Profile} = Auteur] ->\r
+ if length(Repond_a) =/= length(Repond_A_ids) ->\r
+ {erreur, "Un ou plusieurs messages introuvable"};\r
+ true ->\r
+ % comparaison entre la date du dernier poste et maintenant (gestion du flood)\r
+ Delta = delta_date_ms(Auteur#user.date_derniere_connexion, now()),\r
+ Nouvel_indice_flood = Auteur#user.indice_flood + if Delta =< ?DUREE_SPAM -> 2; true -> -1 end,\r
+ Auteur_maj = Auteur#user{\r
+ indice_flood = if Nouvel_indice_flood > ?INDICE_SPAM_MAX -> ?INDICE_SPAM_MAX; Nouvel_indice_flood < 0 -> 0; true -> Nouvel_indice_flood end,\r
+ date_derniere_connexion = now()\r
+ },\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("''" ++ 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
+ mnesia:write(Auteur_maj),\r
+ Id = nouvel_id(minichat),\r
+ inserer_reponses(Id, Repond_A_ids),\r
+ mnesia:write(#minichat{\r
+ id = Id,\r
+ auteur_id = Auteur#user.id,\r
+ date = now(),\r
+ pseudo = Profile#profile.pseudo,\r
+ contenu = Mess,\r
+ racine_id = if Racine_id =:= undefined -> Id; true -> Racine_id end\r
+ }),\r
+ Id\r
+ end\r
+ end;\r
+ _ ->\r
+ {erreur, "L'auteur du message est introuvable"}\r
+ end\r
+ end\r
+ end,\r
+ resultat_transaction(mnesia:transaction(F)).\r
+ \r
+% Définit Id_repondant comme étant la réponse à Ids. Ids est une liste d'id.\r
+inserer_reponses(Id_repondant, [Id_mess | Reste]) ->\r
+ mnesia:write(#reponse_minichat{repondant = Id_repondant, cible = Id_mess}),\r
+ inserer_reponses(Id_repondant, Reste);\r
+inserer_reponses(_, []) ->\r
+ ok.\r
+ \r
+ \r
+% Permet de créer un message système.\r
+% Renvoie l'id du message système\r
+nouveau_message_sys(Mess) ->\r
+ nouveau_message_sys(Mess, undefined).\r
+ \r
+\r
+% Création d'un message système lié à un troll.\r
+nouveau_message_sys(Mess, Troll_id) ->\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 = Profile#profile.pseudo, contenu = Mess, troll_id = Troll_id, racine_id = Id}),\r
+ Id\r
+ end\r
+ )).\r
+ \r
+ \r
+% Renvoie N messages se trouvant sur la première page\r
+messages(N) -> \r
+ messages(N, 1).\r
+\r
+\r
+% Renvoie N messages se trouvant sur la page P\r
+messages(N, P) ->\r
+ F = fun() ->\r
+ C = cursor(\r
+ qlc:keysort(\r
+ #minichat.id, \r
+ q([E#minichat{contenu = contenu_message(E)} || E <- mnesia:table(minichat)]),\r
+ [{order, descending}]\r
+ ),\r
+ [{tmpdir, ?KEY_SORT_TEMP_DIR}]\r
+ ),\r
+ if P > 1 -> qlc:next_answers(C, N * (P - 1));\r
+ true -> ok\r
+ end,\r
+ R = qlc:next_answers(C, N),\r
+ qlc:delete_cursor(C),\r
+ R\r
+ end,\r
+ lists:reverse(resultat_transaction(mnesia:transaction(F))).\r
+\r
+\r
+% Renvoie les messages manquants pour la page P en sachant qu'il y a N message\r
+% par page et que le dernier message que l'on possède est Id\r
+messages(Id, N, P) ->\r
+ lists:filter(fun (M) -> M#minichat.id > Id end, messages(N, P)).\r
+ \r
+ \r
+% Renvoie {ok, #minichat} (voir #minichat de euphorik_bd.hrl) à partir de son id.\r
+message_by_id(Id) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ case mnesia:read({minichat, Id}) of\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 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
+ [] -> E#minichat.contenu;\r
+ [T] -> E#minichat.contenu ++ T#troll.content\r
+ end.\r
+ \r
+\r
+% Renvoie une liste de message (voir #minichat de euphorik_bd.hrl) à partir d'une liste d'id (Ids).\r
+% TODO : optimisations ? serait-ce du O(n) ?\r
+% Bon de toutes façons on s'en fout c'est pas utilisé :)\r
+messages_by_ids(Ids) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ e(qlc:keysort(\r
+ #minichat.id,\r
+ q([E || E <- mnesia:table(minichat), lists:any(fun(Id) -> Id =:= E#minichat.id end, Ids)]),\r
+ [{order, ascending}]\r
+ ),[{tmpdir, ?KEY_SORT_TEMP_DIR}])\r
+ end\r
+ )).\r
+ \r
+\r
+% Est-ce qu'un message existe ? Renvoie un boolean.\r
+% TODO : ya pas plus simple ?\r
+message_existe(Id) ->\r
+ resultat_transaction(mnesia:transaction(fun() ->\r
+ length(e(q([E#minichat.id || E <- mnesia:table(minichat), E#minichat.id =:= Id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])) =:= 1\r
+ end)).\r
+ \r
+ \r
+% Renvoie les reponses (utilisé normalement uniquement pendant le debug).\r
+reponses() ->\r
+ F = fun() ->\r
+ e(q([E || E <- mnesia:table(reponse_minichat)]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])\r
+ end,\r
+ resultat_transaction(mnesia:transaction(F)).\r
+ \r
+ \r
+% Renvoie les messages auquel M_id répond.\r
+parents(M_id) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ e(q(\r
+ [M || R <- mnesia:table(reponse_minichat),\r
+ M <- mnesia:table(minichat),\r
+ R#reponse_minichat.repondant =:= M_id,\r
+ M#minichat.id =:= R#reponse_minichat.cible]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])\r
+ end\r
+ )).\r
+ \r
+ \r
+% Renvoie les message qui repondent à M_id\r
+enfants(M_id) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ e(q(\r
+ [M || R <- mnesia:table(reponse_minichat),\r
+ M <- mnesia:table(minichat),\r
+ R#reponse_minichat.cible =:= M_id,\r
+ M#minichat.id =:= R#reponse_minichat.repondant]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])\r
+ end\r
+ )).\r
+ \r
+ \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_id) ->\r
+ resultat_transaction(mnesia:transaction(fun() ->\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
+% Renvoie les id des enfants d'un message M (les messages qui répondent à M)\r
+% ordrés du plus petit au plus grand.\r
+% @spec enfants_id(integer()) -> [integer()]\r
+enfants_id(M) ->\r
+ resultat_transaction(mnesia:transaction(fun() ->\r
+ e(\r
+ qlc:sort(\r
+ q([E#reponse_minichat.repondant || E <- mnesia:table(reponse_minichat), E#reponse_minichat.cible =:= M]),\r
+ [{order, ascending}]\r
+ ),\r
+ [{tmpdir, ?KEY_SORT_TEMP_DIR}]\r
+ )\r
+ end)).\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
+ 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
+ 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
+possede_message(Id_user, Id_mess) ->\r
+ case mnesia:transaction(\r
+ fun() ->\r
+ e(q([E#minichat.auteur_id || E <- mnesia:table(minichat), E#minichat.id =:= Id_mess]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])\r
+ end\r
+ ) of\r
+ {atomic, [Id_user | []]} -> true;\r
+ _ -> false\r
+ end.\r
+ \r
+ \r
+% renvoie la liste des ip bannies\r
+% liste de {ip, temps_restant(en minutes), Users} ou Users est une liste de {pseudo, login}\r
+% TODO : déterminer la complexité de cette fonction. (Il n'y a pas d'index sur #user.last_ip)\r
+list_ban() ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ Now = now(),\r
+ e(qlc:keysort(1, q([\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([{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
+ ])), [{tmpdir, ?KEY_SORT_TEMP_DIR}])\r
+ end\r
+ )).\r
+\r
+\r
+% Bannie une ip pour un certain temps (en minute).\r
+ban(IP, Duration) ->\r
+ mnesia:transaction(\r
+ fun() ->\r
+ case mnesia:wread({ip_table, IP}) of\r
+ [IP_tuple] ->\r
+ mnesia:write(IP_tuple#ip_table{ban = now(), ban_duration = Duration});\r
+ _ ->\r
+ mnesia:write(#ip_table{ip = IP, ban = now(), ban_duration = Duration})\r
+ end\r
+ end\r
+ ).\r
+ \r
+\r
+% Débanni une ip\r
+deban(IP) ->\r
+ ban(IP, 0).\r
+\r
+ \r
+% Renvoie soit {true, Temps} où Temps est le temps en minutes pendant lequel le user est encore banni\r
+% ou false.\r
+est_banni(User_id) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ case e(q([\r
+ {IP#ip_table.ban, IP#ip_table.ban_duration} ||\r
+ U <- mnesia:table(user),\r
+ U#user.id =:= User_id,\r
+ IP <- mnesia:table(ip_table),\r
+ IP#ip_table.ip =:= U#user.last_ip\r
+ ]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
+ [{Ban, Ban_duration}] ->\r
+ Echeance = date_plus_minutes(Ban, Ban_duration),\r
+ Now = now(),\r
+ if Echeance < Now -> % l'échéance est passée\r
+ false;\r
+ true ->\r
+ {true, delta_date_minute(Echeance, Now)}\r
+ end;\r
+ _ ->\r
+ false\r
+ end\r
+ end\r
+ )).\r
+ \r
+ \r
+% Ban est une date tel que retourner par now().\r
+% Ban_duration est un temps en minutes.\r
+% retourne une date.\r
+date_plus_minutes(Ban, Ban_duration) ->\r
+ Duration_sec = Ban_duration * 60,\r
+ {MegaSec, Sec, MicroSec} = Ban,\r
+ {MegaSec + if Sec + Duration_sec >= 1000000 -> 1; true -> 0 end,(Sec + Duration_sec) rem 1000000, MicroSec}.\r
+ \r
+\r
+% Si deux enregistrements consequtifs de la même IP sont fait en moins d'une seconde alors \r
+% ip_table.nb_try_register est incrémenté de 1 sinon il est décrémenté de 1 (jusqu'a 0).\r
+% Si ip_table.nb_try_register vaut 5 alors l'ip ne peux plus s'enregistrer pour une heure.\r
+can_register(IP) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ case e(q([I || I <- mnesia:table(ip_table), I#ip_table.ip =:= IP]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
+ [] -> \r
+ mnesia:write(#ip_table{ip = IP, date_last_try_register = now()}),\r
+ true;\r
+ [T] ->\r
+ Delta = delta_date_ms(T#ip_table.date_last_try_register, now()),\r
+ if T#ip_table.nb_try_register =:= ?NB_MAX_FLOOD_REGISTER, Delta < ?TEMPS_BAN_FLOOD_REGISTER ->\r
+ false;\r
+ true ->\r
+ mnesia:write(#ip_table{\r
+ ip = IP,\r
+ date_last_try_register = now(),\r
+ 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\r
+ }),\r
+ true\r
+ end\r
+ end\r
+ end\r
+ )).\r
+ \r
+\r
+% Renvoie tous les trolls\r
+trolls() ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ e(qlc:keysort(#troll.id, q([T || T <- mnesia:table(troll)])), [{tmpdir, ?KEY_SORT_TEMP_DIR}])\r
+ end\r
+ )).\r
+ \r
+ \r
+% Renvoie les trolls manquants posté après Last_id.\r
+trolls(Last_id) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ 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}])\r
+ end\r
+ )).\r
+ \r
+ \r
+ % Crée un nouveau troll.\r
+ % Renvoie l'id du nouveau troll\r
+ % ou max_troll_reached_per_user si le nombre de troll posté par l'utilisateur max a été atteind\r
+ % ou max_troll_reached si le nombre de troll posté max a été atteind\r
+ % ou user_unknown\r
+put_troll(User_id, Content) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ % control le nombre de troll déjà posté\r
+ Nb_troll_poste_par_user = length(e(q(\r
+ [\r
+ E#troll.id || E <- mnesia:table(troll),\r
+ E#troll.id_user =:= User_id,\r
+ E#troll.date_post =:= undefined\r
+ ]\r
+ ), [{tmpdir, ?KEY_SORT_TEMP_DIR}])),\r
+ Nb_troll_poste_total = length(e(q(\r
+ [\r
+ E#troll.id || E <- mnesia:table(troll),\r
+ E#troll.date_post =:= undefined\r
+ ]\r
+ ), [{tmpdir, ?KEY_SORT_TEMP_DIR}])),\r
+ User = user_by_id(User_id),\r
+ case User of\r
+ {ok, _} ->\r
+ if Nb_troll_poste_par_user >= ?NB_MAX_TROLL_WAITING_BY_USER ->\r
+ max_troll_reached_per_user;\r
+ Nb_troll_poste_total >= ?NB_MAX_TROLL_WAITING ->\r
+ max_troll_reached;\r
+ true ->\r
+ Id = nouvel_id(minichat),\r
+ mnesia:write(#troll{id = Id, id_user = User_id, date_create = now(), content = Content}),\r
+ Id\r
+ end;\r
+ _ ->\r
+ user_unknown\r
+ end\r
+ end\r
+ )).\r
+\r
+\r
+% renvoie ok | erreur\r
+mod_troll(Troll_id, Content) ->\r
+ mnesia:transaction(\r
+ fun() ->\r
+ case mnesia:wread({troll, Troll_id}) of\r
+ [Troll = #troll{date_post = undefined}] ->\r
+ mnesia:write(Troll#troll{content = Content});\r
+ _ ->\r
+ mnesia:abort("mod_troll: Troll inconnu ou déjà posté")\r
+ end\r
+ end\r
+ ).\r
+ \r
+ \r
+del_troll(Troll_id) ->\r
+ mnesia:transaction(\r
+ fun() ->\r
+ case mnesia:wread({troll, Troll_id}) of\r
+ [#troll{date_post = undefined}] ->\r
+ mnesia:delete({troll, Troll_id});\r
+ _ ->\r
+ mnesia:abort("mod_troll: Troll inconnu ou déjà posté")\r
+ end\r
+ end\r
+ ).\r
+ \r
+ \r
+troll_by_id(Troll_id) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ case e(q([T || T <- mnesia:table(troll), T#troll.id =:= Troll_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
+ [T] -> {ok, T};\r
+ _ ->\r
+ erreur\r
+ end\r
+ end\r
+ )).\r
+ \r
+\r
+% Renvoie le troll actuel qui se trouve sur la page principale.\r
+% Renvois aucun si pas de troll courant.\r
+current_troll() ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ 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
+ R = case qlc:next_answers(C, 1) of\r
+ [T] -> T;\r
+ _ -> aucun\r
+ end,\r
+ qlc:delete_cursor(C),\r
+ R\r
+ end\r
+ )).\r
+\r
+\r
+% Elit un troll au hasard parmis les trolls en attente (leur date_post =:= undefined)\r
+% Un message est posté par 'Sys' et le troll elu est lié à ce message\r
+% met à jour sa date de post.\r
+% renvoie plus_de_trolls si il n'y a aucun troll en attente.\r
+elire_troll() -> \r
+ {A1,A2,A3} = now(),\r
+ random:seed(A1, A2, A3),\r
+ mnesia:transaction(\r
+ fun() ->\r
+ case e(q([T || T <- mnesia:table(troll), T#troll.date_post =:= undefined]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
+ [] ->\r
+ plus_de_trolls;\r
+ Trolls ->\r
+ Troll = lists:nth(random:uniform(length(Trolls)), Trolls),\r
+ Troll2 = Troll#troll{date_post = now()},\r
+ mnesia:write(Troll2),\r
+ nouveau_message_sys("Troll de la semaine : ", Troll2#troll.id)\r
+ end\r
+ end\r
+ ).\r
+ \r
+\r
+% Renvoie l'id du message associé au troll dont l'id est donnée.\r
+% Renvoie undefined si il n'y en a pas.\r
+message_id_associe(Troll_id) ->\r
+ resultat_transaction(mnesia:transaction(\r
+ fun() ->\r
+ case e(q([M#minichat.id || M <- mnesia:table(minichat), M#minichat.troll_id =:= Troll_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of\r
+ [Id] -> Id;\r
+ _ -> undefined\r
+ end\r
+ end\r
+ )).\r
+\r
+ \r
+% Renvoie le résultat d'une transaction (en décomposant le tuple fournit)\r
+resultat_transaction({_, T}) ->\r
+ T.\r
+ \r
+\r
+% Retourne la difference entre deux timestamp (erlang:now()) en miliseconde\r
+delta_date_ms(D1, D2) ->\r
+ 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).\r
+\r
+% Retourne la différence entre deux timestamp (erlang:now) en minutes\r
+delta_date_minute(D1, D2) ->\r
+ trunc(delta_date_ms(D1, D2) / 1000 / 60).\r
+\r
+\r
+% Renvoie un nouvel id pour une table donnée\r
+nouvel_id(Table) ->\r
+ mnesia:dirty_update_counter(counter, Table, 1).\r
+
\ No newline at end of file
projects / euphorik.git / blobdiff