FIX #81 (pas testé)
[euphorik.git] / 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 % texte :
28 get_texte/1,
29 get_texte/2,
30
31 % users :
32 nouveau_user/2,
33 nouveau_user/4,
34 set_profile/4,
35 update_date_derniere_connexion/1,
36 update_ip/2,
37 update_pseudo_user/2,
38 user_by_cookie/1,
39 user_by_id/1,
40 user_by_login/1,
41 user_by_login_password/2,
42 user_by_mess/1,
43 css_from_user_cookie/1,
44 is_ek_master_from_cookie/1,
45
46 % messages :
47 nouveau_message/3,
48 nouveau_message_sys/1,
49 messages/1,
50 messages/2,
51 messages/3,
52 message_by_id/1,
53 messages_by_ids/1,
54 message_existe/1,
55 reponses/0,
56 parents/1,
57 enfants/1,
58 parents_id/1,
59 enfants_id/1,
60 est_une_reponse_a_user/2,
61 a_repondu_a_message/2,
62 possede_message/2,
63
64 % ip :
65 list_ban/0,
66 ban/2,
67 deban/1,
68 est_banni/1,
69 can_register/1,
70
71 % trolls :
72 trolls/0,
73 trolls/1,
74 put_troll/2,
75 mod_troll/2,
76 del_troll/1,
77 troll_by_id/1,
78 current_troll/0,
79 elire_troll/0,
80
81 % utiles :
82 resultat_transaction/1
83 ]).
84 -import(qlc, [e/2, q/1, cursor/2]).
85 -include("../include/euphorik_bd.hrl").
86 -include("../include/euphorik_defines.hrl").
87 -include_lib("stdlib/include/qlc.hrl").
88
89
90 get_texte(Id) ->
91 get_texte(Id, fr).
92
93
94 % TODO : généraliser la langue
95 get_texte(Id, _Lang = fr) ->
96 resultat_transaction(mnesia:transaction(fun() ->
97 case mnesia:read({texte, Id}) of
98 [#texte{fr = Texte}] -> Texte;
99 _ -> "Message " ++ integer_to_list(Id) ++ " unknown"
100 end
101 end)).
102
103
104 % Ajoute un nouveau user et le renvoie
105 nouveau_user(Cookie, Profile) ->
106 F = fun() ->
107 Id = nouvel_id(user),
108 User = #user{id = Id, cookie = Cookie, date_creation = now(), date_derniere_connexion = now(), profile = Profile},
109 mnesia:write(User),
110 User
111 end,
112 resultat_transaction(mnesia:transaction(F)).
113
114
115 % Ajoute un nouveau user et le renvoie
116 nouveau_user(Login, Password, Cookie, Profile) ->
117 F = fun() ->
118 Id = nouvel_id(user),
119 User = #user{id = Id, cookie = Cookie, login = Login, password = Password, date_creation = now(), date_derniere_connexion = now(), profile = Profile#profile{pseudo = Login}},
120 mnesia:write(User),
121 User
122 end,
123 resultat_transaction(mnesia:transaction(F)).
124
125
126 % Définit les données du profile d'une utilisateur.
127 set_profile(Cookie, Login, Password, Profile) ->
128 resultat_transaction(mnesia:transaction(
129 fun() ->
130 case user_by_cookie(Cookie) of
131 {ok, User_existant} ->
132 case user_by_login(Login) of
133 {ok, U} when Login =/= [], U#user.id =/= User_existant#user.id ->
134 login_deja_pris;
135 _ ->
136 mnesia:write(
137 User_existant#user{
138 login = if User_existant#user.login =:= [] -> Login; true -> User_existant#user.login end, % on ne peut pas changer de login sauf si on en a pas !
139 password = if Password =:= [] -> User_existant#user.password; true -> Password end,
140 profile = Profile
141 }
142 ),
143 ok
144 end;
145 _ -> erreur
146 end
147 end
148 )).
149
150
151 % Met à jour la date de la dernière connexion d'un utilisateur à maintenant
152 update_date_derniere_connexion(User_id) ->
153 mnesia:transaction(
154 fun() ->
155 case mnesia:wread({user, User_id}) of
156 [User] ->
157 mnesia:write(User#user{date_derniere_connexion = now()});
158 _ ->
159 mnesia:abort("update_date_derniere_connexion: User inconnu")
160 end
161 end
162 ).
163
164
165 % Met à jour l'ip d'un user
166 update_ip(User_id, IP) ->
167 mnesia:transaction(
168 fun() ->
169 case mnesia:wread({user, User_id}) of
170 [User] ->
171 mnesia:write(User#user{last_ip = IP});
172 _ ->
173 mnesia:abort("update_ip: User inconnu")
174 end
175 end
176 ).
177
178
179 % Met à jour le pseudo du user
180 update_pseudo_user(UserId, Pseudo) ->
181 mnesia:transaction(
182 fun() ->
183 case mnesia:wread({user, UserId}) of
184 [#user{profile = Profile} = User] when Profile#profile.pseudo =/= Pseudo ->
185 mnesia:write(User#user{profile = Profile#profile { pseudo = Pseudo } });
186 _ ->
187 mnesia:abort("update_pseudo_user: User inconnu ou pseudo deja à jour")
188 end
189 end
190 ).
191
192
193 % Est-ce qu'un utilisateur existe en fonction de son cookie ?
194 % Renvoie {ok, User} ou erreur
195 user_by_cookie(Cookie) ->
196 resultat_transaction(mnesia:transaction(
197 fun() ->
198 case mnesia:index_read(user, Cookie, #user.cookie) of
199 [User] -> {ok, User};
200 _ -> erreur
201 end
202 end
203 )).
204
205
206 user_by_id(ID) ->
207 resultat_transaction(mnesia:transaction(
208 fun() ->
209 % case e(q([E || E <- mnesia:table(user), E#user.id =:= ID]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
210 % [User] -> {ok, User};
211 % _ -> erreur
212 % end
213 case mnesia:read({user, ID}) of
214 [User] -> {ok, User};
215 _ -> erreur
216 end
217 end
218 )).
219
220
221 user_by_login(Login) ->
222 resultat_transaction(mnesia:transaction(
223 fun() ->
224 Users = e(q([E || E <- mnesia:table(user), E#user.login =:= Login]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]),
225 case Users of
226 [User] -> {ok, User};
227 _ -> erreur
228 end
229 end
230 )).
231
232
233 % Renvoie une chaine représentant le cookie ou undefined si pas trouvé.
234 css_from_user_cookie(Cookie) ->
235 case user_by_cookie(Cookie) of
236 {ok, #user{profile = Profile}} ->
237 Profile#profile.css;
238 _ ->
239 undefined
240 end.
241
242
243 is_ek_master_from_cookie(Cookie) ->
244 case user_by_cookie(Cookie) of
245 {ok, #user{ek_master = true}} -> true;
246 _ -> false
247 end.
248
249
250 user_by_login_password(Login, Password) ->
251 resultat_transaction(mnesia:transaction(
252 fun() ->
253 case e(q([E || E <- mnesia:table(user), E#user.login =:= Login, E#user.password =:= Password]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
254 [User | _] -> {ok, User};
255 _ -> erreur
256 end
257 end
258 )).
259
260
261 % Renvoie {ok, User} où User est un #user possédant le message donné.
262 user_by_mess(Id) ->
263 resultat_transaction(mnesia:transaction(
264 fun() ->
265 case e(q([U || U <- mnesia:table(user), M <- mnesia:table(minichat), M#minichat.id =:= Id, M#minichat.auteur_id =:= U#user.id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
266 [User | _] -> {ok, User};
267 _ -> erreur
268 end
269 end
270 )).
271
272
273 % Ajoute un message. Repond_A est une liste d'id auquel le message répond
274 % retourne soit l'id du message soit {erreur, <raison>}.
275 nouveau_message(Mess, Auteur_id, Repond_A_ids) ->
276 % regarde si les id 'Repond_A' existent
277 F = fun() ->
278 Repond_a = lists:map(
279 fun(Repond_a_id) ->
280 [M] = mnesia:read({minichat, Repond_a_id}),
281 M
282 end,
283 Repond_A_ids
284 ),
285 Racine_id = case Repond_a of
286 [] -> undefined;
287 [M | _] ->
288 Une_racine = M#minichat.racine_id,
289 % vérification que tout les messages de Repond_a possède la même racine (même conversation)
290 case lists:all(fun(R) -> R#minichat.racine_id =:= Une_racine end, Repond_a) of
291 true ->
292 Une_racine;
293 _ ->
294 {erreur, "Les messages ne font pas partie de la même conversation"}
295 end
296 end,
297 case Racine_id of
298 {erreur, E} -> {erreur, E};
299 _ ->
300 % est-ce que l'auteur existe ?
301 case mnesia:wread({user, Auteur_id}) of
302 [#user{profile = Profile} = Auteur] ->
303 if length(Repond_a) =/= length(Repond_A_ids) ->
304 {erreur, "Un ou plusieurs messages introuvable"};
305 true ->
306 % comparaison entre la date du dernier poste et maintenant (gestion du flood)
307 Now = now(),
308 Delta = euphorik_common:delta_date_ms(Auteur#user.date_derniere_connexion, Now),
309 Nouvel_indice_flood = Auteur#user.indice_flood + if Delta =< ?DUREE_SPAM -> 2; true -> -1 end,
310 Auteur_maj = Auteur#user{
311 indice_flood = if Nouvel_indice_flood > ?INDICE_SPAM_MAX -> ?INDICE_SPAM_MAX; Nouvel_indice_flood < 0 -> 0; true -> Nouvel_indice_flood end,
312 date_derniere_connexion = Now
313 },
314 % est-ce que l'auteur à trop floodé ?
315 if Auteur#user.indice_flood =/= ?INDICE_SPAM_MAX, Auteur_maj#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->
316 mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}),
317 nouveau_message_sys("\"" ++ Profile#profile.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.");
318 Auteur#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM ->
319 {erreur, "Bloqué pour cause de flood"};
320 true ->
321 mnesia:write(Auteur_maj),
322 Id = nouvel_id(minichat),
323 inserer_reponses(Id, Repond_A_ids),
324 mnesia:write(#minichat{
325 id = Id,
326 auteur_id = Auteur#user.id,
327 date = Now,
328 pseudo = Profile#profile.pseudo,
329 contenu = Mess,
330 racine_id = if Racine_id =:= undefined -> Id; true -> Racine_id end
331 }),
332 Id
333 end
334 end;
335 _ ->
336 {erreur, "L'auteur du message est introuvable"}
337 end
338 end
339 end,
340 resultat_transaction(mnesia:transaction(F)).
341
342 % Définit Id_repondant comme étant la réponse à Ids. Ids est une liste d'id.
343 inserer_reponses(Id_repondant, [Id_mess | Reste]) ->
344 mnesia:write(#reponse_minichat{repondant = Id_repondant, cible = Id_mess}),
345 inserer_reponses(Id_repondant, Reste);
346 inserer_reponses(_, []) ->
347 ok.
348
349
350 % Permet de créer un message système.
351 % Renvoie l'id du message système
352 nouveau_message_sys(Mess) ->
353 {ok, #user{profile = Profile}} = user_by_id(0),
354 resultat_transaction(mnesia:transaction(
355 fun() ->
356 Id = nouvel_id(minichat),
357 mnesia:write(#minichat{id = Id, auteur_id = 0, date = now(), pseudo = Profile#profile.pseudo, contenu = Mess, racine_id = Id}),
358 Id
359 end
360 )).
361
362
363 % Renvoie N messages se trouvant sur la première page
364 messages(N) ->
365 messages(N, 1).
366
367
368 % Renvoie N messages se trouvant sur la page P
369 messages(N, P) ->
370 F = fun() ->
371 C = cursor(
372 qlc:keysort(
373 #minichat.id,
374 q([E#minichat{contenu = contenu_message(E)} || E <- mnesia:table(minichat)]),
375 [{order, descending}]
376 ),
377 [{tmpdir, ?KEY_SORT_TEMP_DIR}]
378 ),
379 if P > 1 -> qlc:next_answers(C, N * (P - 1));
380 true -> ok
381 end,
382 R = qlc:next_answers(C, N),
383 qlc:delete_cursor(C),
384 R
385 end,
386 lists:reverse(resultat_transaction(mnesia:transaction(F))).
387
388
389 % Renvoie les messages manquants pour la page P en sachant qu'il y a N message
390 % par page et que le dernier message que l'on possède est Id
391 messages(Id, N, P) ->
392 lists:filter(fun (M) -> M#minichat.id > Id end, messages(N, P)).
393
394
395 % Renvoie {ok, #minichat} (voir #minichat de euphorik_bd.hrl) à partir de son id.
396 message_by_id(Id) ->
397 resultat_transaction(mnesia:transaction(
398 fun() ->
399 case mnesia:read({minichat, Id}) of
400 [M] -> {ok, M#minichat{contenu = contenu_message(M)}};
401 _ -> erreur
402 end
403 end
404 )).
405
406
407 % Renvoie le contenu d'un message donnée en fonction du troll associé, à utiliser à l'intérieur d'une transaction.
408 % TODO : Cette fonction pourrait être remplacée par un "outer-join", est-ce possible avec qlc ?
409 contenu_message(E) ->
410 case mnesia:index_read(troll, E#minichat.id, #troll.id_minichat) of
411 [] -> E#minichat.contenu;
412 [T] -> E#minichat.contenu ++ T#troll.content
413 end.
414
415
416 % Renvoie une liste de message (voir #minichat de euphorik_bd.hrl) à partir d'une liste d'id (Ids).
417 % TODO : optimisations ? serait-ce du O(n) ?
418 % Bon de toutes façons on s'en fout c'est pas utilisé :)
419 messages_by_ids(Ids) ->
420 resultat_transaction(mnesia:transaction(
421 fun() ->
422 e(qlc:keysort(
423 #minichat.id,
424 q([E || E <- mnesia:table(minichat), lists:any(fun(Id) -> Id =:= E#minichat.id end, Ids)]),
425 [{order, ascending}]
426 ),[{tmpdir, ?KEY_SORT_TEMP_DIR}])
427 end
428 )).
429
430
431 % Est-ce qu'un message existe ? Renvoie un boolean.
432 % TODO : ya pas plus simple ?
433 message_existe(Id) ->
434 resultat_transaction(mnesia:transaction(fun() ->
435 length(e(q([E#minichat.id || E <- mnesia:table(minichat), E#minichat.id =:= Id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])) =:= 1
436 end)).
437
438
439 % Renvoie les reponses (utilisé normalement uniquement pendant le debug).
440 reponses() ->
441 F = fun() ->
442 e(q([E || E <- mnesia:table(reponse_minichat)]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
443 end,
444 resultat_transaction(mnesia:transaction(F)).
445
446
447 % Renvoie les messages auquel M_id répond.
448 parents(M_id) ->
449 resultat_transaction(mnesia:transaction(
450 fun() ->
451 e(q(
452 [M || R <- mnesia:table(reponse_minichat),
453 M <- mnesia:table(minichat),
454 R#reponse_minichat.repondant =:= M_id,
455 M#minichat.id =:= R#reponse_minichat.cible]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
456 end
457 )).
458
459
460 % Renvoie les message qui repondent à M_id
461 enfants(M_id) ->
462 resultat_transaction(mnesia:transaction(
463 fun() ->
464 e(q(
465 [M || R <- mnesia:table(reponse_minichat),
466 M <- mnesia:table(minichat),
467 R#reponse_minichat.cible =:= M_id,
468 M#minichat.id =:= R#reponse_minichat.repondant]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
469 end
470 )).
471
472
473 % Renvoie les id des parents d'un message M (les messages auquels répond M)
474 % ordrés du plus petit au plus grand..
475 % On évite d'utiliser qlc pour des raisons de performance
476 % @spec parents_id(integer()) -> [integer()]
477 parents_id(M_id) ->
478 resultat_transaction(mnesia:transaction(fun() ->
479 case mnesia:read({reponse_minichat, M_id}) of
480 Parents when is_list(Parents) ->
481 lists:sort(lists:map(
482 fun(#reponse_minichat{cible = Cible}) -> Cible end,
483 Parents
484 ));
485 _ -> []
486 end
487 end)).
488
489
490 % Renvoie les id des enfants d'un message M (les messages qui répondent à M)
491 % ordrés du plus petit au plus grand.
492 % @spec enfants_id(integer()) -> [integer()]
493 enfants_id(M_id) ->
494 resultat_transaction(mnesia:transaction(fun() ->
495 case mnesia:index_read(reponse_minichat, M_id, #reponse_minichat.cible) of
496 Enfants when is_list(Enfants) ->
497 lists:sort(lists:map(
498 fun(#reponse_minichat{repondant = Repondant}) -> Repondant end,
499 Enfants
500 ));
501 _ -> []
502 end
503 end)).
504
505
506 % Est-ce que le message Id_mess est une réponse d'une message de Id_user ?
507 % On evite d'utiliser qlc (ce qui était fait avant) pour des raisons de performance.
508 est_une_reponse_a_user(Id_user, Id_mess) ->
509 resultat_transaction(mnesia:transaction(
510 fun() ->
511 case mnesia:read({reponse_minichat, Id_mess}) of
512 [] -> false;
513 Cibles ->
514 lists:any(
515 fun(#reponse_minichat{cible = Cible}) ->
516 case mnesia:read({minichat, Cible}) of
517 [#minichat{auteur_id = Id_user}] -> true;
518 _ -> false
519 end
520 end,
521 Cibles
522 )
523 end
524 end
525 )).
526
527
528 % Est-ce que Id_user à répondu au message Id_mess
529 % On evite d'utiliser qlc (ce qui était fait avant) pour des raisons de performance.
530 a_repondu_a_message(Id_user, Id_mess) ->
531 resultat_transaction(mnesia:transaction(
532 fun() ->
533 case mnesia:index_read(reponse_minichat, Id_mess, #reponse_minichat.cible) of
534 [] -> false;
535 Reponses ->
536 lists:any(
537 fun(#reponse_minichat{repondant = Reponse}) ->
538 case mnesia:read({minichat, Reponse}) of
539 [#minichat{auteur_id = Id_user}] -> true;
540 _ -> false
541 end
542 end,
543 Reponses
544 )
545 end
546 end
547 )).
548
549
550 % Est-ce que Id_user possède Id_mess ?
551 possede_message(Id_user, Id_mess) ->
552 case mnesia:transaction(
553 fun() ->
554 e(q([E#minichat.auteur_id || E <- mnesia:table(minichat), E#minichat.id =:= Id_mess]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
555 end
556 ) of
557 {atomic, [Id_user | []]} -> true;
558 _ -> false
559 end.
560
561
562 % renvoie la liste des ip bannies
563 % liste de {ip, temps_restant(en minutes), Users} ou Users est une liste de {pseudo, login}
564 % TODO : déterminer la complexité de cette fonction. (Il n'y a pas d'index sur #user.last_ip)
565 list_ban() ->
566 resultat_transaction(mnesia:transaction(
567 fun() ->
568 Now = now(),
569 e(qlc:keysort(1, q([
570 {
571 IP#ip_table.ip,
572 euphorik_common:delta_date_minute(date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration), Now),
573 e(q([{Profile#profile.pseudo, U#user.login} || #user{profile = Profile} = U <- mnesia:table(user), U#user.last_ip =:= IP#ip_table.ip]), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
574 } ||
575 IP <- mnesia:table(ip_table),
576 if IP#ip_table.ban =:= undefined -> false; true -> date_plus_minutes(IP#ip_table.ban, IP#ip_table.ban_duration) > Now end
577 ])), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
578 end
579 )).
580
581
582 % Bannie une ip pour un certain temps (en minute).
583 ban(IP, Duration) ->
584 mnesia:transaction(
585 fun() ->
586 case mnesia:wread({ip_table, IP}) of
587 [IP_tuple] ->
588 mnesia:write(IP_tuple#ip_table{ban = now(), ban_duration = Duration});
589 _ ->
590 mnesia:write(#ip_table{ip = IP, ban = now(), ban_duration = Duration})
591 end
592 end
593 ).
594
595
596 % Débanni une ip
597 deban(IP) ->
598 ban(IP, 0).
599
600
601 % Renvoie soit {true, Temps} où Temps est le temps en minutes pendant lequel le user est encore banni
602 % ou false.
603 est_banni(User_id) ->
604 resultat_transaction(mnesia:transaction(
605 fun() ->
606 case e(q([
607 {IP#ip_table.ban, IP#ip_table.ban_duration} ||
608 U <- mnesia:table(user),
609 U#user.id =:= User_id,
610 IP <- mnesia:table(ip_table),
611 IP#ip_table.ip =:= U#user.last_ip
612 ]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
613 [{Ban, Ban_duration}] ->
614 Echeance = date_plus_minutes(Ban, Ban_duration),
615 Now = now(),
616 if Echeance < Now -> % l'échéance est passée
617 false;
618 true ->
619 {true, euphorik_common:delta_date_minute(Echeance, Now)}
620 end;
621 _ ->
622 false
623 end
624 end
625 )).
626
627
628 % Ban est une date tel que retourner par now().
629 % Ban_duration est un temps en minutes.
630 % retourne une date.
631 date_plus_minutes(Ban, Ban_duration) ->
632 Duration_sec = Ban_duration * 60,
633 {MegaSec, Sec, MicroSec} = Ban,
634 {MegaSec + if Sec + Duration_sec >= 1000000 -> 1; true -> 0 end,(Sec + Duration_sec) rem 1000000, MicroSec}.
635
636
637 % Si deux enregistrements consequtifs de la même IP sont fait en moins d'une seconde alors
638 % ip_table.nb_try_register est incrémenté de 1 sinon il est décrémenté de 1 (jusqu'a 0).
639 % Si ip_table.nb_try_register vaut 5 alors l'ip ne peux plus s'enregistrer pour une heure.
640 can_register(IP) ->
641 resultat_transaction(mnesia:transaction(
642 fun() ->
643 case e(q([I || I <- mnesia:table(ip_table), I#ip_table.ip =:= IP]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
644 [] ->
645 mnesia:write(#ip_table{ip = IP, date_last_try_register = now()}),
646 true;
647 [T] ->
648 Delta = euphorik_common:delta_date_ms(T#ip_table.date_last_try_register, now()),
649 if T#ip_table.nb_try_register =:= ?NB_MAX_FLOOD_REGISTER, Delta < ?TEMPS_BAN_FLOOD_REGISTER ->
650 false;
651 true ->
652 mnesia:write(T#ip_table{
653 ip = IP,
654 date_last_try_register = now(),
655 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
656 }),
657 true
658 end
659 end
660 end
661 )).
662
663
664 % Renvoie tous les trolls
665 trolls() ->
666 resultat_transaction(mnesia:transaction(
667 fun() ->
668 e(qlc:keysort(#troll.id, q([T || T <- mnesia:table(troll)])), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
669 end
670 )).
671
672
673 % Renvoie les trolls manquants posté après Last_id.
674 trolls(Last_id) ->
675 resultat_transaction(mnesia:transaction(
676 fun() ->
677 e(qlc:keysort(#troll.id, q([T || T <- mnesia:table(troll), T#troll.id > Last_id, T#troll.date_post =:= undefined])), [{tmpdir, ?KEY_SORT_TEMP_DIR}])
678 end
679 )).
680
681
682 % Crée un nouveau troll.
683 % Renvoie l'id du nouveau troll
684 % ou max_troll_reached_per_user si le nombre de troll posté par l'utilisateur max a été atteind
685 % ou max_troll_reached si le nombre de troll posté max a été atteind
686 % ou user_unknown
687 put_troll(User_id, Content) ->
688 resultat_transaction(mnesia:transaction(
689 fun() ->
690 % control le nombre de troll déjà posté
691 Nb_troll_poste_par_user = length(e(q(
692 [
693 E#troll.id || E <- mnesia:table(troll),
694 E#troll.id_user =:= User_id,
695 E#troll.date_post =:= undefined
696 ]
697 ), [{tmpdir, ?KEY_SORT_TEMP_DIR}])),
698 Nb_troll_poste_total = length(e(q(
699 [
700 E#troll.id || E <- mnesia:table(troll),
701 E#troll.date_post =:= undefined
702 ]
703 ), [{tmpdir, ?KEY_SORT_TEMP_DIR}])),
704 User = user_by_id(User_id),
705 case User of
706 {ok, _} ->
707 if Nb_troll_poste_par_user >= ?NB_MAX_TROLL_WAITING_BY_USER ->
708 max_troll_reached_per_user;
709 Nb_troll_poste_total >= ?NB_MAX_TROLL_WAITING ->
710 max_troll_reached;
711 true ->
712 Id = nouvel_id(minichat),
713 mnesia:write(#troll{id = Id, id_user = User_id, date_create = now(), content = Content}),
714 Id
715 end;
716 _ ->
717 user_unknown
718 end
719 end
720 )).
721
722
723 % renvoie ok | erreur
724 mod_troll(Troll_id, Content) ->
725 mnesia:transaction(
726 fun() ->
727 case mnesia:wread({troll, Troll_id}) of
728 [Troll = #troll{date_post = undefined}] ->
729 mnesia:write(Troll#troll{content = Content});
730 _ ->
731 mnesia:abort("mod_troll: Troll inconnu ou déjà posté")
732 end
733 end
734 ).
735
736
737 del_troll(Troll_id) ->
738 mnesia:transaction(
739 fun() ->
740 case mnesia:wread({troll, Troll_id}) of
741 [#troll{date_post = undefined}] ->
742 mnesia:delete({troll, Troll_id});
743 _ ->
744 mnesia:abort("mod_troll: Troll inconnu ou déjà posté")
745 end
746 end
747 ).
748
749
750 troll_by_id(Troll_id) ->
751 resultat_transaction(mnesia:transaction(
752 fun() ->
753 case e(q([T || T <- mnesia:table(troll), T#troll.id =:= Troll_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
754 [T] -> {ok, T};
755 _ ->
756 erreur
757 end
758 end
759 )).
760
761
762 % Renvoie le troll actuel qui se trouve sur la page principale.
763 % Renvois aucun si pas de troll courant.
764 current_troll() ->
765 resultat_transaction(mnesia:transaction(
766 fun() ->
767 C = cursor(qlc:keysort(#troll.date_post, q([T || T <- mnesia:table(troll), T#troll.date_post =/= undefined]), [{order, descending}]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]),
768 R = case qlc:next_answers(C, 1) of
769 [T] -> T;
770 _ -> aucun
771 end,
772 qlc:delete_cursor(C),
773 R
774 end
775 )).
776
777
778 % Elit un troll au hasard parmis les trolls en attente (leur date_post =:= undefined)
779 % Un message est posté par 'Sys' et le troll elu est lié à ce message
780 % met à jour sa date de post.
781 % renvoie plus_de_trolls si il n'y a aucun troll en attente.
782 elire_troll() ->
783 {A1,A2,A3} = now(),
784 random:seed(A1, A2, A3),
785 mnesia:transaction(
786 fun() ->
787 case e(q([T || T <- mnesia:table(troll), T#troll.date_post =:= undefined]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
788 [] ->
789 plus_de_trolls;
790 Trolls ->
791 Troll = lists:nth(random:uniform(length(Trolls)), Trolls),
792 Id_message = nouveau_message_sys("Troll de la semaine : "),
793 Troll2 = Troll#troll{date_post = now(), id_minichat = Id_message},
794 mnesia:write(Troll2)
795 end
796 end
797 ).
798
799
800 % Renvoie le résultat d'une transaction (en décomposant le tuple fournit)
801 resultat_transaction({_, T}) ->
802 T.
803
804
805 % Renvoie un nouvel id pour une table donnée
806 nouvel_id(Table) ->
807 mnesia:dirty_update_counter(counter, Table, 1).
808