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