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    % gestion :
  28    create/0,
  29    connect/0,
  30    connect/1,
  31    reset/0,
  32    update/0,
  33
  34    % users :
  35    nouveau_user/2,
  36    nouveau_user/3,
  37    set_profile/11,
  38    update_date_derniere_connexion/1,
  39    update_ip/2,
  40    update_pseudo_user/2,
  41    print_users/0,
  42    print_users/1,
  43    print_user/1,
  44    user_by_cookie/1,
  45    user_by_id/1,
  46    user_by_login/1,
  47    user_by_login_password/2,
  48    user_by_mess/1,
  49    toggle_ek_master/1,
  50    css_from_user_cookie/1,
  51
  52    % messages :
  53    nouveau_message/3,
  54    nouveau_message_sys/1,
  55    nouveau_message_sys/2,
  56    messages/1,
  57    messages/2,
  58    messages/3,
  59    message_by_id/1,
  60    messages_by_ids/1,
  61    message_existe/1,
  62    reponses/0,
  63    repond_a/1,
  64    est_une_reponse_a_user/2,
  65    a_repondu_a_message/2,
  66    possede_message/2,
  67
  68    % ip :
  69    list_ban/0,
  70    ban/2,
  71    deban/1,
  72    est_banni/1,
  73    can_register/1,
  74
  75    % trolls :
  76    trolls/0,
  77    trolls/1,
  78    put_troll/2,
  79    mod_troll/2,
  80    del_troll/1,
  81    troll_by_id/1,
  82    current_troll/0,
  83    elire_troll/0,
  84    message_id_associe/1,
  85
  86    % versions :
  87    update_version/1,
  88
  89    % utiles :
  90    resultat_transaction/1
  91 ]).
  92 -import(qlc, [e/2, q/1, cursor/2]).
  93 -include("../include/euphorik_bd.hrl").
  94 -include("../include/euphorik_defines.hrl").
  95 -include_lib("stdlib/include/qlc.hrl").
  96
  97
  98 % Instructions pour créer une nouvelle base :
  99 % $erl -sname yaws -mnesia dir '"/projets/euphorik/BD"'
 100 % voir doc/installation.txt
 101 % >l(euphorik_bd).
 102 % >euphorik_bd:create().
 103 create() ->
 104    mnesia:stop(),
 105    mnesia:delete_schema([node()]),
 106    mnesia:create_schema([node()]), % nécessaire pour les tables sur disc
 107    mnesia:start(),
 108    create_tables(),
 109    reset().
 110
 111 create_tables() ->
 112    mnesia:create_table(counter, [
 113       {attributes, record_info(fields, counter)},
 114       {disc_copies, [node()]}
 115    ]),
 116    mnesia:create_table(proprietes, [
 117       {attributes, record_info(fields, proprietes)},
 118       {disc_copies, [node()]}
 119    ]),
 120    mnesia:create_table(minichat, [
 121       {attributes, record_info(fields, minichat)},
 122       {index, [auteur_id, troll_id]},
 123       {disc_copies, [node()]}
 124    ]),
 125    mnesia:create_table(reponse_minichat, [
 126       {type, bag},
 127       {attributes, record_info(fields, reponse_minichat)},
 128       {index, [cible]},
 129       {disc_copies, [node()]}
 130    ]),
 131    mnesia:create_table(user, [
 132       {attributes, record_info(fields, user)},
 133       {index, [cookie, login]},
 134       {disc_copies, [node()]}
 135    ]),
 136    mnesia:create_table(ip_table, [
 137       {attributes, record_info(fields, ip_table)},
 138       {disc_copies, [node()]}
 139    ]),
 140    mnesia:create_table(troll, [
 141       {attributes, record_info(fields, troll)},
 142       {index, [date_post]},
 143       {disc_copies, [node()]}
 144    ]).
 145
 146
 147 % Connexion à la base de données de yaws sur overnux
 148 connect() ->
 149    connect(yaws@flynux).
 150 connect(Node) ->
 151    mnesia:start(),
 152    mnesia:change_config(extra_db_nodes, [Node]).
 153
 154
 155 % Efface tous les users, minichat_reponse et minichat.
 156 reset() ->
 157    mnesia:clear_table(counter),
 158    mnesia:clear_table(proprietes),
 159    mnesia:clear_table(user),
 160    mnesia:clear_table(reponse_minichat),
 161    mnesia:clear_table(minichat),
 162    mnesia:clear_table(troll),
 163    mnesia:clear_table(ip_table),
 164    % crée l'utilisateur root
 165    mnesia:transaction(fun() ->
 166       mnesia:write(#proprietes{nom = version, valeur = ?VERSION_BD}),
 167       User = #user{id = 0, pseudo = "Sys", login = "Sys", date_creation = now(), date_derniere_connexion = now(), ek_master = true},
 168       mnesia:write(User),
 169       User
 170    end).
 171
 172
 173 % Met à jour la bd, compare ?VERSION_BD avec la version dans la table 'proprietes'
 174 % et exécute les patchs nécessaires.
 175 update() ->
 176    mnesia:transaction(
 177       fun() ->
 178          case mnesia:read({proprietes, version}) of
 179             [#proprietes{nom = Version}] ->
 180                update(Version);
 181             _ ->
 182                erreur
 183          end
 184       end
 185    ).
 186
 187
 188 % Mise à jour de la BD.
 189 % attention : il est nécessaire de se trouver dans une transaction.
 190 update(?VERSION_BD) -> fini;
 191 update(Version) ->
 192    case mnesia:backup(mnesia:system_info(directory) ++ "/backup" ++ integer_to_list(Version)) of
 193       ok ->
 194          patch(Version),
 195          update(Version + 1);
 196       {error, Raison} -> {error, lists:flatten(io_lib:format("Erreur de création du backup de la version ~w : ~w", [Version, Raison]))}
 197    end.
 198
 199
 200 % Applique une modification de la BD pour passer d'une version à la suivante.
 201 % crée un backup avant l'application du patch
 202 % dans BD/backups nommé "backup<num>" où <num> et le numéro de la version.
 203 % 1 -> 2
 204 patch(1) ->
 205    ok.
 206 % 2 -> 3
 207 %patch(2) ->
 208
 209
 210 % Reviens à une version précédente de la base de données
 211 % (les insérés durant les versions plus récentes sont perdues données sont perdues)
 212 restore(N) ->
 213    mnesia:restore(mnesia:system_info(directory) ++ "/backup" ++ integer_to_list(N)).
 214
 215
 216 % Ajoute un nouveau user et le renvoie
 217 nouveau_user(Pseudo, Cookie) ->
 218    F = fun() ->
 219       Id = nouvel_id(user),
 220       User = #user{id = Id, cookie = Cookie, pseudo = Pseudo, date_creation = now(), date_derniere_connexion = now()},
 221       mnesia:write(User),
 222       User
 223    end,
 224   resultat_transaction(mnesia:transaction(F)).
 225
 226
 227 % Ajoute un nouveau user et le renvoie
 228 nouveau_user(Login, Password, Cookie) ->
 229    F = fun() ->
 230       Id = nouvel_id(user),
 231       User = #user{id = Id, cookie = Cookie, pseudo = Login, login = Login, password = Password, date_creation = now(), date_derniere_connexion = now()},
 232       mnesia:write(User),
 233       User
 234    end,
 235   resultat_transaction(mnesia:transaction(F)).
 236
 237
 238 % Mise à par Cookie les autres peuvent être undefined ce qui veut dire qu'ils ne seront pas modifié.
 239 set_profile(Cookie, Login, Password, Pseudo, Email, Css, Nick_format, View_times, View_tooltips, Page_principale, Conversations) ->
 240    if Nick_format =:= nick; Nick_format =:= login; Nick_format =:= nick_login ->
 241          resultat_transaction(mnesia:transaction(
 242             fun() ->
 243                case user_by_cookie(Cookie) of
 244                   {ok, User} ->
 245                      case user_by_login(Login) of
 246                         {ok, U} when Login =/= [], U#user.id =/= User#user.id ->
 247                            login_deja_pris;
 248                         _ ->
 249                            User_modifie = User#user{
 250                               % TODO : pourquoi ne pas tester avec la valeur "undefined" plutôt qu'avec "is_list" ?
 251                               % TODO : validation plus strict des données (pas de page négative dans les conv par exemple)
 252                               login = if is_list(Login) -> Login; true -> User#user.login end,
 253                               password = if is_list(Password) andalso Password =/= [] -> Password; true -> User#user.password end,
 254                               pseudo = if is_list(Pseudo) -> Pseudo; true -> User#user.pseudo end,
 255                               email = if is_list(Email) -> Email; true -> User#user.email end,
 256                               css = if is_list(Css) -> Css; true -> User#user.css end,
 257                               nick_format = Nick_format,
 258                               view_times = View_times,
 259                               view_tooltips = View_tooltips,
 260                               page_principale = if is_integer(Page_principale), Page_principale > 0 -> Page_principale; true -> User#user.page_principale end,
 261                               conversations = if is_list(Conversations) -> Conversations; true -> User#user.conversations end
 262                            },
 263                            mnesia:write(User_modifie),
 264                            ok
 265                      end;
 266                   _ -> erreur
 267                end
 268             end
 269          ));
 270       true ->
 271          erreur
 272    end.
 273
 274
 275 % Met à jour la date de la dernière connexion d'un utilisateur à maintenant
 276 update_date_derniere_connexion(User_id) ->
 277    mnesia:transaction(
 278       fun() ->
 279          case mnesia:wread({user, User_id}) of
 280             [User] ->
 281                mnesia:write(User#user{date_derniere_connexion = now()});
 282             _ ->
 283                mnesia:abort("update_date_derniere_connexion: User inconnu")
 284           end
 285       end
 286    ).
 287
 288
 289 % Met à jour l'ip d'un user
 290 update_ip(User_id, IP) ->
 291    mnesia:transaction(
 292       fun() ->
 293          case mnesia:wread({user, User_id}) of
 294             [User] ->
 295                mnesia:write(User#user{last_ip = IP});
 296             _ ->
 297                mnesia:abort("update_ip: User inconnu")
 298           end
 299       end
 300    ).
 301
 302
 303 % Met à jour le pseudo du user
 304 update_pseudo_user(UserId, Pseudo) ->
 305    mnesia:transaction(
 306       fun() ->
 307          case mnesia:wread({user, UserId}) of
 308             [User] when User#user.pseudo =/= Pseudo ->
 309                mnesia:write(User#user{pseudo = Pseudo});
 310             _ ->
 311                mnesia:abort("update_pseudo_user: User inconnu ou pseudo deja à jour")
 312           end
 313       end
 314    ).
 315
 316
 317 % Affiche N user trié par leur date de dernière connexion.
 318 % Opt est une liste d'option d'affichage :
 319 %  * ekmaster : n'affiche que les admins
 320 print_users(N, Opt) ->
 321    AfficheQueLesEkMaster = lists:any(fun(O) -> O =:= ekmaster end, Opt),
 322    resultat_transaction(mnesia:transaction(fun() ->
 323       C = cursor(
 324          qlc:keysort(
 325             #user.date_derniere_connexion,
 326             if AfficheQueLesEkMaster ->
 327                q([E || E <- mnesia:table(user), E#user.ek_master =:= true]);
 328             true ->
 329                q([E || E <- mnesia:table(user)])
 330             end,
 331             [{order, descending}]
 332          ),
 333          [{tmpdir, ?KEY_SORT_TEMP_DIR}]
 334       ),
 335       Users = qlc:next_answers(C, N),
 336       lists:foreach(
 337          fun(U) ->
 338             print_user(U)
 339          end,
 340          Users
 341       ),
 342       qlc:delete_cursor(C)
 343    end)).
 344
 345
 346 % Affiche tous les users.
 347 print_users(Opt) ->
 348    print_users(all_remaining, Opt).
 349
 350 % Affiche tous les users.
 351 print_users() ->
 352    print_users(all_remaining, []).
 353
 354 print_user(User) when is_record(User, user) ->
 355    #user{id = Id, pseudo = Pseudo, login = Login, ek_master = Ek_master, date_derniere_connexion = Date, last_ip = IP} = User,
 356    {{Annee, Mois, Jour}, {Heure, Min, _}} = calendar:now_to_local_time(Date),
 357    io:format(
 358       % id        pseudo     (login)        IP  Jour  Mois   Année  Heure Minute
 359       "~4w : ~10.10..s(~10.10..s) ~s ~2w.~2.2.0w.~w - ~2wh~2.2.0w~n",
 360       [
 361          Id,
 362          if Ek_master -> "*"; true -> "" end ++ Pseudo,
 363          Login,
 364          euphorik_common:serialize_ip(IP),
 365          Jour, Mois, Annee, Heure, Min
 366       ]
 367    );
 368 % Affichage d'un user en fonction de son login
 369 print_user(Login) when is_list(Login) ->
 370    case user_by_login(Login) of
 371       {ok, User} ->
 372          print_user(User);
 373       _ ->
 374          {erreur, "Login pas trouvé : " ++ Login}
 375    end;
 376 % Affichage d'un user en fonction de son id
 377 print_user(Id) when is_integer(Id) ->
 378    case user_by_id(Id) of
 379       {ok, User} ->
 380          print_user(User);
 381       _ ->
 382          {erreur, "Id pas trouvé : " ++ integer_to_list(Id)}
 383    end.
 384
 385
 386 % Est-ce qu'un utilisateur existe en fonction de son cookie ?
 387 % Renvoie {ok, User} ou erreur
 388 user_by_cookie(Cookie) ->
 389    resultat_transaction(mnesia:transaction(
 390       fun() ->
 391          case e(q([E || E <- mnesia:table(user), E#user.cookie =:= Cookie]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 392             [User] -> {ok, User};
 393             _ -> erreur
 394          end
 395       end
 396    )).
 397
 398
 399 user_by_id(ID) ->
 400    resultat_transaction(mnesia:transaction(
 401       fun() ->
 402          case e(q([E || E <- mnesia:table(user), E#user.id =:= ID]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 403             [User] -> {ok, User};
 404             _ -> erreur
 405          end
 406       end
 407    )).
 408
 409
 410 user_by_login(Login) ->
 411    resultat_transaction(mnesia:transaction(
 412       fun() ->
 413          Users = e(q([E || E <- mnesia:table(user), E#user.login =:= Login]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]),
 414          case Users of
 415             [User] -> {ok, User};
 416             _ -> erreur
 417          end
 418       end
 419    )).
 420
 421
 422 toggle_ek_master(User_id) ->
 423    resultat_transaction(mnesia:transaction(
 424       fun() ->
 425          Users = e(q([E || E <- mnesia:table(user), E#user.id =:= User_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]),
 426          case Users of
 427             [User] ->
 428                mnesia:write(User#user{ek_master = not User#user.ek_master});
 429             _ -> erreur
 430          end
 431       end
 432    )).
 433
 434
 435 % Renvoie une chaine représentant le cookie ou undefined si pas trouvé.
 436 css_from_user_cookie(Cookie) ->
 437    case user_by_cookie(Cookie) of
 438       {ok, User} ->
 439          User#user.css;
 440       _ ->
 441          undefined
 442    end.
 443
 444
 445 user_by_login_password(Login, Password) ->
 446    resultat_transaction(mnesia:transaction(
 447       fun() ->
 448          case e(q([E || E <- mnesia:table(user), E#user.login =:= Login, E#user.password =:= Password]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 449             [User | _] -> {ok, User};
 450             _ -> erreur
 451          end
 452       end
 453    )).
 454
 455
 456 % Renvoie {ok, User} où User est un #user possédant le message donné.
 457 user_by_mess(Id) ->
 458    resultat_transaction(mnesia:transaction(
 459       fun() ->
 460          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
 461             [User | _] -> {ok, User};
 462             _ -> erreur
 463          end
 464       end
 465    )).
 466
 467
 468 % Ajoute un message. Repond_A est une liste d'id auquel le message répond
 469 % retourne soit l'id du message soit {erreur, <raison>}.
 470 nouveau_message(Mess, Auteur_id, Repond_A) ->
 471    % regarde si les id 'Repond_A' existent
 472    F = fun() ->
 473       Nb_id_trouve = length(e(q([E#minichat.id || E <- mnesia:table(minichat), lists:member(E#minichat.id, Repond_A)]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])),
 474       % est-ce que l'auteur existe ?
 475       case e(q([E || E <- mnesia:table(user), E#user.id =:= Auteur_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 476          [Auteur] ->
 477             if Nb_id_trouve =/= length(Repond_A) ->
 478                   {erreur, "Un ou plusieurs messages introuvable"};
 479                true ->
 480                   % comparaison entre la date du dernier poste et maintenant (gestion du flood)
 481                   Delta = delta_date_ms(Auteur#user.date_derniere_connexion, now()),
 482                   Nouvel_indice_flood = Auteur#user.indice_flood + if Delta =< ?DUREE_SPAM -> 2; true -> -1 end,
 483                   Auteur_maj = Auteur#user{
 484                      indice_flood = if Nouvel_indice_flood > ?INDICE_SPAM_MAX -> ?INDICE_SPAM_MAX; Nouvel_indice_flood < 0 -> 0; true -> Nouvel_indice_flood end,
 485                      date_derniere_connexion = now()
 486                   },
 487                   % est-ce que l'auteur à trop floodé ?
 488                   if Auteur#user.indice_flood =/= ?INDICE_SPAM_MAX, Auteur_maj#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->
 489                      mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}),
 490                      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.");
 491                   Auteur#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->
 492                      {erreur, "Bloqué pour cause de flood"};
 493                   true ->
 494                      mnesia:write(Auteur_maj),
 495                      Id = nouvel_id(minichat),
 496                      inserer_reponses(Id, Repond_A),
 497                      mnesia:write(#minichat{id=Id, auteur_id=Auteur#user.id, date=now(), pseudo=Auteur#user.pseudo, contenu=Mess}),
 498                      Id
 499                   end
 500             end;
 501          _ ->
 502             {erreur, "L'auteur du message est introuvable"}
 503       end
 504    end,
 505    resultat_transaction(mnesia:transaction(F)).
 506
 507 % Définit Id_repondant comme étant la réponse à Ids. Ids est une liste d'id.
 508 inserer_reponses(Id_repondant, [Id_mess | Reste]) ->
 509    mnesia:write(#reponse_minichat{repondant = Id_repondant, cible = Id_mess}),
 510    inserer_reponses(Id_repondant, Reste);
 511 inserer_reponses(_, []) ->
 512    ok.
 513
 514
 515 % Permet de créer un message système.
 516 % Renvoie l'id du message système
 517 nouveau_message_sys(Mess) ->
 518    nouveau_message_sys(Mess, undefined).
 519
 520
 521 % Création d'un message système lié à un troll.
 522 nouveau_message_sys(Mess, Troll_id) ->
 523    {ok, Root} = user_by_id(0),
 524    resultat_transaction(mnesia:transaction(
 525       fun() ->
 526          Id = nouvel_id(minichat),
 527          mnesia:write(#minichat{id=Id, auteur_id=0, date=now(), pseudo=Root#user.pseudo, contenu=Mess, troll_id=Troll_id}),
 528          Id
 529       end
 530    )).
 531
 532
 533 % Renvoie N messages se trouvant sur la première page
 534 messages(N) ->
 535    messages(N, 1).
 536
 537
 538 % Renvoie N messages se trouvant sur la page P
 539 messages(N, P) ->
 540    F = fun() ->
 541       C = cursor(
 542          qlc:keysort(
 543             #minichat.id,
 544             q([E#minichat{contenu = contenu_message(E)} || E <- mnesia:table(minichat)]),
 545             [{order, descending}]
 546          ),
 547          [{tmpdir, ?KEY_SORT_TEMP_DIR}]
 548       ),
 549       if P > 1 -> qlc:next_answers(C, N * (P - 1));
 550          true -> ok
 551       end,
 552       R = qlc:next_answers(C, N),
 553       qlc:delete_cursor(C),
 554       R
 555    end,
 556    lists:reverse(resultat_transaction(mnesia:transaction(F))).
 557
 558
 559 % Renvoie les messages manquants pour la page P en sachant qu'il y a N message
 560 % par page et que le dernier message que l'on possède est Id
 561 messages(Id, N, P) ->
 562    lists:filter(fun (M) -> M#minichat.id > Id end, messages(N, P)).
 563
 564
 565 % Renvoie {ok, #minichat} (voir #minichat de euphorik_bd.hrl) à partir de son id.
 566 message_by_id(Id) ->
 567    resultat_transaction(mnesia:transaction(
 568       fun() ->
 569          case mnesia:read({minichat, Id}) of
 570             [] -> erreur;
 571             [M] ->
 572                {ok, M#minichat{contenu = contenu_message(M)}}
 573          end
 574       end
 575    )).
 576
 577
 578 % Renvoie le contenu d'un message donnée, à utiliser à l'intérieur d'une transaction.
 579 % TODO : Cette fonction pourrait être remplacé par un "outer-join", est-ce possible avec qlc ?
 580 contenu_message(E) ->
 581    case mnesia:read({troll, E#minichat.troll_id}) of
 582       [] -> E#minichat.contenu;
 583       [T] -> E#minichat.contenu ++ T#troll.content
 584    end.
 585
 586
 587 % Renvoie une liste de message (voir #minichat de euphorik_bd.hrl) à partir d'une liste d'id (Ids).
 588 % TODO : optimisations ? serait-ce du O(n) ?
 589 % Bon de toutes façons on s'en fout c'est pas utilisé :)
 590 messages_by_ids(Ids) ->
 591    resultat_transaction(mnesia:transaction(
 592       fun() ->
 593          e(qlc:keysort(
 594             #minichat.id,
 595             q([E || E <- mnesia:table(minichat), lists:any(fun(Id) -> Id =:= E#minichat.id end, Ids)]),
 596             [{order, ascending}]
 597          ),[{tmpdir, ?KEY_SORT_TEMP_DIR}])
 598       end
 599    )).
 600
 601
 602 % Est-ce qu'un message existe ? Renvoie un boolean.
 603 % TODO : ya pas plus simple ?
 604 message_existe(Id) ->
 605    resultat_transaction(mnesia:transaction(fun() ->
 606       length(e(q([E#minichat.id || E <- mnesia:table(minichat), E#minichat.id =:= Id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])) =:= 1
 607    end)).
 608
 609
 610 % Renvoie les reponses (utilisé normalement uniquement pendant le debug).
 611 reponses() ->
 612    F = fun() ->
 613       e(q([E || E <- mnesia:table(reponse_minichat)]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
 614    end,
 615    resultat_transaction(mnesia:transaction(F)).
 616
 617
 618 % Renvoie les messages auquel M_id répond.
 619 repond_a(M_id) ->
 620    resultat_transaction(mnesia:transaction(
 621       fun() ->
 622          e(q(
 623             [M || E <- mnesia:table(reponse_minichat),
 624             M <- mnesia:table(minichat),
 625             E#reponse_minichat.repondant =:= M_id,
 626             M#minichat.id =:= E#reponse_minichat.cible]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
 627       end
 628    )).
 629
 630
 631 % Est-ce que le message Id_mess est une réponse d'une message de Id_user ?
 632 est_une_reponse_a_user(Id_user, Id_mess) ->
 633    case mnesia:transaction(
 634             fun() ->
 635                e(q([
 636                   M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat),
 637                   M#minichat.auteur_id =:= Id_user, M#minichat.id =:= R#reponse_minichat.cible, R#reponse_minichat.repondant =:= Id_mess
 638                ]), [{unique_all, true}, {tmpdir, ?KEY_SORT_TEMP_DIR}])
 639             end
 640          ) of
 641       {atomic, [_]} -> true;
 642       _ -> false
 643    end.
 644
 645
 646 % Est-ce que Id_user à répondu au message Id_mess
 647 a_repondu_a_message(Id_user, Id_mess) ->
 648    case mnesia:transaction(
 649             fun() ->
 650                e(q([
 651                   M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat),
 652                   R#reponse_minichat.cible =:= Id_mess, R#reponse_minichat.repondant =:= M#minichat.id, M#minichat.auteur_id =:= Id_user
 653                ]), [{unique_all, true}, {tmpdir, ?KEY_SORT_TEMP_DIR}])
 654             end
 655          ) of
 656       {atomic, [_]} -> true;
 657       _ -> false
 658    end.
 659
 660
 661 % Est-ce que Id_user possède Id_mess ?
 662 possede_message(Id_user, Id_mess) ->
 663    case mnesia:transaction(
 664             fun() ->
 665                e(q([E#minichat.auteur_id || E <- mnesia:table(minichat), E#minichat.id =:= Id_mess]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
 666             end
 667          ) of
 668       {atomic, [Id_user | []]} -> true;
 669       _ -> false
 670    end.
 671
 672
 673 % renvoie la liste des ip bannies
 674 % liste de {ip, temps_restant(en minutes), Users} ou Users est une liste de {pseudo, login}
 675 % TODO : déterminer la complexité de cette fonction. (Il n'y a pas d'index sur #user.last_ip)
 676 list_ban() ->
 677    resultat_transaction(mnesia:transaction(
 678       fun() ->
 679          Now = now(),
 680          e(qlc:keysort(1, q([
 681             {
 682                IP#ip_table.ip,
 683                delta_date_minute(date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration), Now),
 684                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}])
 685             } ||
 686             IP <- mnesia:table(ip_table),
 687             if IP#ip_table.ban =:= undefined -> false; true -> date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration) > Now end
 688          ])), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
 689       end
 690    )).
 691
 692
 693 % Bannie une ip pour un certain temps (en minute).
 694 ban(IP, Duration) ->
 695    mnesia:transaction(
 696       fun() ->
 697          case mnesia:wread({ip_table, IP}) of
 698             [IP_tuple] ->
 699                mnesia:write(IP_tuple#ip_table{ban = now(), ban_duration = Duration});
 700             _ ->
 701                mnesia:write(#ip_table{ip = IP, ban = now(), ban_duration = Duration})
 702           end
 703       end
 704    ).
 705
 706
 707 % Débanni une ip
 708 deban(IP) ->
 709    ban(IP, 0).
 710
 711
 712 % Renvoie soit {true, Temps} où Temps est le temps en minutes pendant lequel le user est encore banni
 713 % ou false.
 714 est_banni(User_id) ->
 715    resultat_transaction(mnesia:transaction(
 716       fun() ->
 717          case e(q([
 718             {IP#ip_table.ban, IP#ip_table.ban_duration} ||
 719             U <- mnesia:table(user),
 720             U#user.id =:= User_id,
 721             IP <- mnesia:table(ip_table),
 722             IP#ip_table.ip =:= U#user.last_ip
 723          ]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 724             [{Ban, Ban_duration}] ->
 725                Echeance = date_plus_minutes(Ban, Ban_duration),
 726                Now = now(),
 727                if Echeance < Now -> % l'échéance est passée
 728                      false;
 729                   true ->
 730                      {true, delta_date_minute(Echeance, Now)}
 731                end;
 732             _ ->
 733                false
 734          end
 735       end
 736    )).
 737
 738
 739 % Ban est une date tel que retourner par now().
 740 % Ban_duration est un temps en minutes.
 741 % retourne une date.
 742 date_plus_minutes(Ban, Ban_duration) ->
 743    Duration_sec = Ban_duration * 60,
 744    {MegaSec, Sec, MicroSec} = Ban,
 745    {MegaSec + if Sec + Duration_sec >= 1000000 -> 1; true -> 0 end,(Sec + Duration_sec) rem 1000000, MicroSec}.
 746
 747
 748 % Si deux enregistrements consequtifs de la même IP sont fait en moins d'une seconde alors
 749 % ip_table.nb_try_register est incrémenté de 1 sinon il est décrémenté de 1 (jusqu'a 0).
 750 % Si ip_table.nb_try_register vaut 5 alors l'ip ne peux plus s'enregistrer pour une heure.
 751 can_register(IP) ->
 752    resultat_transaction(mnesia:transaction(
 753       fun() ->
 754          case e(q([I || I <- mnesia:table(ip_table), I#ip_table.ip =:= IP]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 755             [] ->
 756                mnesia:write(#ip_table{ip = IP, date_last_try_register = now()}),
 757                true;
 758             [T] ->
 759                Delta = delta_date_ms(T#ip_table.date_last_try_register, now()),
 760                if T#ip_table.nb_try_register =:= ?NB_MAX_FLOOD_REGISTER, Delta < ?TEMPS_BAN_FLOOD_REGISTER ->
 761                      false;
 762                   true ->
 763                      mnesia:write(#ip_table{
 764                         ip = IP,
 765                         date_last_try_register = now(),
 766                         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
 767                      }),
 768                      true
 769                end
 770          end
 771       end
 772    )).
 773
 774
 775 % Renvoie tous les trolls
 776 trolls() ->
 777    resultat_transaction(mnesia:transaction(
 778       fun() ->
 779          e(qlc:keysort(#troll.id, q([T || T <- mnesia:table(troll)])), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
 780       end
 781    )).
 782
 783
 784 % Renvoie les trolls manquants posté après Last_id.
 785 trolls(Last_id) ->
 786    resultat_transaction(mnesia:transaction(
 787       fun() ->
 788          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}])
 789       end
 790    )).
 791
 792
 793  % Crée un nouveau troll.
 794  % Renvoie l'id du nouveau troll
 795  % ou max_troll_reached_per_user si le nombre de troll posté par l'utilisateur max a été atteind
 796  % ou max_troll_reached si le nombre de troll posté max a été atteind
 797  % ou user_unknown
 798 put_troll(User_id, Content) ->
 799    resultat_transaction(mnesia:transaction(
 800       fun() ->
 801          % control le nombre de troll déjà posté
 802          Nb_troll_poste_par_user = length(e(q(
 803             [
 804                E#troll.id || E <- mnesia:table(troll),
 805                E#troll.id_user =:= User_id,
 806                E#troll.date_post =:= undefined
 807             ]
 808          ), [{tmpdir, ?KEY_SORT_TEMP_DIR}])),
 809          Nb_troll_poste_total = length(e(q(
 810             [
 811                E#troll.id || E <- mnesia:table(troll),
 812                E#troll.date_post =:= undefined
 813             ]
 814          ), [{tmpdir, ?KEY_SORT_TEMP_DIR}])),
 815          User = user_by_id(User_id),
 816          case User of
 817             {ok, _} ->
 818                if Nb_troll_poste_par_user >= ?NB_MAX_TROLL_WAITING_BY_USER ->
 819                      max_troll_reached_per_user;
 820                   Nb_troll_poste_total >= ?NB_MAX_TROLL_WAITING ->
 821                      max_troll_reached;
 822                   true ->
 823                      Id = nouvel_id(minichat),
 824                      mnesia:write(#troll{id = Id, id_user = User_id, date_create = now(), content = Content}),
 825                      Id
 826                end;
 827             _ ->
 828                user_unknown
 829          end
 830       end
 831    )).
 832
 833
 834 % renvoie ok | erreur
 835 mod_troll(Troll_id, Content) ->
 836    mnesia:transaction(
 837       fun() ->
 838          case mnesia:wread({troll, Troll_id}) of
 839             [Troll = #troll{date_post = undefined}] ->
 840                mnesia:write(Troll#troll{content = Content});
 841             _ ->
 842                mnesia:abort("mod_troll: Troll inconnu ou déjà posté")
 843           end
 844       end
 845    ).
 846
 847
 848 del_troll(Troll_id) ->
 849    mnesia:transaction(
 850       fun() ->
 851          case mnesia:wread({troll, Troll_id}) of
 852             [#troll{date_post = undefined}] ->
 853                mnesia:delete({troll, Troll_id});
 854             _ ->
 855                mnesia:abort("mod_troll: Troll inconnu ou déjà posté")
 856           end
 857       end
 858    ).
 859
 860
 861 troll_by_id(Troll_id) ->
 862    resultat_transaction(mnesia:transaction(
 863       fun() ->
 864          case e(q([T || T <- mnesia:table(troll), T#troll.id =:= Troll_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 865             [T] -> {ok, T};
 866             _ ->
 867                erreur
 868          end
 869       end
 870    )).
 871
 872
 873 % Renvoie le troll actuel qui se trouve sur la page principale.
 874 % Renvois aucun si pas de troll courant.
 875 current_troll() ->
 876    resultat_transaction(mnesia:transaction(
 877       fun() ->
 878          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}]),
 879          R = case qlc:next_answers(C, 1) of
 880             [T] -> T;
 881             _ -> aucun
 882          end,
 883          qlc:delete_cursor(C),
 884          R
 885       end
 886    )).
 887
 888
 889 % Elit un troll au hasard parmis les trolls en attente (leur date_post =:= undefined)
 890 % Un message est posté par 'Sys' et le troll elu est lié à ce message
 891 % met à jour sa date de post.
 892 % renvoie plus_de_trolls si il n'y a aucun troll en attente.
 893 elire_troll() ->
 894    {A1,A2,A3} = now(),
 895    random:seed(A1, A2, A3),
 896    mnesia:transaction(
 897       fun() ->
 898          case e(q([T || T <- mnesia:table(troll), T#troll.date_post =:= undefined]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 899             [] ->
 900                plus_de_trolls;
 901             Trolls ->
 902                Troll = lists:nth(random:uniform(length(Trolls)), Trolls),
 903                Troll2 = Troll#troll{date_post = now()},
 904                mnesia:write(Troll2),
 905                nouveau_message_sys("Troll de la semaine : ", Troll2#troll.id)
 906          end
 907       end
 908    ).
 909
 910
 911 % Renvoie l'id du message associé au troll dont l'id est donnée.
 912 % Renvoie undefined si il n'y en a pas.
 913 message_id_associe(Troll_id) ->
 914    resultat_transaction(mnesia:transaction(
 915       fun() ->
 916          case e(q([M#minichat.id || M <- mnesia:table(minichat), M#minichat.troll_id =:= Troll_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
 917             [Id] -> Id;
 918             _ -> undefined
 919          end
 920       end
 921    )).
 922
 923
 924 update_version(1) ->
 925    mnesia:transform_table(
 926       ip_table,
 927       fun() -> null end,
 928       record_info(fields, ip_table),
 929       ip_table
 930    ).
 931
 932
 933 % Renvoie le résultat d'une transaction (en décomposant le tuple fournit)
 934 resultat_transaction({_, T}) ->
 935    T.
 936
 937
 938 % Retourne la difference entre deux timestamp (erlang:now()) en miliseconde
 939 delta_date_ms(D1, D2) ->
 940    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).
 941
 942 % Retourne la différence entre deux timestamp (erlang:now) en minutes
 943 delta_date_minute(D1, D2) ->
 944    trunc(delta_date_ms(D1, D2) / 1000 / 60).
 945
 946
 947 % Bizarre, cette fonction n'existe pas dans la stdlib.
 948 % Pas utilisé mais bon ca me fait de la peine de l'enlever.
 949 ceiling(X) ->
 950    T = trunc(X),
 951    case (X - T) of
 952       Neg when Neg < 0 -> T;
 953       Pos when Pos > 0 -> T + 1;
 954       _ -> T
 955    end.
 956
 957
 958 % Renvoie un nouvel id pour une table donnée
 959 nouvel_id(Table) ->
 960    mnesia:dirty_update_counter(counter, Table, 1).
 961