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