modules/erl/euphorik_bd.erl

   1 % coding: utf-8
   2 % Ce module permet de gérer les données persistantes lié au site d'euphorik.ch.
   3 % Il permet d'ajouter des message, de demande les messages sur une page donnée, etc..
   4 % Ce module utilise une base mnesia.
   5 % @author G.Burri
   6
   7 -module(euphorik_bd).
   8 -export([
   9    % gestion :
  10    create/0,
  11    connect/0,
  12    connect/1,
  13    reset/0,
  14
  15    % users :
  16    nouveau_user/2,
  17    nouveau_user/3,
  18    set_profile/9,
  19    update_date_derniere_connexion/1,
  20    update_ip/2,
  21    update_pseudo_user/2,
  22    users/0,
  23    user_by_cookie/1,
  24    user_by_id/1,
  25    user_by_login/1,
  26    user_by_login_password/2,
  27    user_by_mess/1,
  28    toggle_ek_master/1,
  29
  30    % messages :
  31    nouveau_message/3,
  32    nouveau_message_sys/1,
  33    messages/1,
  34    messages/2,
  35    messages/3,
  36    message_by_id/1,
  37    messages_by_ids/1,
  38    message_existe/1,
  39    reponses/0,
  40    repond_a/1,
  41    est_une_reponse_a_user/2,
  42    a_repondu_a_message/2,
  43    possede_message/2,
  44
  45    % ip :
  46    ip_table/0,
  47    ban/2,
  48    deban/1,
  49    est_banni/1,
  50    can_register/1,
  51
  52    % trolls :
  53    trolls/0,
  54    trolls/1,
  55    put_troll/2,
  56    mod_troll/2,
  57    del_troll/1,
  58
  59    % versions :
  60    update_version/1,
  61
  62    % utiles :
  63    resultat_transaction/1
  64 ]).
  65
  66 -include("../include/euphorik_bd.hrl").
  67 -include("../include/euphorik_defines.hrl").
  68 -include_lib("stdlib/include/qlc.hrl").
  69
  70
  71 % Instructions pour créer une nouvelle base :
  72 % $erl -sname yaws -mnesia dir '"/projets/euphorik/BD"'
  73 % voir doc/installation.txt
  74 % >l(euphorik_bd).
  75 % >euphorik_bd:create().
  76 create() ->
  77    mnesia:stop(),
  78    mnesia:delete_schema([node()]),
  79    mnesia:create_schema([node()]), % nécessaire pour les tables sur disc
  80    mnesia:start(),
  81    create_tables(),
  82    reset().
  83 create_tables() ->
  84    mnesia:create_table(counter, [
  85       {attributes, record_info(fields, counter)},
  86       {disc_copies, [node()]}
  87    ]),
  88    mnesia:create_table(minichat, [
  89       {attributes, record_info(fields, minichat)},
  90       {index, [auteur_id]},
  91       {disc_copies, [node()]}
  92    ]),
  93    mnesia:create_table(reponse_minichat, [
  94       {type, bag},
  95       {attributes, record_info(fields, reponse_minichat)},
  96       {index, [cible]},
  97       {disc_copies, [node()]}
  98    ]),
  99    mnesia:create_table(user, [
 100       {attributes, record_info(fields, user)},
 101       {index, [cookie, login]},
 102       {disc_copies, [node()]}
 103    ]),
 104    mnesia:create_table(ip_table, [
 105       {attributes, record_info(fields, ip_table)},
 106       {disc_copies, [node()]}
 107    ]),
 108    mnesia:create_table(troll, [
 109       {attributes, record_info(fields, troll)},
 110       {index, [date_post]},
 111       {disc_copies, [node()]}
 112    ]).
 113
 114
 115 % Connexion à la base de données de yaws sur overnux
 116 connect() ->
 117    connect(yaws@flynux).
 118 connect(Node) ->
 119    mnesia:start(),
 120    mnesia:change_config(extra_db_nodes, [Node]).
 121
 122
 123 % Efface tous les users, minichat_reponse et minichat.
 124 reset() ->
 125    mnesia:clear_table(counter),
 126    mnesia:clear_table(user),
 127    mnesia:clear_table(reponse_minichat),
 128    mnesia:clear_table(minichat),
 129    mnesia:clear_table(troll),
 130    mnesia:clear_table(ip_table),
 131    % crée l'utilisateur root
 132    mnesia:transaction(fun() ->
 133       User = #user{id = 0, pseudo = "Sys", login = "Sys", date_creation = now(), date_derniere_connexion = now(), ek_master = true},
 134       mnesia:write(User),
 135       User
 136    end).
 137
 138
 139 % Ajoute un nouveau user et le renvoie
 140 nouveau_user(Pseudo, Cookie) ->
 141    F = fun() ->
 142       Id = nouvel_id(user),
 143       User = #user{id = Id, cookie = Cookie, pseudo = Pseudo, date_creation = now(), date_derniere_connexion = now()},
 144       mnesia:write(User),
 145       User
 146    end,
 147   resultat_transaction(mnesia:transaction(F)).
 148
 149
 150 % Ajoute un nouveau user et le renvoie
 151 nouveau_user(Login, Password, Cookie) ->
 152    F = fun() ->
 153       Id = nouvel_id(user),
 154       User = #user{id = Id, cookie = Cookie, pseudo = Login, login = Login, password = Password, date_creation = now(), date_derniere_connexion = now()},
 155       mnesia:write(User),
 156       User
 157    end,
 158   resultat_transaction(mnesia:transaction(F)).
 159
 160
 161 % Mise à par Cookie les autres peuvent être undefined ce qui veut dire qu'ils ne seront pas modifié.
 162 set_profile(Cookie, Login, Password, Pseudo, Email, Css, Nick_format, Page_principale, Conversations) ->
 163    if Nick_format =:= nick; Nick_format =:= login; Nick_format =:= nick_login ->
 164          resultat_transaction(mnesia:transaction(
 165             fun() ->
 166                case user_by_cookie(Cookie) of
 167                   {ok, User} ->
 168                      case user_by_login(Login) of
 169                         {ok, U} when User#user.login =/= [], U#user.id =/= User#user.id ->
 170                            login_deja_pris;
 171                         _ ->
 172                            User_modifie = User#user{
 173                               % TODO : pourquoi ne pas tester avec la valeur "undefined" plutôt qu'avec "is_list" ?
 174                               % TODO : validation plus strict des données (pas de page négative dans les conv par exemple)
 175                               login = if is_list(Login) -> Login; true -> User#user.login end,
 176                               password = if is_list(Password) andalso Password =/= [] -> Password; true -> User#user.password end,
 177                               pseudo = if is_list(Pseudo) -> Pseudo; true -> User#user.pseudo end,
 178                               email = if is_list(Email) -> Email; true -> User#user.email end,
 179                               css = if is_list(Css) -> Css; true -> User#user.css end,
 180                               nick_format = Nick_format,
 181                               page_principale = if is_integer(Page_principale), Page_principale > 0 -> Page_principale; true -> User#user.page_principale end,
 182                               conversations = if is_list(Conversations) -> Conversations; true -> User#user.conversations end
 183                            },
 184                            mnesia:write(User_modifie),
 185                            ok
 186                      end;
 187                   _ -> erreur
 188                end
 189             end
 190          ));
 191       true ->
 192          erreur
 193    end.
 194
 195
 196 % Met à jour la date de la dernière connexion d'un utilisateur à maintenant
 197 update_date_derniere_connexion(User_id) ->
 198    mnesia:transaction(
 199       fun() ->
 200          case mnesia:wread({user, User_id}) of
 201             [User] ->
 202                mnesia:write(User#user{date_derniere_connexion = now()});
 203             _ ->
 204                mnesia:abort("update_date_derniere_connexion: User inconnu")
 205           end
 206       end
 207    ).
 208
 209
 210 % Met à jour l'ip d'un user
 211 update_ip(User_id, IP) ->
 212    mnesia:transaction(
 213       fun() ->
 214          case mnesia:wread({user, User_id}) of
 215             [User] ->
 216                mnesia:write(User#user{last_ip = IP});
 217             _ ->
 218                mnesia:abort("update_ip: User inconnu")
 219           end
 220       end
 221    ).
 222
 223
 224 % Met à jour le pseudo du user
 225 update_pseudo_user(UserId, Pseudo) ->
 226    mnesia:transaction(
 227       fun() ->
 228          case mnesia:wread({user, UserId}) of
 229             [User] when User#user.pseudo =/= Pseudo ->
 230                mnesia:write(User#user{pseudo = Pseudo});
 231             _ ->
 232                mnesia:abort("update_pseudo_user: User inconnu ou pseudo deja à jour")
 233           end
 234       end
 235    ).
 236
 237
 238 % Renvoie tous les users.
 239 users() ->
 240    resultat_transaction(mnesia:transaction(fun() ->
 241       qlc:e(qlc:q([E || E <- mnesia:table(user)]))
 242    end)).
 243
 244
 245 % Est-ce qu'un utilisateur existe en fonction de son cookie ?
 246 % Renvoie {ok, User} ou erreur
 247 user_by_cookie(Cookie) ->
 248    resultat_transaction(mnesia:transaction(
 249       fun() ->
 250          Users = qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.cookie =:= Cookie])),
 251          case Users of
 252             [User] -> {ok, User};
 253             _ -> erreur
 254          end
 255       end
 256    )).
 257
 258
 259 user_by_id(ID) ->
 260    resultat_transaction(mnesia:transaction(
 261       fun() ->
 262          Users = qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.id =:= ID])),
 263          case Users of
 264             [User] -> {ok, User};
 265             _ -> erreur
 266          end
 267       end
 268    )).
 269
 270
 271 user_by_login(Login) ->
 272    resultat_transaction(mnesia:transaction(
 273       fun() ->
 274          Users = qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.login =:= Login])),
 275          case Users of
 276             [User] -> {ok, User};
 277             _ -> erreur
 278          end
 279       end
 280    )).
 281
 282
 283 toggle_ek_master(User_id) ->
 284    resultat_transaction(mnesia:transaction(
 285       fun() ->
 286          Users = qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.id =:= User_id])),
 287          case Users of
 288             [User] ->
 289                mnesia:write(User#user{ek_master = not User#user.ek_master});
 290             _ -> erreur
 291          end
 292       end
 293    )).
 294
 295
 296 user_by_login_password(Login, Password) ->
 297    resultat_transaction(mnesia:transaction(
 298       fun() ->
 299          Users = qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.login =:= Login, E#user.password =:= Password])),
 300          case Users of
 301             [User] -> {ok, User};
 302             _ -> erreur
 303          end
 304       end
 305    )).
 306
 307
 308 % Renvoie {ok, User} où User est un #user possédant le message donné.
 309 user_by_mess(Id) ->
 310    resultat_transaction(mnesia:transaction(
 311       fun() ->
 312          case qlc:e(qlc:q([U || U <- mnesia:table(user), M <- mnesia:table(minichat), M#minichat.id =:= Id, M#minichat.auteur_id =:= U#user.id])) of
 313             [User] -> {ok, User};
 314             _ -> erreur
 315          end
 316       end
 317    )).
 318
 319
 320 % Ajoute un message. Repond_A est une liste d'id auquel le message répond
 321 % retourne soit l'id du message soit erreur.
 322 nouveau_message(Mess, Auteur_id, Repond_A) ->
 323    % regarde si les id 'Repond_A' existent
 324    F = fun() ->
 325       Nb_id_trouve = length(qlc:e(qlc:q([E#minichat.id || E <- mnesia:table(minichat), lists:member(E#minichat.id, Repond_A)]))),
 326       % est-ce que l'auteur existe ?
 327       Auteur = case qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.id =:= Auteur_id])) of
 328          [A] -> A;
 329          _ -> throw("L'auteur du message est introuvable")
 330       end,
 331       if Nb_id_trouve =/= length(Repond_A) -> throw("Un ou plusieurs messages introuvable");
 332          true -> ok
 333       end,
 334       % compare les dernière
 335       Delta = delta_date_ms(Auteur#user.date_derniere_connexion, now()),
 336       Nouvel_indice_flood = Auteur#user.indice_flood + if Delta =< ?DUREE_SPAM -> 2; true -> -1 end,
 337       Auteur_maj = Auteur#user{
 338          indice_flood = if Nouvel_indice_flood > ?INDICE_SPAM_MAX -> ?INDICE_SPAM_MAX; Nouvel_indice_flood < 0 -> 0; true -> Nouvel_indice_flood end,
 339          date_derniere_connexion = now()
 340       },
 341       % est-ce que l'auteur à trop floodé ?
 342       if Auteur#user.indice_flood =/= ?INDICE_SPAM_MAX, Auteur_maj#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->
 343          mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}),
 344          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.");
 345       Auteur#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->
 346          erreur;
 347       true ->
 348          mnesia:write(Auteur_maj),
 349          Id = nouvel_id(minichat),
 350          inserer_reponses(Id, Repond_A),
 351          mnesia:write(#minichat{id=Id, auteur_id=Auteur#user.id, date=now(), pseudo=Auteur#user.pseudo, contenu=Mess}),
 352          Id
 353       end
 354    end,
 355    resultat_transaction(mnesia:transaction(F)).
 356
 357 % Définit Id_repondant comme étant la réponse à Ids. Ids est une liste d'id.
 358 inserer_reponses(Id_repondant, [Id_mess | Reste]) ->
 359    mnesia:write(#reponse_minichat{repondant = Id_repondant, cible = Id_mess}),
 360    inserer_reponses(Id_repondant, Reste);
 361 inserer_reponses(_, []) ->
 362    ok.
 363
 364
 365 % Permet de créer un message système.
 366 % Renvoie l'id du message système
 367 nouveau_message_sys(Mess) ->
 368    {ok, Root} = user_by_id(0),
 369    mnesia:transaction(
 370       fun() ->
 371          Id = nouvel_id(minichat),
 372          mnesia:write(#minichat{id=Id, auteur_id=0, date=now(), pseudo=Root#user.pseudo, contenu=Mess})
 373       end
 374    ).
 375
 376
 377 % Renvoie N messages se trouvant sur la première page
 378 messages(N) ->
 379    messages(N, 1).
 380
 381
 382 % Renvoie N messages se trouvant sur la page P
 383 messages(N, P) ->
 384    F = fun() ->
 385       C = qlc:cursor(qlc:q([E || E <- qlc:keysort(2, mnesia:table(minichat), [{order, descending}])])),
 386       if P > 1 -> qlc:next_answers(C, N * (P - 1));
 387          true -> ok
 388       end,
 389       R = qlc:next_answers(C, N),
 390       qlc:delete_cursor(C),
 391       lists:reverse(R)
 392    end,
 393    resultat_transaction(mnesia:transaction(F)).
 394
 395
 396 % Renvoie les messages manquants pour la page P en sachant qu'il y a N message
 397 % par page et que le dernier message que l'on possède est Id
 398 messages(Id, N, P) ->
 399    lists:filter(fun (M) -> M#minichat.id > Id end, messages(N, P)).
 400
 401
 402 % Renvoie {ok, #minichat} (voir #minichat de euphorik_bd.hrl) à partir de son id.
 403 message_by_id(Id) ->
 404    case resultat_transaction(mnesia:transaction(
 405       fun() ->
 406          qlc:e(qlc:q([E || E <- qlc:keysort(2, mnesia:table(minichat), [{order, ascending}]), Id =:= E#minichat.id]))
 407       end
 408    )) of
 409       [M] -> {ok, M};
 410       _ -> erreur
 411    end.
 412
 413
 414 % Renvoie une liste de message (voir #minichat de euphorik_bd.hrl) à partir d'une liste d'id (Ids).
 415 messages_by_ids(Ids) ->
 416    resultat_transaction(mnesia:transaction(
 417       fun() ->
 418          % TODO : optimisations ? serait-ce du O(n) ?
 419          Query = qlc:q([E || E <- qlc:keysort(2, mnesia:table(minichat), [{order, ascending}]), lists:any(fun(Id) -> Id =:= E#minichat.id end, Ids)]),
 420          qlc:e(Query)
 421       end
 422    )).
 423
 424
 425 % Est-ce qu'un message existe ? Renvoie un boolean.
 426 % TODO : ya pas plus simple ?
 427 message_existe(Id) ->
 428    resultat_transaction(mnesia:transaction(fun() ->
 429       length(qlc:e(qlc:q([E#minichat.id || E <- mnesia:table(minichat), E#minichat.id =:= Id]))) =:= 1
 430    end)).
 431
 432
 433 % Renvoie les reponses (utilisé normalement uniquement pendant le debug).
 434 reponses() ->
 435    F = fun() ->
 436       qlc:e(qlc:q([E || E <- mnesia:table(reponse_minichat)]))
 437    end,
 438    resultat_transaction(mnesia:transaction(F)).
 439
 440
 441 % Renvoie les messages auquel M_id répond.
 442 repond_a(M_id) ->
 443    resultat_transaction(mnesia:transaction(
 444       fun() ->
 445          qlc:e(qlc:q(
 446             [M || E <- mnesia:table(reponse_minichat),
 447             M <- mnesia:table(minichat),
 448             E#reponse_minichat.repondant =:= M_id,
 449             M#minichat.id =:= E#reponse_minichat.cible]))
 450       end
 451    )).
 452
 453
 454 % Est-ce que le message Id_mess est une réponse d'une message de Id_user ?
 455 est_une_reponse_a_user(Id_user, Id_mess) ->
 456    case mnesia:transaction(
 457             fun() ->
 458                qlc:e(qlc:q([
 459                   M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat),
 460                   M#minichat.auteur_id =:= Id_user, M#minichat.id =:= R#reponse_minichat.cible, R#reponse_minichat.repondant =:= Id_mess
 461                ]), [{unique_all, true}])
 462             end
 463          ) of
 464       {atomic, [_]} -> true;
 465       _ -> false
 466    end.
 467
 468
 469 % Est-ce que Id_user à répondu au message Id_mess
 470 a_repondu_a_message(Id_user, Id_mess) ->
 471    case mnesia:transaction(
 472             fun() ->
 473                qlc:e(qlc:q([
 474                   M#minichat.auteur_id || M <- mnesia:table(minichat), R <- mnesia:table(reponse_minichat),
 475                   R#reponse_minichat.cible =:= Id_mess, R#reponse_minichat.repondant =:= M#minichat.id, M#minichat.auteur_id =:= Id_user
 476                ]), [{unique_all, true}])
 477             end
 478          ) of
 479       {atomic, [_]} -> true;
 480       _ -> false
 481    end.
 482
 483
 484 % Est-ce que Id_user possède Id_mess ?
 485 possede_message(Id_user, Id_mess) ->
 486    case mnesia:transaction(
 487             fun() ->
 488                qlc:e(qlc:q([E#minichat.auteur_id || E <- mnesia:table(minichat), E#minichat.id =:= Id_mess]))
 489             end
 490          ) of
 491       {atomic, [Id_user | []]} -> true;
 492       _ -> false
 493    end.
 494
 495
 496 ip_table() ->
 497    resultat_transaction(mnesia:transaction(
 498       fun() ->
 499          qlc:e(qlc:q([IP || IP <- mnesia:table(ip_table)]))
 500       end
 501    )).
 502
 503
 504 % Bannie une ip pour un certain temps (en minute).
 505 ban(IP, Duration) ->
 506    mnesia:transaction(
 507       fun() ->
 508          case mnesia:wread({ip_table, IP}) of
 509             [IP_tuple] ->
 510                mnesia:write(IP_tuple#ip_table{ban = now(), ban_duration = Duration});
 511             _ ->
 512                mnesia:write(#ip_table{ip = IP, ban = now(), ban_duration = Duration})
 513           end
 514       end
 515    ).
 516
 517
 518 % Débanni une ip
 519 deban(IP) ->
 520    ban(IP, 0).
 521
 522
 523 % Renvoie soit {true, Temps} où Temps est le temps en minutes pendant lequel le user est encore banni
 524 % ou false.
 525 est_banni(User_id) ->
 526    resultat_transaction(mnesia:transaction(
 527       fun() ->
 528          case qlc:e(qlc:q([
 529             {IP#ip_table.ban, IP#ip_table.ban_duration} ||
 530             U <- mnesia:table(user),
 531             U#user.id =:= User_id,
 532             IP <- mnesia:table(ip_table),
 533             IP#ip_table.ip =:= U#user.last_ip
 534          ])) of
 535             [{Ban, Ban_duration}] ->
 536                Echeance = date_plus_minutes(Ban, Ban_duration),
 537                Now = now(),
 538                if Echeance < Now -> % l'échéance est passée
 539                      false;
 540                   true ->
 541                      {true, trunc(delta_date_ms(Echeance, Now) / 1000 / 60)}
 542                end;
 543             _ ->
 544                false
 545          end
 546       end
 547    )).
 548
 549
 550 % Ban est une date tel que retourner par now().
 551 % Ban_duration est un temps en minutes.
 552 % retourne une date.
 553 date_plus_minutes(Ban, Ban_duration) ->
 554    Duration_sec = Ban_duration * 60,
 555    {MegaSec, Sec, MicroSec} = Ban,
 556    {MegaSec + if Sec + Duration_sec >= 1000000 -> 1; true -> 0 end,(Sec + Duration_sec) rem 1000000, MicroSec}.
 557
 558
 559 % Si deux enregistrements consequtifs de la même IP sont fait en moins d'une seconde alors
 560 % ip_table.nb_try_register est incrémenté de 1 sinon il est décrémenté de 1 (jusqu'a 0).
 561 % Si ip_table.nb_try_register vaut 5 alors l'ip ne peux plus s'enregistrer pour une heure.
 562 can_register(IP) ->
 563    resultat_transaction(mnesia:transaction(
 564       fun() ->
 565          case qlc:e(qlc:q([I || I <- mnesia:table(ip_table), I#ip_table.ip =:= IP])) of
 566             [] ->
 567                mnesia:write(#ip_table{ip = IP, date_last_try_register = now()}),
 568                true;
 569             [T] ->
 570                Delta = delta_date_ms(T#ip_table.date_last_try_register, now()),
 571                if T#ip_table.nb_try_register =:= ?NB_MAX_FLOOD_REGISTER, Delta < ?TEMPS_BAN_FLOOD_REGISTER ->
 572                      false;
 573                   true ->
 574                      mnesia:write(#ip_table{
 575                         ip = IP,
 576                         date_last_try_register = now(),
 577                         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
 578                      }),
 579                      true
 580                end
 581          end
 582       end
 583    )).
 584
 585
 586 % Renvoie tous les trolls.
 587 trolls() ->
 588    resultat_transaction(mnesia:transaction(
 589       fun() ->
 590          qlc:e(qlc:q([T || T <- qlc:keysort(2, mnesia:table(troll))]))
 591       end
 592    )).
 593
 594 % Renvoie les trolls manquants posté après Last_id.
 595 % Si pas de trolls alors attend un événement tel qu'un ajout, une modification ou une suppression.
 596 % renvoie :
 597 %  {mod, Troll}
 598 % ou {add, [Trolls]}
 599 % ou {del, Troll_id}
 600 % ou timeout
 601 trolls(Last_id) ->
 602    case mnesia:subscribe({table, troll, detailed}) of
 603       {error, E} = E ->
 604          E;
 605       _ ->
 606          case resultat_transaction(mnesia:transaction(
 607                fun() ->
 608                   qlc:e(qlc:q([T || T <- qlc:keysort(2, mnesia:table(troll)), T#troll.id > Last_id, T#troll.date_post =:= undefined]))
 609                end
 610             )) of
 611                [] -> % pas de trolls
 612                   attend_evenement_troll();
 613                Trolls ->
 614                   {add, Trolls}
 615          end
 616    end.
 617
 618 attend_evenement_troll() ->
 619    % s'il n'y a pas de trolls que l'utilisateur n'a pas connaissance alors on attend un événement
 620    receive
 621       {mnesia_table_event, {write, troll, Troll, [_Old_troll | _], _}} ->
 622          {mod, Troll};
 623       {mnesia_table_event, {write, troll, Troll, [], _}} ->
 624          {add, [Troll]};
 625       {mnesia_table_event, {delete, troll, {troll, Id}, _, _}} ->
 626          {del, Id};
 627       _ ->
 628          attend_evenement_troll()
 629    after 1000 * 60 * 60 ->
 630       timeout
 631    end.
 632
 633
 634  % Renvoie l'id du nouveau troll
 635  % ou max_troll_reached_per_user si le nombre de troll posté par l'utilisateur max a été atteind
 636  % ou max_troll_reached si le nombre de troll posté max a été atteind
 637  % ou user_unknown
 638 put_troll(User_id, Content) ->
 639    resultat_transaction(mnesia:transaction(
 640       fun() ->
 641          % control le nombre de troll déjà posté
 642          Nb_troll_poste_par_user = length(qlc:e(qlc:q(
 643             [
 644                E#troll.id || E <- mnesia:table(troll),
 645                E#troll.id_user =:= User_id,
 646                E#troll.date_post =:= undefined
 647             ]
 648          ))),
 649          Nb_troll_poste_total = length(qlc:e(qlc:q(
 650             [
 651                E#troll.id || E <- mnesia:table(troll),
 652                E#troll.date_post =:= undefined
 653             ]
 654          ))),
 655          User = user_by_id(User_id),
 656          case User of
 657             {ok, _} ->
 658                if Nb_troll_poste_par_user >= ?NB_MAX_TROLL_WAITING_BY_USER ->
 659                      max_troll_reached_per_user;
 660                   Nb_troll_poste_total >= ?NB_MAX_TROLL_WAITING ->
 661                      max_troll_reached;
 662                   true ->
 663                      Id = nouvel_id(minichat),
 664                      mnesia:write(#troll{id = Id, id_user = User_id, date_create = now(), content = Content}),
 665                      Id
 666                end;
 667             _ ->
 668                user_unknown
 669          end
 670       end
 671    )).
 672
 673
 674 % renvoie ok | erreur
 675 mod_troll(Troll_id, Content) ->
 676    mnesia:transaction(
 677       fun() ->
 678          case mnesia:wread({troll, Troll_id}) of
 679             [Troll = #troll{date_post = undefined}] ->
 680                mnesia:write(Troll#troll{content = Content});
 681             _ ->
 682                mnesia:abort("mod_troll: Troll inconnu ou déjà posté")
 683           end
 684       end
 685    ).
 686
 687
 688 del_troll(Troll_id) ->
 689    mnesia:transaction(
 690       fun() ->
 691          case mnesia:wread({troll, Troll_id}) of
 692             [#troll{date_post = undefined}] ->
 693                mnesia:delete({troll, Troll_id});
 694             _ ->
 695                mnesia:abort("mod_troll: Troll inconnu ou déjà posté")
 696           end
 697       end
 698    ).
 699
 700
 701 update_version(1) ->
 702    mnesia:transform_table(
 703       ip_table,
 704       fun() -> null end,
 705       record_info(fields, ip_table),
 706       ip_table
 707    ).
 708
 709
 710 % Renvoie le résultat d'une transaction (en décomposant le tuple fournit)
 711 resultat_transaction({_, T}) ->
 712    T.
 713
 714
 715 % Retourne la difference entre deux timestamp (erlang:now()) en miliseconde
 716 delta_date_ms(D1, D2) ->
 717    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).
 718
 719
 720 % Bizarre, cette fonction n'existe pas dans la stdlib.
 721 % Pas utilisé mais bon ca me fait de la peine de l'enlever.
 722 ceiling(X) ->
 723    T = trunc(X),
 724    case (X - T) of
 725       Neg when Neg < 0 -> T;
 726       Pos when Pos > 0 -> T + 1;
 727       _ -> T
 728    end.
 729
 730
 731 % Renvoie un nouvel id pour une table donnée
 732 nouvel_id(Table) ->
 733    mnesia:dirty_update_counter(counter, Table, 1).
 734