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