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