MOD amélioration de la complexité de la fonction euphorik_bd:nouveau_message/3 mais...
[euphorik.git] / modules / erl / euphorik_bd.erl
index 196a26e..81a36a0 100755 (executable)
 \r
 -module(euphorik_bd).\r
 -export([   \r
+   % texte :\r
+   get_texte/1,\r
+   get_texte/2,\r
+   \r
    % users :\r
    nouveau_user/2,\r
    nouveau_user/4,\r
@@ -39,7 +43,7 @@
    css_from_user_cookie/1,\r
    is_ek_master_from_cookie/1,\r
    \r
-   % messages :e\r
+   % messages :\r
    nouveau_message/3,\r
    nouveau_message_sys/1,\r
    nouveau_message_sys/2,\r
 -include_lib("stdlib/include/qlc.hrl").\r
 \r
 \r
+get_texte(Id) ->\r
+   get_texte(Id, fr).\r
+\r
+\r
+% TODO : généraliser la langue\r
+get_texte(Id, _Lang = fr) ->\r
+   resultat_transaction(mnesia:transaction(fun() ->\r
+      case mnesia:read({texte, Id}) of\r
+         [#texte{fr = Texte}] -> Texte;\r
+         _ -> "Message " ++ integer_to_list(Id) ++ " unknown"\r
+      end\r
+   end)).\r
+\r
+\r
 % Ajoute un nouveau user et le renvoie\r
 nouveau_user(Cookie, Profile) ->\r
    F = fun() ->\r
@@ -179,7 +197,7 @@ update_pseudo_user(UserId, Pseudo) ->
 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
+         case mnesia:index_read(user, Cookie, #user.cookie) of\r
             [User] -> {ok, User};\r
             _ -> erreur\r
          end\r
@@ -190,7 +208,11 @@ user_by_cookie(Cookie) ->
 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
+%         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
@@ -254,14 +276,17 @@ user_by_mess(Id) ->
 % 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
+   F = fun() ->         \r
+      Repond_a = lists:map(\r
+         fun(Repond_a_id) ->\r
+            [M] = mnesia:wread({minichat, Repond_a_id}),\r
+            M\r
+         end,\r
+         Repond_A_ids\r
+      ),      \r
       Racine_id = case Repond_a of\r
          [] -> undefined;\r
-         [M | _ ] -> \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
@@ -270,42 +295,43 @@ nouveau_message(Mess, Auteur_id, Repond_A_ids) ->
                _ ->\r
                   {erreur, "Les messages ne font pas partie de la même conversation"}\r
             end\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
+            case mnesia:wread({user, Auteur_id}) 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
+                        Now = now(),\r
+                        Delta = euphorik_common: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
+                           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
+                              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
@@ -378,15 +404,14 @@ message_by_id(Id) ->
    resultat_transaction(mnesia:transaction(\r
       fun() ->\r
          case mnesia:read({minichat, Id}) of\r
-            [] -> erreur;\r
-            [M] ->\r
-               {ok, M#minichat{contenu = contenu_message(M)}}\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, à utiliser à l'intérieur d'une transaction.\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
@@ -452,64 +477,81 @@ enfants(M_id) ->
    )).\r
    \r
    \r
-% Renvoie les parents d'un message M (les messages auquels répond M)\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) ->\r
+parents_id(M_id) ->\r
    resultat_transaction(mnesia:transaction(fun() ->\r
-      e(\r
-         qlc:sort(\r
-            q([E#reponse_minichat.cible || E <- mnesia:table(reponse_minichat), E#reponse_minichat.repondant =:= M]),\r
-            [{order, ascending}]\r
-         ),\r
-         [{tmpdir, ?KEY_SORT_TEMP_DIR}]\r
-      )\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
+enfants_id(M_id) ->\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
+      case mnesia:index_read(reponse_minichat, M_id, #reponse_minichat.cible) of\r
+         Enfants when is_list(Enfants) ->\r
+            lists:sort(lists:map(\r
+               fun(#reponse_minichat{repondant = Repondant}) -> Repondant end,\r
+               Enfants\r
+            ));\r
+         _ -> []            \r
+      end\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
-   case mnesia:transaction(\r
-            fun() ->\r
-               e(q([\r
-                  M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat),\r
-                  M#minichat.auteur_id =:= Id_user, M#minichat.id =:= R#reponse_minichat.cible, R#reponse_minichat.repondant =:= Id_mess\r
-               ]), [{unique_all, true}, {tmpdir, ?KEY_SORT_TEMP_DIR}])\r
-            end\r
-         ) of\r
-      {atomic, [_]} -> true;\r
-      _ -> false\r
-   end.\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
-   case mnesia:transaction(\r
-            fun() ->\r
-               e(q([\r
-                  M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat),\r
-                  R#reponse_minichat.cible =:= Id_mess, R#reponse_minichat.repondant =:= M#minichat.id, M#minichat.auteur_id =:= Id_user\r
-               ]), [{unique_all, true}, {tmpdir, ?KEY_SORT_TEMP_DIR}])\r
-            end\r
-         ) of\r
-      {atomic, [_]} -> true;\r
-      _ -> false\r
-   end.\r
+   resultat_transaction(mnesia:transaction(\r
+      fun() ->\r
+         case mnesia:index_read(reponse_minichat, Id_mess, #reponse_minichat.cible) 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
@@ -534,7 +576,7 @@ list_ban() ->
          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
+               euphorik_common: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
@@ -581,7 +623,7 @@ est_banni(User_id) ->
                if Echeance < Now -> % l'échéance est passée\r
                      false;\r
                   true ->\r
-                     {true, delta_date_minute(Echeance, Now)}\r
+                     {true, euphorik_common:delta_date_minute(Echeance, Now)}\r
                end;\r
             _ ->\r
                false\r
@@ -610,11 +652,11 @@ can_register(IP) ->
                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
+               Delta = euphorik_common: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
+                     mnesia:write(T#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
@@ -778,15 +820,6 @@ message_id_associe(Troll_id) ->
 % 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