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