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