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