ADD Avancement sur la page d'administration (trolls)
[euphorik.git] / modules / erl / euphorik_bd.erl
index 9ad6da2..8875973 100755 (executable)
@@ -4,7 +4,7 @@
 % Ce module utilise une base mnesia.
 % @author G.Burri
 
--module(euphorik_minichat).\r
+-module(euphorik_bd).\r
 -export([
    % gestion :
    create/0,
@@ -17,6 +17,7 @@
    nouveau_user/3,
    set_profile/9,
    update_date_derniere_connexion/1,
+   update_ip/2,
    update_pseudo_user/2,
    users/0,
    user_by_cookie/1,   
    user_by_login/1,
    user_by_login_password/2,
    user_by_mess/1,
+   toggle_ek_master/1,
    
    % messages :
    nouveau_message/3,
+   nouveau_message_sys/1,
    messages/1,
    messages/2,\r
    messages/3,
    a_repondu_a_message/2,
    possede_message/2,
    
-   % utiles :
+   % ip :
+   ip_table/0,
+   ban/2,
+   deban/1,
+   est_banni/1,
    can_register/1,
+   
+   % trolls :
+   trolls/0,
+   trolls/1,
+   put_troll/2,
+   mod_troll/2,
+   del_troll/1,
+   
+   % versions :
+   update_version/1,
+   
+   % utiles :
    resultat_transaction/1\r
 ]).\r
 \r
@@ -88,6 +107,7 @@ create_tables() ->
    ]),
    mnesia:create_table(troll, [
       {attributes, record_info(fields, troll)},
+      {index, [date_post]},
       {disc_copies, [node()]}
    ]).
    
@@ -105,10 +125,12 @@ reset() ->
    mnesia:clear_table(counter),\r
    mnesia:clear_table(user),\r
    mnesia:clear_table(reponse_minichat),\r
-   mnesia:clear_table(minichat),\r
+   mnesia:clear_table(minichat),
+   mnesia:clear_table(troll),
+   mnesia:clear_table(ip_table),\r
    % crée l'utilisateur root\r
    mnesia:transaction(fun() ->\r
-      User = #user{id = 0, pseudo = "Sys", login = "Sys", date_creation = now(), date_derniere_connexion = now()},\r
+      User = #user{id = 0, pseudo = "Sys", login = "Sys", date_creation = now(), date_derniere_connexion = now(), ek_master = true},\r
       mnesia:write(User),\r
       User\r
    end).
@@ -151,7 +173,7 @@ set_profile(Cookie, Login, Password, Pseudo, Email, Css, Nick_format, Page_princ
                               % TODO : pourquoi ne pas tester avec la valeur "undefined" plutôt qu'avec "is_list" ?
                               % TODO : validation plus strict des données (pas de page négative dans les conv par exemple)
                               login = if is_list(Login) -> Login; true -> User#user.login end,
-                              password = if is_list(Password) -> Password; true -> User#user.password end,
+                              password = if is_list(Password) andalso Password =/= [] -> Password; true -> User#user.password end,
                               pseudo = if is_list(Pseudo) -> Pseudo; true -> User#user.pseudo end,
                               email = if is_list(Email) -> Email; true -> User#user.email end,
                               css = if is_list(Css) -> Css; true -> User#user.css end,
@@ -172,10 +194,10 @@ set_profile(Cookie, Login, Password, Pseudo, Email, Css, Nick_format, Page_princ
 
 
 % Met à jour la date de la dernière connexion d'un utilisateur à maintenant
-update_date_derniere_connexion(UserId) ->
+update_date_derniere_connexion(User_id) ->
    mnesia:transaction(
       fun() ->
-         case mnesia:wread({user, UserId}) of
+         case mnesia:wread({user, User_id}) of
             [User] ->
                mnesia:write(User#user{date_derniere_connexion = now()});
             _ ->
@@ -183,7 +205,21 @@ update_date_derniere_connexion(UserId) ->
           end
       end
    ).   
-  
+
+
+% Met à jour l'ip d'un user
+update_ip(User_id, IP) ->
+   mnesia:transaction(
+      fun() ->
+         case mnesia:wread({user, User_id}) of
+            [User] ->
+               mnesia:write(User#user{last_ip = IP});
+            _ ->
+               mnesia:abort("update_ip: User inconnu")
+          end
+      end
+   ).   
+   
   
 % Met à jour le pseudo du user
 update_pseudo_user(UserId, Pseudo) ->
@@ -244,6 +280,19 @@ user_by_login(Login) ->
    )).\r
    
    
+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() ->
@@ -267,12 +316,6 @@ user_by_mess(Id) ->
       end
    )).\r
    \r
-   \r
-% Renvoie l'utilisateur root\r
-root() ->\r
-   {ok, User} = user_by_id(0),\r
-   User.
-   
    
 % Ajoute un message. Repond_A est une liste d'id auquel le message répond
 % retourne soit l'id du message soit erreur.
@@ -288,7 +331,6 @@ nouveau_message(Mess, Auteur_id, Repond_A) ->
       if Nb_id_trouve =/= length(Repond_A) -> throw("Un ou plusieurs messages introuvable");
          true -> ok
       end,
-      Id = nouvel_id(minichat),
       % compare les dernière
       Delta = delta_date_ms(Auteur#user.date_derniere_connexion, now()),
       Nouvel_indice_flood = Auteur#user.indice_flood + if Delta =< ?DUREE_SPAM -> 2; true -> -1 end,
@@ -298,14 +340,13 @@ nouveau_message(Mess, Auteur_id, Repond_A) ->
       },
       % est-ce que l'auteur à trop floodé ?
       if Auteur#user.indice_flood =/= ?INDICE_SPAM_MAX, Auteur_maj#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->
-         Root = root(),
          mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}),
-         mnesia:write(#minichat{id=Id, auteur_id=Root#user.id, date=now(), pseudo=Root#user.pseudo, contenu=Auteur#user.pseudo ++ "(" ++ Auteur#user.login ++ ") est bloqué pour " ++ integer_to_list(trunc(?DUREE_BLOCAGE_SPAM / 1000)) ++ " secondes pour cause de flood.."}),
-         Id;
+         nouveau_message_sys(Auteur#user.pseudo ++ if Auteur#user.login =/= [] -> " (" ++ Auteur#user.login ++ ")"; true -> "" end ++ " est bloqué pour " ++ integer_to_list(trunc(?DUREE_BLOCAGE_SPAM / 1000)) ++ " secondes pour cause de flood.");
       Auteur#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->
          erreur;
       true ->     
          mnesia:write(Auteur_maj),
+         Id = nouvel_id(minichat),
          inserer_reponses(Id, Repond_A),
          mnesia:write(#minichat{id=Id, auteur_id=Auteur#user.id, date=now(), pseudo=Auteur#user.pseudo, contenu=Mess}),
          Id
@@ -320,6 +361,18 @@ inserer_reponses(Id_repondant, [Id_mess | Reste]) ->
 inserer_reponses(_, []) ->
    ok.
    
+% Permet de créer un message système.
+% Renvoie l'id du message système
+nouveau_message_sys(Mess) ->
+   {ok, Root} = user_by_id(0),
+   mnesia:transaction(
+      fun() ->
+         Id = nouvel_id(minichat),
+         mnesia:write(#minichat{id=Id, auteur_id=0, date=now(), pseudo=Root#user.pseudo, contenu=Mess})
+      end
+   ).
+   
    
 % Renvoie N messages se trouvant sur la première page
 messages(N) ->  
@@ -364,7 +417,6 @@ messages_by_ids(Ids) ->
       fun() ->
          % TODO : optimisations ? serait-ce du O(n) ?
          Query = qlc:q([E || E <- qlc:keysort(2, mnesia:table(minichat), [{order, ascending}]), lists:any(fun(Id) -> Id =:= E#minichat.id end, Ids)]),
-         %io:format("~s~n", [qlc:info(Query)]),
          qlc:e(Query)
       end
    )).
@@ -440,10 +492,73 @@ possede_message(Id_user, Id_mess) ->
       _ -> 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.\r
+% 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() ->
@@ -467,6 +582,130 @@ can_register(IP) ->
       end
    )).
    
+
+% Renvoie tous les trolls.
+trolls() ->
+   resultat_transaction(mnesia:transaction(
+      fun() ->
+         qlc:e(qlc:q([T || T <- qlc:keysort(2, mnesia:table(troll))]))
+      end
+   )).
+   
+% 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(Last_id) ->
+   case mnesia:subscribe({table, troll, detailed}) of
+      {error, E} = E ->
+         E;
+      _ ->
+         case resultat_transaction(mnesia:transaction(
+               fun() ->
+                  qlc:e(qlc: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
+   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
+      {mnesia_table_event, {write, troll, Troll, [_Old_troll | _], _}} ->
+         {mod, Troll};
+      {mnesia_table_event, {write, troll, Troll, [], _}} ->
+         {add, [Troll]};
+      {mnesia_table_event, {delete, troll, {troll, Id}, _, _}} ->
+         {del, Id};
+      _ ->
+         attend_evenement_troll()
+   after 1000 * 60 * 60 -> 
+      timeout
+   end.
+   
+ % Renvoie l'id du nouveau troll
+ % ou max_troll_reached_per_user si le nombre de troll posté par l'utilisateur max a été atteind
+ % ou max_troll_reached si le nombre de troll posté max a été atteind
+ % ou user_unknown
+put_troll(User_id, Content) ->
+   resultat_transaction(mnesia:transaction(
+      fun() ->
+         % control le nombre de troll déjà posté
+         Nb_troll_poste_par_user = length(qlc:e(qlc:q(
+            [
+               E#troll.id || E <- mnesia:table(troll),
+               E#troll.id_user =:= User_id,
+               E#troll.date_post =:= undefined
+            ]
+         ))),
+         Nb_troll_poste_total = length(qlc:e(qlc: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
+   ).
+   
+update_version(1) ->
+   mnesia:transform_table(
+      ip_table,
+      fun() -> null end,
+      record_info(fields, ip_table),
+      ip_table
+   ).
+   
    
 % Renvoie le résultat d'une transaction (en décomposant le tuple fournit)
 resultat_transaction({_, T}) ->
@@ -479,6 +718,7 @@ delta_date_ms(D1, D2) ->
 
 
 % Bizarre, cette fonction n'existe pas dans la stdlib.
+% Pas utilisé mais bon ca me fait de la peine de l'enlever.
 ceiling(X) ->
    T = trunc(X),
    case (X - T) of