modules/erl/euphorik_bd.erl

   1 % coding: utf-8
   2 % Copyright 2008 Grégory Burri
   3 %
   4 % This file is part of Euphorik.
   5 %
   6 % Euphorik is free software: you can redistribute it and/or modify
   7 % it under the terms of the GNU General Public License as published by
   8 % the Free Software Foundation, either version 3 of the License, or
   9 % (at your option) any later version.
  10 %
  11 % Euphorik is distributed in the hope that it will be useful,
  12 % but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 % MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 % GNU General Public License for more details.
  15 %
  16 % You should have received a copy of the GNU General Public License
  17 % along with Euphorik.  If not, see <http://www.gnu.org/licenses/>.
  18 %
  19 % Ce module permet de gérer les données persistantes lié au site d'euphorik.ch.
  20 % Il permet d'ajouter des message, de demande les messages sur une page donnée, etc..
  21 % Ce module utilise une base mnesia.
  22 % @author G.Burri
  23
  24
  25 -module(euphorik_bd).
  26 -export([
  27    % gestion :
  28    create/0,
  29    connect/0,
  30    connect/1,
  31    reset/0,
  32    update/0,
  33
  34    % users :
  35    nouveau_user/2,
  36    nouveau_user/3,
  37    set_profile/11,
  38    update_date_derniere_connexion/1,
  39    update_ip/2,
  40    update_pseudo_user/2,
  41    print_users/0,
  42    print_users/1,
  43    print_user/1,
  44    user_by_cookie/1,
  45    user_by_id/1,
  46    user_by_login/1,
  47    user_by_login_password/2,
  48    user_by_mess/1,
  49    toggle_ek_master/1,
  50    css_from_user_cookie/1,
  51
  52    % messages :
  53    nouveau_message/3,
  54    nouveau_message_sys/1,
  55    nouveau_message_sys/2,
  56    messages/1,
  57    messages/2,
  58    messages/3,
  59    message_by_id/1,
  60    message_by_id_sans_transaction/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/1, e/2, q/1, cursor/1]).
  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    patch(Version),
 194    update(Version + 1).
 195
 196
 197 % Applique une modification de la BD pour passer d'une version à la suivante.
 198 % 1 -> 2
 199 patch(1) ->
 200    ok.
 201 % 2 -> 3
 202 %patch(2) ->
 203
 204
 205 % Ajoute un nouveau user et le renvoie
 206 nouveau_user(Pseudo, Cookie) ->
 207    F = fun() ->
 208       Id = nouvel_id(user),
 209       User = #user{id = Id, cookie = Cookie, pseudo = Pseudo, date_creation = now(), date_derniere_connexion = now()},
 210       mnesia:write(User),
 211       User
 212    end,
 213   resultat_transaction(mnesia:transaction(F)).
 214
 215
 216 % Ajoute un nouveau user et le renvoie
 217 nouveau_user(Login, Password, Cookie) ->
 218    F = fun() ->
 219       Id = nouvel_id(user),
 220       User = #user{id = Id, cookie = Cookie, pseudo = Login, login = Login, password = Password, date_creation = now(), date_derniere_connexion = now()},
 221       mnesia:write(User),
 222       User
 223    end,
 224   resultat_transaction(mnesia:transaction(F)).
 225
 226
 227 % Mise à par Cookie les autres peuvent être undefined ce qui veut dire qu'ils ne seront pas modifié.
 228 set_profile(Cookie, Login, Password, Pseudo, Email, Css, Nick_format, View_times, View_tooltips, Page_principale, Conversations) ->
 229    if Nick_format =:= nick; Nick_format =:= login; Nick_format =:= nick_login ->
 230          resultat_transaction(mnesia:transaction(
 231             fun() ->
 232                case user_by_cookie(Cookie) of
 233                   {ok, User} ->
 234                      case user_by_login(Login) of
 235                         {ok, U} when Login =/= [], U#user.id =/= User#user.id ->
 236                            login_deja_pris;
 237                         _ ->
 238                            User_modifie = User#user{
 239                               % TODO : pourquoi ne pas tester avec la valeur "undefined" plutôt qu'avec "is_list" ?
 240                               % TODO : validation plus strict des données (pas de page négative dans les conv par exemple)
 241                               login = if is_list(Login) -> Login; true -> User#user.login end,
 242                               password = if is_list(Password) andalso Password =/= [] -> Password; true -> User#user.password end,
 243                               pseudo = if is_list(Pseudo) -> Pseudo; true -> User#user.pseudo end,
 244                               email = if is_list(Email) -> Email; true -> User#user.email end,
 245                               css = if is_list(Css) -> Css; true -> User#user.css end,
 246                               nick_format = Nick_format,
 247                               view_times = View_times,
 248                               view_tooltips = View_tooltips,
 249                               page_principale = if is_integer(Page_principale), Page_principale > 0 -> Page_principale; true -> User#user.page_principale end,
 250                               conversations = if is_list(Conversations) -> Conversations; true -> User#user.conversations end
 251                            },
 252                            mnesia:write(User_modifie),
 253                            ok
 254                      end;
 255                   _ -> erreur
 256                end
 257             end
 258          ));
 259       true ->
 260          erreur
 261    end.
 262
 263
 264 % Met à jour la date de la dernière connexion d'un utilisateur à maintenant
 265 update_date_derniere_connexion(User_id) ->
 266    mnesia:transaction(
 267       fun() ->
 268          case mnesia:wread({user, User_id}) of
 269             [User] ->
 270                mnesia:write(User#user{date_derniere_connexion = now()});
 271             _ ->
 272                mnesia:abort("update_date_derniere_connexion: User inconnu")
 273           end
 274       end
 275    ).
 276
 277
 278 % Met à jour l'ip d'un user
 279 update_ip(User_id, IP) ->
 280    mnesia:transaction(
 281       fun() ->
 282          case mnesia:wread({user, User_id}) of
 283             [User] ->
 284                mnesia:write(User#user{last_ip = IP});
 285             _ ->
 286                mnesia:abort("update_ip: User inconnu")
 287           end
 288       end
 289    ).
 290
 291
 292 % Met à jour le pseudo du user
 293 update_pseudo_user(UserId, Pseudo) ->
 294    mnesia:transaction(
 295       fun() ->
 296          case mnesia:wread({user, UserId}) of
 297             [User] when User#user.pseudo =/= Pseudo ->
 298                mnesia:write(User#user{pseudo = Pseudo});
 299             _ ->
 300                mnesia:abort("update_pseudo_user: User inconnu ou pseudo deja à jour")
 301           end
 302       end
 303    ).
 304
 305
 306 % Affiche N user trié par leur date de dernière connexion.
 307 % Attention : pas d'index sur ce champs (la date)
 308 print_users(N) ->
 309    resultat_transaction(mnesia:transaction(fun() ->
 310       C = cursor(
 311          qlc:keysort(
 312             9,
 313             q([E || E <- mnesia:table(user)]),
 314             [{order, descending}]
 315          )
 316       ),
 317       Users = qlc:next_answers(C, N),
 318       lists:foreach(
 319          fun(U) ->
 320             print_user(U)
 321          end,
 322          Users
 323       ),
 324       qlc:delete_cursor(C)
 325    end)).
 326
 327
 328 % Affiche tous les users.
 329 print_users() ->
 330    print_users(all_remaining).
 331
 332
 333 print_user(User) when is_record(User, user) ->
 334    #user{id = Id, pseudo = Pseudo, login = Login, ek_master = Ek_master, date_derniere_connexion = Date} = User,
 335    {{Annee, Mois, Jour}, {Heure, Min, _}} = calendar:now_to_local_time(Date),
 336    io:format(
 337       "~4w : ~10.10..s~10.10..s ~2w.~2.2.0w.~w - ~2wh~2.2.0w~n",
 338       [
 339          Id,
 340          if Ek_master -> "*"; true -> "" end ++ Pseudo,
 341          "(" ++ Login ++ ")",
 342          Jour, Mois, Annee, Heure, Min
 343       ]
 344    );
 345 % Affichage d'un user en fonction de son login
 346 print_user(Login) when is_list(Login) ->
 347    case user_by_login(Login) of
 348       {ok, User} ->
 349          print_user(User);
 350       _ ->
 351          {erreur, "Login pas trouvé : " ++ Login}
 352    end;
 353 % Affichage d'un user en fonction de son id
 354 print_user(Id) when is_integer(Id) ->
 355    case user_by_id(Id) of
 356       {ok, User} ->
 357          print_user(User);
 358       _ ->
 359          {erreur, "Id pas trouvé : " ++ integer_to_list(Id)}
 360    end.
 361
 362
 363 % Est-ce qu'un utilisateur existe en fonction de son cookie ?
 364 % Renvoie {ok, User} ou erreur
 365 user_by_cookie(Cookie) ->
 366    resultat_transaction(mnesia:transaction(
 367       fun() ->
 368          case e(q([E || E <- mnesia:table(user), E#user.cookie =:= Cookie])) of
 369             [User] -> {ok, User};
 370             _ -> erreur
 371          end
 372       end
 373    )).
 374
 375
 376 user_by_id(ID) ->
 377    resultat_transaction(mnesia:transaction(
 378       fun() ->
 379          case e(q([E || E <- mnesia:table(user), E#user.id =:= ID])) of
 380             [User] -> {ok, User};
 381             _ -> erreur
 382          end
 383       end
 384    )).
 385
 386
 387 user_by_login(Login) ->
 388    resultat_transaction(mnesia:transaction(
 389       fun() ->
 390          Users = e(q([E || E <- mnesia:table(user), E#user.login =:= Login])),
 391          case Users of
 392             [User] -> {ok, User};
 393             _ -> erreur
 394          end
 395       end
 396    )).
 397
 398
 399 toggle_ek_master(User_id) ->
 400    resultat_transaction(mnesia:transaction(
 401       fun() ->
 402          Users = e(q([E || E <- mnesia:table(user), E#user.id =:= User_id])),
 403          case Users of
 404             [User] ->
 405                mnesia:write(User#user{ek_master = not User#user.ek_master});
 406             _ -> erreur
 407          end
 408       end
 409    )).
 410
 411
 412 % Renvoie une chaine représentant le cookie ou undefined si pas trouvé.
 413 css_from_user_cookie(Cookie) ->
 414    case user_by_cookie(Cookie) of
 415       {ok, User} ->
 416          User#user.css;
 417       _ ->
 418          undefined
 419    end.
 420
 421
 422
 423
 424 user_by_login_password(Login, Password) ->
 425    resultat_transaction(mnesia:transaction(
 426       fun() ->
 427          case e(q([E || E <- mnesia:table(user), E#user.login =:= Login, E#user.password =:= Password])) of
 428             [User | _] -> {ok, User};
 429             _ -> erreur
 430          end
 431       end
 432    )).
 433
 434
 435 % Renvoie {ok, User} où User est un #user possédant le message donné.
 436 user_by_mess(Id) ->
 437    resultat_transaction(mnesia:transaction(
 438       fun() ->
 439          case e(q([U || U <- mnesia:table(user), M <- mnesia:table(minichat), M#minichat.id =:= Id, M#minichat.auteur_id =:= U#user.id])) of
 440             [User | _] -> {ok, User};
 441             _ -> erreur
 442          end
 443       end
 444    )).
 445
 446
 447 % Ajoute un message. Repond_A est une liste d'id auquel le message répond
 448 % retourne soit l'id du message soit {erreur, <raison>}.
 449 nouveau_message(Mess, Auteur_id, Repond_A) ->
 450    % regarde si les id 'Repond_A' existent
 451    F = fun() ->
 452       Nb_id_trouve = length(e(q([E#minichat.id || E <- mnesia:table(minichat), lists:member(E#minichat.id, Repond_A)]))),
 453       % est-ce que l'auteur existe ?
 454       Auteur = case e(q([E || E <- mnesia:table(user), E#user.id =:= Auteur_id])) of
 455          [A] -> A;
 456          _ -> {erreur, "L'auteur du message est introuvable"}
 457       end,
 458       if Nb_id_trouve =/= length(Repond_A) ->
 459             {erreur, "Un ou plusieurs messages introuvable"};
 460          true ->
 461             % comparaison entre la date du dernier poste et maintenant (gestion du flood)
 462             Delta = delta_date_ms(Auteur#user.date_derniere_connexion, now()),
 463             Nouvel_indice_flood = Auteur#user.indice_flood + if Delta =< ?DUREE_SPAM -> 2; true -> -1 end,
 464             Auteur_maj = Auteur#user{
 465                indice_flood = if Nouvel_indice_flood > ?INDICE_SPAM_MAX -> ?INDICE_SPAM_MAX; Nouvel_indice_flood < 0 -> 0; true -> Nouvel_indice_flood end,
 466                date_derniere_connexion = now()
 467             },
 468             % est-ce que l'auteur à trop floodé ?
 469             if Auteur#user.indice_flood =/= ?INDICE_SPAM_MAX, Auteur_maj#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->
 470                mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}),
 471                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.");
 472             Auteur#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->
 473                {erreur, "Bloqué pour cause de flood"};
 474             true ->
 475                mnesia:write(Auteur_maj),
 476                Id = nouvel_id(minichat),
 477                inserer_reponses(Id, Repond_A),
 478                mnesia:write(#minichat{id=Id, auteur_id=Auteur#user.id, date=now(), pseudo=Auteur#user.pseudo, contenu=Mess}),
 479                Id
 480             end
 481       end
 482    end,
 483    resultat_transaction(mnesia:transaction(F)).
 484
 485 % Définit Id_repondant comme étant la réponse à Ids. Ids est une liste d'id.
 486 inserer_reponses(Id_repondant, [Id_mess | Reste]) ->
 487    mnesia:write(#reponse_minichat{repondant = Id_repondant, cible = Id_mess}),
 488    inserer_reponses(Id_repondant, Reste);
 489 inserer_reponses(_, []) ->
 490    ok.
 491
 492
 493 % Permet de créer un message système.
 494 % Renvoie l'id du message système
 495 nouveau_message_sys(Mess) ->
 496    nouveau_message_sys(Mess, undefined).
 497
 498
 499 % Création d'un message système lié à un troll.
 500 nouveau_message_sys(Mess, Troll_id) ->
 501    {ok, Root} = user_by_id(0),
 502    resultat_transaction(mnesia:transaction(
 503       fun() ->
 504          Id = nouvel_id(minichat),
 505          mnesia:write(#minichat{id=Id, auteur_id=0, date=now(), pseudo=Root#user.pseudo, contenu=Mess, troll_id=Troll_id}),
 506          Id
 507       end
 508    )).
 509
 510
 511 % Renvoie N messages se trouvant sur la première page
 512 messages(N) ->
 513    messages(N, 1).
 514
 515
 516 % Renvoie N messages se trouvant sur la page P
 517 messages(N, P) ->
 518    F = fun() ->
 519       C = cursor(
 520          qlc:keysort(
 521             2,
 522             q([E#minichat{contenu = contenu_message(E)} || E <- mnesia:table(minichat)]),
 523             [{order, descending}]
 524          )
 525       ),
 526       if P > 1 -> qlc:next_answers(C, N * (P - 1));
 527          true -> ok
 528       end,
 529       R = qlc:next_answers(C, N),
 530       qlc:delete_cursor(C),
 531       lists:reverse(R)
 532    end,
 533    resultat_transaction(mnesia:transaction(F)).
 534
 535
 536 % Renvoie les messages manquants pour la page P en sachant qu'il y a N message
 537 % par page et que le dernier message que l'on possède est Id
 538 messages(Id, N, P) ->
 539    lists:filter(fun (M) -> M#minichat.id > Id end, messages(N, P)).
 540
 541
 542 % Renvoie {ok, #minichat} (voir #minichat de euphorik_bd.hrl) à partir de son id.
 543 message_by_id(Id) ->
 544    resultat_transaction(mnesia:transaction(
 545       fun() ->
 546          message_by_id_sans_transaction(Id)
 547       end
 548    )).
 549 message_by_id_sans_transaction(Id) ->
 550    case mnesia:read({minichat, Id}) of
 551       [] -> erreur;
 552       [M] ->
 553          {ok, M#minichat{contenu = contenu_message(M)}}
 554    end.
 555
 556
 557 % Renvoie le contenu d'un message donnée, à utiliser à l'intérieur d'une transaction.
 558 % TODO : Cette fonction pourrait être remplacé par un "outer-join", est-ce possible avec qlc ?
 559 contenu_message(E) ->
 560    case mnesia:read({troll, E#minichat.troll_id}) of
 561       [] -> E#minichat.contenu;
 562       [T] -> E#minichat.contenu ++ T#troll.content
 563    end.
 564
 565
 566 % Renvoie une liste de message (voir #minichat de euphorik_bd.hrl) à partir d'une liste d'id (Ids).
 567 messages_by_ids(Ids) ->
 568    resultat_transaction(mnesia:transaction(
 569       fun() ->
 570          % TODO : optimisations ? serait-ce du O(n) ?
 571          e(qlc:keysort(
 572             2,
 573             q([E || E <- mnesia:table(minichat), lists:any(fun(Id) -> Id =:= E#minichat.id end, Ids)]),
 574             [{order, ascending}]
 575          ))
 576       end
 577    )).
 578
 579
 580 % Est-ce qu'un message existe ? Renvoie un boolean.
 581 % TODO : ya pas plus simple ?
 582 message_existe(Id) ->
 583    resultat_transaction(mnesia:transaction(fun() ->
 584       length(e(q([E#minichat.id || E <- mnesia:table(minichat), E#minichat.id =:= Id]))) =:= 1
 585    end)).
 586
 587
 588 % Renvoie les reponses (utilisé normalement uniquement pendant le debug).
 589 reponses() ->
 590    F = fun() ->
 591       e(q([E || E <- mnesia:table(reponse_minichat)]))
 592    end,
 593    resultat_transaction(mnesia:transaction(F)).
 594
 595
 596 % Renvoie les messages auquel M_id répond.
 597 repond_a(M_id) ->
 598    resultat_transaction(mnesia:transaction(
 599       fun() ->
 600          e(q(
 601             [M || E <- mnesia:table(reponse_minichat),
 602             M <- mnesia:table(minichat),
 603             E#reponse_minichat.repondant =:= M_id,
 604             M#minichat.id =:= E#reponse_minichat.cible]))
 605       end
 606    )).
 607
 608
 609 % Est-ce que le message Id_mess est une réponse d'une message de Id_user ?
 610 est_une_reponse_a_user(Id_user, Id_mess) ->
 611    case mnesia:transaction(
 612             fun() ->
 613                e(q([
 614                   M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat),
 615                   M#minichat.auteur_id =:= Id_user, M#minichat.id =:= R#reponse_minichat.cible, R#reponse_minichat.repondant =:= Id_mess
 616                ]), [{unique_all, true}])
 617             end
 618          ) of
 619       {atomic, [_]} -> true;
 620       _ -> false
 621    end.
 622
 623
 624 % Est-ce que Id_user à répondu au message Id_mess
 625 a_repondu_a_message(Id_user, Id_mess) ->
 626    case mnesia:transaction(
 627             fun() ->
 628                e(q([
 629                   M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat),
 630                   R#reponse_minichat.cible =:= Id_mess, R#reponse_minichat.repondant =:= M#minichat.id, M#minichat.auteur_id =:= Id_user
 631                ]), [{unique_all, true}])
 632             end
 633          ) of
 634       {atomic, [_]} -> true;
 635       _ -> false
 636    end.
 637
 638
 639 % Est-ce que Id_user possède Id_mess ?
 640 possede_message(Id_user, Id_mess) ->
 641    case mnesia:transaction(
 642             fun() ->
 643                e(q([E#minichat.auteur_id || E <- mnesia:table(minichat), E#minichat.id =:= Id_mess]))
 644             end
 645          ) of
 646       {atomic, [Id_user | []]} -> true;
 647       _ -> false
 648    end.
 649
 650
 651 % renvoie la liste des ip bannies
 652 % liste de {ip, temps_restant(en minutes), Users} ou Users est une liste de {pseudo, login}
 653 % TODO : déterminer la complexité de cette fonction. (Il n'y a pas d'index sur #user.last_ip)
 654 list_ban() ->
 655    resultat_transaction(mnesia:transaction(
 656       fun() ->
 657          Now = now(),
 658          e(qlc:keysort(1, q([
 659             {
 660                IP#ip_table.ip,
 661                delta_date_minute(date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration), Now),
 662                e(q([{U#user.pseudo, U#user.login} || U <- mnesia:table(user), U#user.last_ip =:= IP#ip_table.ip]))
 663             } ||
 664             IP <- mnesia:table(ip_table),
 665             if IP#ip_table.ban =:= undefined -> false; true -> date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration) > Now end
 666          ])))
 667       end
 668    )).
 669
 670
 671 % Bannie une ip pour un certain temps (en minute).
 672 ban(IP, Duration) ->
 673    mnesia:transaction(
 674       fun() ->
 675          case mnesia:wread({ip_table, IP}) of
 676             [IP_tuple] ->
 677                mnesia:write(IP_tuple#ip_table{ban = now(), ban_duration = Duration});
 678             _ ->
 679                mnesia:write(#ip_table{ip = IP, ban = now(), ban_duration = Duration})
 680           end
 681       end
 682    ).
 683
 684
 685 % Débanni une ip
 686 deban(IP) ->
 687    ban(IP, 0).
 688
 689
 690 % Renvoie soit {true, Temps} où Temps est le temps en minutes pendant lequel le user est encore banni
 691 % ou false.
 692 est_banni(User_id) ->
 693    resultat_transaction(mnesia:transaction(
 694       fun() ->
 695          case e(q([
 696             {IP#ip_table.ban, IP#ip_table.ban_duration} ||
 697             U <- mnesia:table(user),
 698             U#user.id =:= User_id,
 699             IP <- mnesia:table(ip_table),
 700             IP#ip_table.ip =:= U#user.last_ip
 701          ])) of
 702             [{Ban, Ban_duration}] ->
 703                Echeance = date_plus_minutes(Ban, Ban_duration),
 704                Now = now(),
 705                if Echeance < Now -> % l'échéance est passée
 706                      false;
 707                   true ->
 708                      {true, delta_date_minute(Echeance, Now)}
 709                end;
 710             _ ->
 711                false
 712          end
 713       end
 714    )).
 715
 716
 717 % Ban est une date tel que retourner par now().
 718 % Ban_duration est un temps en minutes.
 719 % retourne une date.
 720 date_plus_minutes(Ban, Ban_duration) ->
 721    Duration_sec = Ban_duration * 60,
 722    {MegaSec, Sec, MicroSec} = Ban,
 723    {MegaSec + if Sec + Duration_sec >= 1000000 -> 1; true -> 0 end,(Sec + Duration_sec) rem 1000000, MicroSec}.
 724
 725
 726 % Si deux enregistrements consequtifs de la même IP sont fait en moins d'une seconde alors
 727 % ip_table.nb_try_register est incrémenté de 1 sinon il est décrémenté de 1 (jusqu'a 0).
 728 % Si ip_table.nb_try_register vaut 5 alors l'ip ne peux plus s'enregistrer pour une heure.
 729 can_register(IP) ->
 730    resultat_transaction(mnesia:transaction(
 731       fun() ->
 732          case e(q([I || I <- mnesia:table(ip_table), I#ip_table.ip =:= IP])) of
 733             [] ->
 734                mnesia:write(#ip_table{ip = IP, date_last_try_register = now()}),
 735                true;
 736             [T] ->
 737                Delta = delta_date_ms(T#ip_table.date_last_try_register, now()),
 738                if T#ip_table.nb_try_register =:= ?NB_MAX_FLOOD_REGISTER, Delta < ?TEMPS_BAN_FLOOD_REGISTER ->
 739                      false;
 740                   true ->
 741                      mnesia:write(#ip_table{
 742                         ip = IP,
 743                         date_last_try_register = now(),
 744                         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
 745                      }),
 746                      true
 747                end
 748          end
 749       end
 750    )).
 751
 752
 753 % Renvoie tous les trolls
 754 trolls() ->
 755    resultat_transaction(mnesia:transaction(
 756       fun() ->
 757          e(qlc:keysort(2, q([T || T <- mnesia:table(troll)])))
 758       end
 759    )).
 760
 761
 762 % Renvoie les trolls manquants posté après Last_id.
 763 trolls(Last_id) ->
 764    resultat_transaction(mnesia:transaction(
 765       fun() ->
 766          e(qlc:keysort(2, q([T || T <- mnesia:table(troll), T#troll.id > Last_id, T#troll.date_post =:= undefined])))
 767       end
 768    )).
 769
 770
 771  % Crée un nouveau troll.
 772  % Renvoie l'id du nouveau troll
 773  % ou max_troll_reached_per_user si le nombre de troll posté par l'utilisateur max a été atteind
 774  % ou max_troll_reached si le nombre de troll posté max a été atteind
 775  % ou user_unknown
 776 put_troll(User_id, Content) ->
 777    resultat_transaction(mnesia:transaction(
 778       fun() ->
 779          % control le nombre de troll déjà posté
 780          Nb_troll_poste_par_user = length(e(q(
 781             [
 782                E#troll.id || E <- mnesia:table(troll),
 783                E#troll.id_user =:= User_id,
 784                E#troll.date_post =:= undefined
 785             ]
 786          ))),
 787          Nb_troll_poste_total = length(e(q(
 788             [
 789                E#troll.id || E <- mnesia:table(troll),
 790                E#troll.date_post =:= undefined
 791             ]
 792          ))),
 793          User = user_by_id(User_id),
 794          case User of
 795             {ok, _} ->
 796                if Nb_troll_poste_par_user >= ?NB_MAX_TROLL_WAITING_BY_USER ->
 797                      max_troll_reached_per_user;
 798                   Nb_troll_poste_total >= ?NB_MAX_TROLL_WAITING ->
 799                      max_troll_reached;
 800                   true ->
 801                      Id = nouvel_id(minichat),
 802                      mnesia:write(#troll{id = Id, id_user = User_id, date_create = now(), content = Content}),
 803                      Id
 804                end;
 805             _ ->
 806                user_unknown
 807          end
 808       end
 809    )).
 810
 811
 812 % renvoie ok | erreur
 813 mod_troll(Troll_id, Content) ->
 814    mnesia:transaction(
 815       fun() ->
 816          case mnesia:wread({troll, Troll_id}) of
 817             [Troll = #troll{date_post = undefined}] ->
 818                mnesia:write(Troll#troll{content = Content});
 819             _ ->
 820                mnesia:abort("mod_troll: Troll inconnu ou déjà posté")
 821           end
 822       end
 823    ).
 824
 825
 826 del_troll(Troll_id) ->
 827    mnesia:transaction(
 828       fun() ->
 829          case mnesia:wread({troll, Troll_id}) of
 830             [#troll{date_post = undefined}] ->
 831                mnesia:delete({troll, Troll_id});
 832             _ ->
 833                mnesia:abort("mod_troll: Troll inconnu ou déjà posté")
 834           end
 835       end
 836    ).
 837
 838
 839 troll_by_id(Troll_id) ->
 840    resultat_transaction(mnesia:transaction(
 841       fun() ->
 842          case e(q([T || T <- mnesia:table(troll), T#troll.id =:= Troll_id])) of
 843             [T] -> {ok, T};
 844             _ ->
 845                erreur
 846          end
 847       end
 848    )).
 849
 850
 851 % Renvoie le troll actuel qui se trouve sur la page principale.
 852 % Renvois aucun si pas de troll courant.
 853 current_troll() ->
 854    resultat_transaction(mnesia:transaction(
 855       fun() ->
 856          % TODO : ya pas moyen de désigner le champs plutot qu'avec un nombre pour le tri ?
 857          C = cursor(qlc:keysort(5, q([T || T <- mnesia:table(troll), T#troll.date_post =/= undefined]), [{order, descending}])),
 858          R = case qlc:next_answers(C, 1) of
 859             [T] -> T;
 860             _ -> aucun
 861          end,
 862          qlc:delete_cursor(C),
 863          R
 864       end
 865    )).
 866
 867
 868 % Elit un troll au hasard parmis les trolls en attente (leur date_post =:= undefined)
 869 % Un message est posté par 'Sys' et le troll elu est lié à ce message
 870 % met à jour sa date de post.
 871 % renvoie plus_de_trolls si il n'y a aucun troll en attente.
 872 elire_troll() ->
 873    {A1,A2,A3} = now(),
 874    random:seed(A1, A2, A3),
 875    mnesia:transaction(
 876       fun() ->
 877          case e(q([T || T <- mnesia:table(troll), T#troll.date_post =:= undefined])) of
 878             [] ->
 879                plus_de_trolls;
 880             Trolls ->
 881                Troll = lists:nth(random:uniform(length(Trolls)), Trolls),
 882                Troll2 = Troll#troll{date_post = now()},
 883                mnesia:write(Troll2),
 884                nouveau_message_sys("Troll de la semaine : ", Troll2#troll.id)
 885          end
 886       end
 887    ).
 888
 889
 890 % Renvoie l'id du message associé au troll dont l'id est donnée.
 891 % Renvoie undefined si il n'y en a pas.
 892 message_id_associe(Troll_id) ->
 893    resultat_transaction(mnesia:transaction(
 894       fun() ->
 895          case e(q([M#minichat.id || M <- mnesia:table(minichat), M#minichat.troll_id =:= Troll_id])) of
 896             [Id] -> Id;
 897             _ -> undefined
 898          end
 899       end
 900    )).
 901
 902
 903 update_version(1) ->
 904    mnesia:transform_table(
 905       ip_table,
 906       fun() -> null end,
 907       record_info(fields, ip_table),
 908       ip_table
 909    ).
 910
 911
 912 % Renvoie le résultat d'une transaction (en décomposant le tuple fournit)
 913 resultat_transaction({_, T}) ->
 914    T.
 915
 916
 917 % Retourne la difference entre deux timestamp (erlang:now()) en miliseconde
 918 delta_date_ms(D1, D2) ->
 919    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).
 920
 921 % Retourne la différence entre deux timestamp (erlang:now) en minutes
 922 delta_date_minute(D1, D2) ->
 923    trunc(delta_date_ms(D1, D2) / 1000 / 60).
 924
 925
 926 % Bizarre, cette fonction n'existe pas dans la stdlib.
 927 % Pas utilisé mais bon ca me fait de la peine de l'enlever.
 928 ceiling(X) ->
 929    T = trunc(X),
 930    case (X - T) of
 931       Neg when Neg < 0 -> T;
 932       Pos when Pos > 0 -> T + 1;
 933       _ -> T
 934    end.
 935
 936
 937 % Renvoie un nouvel id pour une table donnée
 938 nouvel_id(Table) ->
 939    mnesia:dirty_update_counter(counter, Table, 1).
 940