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