X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=modules%2Ferl%2Feuphorik_bd.erl;h=7fac81d4a89df771598dca1f5aa9f0e8ba3d8a4c;hb=20c5327d1229313b2a6bda46a74459f70614306d;hp=87dab2f44520c8b064d4d26eaa678aa1c8e4bbd5;hpb=01922222ac686c2507052c0cc1d755495145d154;p=euphorik.git

diff --git a/modules/erl/euphorik_bd.erl b/modules/erl/euphorik_bd.erl
index 87dab2f..7fac81d 100755
--- a/modules/erl/euphorik_bd.erl
+++ b/modules/erl/euphorik_bd.erl
@@ -19,7 +19,8 @@
    update_date_derniere_connexion/1,
    update_ip/2,
    update_pseudo_user/2,
-   users/0,
+   print_users/0,
+   print_users/1,
    user_by_cookie/1,   
    user_by_id/1,   
    user_by_login/1,
@@ -49,12 +50,25 @@
    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").
@@ -100,6 +114,7 @@ create_tables() ->
    ]),
    mnesia:create_table(troll, [
       {attributes, record_info(fields, troll)},
+      {index, [date_post]},
       {disc_copies, [node()]}
    ]).
    
@@ -118,6 +133,7 @@ reset() ->
    mnesia:clear_table(user),
    mnesia:clear_table(reponse_minichat),
    mnesia:clear_table(minichat),
+   mnesia:clear_table(troll),
    mnesia:clear_table(ip_table),
    % crée l'utilisateur root
    mnesia:transaction(fun() ->
@@ -157,9 +173,9 @@ 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)
@@ -225,12 +241,36 @@ update_pseudo_user(UserId, Pseudo) ->
       end
    ).
    
-
-% Renvoie tous les users.
-users() ->
+   
+% Affiche N user trié par leur date de dernière connexion.
+% Attention : pas d'index sur ce champs (la date)   
+print_users(N) ->
    resultat_transaction(mnesia:transaction(fun() ->
-      qlc:e(qlc:q([E || E <- mnesia:table(user)]))
-   end)).   
+      C = cursor(q([E || E <- qlc:keysort(9, mnesia:table(user), [{order, descending}])])),
+      Users = qlc:next_answers(C, N),
+      lists:foreach(
+         fun(U) ->
+            #user{id = Id, pseudo = Pseudo, login = Login, ek_master = Ek_master, date_derniere_connexion = Date} = U,
+            {{Annee, Mois, Jour}, {Heure, Min, _}} = calendar:now_to_local_time(Date),
+            io:format(
+               "~4w : ~10.10..s~10.10..s ~2w.~2.2.0w.~w - ~2wh~2.2.0w~n",
+               [  
+                  Id,
+                  if Ek_master -> "*"; true -> "" end ++ Pseudo,
+                  "(" ++ Login ++ ")",
+                  Jour, Mois, Annee, Heure, Min
+               ]
+            )
+         end,
+         Users
+      ),
+      qlc:delete_cursor(C)
+   end)).
+   
+
+% Affiche tous les users.
+print_users() ->
+   print_users(all_remaining).
    
 
 % Est-ce qu'un utilisateur existe en fonction de son cookie ?
@@ -238,8 +278,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
@@ -250,8 +289,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
@@ -262,7 +300,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
@@ -274,7 +312,7 @@ user_by_login(Login) ->
 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])),
+         Users = e(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});
@@ -287,9 +325,8 @@ toggle_ek_master(User_id) ->
 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};
+         case e(q([E || E <- mnesia:table(user), E#user.login =:= Login, E#user.password =:= Password])) of
+            [User | _] -> {ok, User};
             _ -> erreur
          end
       end
@@ -300,8 +337,8 @@ 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
@@ -309,38 +346,39 @@ user_by_mess(Id) ->
    
    
 % Ajoute un message. Repond_A est une liste d'id auquel le message répond
-% retourne soit l'id du message soit erreur.
+% retourne soit l'id du message soit {erreur, <raison>}.
 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")
+         _ -> {erreur, "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
+      if Nb_id_trouve =/= length(Repond_A) ->
+            {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(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, "Bloqué pour cause de flood"};
+            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
    end,
    resultat_transaction(mnesia:transaction(F)).
@@ -373,7 +411,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,
@@ -394,7 +432,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};
@@ -407,8 +445,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)]),
-         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
    )).
    
@@ -417,14 +455,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)).
    
@@ -433,7 +471,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,
@@ -446,7 +484,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}])
@@ -461,7 +499,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}])
@@ -476,7 +514,7 @@ 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;
@@ -487,7 +525,7 @@ possede_message(Id_user, Id_mess) ->
 ip_table() ->
    resultat_transaction(mnesia:transaction(
       fun() ->
-         qlc:e(qlc:q([IP || IP <- mnesia:table(ip_table)]))
+         e(q([IP || IP <- mnesia:table(ip_table)]))
       end
    )).
 
@@ -516,7 +554,7 @@ deban(IP) ->
 est_banni(User_id) ->
    resultat_transaction(mnesia:transaction(
       fun() ->
-         case qlc:e(qlc:q([
+         case e(q([
             {IP#ip_table.ban, IP#ip_table.ban_duration} ||
             U <- mnesia:table(user),
             U#user.id =:= User_id,
@@ -553,7 +591,7 @@ date_plus_minutes(Ban, Ban_duration) ->
 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;
@@ -573,6 +611,188 @@ 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};
+      {tcp_closed, _} ->
+         exit(normal);
+      _ ->
+         attend_evenement_troll()
+   % 60 minutes de timeout (on ne sais jamais)
+   % Après 60 minutes de connexion, le client doit donc reétablir une connexion
+   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(