modules/erl/euphorik_bd.erl

   1 % coding: utf-8
   2 % Copyright 2008 Grégory Burri
   3 %
   4 % This file is part of Euphorik.
   5 %
   6 % Euphorik is free software: you can redistribute it and/or modify
   7 % it under the terms of the GNU General Public License as published by
   8 % the Free Software Foundation, either version 3 of the License, or
   9 % (at your option) any later version.
  10 %
  11 % Euphorik is distributed in the hope that it will be useful,
  12 % but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 % MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 % GNU General Public License for more details.
  15 %
  16 % You should have received a copy of the GNU General Public License
  17 % along with Euphorik.  If not, see <http://www.gnu.org/licenses/>.
  18 %
  19 % Ce module permet de gérer les données persistantes lié au site d'euphorik.ch.
  20 % Il permet d'ajouter des message, de demande les messages sur une page donnée, etc..
  21 % Ce module utilise une base mnesia.
  22 % @author G.Burri
  23
  24
  25 -module(euphorik_bd).
  26 -export([
  27    % users :
  28    nouveau_user/3,
  29    nouveau_user/4,
  30    set_profile/1,
  31    update_date_derniere_connexion/1,
  32    update_ip/2,
  33    update_pseudo_user/2,
  34    print_users/0,
  35    print_users/1,
  36    print_user/1,
  37    user_by_cookie/1,
  38    user_by_id/1,
  39    user_by_login/1,
  40    user_by_login_password/2,
  41    user_by_mess/1,
  42    toggle_ek_master/1,
  43    css_from_user_cookie/1,
  44    is_ek_master_from_cookie/1,
  45
  46    % messages :e
  47    nouveau_message/3,
  48    nouveau_message_sys/1,
  49    nouveau_message_sys/2,
  50    messages/1,
  51    messages/2,
  52    messages/3,
  53    message_by_id/1,
  54    messages_by_ids/1,
  55    message_existe/1,
  56    reponses/0,
  57    parents/1,
  58    enfants/1,
  59    parents_id/1,
  60    enfants_id/1,
  61    est_une_reponse_a_user/2,
  62    a_repondu_a_message/2,
  63    possede_message/2,
  64
  65    % ip :
  66    list_ban/0,
  67    ban/2,
  68    deban/1,
  69    est_banni/1,
  70    can_register/1,
  71
  72    % trolls :
  73    trolls/0,
  74    trolls/1,
  75    put_troll/2,
  76    mod_troll/2,
  77    del_troll/1,
  78    troll_by_id/1,
  79    current_troll/0,
  80    elire_troll/0,
  81    message_id_associe/1,
  82
  83    % utiles :
  84    resultat_transaction/1
  85 ]).
  86 -import(qlc, [e/2, q/1, cursor/2]).
  87 -include("../include/euphorik_bd.hrl").
  88 -include("../include/euphorik_defines.hrl").
  89 -include_lib("stdlib/include/qlc.hrl").
  90
  91
  92 % Ajoute un nouveau user et le renvoie
  93 % User_infos représente le profile de l'utilisateur, il est possibe qu'il soit vide
  94 nouveau_user(Pseudo, Cookie, User_infos) ->
  95    F = fun() ->
  96       Id = nouvel_id(user),
  97       User = User_infos#user{id = Id, cookie = Cookie, pseudo = Pseudo, date_creation = now(), date_derniere_connexion = now()},
  98       mnesia:write(User),
  99       User
 100    end,
 101   resultat_transaction(mnesia:transaction(F)).
 102
 103
 104 % Ajoute un nouveau user et le renvoie
 105 % User_infos représente le profile de l'utilisateur, il est possibe qu'il soit vide
 106 nouveau_user(Login, Password, Cookie, User_infos) ->
 107    F = fun() ->
 108       Id = nouvel_id(user),
 109       User = User_infos#user{id = Id, cookie = Cookie, pseudo = Login, login = Login, password = Password, date_creation = now(), date_derniere_connexion = now()},
 110       mnesia:write(User),
 111       User
 112    end,
 113   resultat_transaction(mnesia:transaction(F)).
 114
 115
 116 % Définit les données d'une utilisateur.
 117 set_profile(User) ->
 118    resultat_transaction(mnesia:transaction(
 119       fun() ->
 120          case user_by_cookie(User#user.cookie) of
 121             {ok, User_existant} ->
 122                case user_by_login(User#user.login) of
 123                   {ok, U} when User#user.login =/= [], U#user.id =/= User_existant#user.id ->
 124                      login_deja_pris;
 125                   _ ->
 126                      mnesia:write(
 127                         User#user{
 128                            id = User_existant#user.id,
 129                            login = if User_existant#user.login =:= [] -> User#user.login; true -> User_existant#user.login end, % on ne peut pas changer de login sauf si on en a pas !
 130                            password = if User#user.password =:= [] -> User_existant#user.password; true -> User#user.password end
 131                         }
 132                      ),
 133                      ok
 134                end;
 135             _ -> erreur
 136          end
 137       end
 138    )).
 139
 140
 141 % Met à jour la date de la dernière connexion d'un utilisateur à maintenant
 142 update_date_derniere_connexion(User_id) ->
 143    mnesia:transaction(
 144       fun() ->
 145          case mnesia:wread({user, User_id}) of
 146             [User] ->
 147                mnesia:write(User#user{date_derniere_connexion = now()});
 148             _ ->
 149                mnesia:abort("update_date_derniere_connexion: User inconnu")
 150           end
 151       end
 152    ).
 153
 154
 155 % Met à jour l'ip d'un user
 156 update_ip(User_id, IP) ->
 157    mnesia:transaction(
 158       fun() ->
 159          case mnesia:wread({user, User_id}) of
 160             [User] ->
 161                mnesia:write(User#user{last_ip = IP});
 162             _ ->
 163                mnesia:abort("update_ip: User inconnu")
 164           end
 165       end
 166    ).
 167
 168
 169 % Met à jour le pseudo du user
 170 update_pseudo_user(UserId, Pseudo) ->
 171    mnesia:transaction(
 172       fun() ->
 173          case mnesia:wread({user, UserId}) of
 174             [User] when User#user.pseudo =/= Pseudo ->
 175                mnesia:write(User#user{pseudo = Pseudo});
 176             _ ->
 177                mnesia:abort("update_pseudo_user: User inconnu ou pseudo deja à jour")
 178           end
 179       end
 180    ).
 181
 182
 183 % Affiche N user trié par leur date de dernière connexion.
 184 % Opt est une liste d'option d'affichage :
 185 %  * ekmaster : n'affiche que les admins
 186 print_users(N, Opt) ->
 187    AfficheQueLesEkMaster = lists:any(fun(O) -> O =:= ekmaster end, Opt),
 188    resultat_transaction(mnesia:transaction(fun() ->
 189       C = cursor(
 190          qlc:keysort(
 191             #user.date_derniere_connexion,
 192             if AfficheQueLesEkMaster ->
 193                q([E || E <- mnesia:table(user), E#user.ek_master =:= true]);
 194             true ->
 195                q([E || E <- mnesia:table(user)])
 196             end,
 197             [{order, descending}]
 198          ),
 199          [{tmpdir, ?KEY_SORT_TEMP_DIR}]
 200       ),
 201       Users = qlc:next_answers(C, N),
 202       lists:foreach(
 203          fun(U) ->
 204             print_user(U)
 205          end,
 206          Users
 207       ),
 208       qlc:delete_cursor(C)
 209    end)).
 210
 211
 212 % Affiche tous les users.
 213 print_users(Opt) ->
 214    print_users(all_remaining, Opt).
 215
 216 % Affiche tous les users.
 217 print_users() ->
 218    print_users(all_remaining, []).
 219
 220 print_user(User) when is_record(User, user) ->
 221    #user{id = Id, pseudo = Pseudo, login = Login, ek_master = Ek_master, date_derniere_connexion = Date, last_ip = IP} = User,
 222    {{Annee, Mois, Jour}, {Heure, Min, _}} = calendar:now_to_local_time(Date),
 223    io:format(
 224       % id        pseudo     (login)        IP  Jour  Mois   Année  Heure Minute
 225       "~4w : ~10.10..s(~10.10..s) ~s ~2w.~2.2.0w.~w - ~2wh~2.2.0w~n",
 226       [
 227          Id,
 228          if Ek_master -> "*"; true -> "" end ++ Pseudo,
 229          Login,
 230          euphorik_common:serialize_ip(IP),
 231          Jour, Mois, Annee, Heure, Min
 232       ]
 233    );
 234 % Affichage d'un user en fonction de son login
 235 print_user(Login) when is_list(Login) ->
 236    case user_by_login(Login) of
 237       {ok, User} ->
 238          print_user(User);
 239       _ ->
 240          {erreur, "Login pas trouvé : " ++ Login}
 241    end;
 242 % Affichage d'un user en fonction de son id
 243 print_user(Id) when is_integer(Id) ->
 244    case user_by_id(Id) of
 245       {ok, User} ->
 246          print_user(User);
 247       _ ->
 248          {erreur, "Id pas trouvé : " ++ integer_to_list(Id)}
 249    end.
 250
 251
 252 % Est-ce qu'un utilisateur existe en fonction de son cookie ?
 253 % Renvoie {ok, User} ou erreur
 254 user_by_cookie(Cookie) ->
 255    resultat_transaction(mnesia:transaction(
 256       fun() ->
 257          case e(q([E || E <- mnesia:table(user), E#user.cookie =:= Cookie]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 258             [User] -> {ok, User};
 259             _ -> erreur
 260          end
 261       end
 262    )).
 263
 264
 265 user_by_id(ID) ->
 266    resultat_transaction(mnesia:transaction(
 267       fun() ->
 268          case e(q([E || E <- mnesia:table(user), E#user.id =:= ID]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 269             [User] -> {ok, User};
 270             _ -> erreur
 271          end
 272       end
 273    )).
 274
 275
 276 user_by_login(Login) ->
 277    resultat_transaction(mnesia:transaction(
 278       fun() ->
 279          Users = e(q([E || E <- mnesia:table(user), E#user.login =:= Login]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]),
 280          case Users of
 281             [User] -> {ok, User};
 282             _ -> erreur
 283          end
 284       end
 285    )).
 286
 287
 288 toggle_ek_master(User_id) ->
 289    resultat_transaction(mnesia:transaction(
 290       fun() ->
 291          Users = e(q([E || E <- mnesia:table(user), E#user.id =:= User_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]),
 292          case Users of
 293             [User] ->
 294                mnesia:write(User#user{ek_master = not User#user.ek_master});
 295             _ -> erreur
 296          end
 297       end
 298    )).
 299
 300
 301 % Renvoie une chaine représentant le cookie ou undefined si pas trouvé.
 302 css_from_user_cookie(Cookie) ->
 303    case user_by_cookie(Cookie) of
 304       {ok, User} ->
 305          User#user.css;
 306       _ ->
 307          undefined
 308    end.
 309
 310
 311 is_ek_master_from_cookie(Cookie) ->
 312    case user_by_cookie(Cookie) of
 313       {ok, #user{ek_master = true}} -> true;
 314       _ -> false
 315    end.
 316
 317
 318 user_by_login_password(Login, Password) ->
 319    resultat_transaction(mnesia:transaction(
 320       fun() ->
 321          case e(q([E || E <- mnesia:table(user), E#user.login =:= Login, E#user.password =:= Password]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 322             [User | _] -> {ok, User};
 323             _ -> erreur
 324          end
 325       end
 326    )).
 327
 328
 329 % Renvoie {ok, User} où User est un #user possédant le message donné.
 330 user_by_mess(Id) ->
 331    resultat_transaction(mnesia:transaction(
 332       fun() ->
 333          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
 334             [User | _] -> {ok, User};
 335             _ -> erreur
 336          end
 337       end
 338    )).
 339
 340
 341 % Ajoute un message. Repond_A est une liste d'id auquel le message répond
 342 % retourne soit l'id du message soit {erreur, <raison>}.
 343 nouveau_message(Mess, Auteur_id, Repond_A_ids) ->
 344    % regarde si les id 'Repond_A' existent
 345    F = fun() ->
 346       Repond_a = e(
 347          q([M || M <- mnesia:table(minichat), lists:member(M#minichat.id, Repond_A_ids)]),
 348          [{tmpdir, ?KEY_SORT_TEMP_DIR}]
 349       ),
 350       Racine_id = case Repond_a of
 351          [] -> undefined;
 352          [M | _ ] ->
 353             Une_racine = M#minichat.racine_id,
 354             % vérification que tout les messages de Repond_a possède la même racine (même conversation)
 355             case lists:all(fun(R) -> R#minichat.racine_id =:= Une_racine end, Repond_a) of
 356                true ->
 357                   Une_racine;
 358                _ ->
 359                   {erreur, "Les messages ne font pas partie de la même conversation"}
 360             end
 361       end,
 362       case Racine_id of
 363          {erreur, E} -> {erreur, E};
 364          _ ->
 365             % est-ce que l'auteur existe ?
 366             case e(q([E || E <- mnesia:table(user), E#user.id =:= Auteur_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 367                [Auteur] ->
 368                   if length(Repond_a) =/= length(Repond_A_ids) ->
 369                         {erreur, "Un ou plusieurs messages introuvable"};
 370                      true ->
 371                         % comparaison entre la date du dernier poste et maintenant (gestion du flood)
 372                         Delta = delta_date_ms(Auteur#user.date_derniere_connexion, now()),
 373                         Nouvel_indice_flood = Auteur#user.indice_flood + if Delta =< ?DUREE_SPAM -> 2; true -> -1 end,
 374                         Auteur_maj = Auteur#user{
 375                            indice_flood = if Nouvel_indice_flood > ?INDICE_SPAM_MAX -> ?INDICE_SPAM_MAX; Nouvel_indice_flood < 0 -> 0; true -> Nouvel_indice_flood end,
 376                            date_derniere_connexion = now()
 377                         },
 378                         % est-ce que l'auteur à trop floodé ?
 379                         if Auteur#user.indice_flood =/= ?INDICE_SPAM_MAX, Auteur_maj#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->
 380                            mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}),
 381                            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.");
 382                         Auteur#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->
 383                            {erreur, "Bloqué pour cause de flood"};
 384                         true ->
 385                            mnesia:write(Auteur_maj),
 386                            Id = nouvel_id(minichat),
 387                            inserer_reponses(Id, Repond_A_ids),
 388                            mnesia:write(#minichat{
 389                               id = Id,
 390                               auteur_id = Auteur#user.id,
 391                               date = now(),
 392                               pseudo = Auteur#user.pseudo,
 393                               contenu = Mess,
 394                               racine_id = if Racine_id =:= undefined -> Id; true -> Racine_id end
 395                            }),
 396                            Id
 397                         end
 398                   end;
 399                _ ->
 400                   {erreur, "L'auteur du message est introuvable"}
 401             end
 402       end
 403    end,
 404    resultat_transaction(mnesia:transaction(F)).
 405
 406 % Définit Id_repondant comme étant la réponse à Ids. Ids est une liste d'id.
 407 inserer_reponses(Id_repondant, [Id_mess | Reste]) ->
 408    mnesia:write(#reponse_minichat{repondant = Id_repondant, cible = Id_mess}),
 409    inserer_reponses(Id_repondant, Reste);
 410 inserer_reponses(_, []) ->
 411    ok.
 412
 413
 414 % Permet de créer un message système.
 415 % Renvoie l'id du message système
 416 nouveau_message_sys(Mess) ->
 417    nouveau_message_sys(Mess, undefined).
 418
 419
 420 % Création d'un message système lié à un troll.
 421 nouveau_message_sys(Mess, Troll_id) ->
 422    {ok, Root} = user_by_id(0),
 423    resultat_transaction(mnesia:transaction(
 424       fun() ->
 425          Id = nouvel_id(minichat),
 426          mnesia:write(#minichat{id=Id, auteur_id=0, date=now(), pseudo=Root#user.pseudo, contenu=Mess, troll_id=Troll_id, racine_id=Id}),
 427          Id
 428       end
 429    )).
 430
 431
 432 % Renvoie N messages se trouvant sur la première page
 433 messages(N) ->
 434    messages(N, 1).
 435
 436
 437 % Renvoie N messages se trouvant sur la page P
 438 messages(N, P) ->
 439    F = fun() ->
 440       C = cursor(
 441          qlc:keysort(
 442             #minichat.id,
 443             q([E#minichat{contenu = contenu_message(E)} || E <- mnesia:table(minichat)]),
 444             [{order, descending}]
 445          ),
 446          [{tmpdir, ?KEY_SORT_TEMP_DIR}]
 447       ),
 448       if P > 1 -> qlc:next_answers(C, N * (P - 1));
 449          true -> ok
 450       end,
 451       R = qlc:next_answers(C, N),
 452       qlc:delete_cursor(C),
 453       R
 454    end,
 455    lists:reverse(resultat_transaction(mnesia:transaction(F))).
 456
 457
 458 % Renvoie les messages manquants pour la page P en sachant qu'il y a N message
 459 % par page et que le dernier message que l'on possède est Id
 460 messages(Id, N, P) ->
 461    lists:filter(fun (M) -> M#minichat.id > Id end, messages(N, P)).
 462
 463
 464 % Renvoie {ok, #minichat} (voir #minichat de euphorik_bd.hrl) à partir de son id.
 465 message_by_id(Id) ->
 466    resultat_transaction(mnesia:transaction(
 467       fun() ->
 468          case mnesia:read({minichat, Id}) of
 469             [] -> erreur;
 470             [M] ->
 471                {ok, M#minichat{contenu = contenu_message(M)}}
 472          end
 473       end
 474    )).
 475
 476
 477 % Renvoie le contenu d'un message donnée, à utiliser à l'intérieur d'une transaction.
 478 % TODO : Cette fonction pourrait être remplacé par un "outer-join", est-ce possible avec qlc ?
 479 contenu_message(E) ->
 480    case mnesia:read({troll, E#minichat.troll_id}) of
 481       [] -> E#minichat.contenu;
 482       [T] -> E#minichat.contenu ++ T#troll.content
 483    end.
 484
 485
 486 % Renvoie une liste de message (voir #minichat de euphorik_bd.hrl) à partir d'une liste d'id (Ids).
 487 % TODO : optimisations ? serait-ce du O(n) ?
 488 % Bon de toutes façons on s'en fout c'est pas utilisé :)
 489 messages_by_ids(Ids) ->
 490    resultat_transaction(mnesia:transaction(
 491       fun() ->
 492          e(qlc:keysort(
 493             #minichat.id,
 494             q([E || E <- mnesia:table(minichat), lists:any(fun(Id) -> Id =:= E#minichat.id end, Ids)]),
 495             [{order, ascending}]
 496          ),[{tmpdir, ?KEY_SORT_TEMP_DIR}])
 497       end
 498    )).
 499
 500
 501 % Est-ce qu'un message existe ? Renvoie un boolean.
 502 % TODO : ya pas plus simple ?
 503 message_existe(Id) ->
 504    resultat_transaction(mnesia:transaction(fun() ->
 505       length(e(q([E#minichat.id || E <- mnesia:table(minichat), E#minichat.id =:= Id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])) =:= 1
 506    end)).
 507
 508
 509 % Renvoie les reponses (utilisé normalement uniquement pendant le debug).
 510 reponses() ->
 511    F = fun() ->
 512       e(q([E || E <- mnesia:table(reponse_minichat)]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
 513    end,
 514    resultat_transaction(mnesia:transaction(F)).
 515
 516
 517 % Renvoie les messages auquel M_id répond.
 518 parents(M_id) ->
 519    resultat_transaction(mnesia:transaction(
 520       fun() ->
 521          e(q(
 522             [M || R <- mnesia:table(reponse_minichat),
 523             M <- mnesia:table(minichat),
 524             R#reponse_minichat.repondant =:= M_id,
 525             M#minichat.id =:= R#reponse_minichat.cible]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
 526       end
 527    )).
 528
 529
 530 % Renvoie les message qui repondent à M_id
 531 enfants(M_id) ->
 532    resultat_transaction(mnesia:transaction(
 533       fun() ->
 534          e(q(
 535             [M || R <- mnesia:table(reponse_minichat),
 536             M <- mnesia:table(minichat),
 537             R#reponse_minichat.cible =:= M_id,
 538             M#minichat.id =:= R#reponse_minichat.repondant]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
 539       end
 540    )).
 541
 542
 543 % Renvoie les parents d'un message M (les messages auquels répond M)
 544 % ordrés du plus petit au plus grand..
 545 % @spec parents_id(integer()) -> [integer()]
 546 parents_id(M) ->
 547    resultat_transaction(mnesia:transaction(fun() ->
 548       e(
 549          qlc:sort(
 550             q([E#reponse_minichat.cible || E <- mnesia:table(reponse_minichat), E#reponse_minichat.repondant =:= M]),
 551             [{order, ascending}]
 552          ),
 553          [{tmpdir, ?KEY_SORT_TEMP_DIR}]
 554       )
 555    end)).
 556
 557
 558 % Renvoie les id des enfants d'un message M (les messages qui répondent à M)
 559 % ordrés du plus petit au plus grand.
 560 % @spec enfants_id(integer()) -> [integer()]
 561 enfants_id(M) ->
 562    resultat_transaction(mnesia:transaction(fun() ->
 563       e(
 564          qlc:sort(
 565             q([E#reponse_minichat.repondant || E <- mnesia:table(reponse_minichat), E#reponse_minichat.cible =:= M]),
 566             [{order, ascending}]
 567          ),
 568          [{tmpdir, ?KEY_SORT_TEMP_DIR}]
 569       )
 570    end)).
 571
 572
 573 % Est-ce que le message Id_mess est une réponse d'une message de Id_user ?
 574 est_une_reponse_a_user(Id_user, Id_mess) ->
 575    case mnesia:transaction(
 576             fun() ->
 577                e(q([
 578                   M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat),
 579                   M#minichat.auteur_id =:= Id_user, M#minichat.id =:= R#reponse_minichat.cible, R#reponse_minichat.repondant =:= Id_mess
 580                ]), [{unique_all, true}, {tmpdir, ?KEY_SORT_TEMP_DIR}])
 581             end
 582          ) of
 583       {atomic, [_]} -> true;
 584       _ -> false
 585    end.
 586
 587
 588 % Est-ce que Id_user à répondu au message Id_mess
 589 a_repondu_a_message(Id_user, Id_mess) ->
 590    case mnesia:transaction(
 591             fun() ->
 592                e(q([
 593                   M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat),
 594                   R#reponse_minichat.cible =:= Id_mess, R#reponse_minichat.repondant =:= M#minichat.id, M#minichat.auteur_id =:= Id_user
 595                ]), [{unique_all, true}, {tmpdir, ?KEY_SORT_TEMP_DIR}])
 596             end
 597          ) of
 598       {atomic, [_]} -> true;
 599       _ -> false
 600    end.
 601
 602
 603 % Est-ce que Id_user possède Id_mess ?
 604 possede_message(Id_user, Id_mess) ->
 605    case mnesia:transaction(
 606             fun() ->
 607                e(q([E#minichat.auteur_id || E <- mnesia:table(minichat), E#minichat.id =:= Id_mess]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
 608             end
 609          ) of
 610       {atomic, [Id_user | []]} -> true;
 611       _ -> false
 612    end.
 613
 614
 615 % renvoie la liste des ip bannies
 616 % liste de {ip, temps_restant(en minutes), Users} ou Users est une liste de {pseudo, login}
 617 % TODO : déterminer la complexité de cette fonction. (Il n'y a pas d'index sur #user.last_ip)
 618 list_ban() ->
 619    resultat_transaction(mnesia:transaction(
 620       fun() ->
 621          Now = now(),
 622          e(qlc:keysort(1, q([
 623             {
 624                IP#ip_table.ip,
 625                delta_date_minute(date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration), Now),
 626                e(q([{U#user.pseudo, U#user.login} || U <- mnesia:table(user), U#user.last_ip =:= IP#ip_table.ip]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
 627             } ||
 628             IP <- mnesia:table(ip_table),
 629             if IP#ip_table.ban =:= undefined -> false; true -> date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration) > Now end
 630          ])), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
 631       end
 632    )).
 633
 634
 635 % Bannie une ip pour un certain temps (en minute).
 636 ban(IP, Duration) ->
 637    mnesia:transaction(
 638       fun() ->
 639          case mnesia:wread({ip_table, IP}) of
 640             [IP_tuple] ->
 641                mnesia:write(IP_tuple#ip_table{ban = now(), ban_duration = Duration});
 642             _ ->
 643                mnesia:write(#ip_table{ip = IP, ban = now(), ban_duration = Duration})
 644           end
 645       end
 646    ).
 647
 648
 649 % Débanni une ip
 650 deban(IP) ->
 651    ban(IP, 0).
 652
 653
 654 % Renvoie soit {true, Temps} où Temps est le temps en minutes pendant lequel le user est encore banni
 655 % ou false.
 656 est_banni(User_id) ->
 657    resultat_transaction(mnesia:transaction(
 658       fun() ->
 659          case e(q([
 660             {IP#ip_table.ban, IP#ip_table.ban_duration} ||
 661             U <- mnesia:table(user),
 662             U#user.id =:= User_id,
 663             IP <- mnesia:table(ip_table),
 664             IP#ip_table.ip =:= U#user.last_ip
 665          ]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 666             [{Ban, Ban_duration}] ->
 667                Echeance = date_plus_minutes(Ban, Ban_duration),
 668                Now = now(),
 669                if Echeance < Now -> % l'échéance est passée
 670                      false;
 671                   true ->
 672                      {true, delta_date_minute(Echeance, Now)}
 673                end;
 674             _ ->
 675                false
 676          end
 677       end
 678    )).
 679
 680
 681 % Ban est une date tel que retourner par now().
 682 % Ban_duration est un temps en minutes.
 683 % retourne une date.
 684 date_plus_minutes(Ban, Ban_duration) ->
 685    Duration_sec = Ban_duration * 60,
 686    {MegaSec, Sec, MicroSec} = Ban,
 687    {MegaSec + if Sec + Duration_sec >= 1000000 -> 1; true -> 0 end,(Sec + Duration_sec) rem 1000000, MicroSec}.
 688
 689
 690 % Si deux enregistrements consequtifs de la même IP sont fait en moins d'une seconde alors
 691 % ip_table.nb_try_register est incrémenté de 1 sinon il est décrémenté de 1 (jusqu'a 0).
 692 % Si ip_table.nb_try_register vaut 5 alors l'ip ne peux plus s'enregistrer pour une heure.
 693 can_register(IP) ->
 694    resultat_transaction(mnesia:transaction(
 695       fun() ->
 696          case e(q([I || I <- mnesia:table(ip_table), I#ip_table.ip =:= IP]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 697             [] ->
 698                mnesia:write(#ip_table{ip = IP, date_last_try_register = now()}),
 699                true;
 700             [T] ->
 701                Delta = delta_date_ms(T#ip_table.date_last_try_register, now()),
 702                if T#ip_table.nb_try_register =:= ?NB_MAX_FLOOD_REGISTER, Delta < ?TEMPS_BAN_FLOOD_REGISTER ->
 703                      false;
 704                   true ->
 705                      mnesia:write(#ip_table{
 706                         ip = IP,
 707                         date_last_try_register = now(),
 708                         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
 709                      }),
 710                      true
 711                end
 712          end
 713       end
 714    )).
 715
 716
 717 % Renvoie tous les trolls
 718 trolls() ->
 719    resultat_transaction(mnesia:transaction(
 720       fun() ->
 721          e(qlc:keysort(#troll.id, q([T || T <- mnesia:table(troll)])), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
 722       end
 723    )).
 724
 725
 726 % Renvoie les trolls manquants posté après Last_id.
 727 trolls(Last_id) ->
 728    resultat_transaction(mnesia:transaction(
 729       fun() ->
 730          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}])
 731       end
 732    )).
 733
 734
 735  % Crée un nouveau troll.
 736  % Renvoie l'id du nouveau troll
 737  % ou max_troll_reached_per_user si le nombre de troll posté par l'utilisateur max a été atteind
 738  % ou max_troll_reached si le nombre de troll posté max a été atteind
 739  % ou user_unknown
 740 put_troll(User_id, Content) ->
 741    resultat_transaction(mnesia:transaction(
 742       fun() ->
 743          % control le nombre de troll déjà posté
 744          Nb_troll_poste_par_user = length(e(q(
 745             [
 746                E#troll.id || E <- mnesia:table(troll),
 747                E#troll.id_user =:= User_id,
 748                E#troll.date_post =:= undefined
 749             ]
 750          ), [{tmpdir, ?KEY_SORT_TEMP_DIR}])),
 751          Nb_troll_poste_total = length(e(q(
 752             [
 753                E#troll.id || E <- mnesia:table(troll),
 754                E#troll.date_post =:= undefined
 755             ]
 756          ), [{tmpdir, ?KEY_SORT_TEMP_DIR}])),
 757          User = user_by_id(User_id),
 758          case User of
 759             {ok, _} ->
 760                if Nb_troll_poste_par_user >= ?NB_MAX_TROLL_WAITING_BY_USER ->
 761                      max_troll_reached_per_user;
 762                   Nb_troll_poste_total >= ?NB_MAX_TROLL_WAITING ->
 763                      max_troll_reached;
 764                   true ->
 765                      Id = nouvel_id(minichat),
 766                      mnesia:write(#troll{id = Id, id_user = User_id, date_create = now(), content = Content}),
 767                      Id
 768                end;
 769             _ ->
 770                user_unknown
 771          end
 772       end
 773    )).
 774
 775
 776 % renvoie ok | erreur
 777 mod_troll(Troll_id, Content) ->
 778    mnesia:transaction(
 779       fun() ->
 780          case mnesia:wread({troll, Troll_id}) of
 781             [Troll = #troll{date_post = undefined}] ->
 782                mnesia:write(Troll#troll{content = Content});
 783             _ ->
 784                mnesia:abort("mod_troll: Troll inconnu ou déjà posté")
 785           end
 786       end
 787    ).
 788
 789
 790 del_troll(Troll_id) ->
 791    mnesia:transaction(
 792       fun() ->
 793          case mnesia:wread({troll, Troll_id}) of
 794             [#troll{date_post = undefined}] ->
 795                mnesia:delete({troll, Troll_id});
 796             _ ->
 797                mnesia:abort("mod_troll: Troll inconnu ou déjà posté")
 798           end
 799       end
 800    ).
 801
 802
 803 troll_by_id(Troll_id) ->
 804    resultat_transaction(mnesia:transaction(
 805       fun() ->
 806          case e(q([T || T <- mnesia:table(troll), T#troll.id =:= Troll_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 807             [T] -> {ok, T};
 808             _ ->
 809                erreur
 810          end
 811       end
 812    )).
 813
 814
 815 % Renvoie le troll actuel qui se trouve sur la page principale.
 816 % Renvois aucun si pas de troll courant.
 817 current_troll() ->
 818    resultat_transaction(mnesia:transaction(
 819       fun() ->
 820          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}]),
 821          R = case qlc:next_answers(C, 1) of
 822             [T] -> T;
 823             _ -> aucun
 824          end,
 825          qlc:delete_cursor(C),
 826          R
 827       end
 828    )).
 829
 830
 831 % Elit un troll au hasard parmis les trolls en attente (leur date_post =:= undefined)
 832 % Un message est posté par 'Sys' et le troll elu est lié à ce message
 833 % met à jour sa date de post.
 834 % renvoie plus_de_trolls si il n'y a aucun troll en attente.
 835 elire_troll() ->
 836    {A1,A2,A3} = now(),
 837    random:seed(A1, A2, A3),
 838    mnesia:transaction(
 839       fun() ->
 840          case e(q([T || T <- mnesia:table(troll), T#troll.date_post =:= undefined]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 841             [] ->
 842                plus_de_trolls;
 843             Trolls ->
 844                Troll = lists:nth(random:uniform(length(Trolls)), Trolls),
 845                Troll2 = Troll#troll{date_post = now()},
 846                mnesia:write(Troll2),
 847                nouveau_message_sys("Troll de la semaine : ", Troll2#troll.id)
 848          end
 849       end
 850    ).
 851
 852
 853 % Renvoie l'id du message associé au troll dont l'id est donnée.
 854 % Renvoie undefined si il n'y en a pas.
 855 message_id_associe(Troll_id) ->
 856    resultat_transaction(mnesia:transaction(
 857       fun() ->
 858          case e(q([M#minichat.id || M <- mnesia:table(minichat), M#minichat.troll_id =:= Troll_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 859             [Id] -> Id;
 860             _ -> undefined
 861          end
 862       end
 863    )).
 864
 865
 866 % Renvoie le résultat d'une transaction (en décomposant le tuple fournit)
 867 resultat_transaction({_, T}) ->
 868    T.
 869
 870
 871 % Retourne la difference entre deux timestamp (erlang:now()) en miliseconde
 872 delta_date_ms(D1, D2) ->
 873    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).
 874
 875 % Retourne la différence entre deux timestamp (erlang:now) en minutes
 876 delta_date_minute(D1, D2) ->
 877    trunc(delta_date_ms(D1, D2) / 1000 / 60).
 878
 879
 880 % Renvoie un nouvel id pour une table donnée
 881 nouvel_id(Table) ->
 882    mnesia:dirty_update_counter(counter, Table, 1).
 883