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