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