X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=modules%2Ferl%2Feuphorik_bd.erl;h=88759736469e5580b2185af9c8fe9f0798300e53;hb=6f1e9a972293544a6c92a99c9a04a509b2c7648a;hp=9ad6da2caf10c0736d404b2941458601455fc90e;hpb=2233e8f8760b118c3cacf541ef32ea456a8e1527;p=euphorik.git diff --git a/modules/erl/euphorik_bd.erl b/modules/erl/euphorik_bd.erl index 9ad6da2..8875973 100755 --- a/modules/erl/euphorik_bd.erl +++ b/modules/erl/euphorik_bd.erl @@ -4,7 +4,7 @@ % Ce module utilise une base mnesia. % @author G.Burri --module(euphorik_minichat). +-module(euphorik_bd). -export([ % gestion : create/0, @@ -17,6 +17,7 @@ nouveau_user/3, set_profile/9, update_date_derniere_connexion/1, + update_ip/2, update_pseudo_user/2, users/0, user_by_cookie/1, @@ -24,9 +25,11 @@ user_by_login/1, user_by_login_password/2, user_by_mess/1, + toggle_ek_master/1, % messages : nouveau_message/3, + nouveau_message_sys/1, messages/1, messages/2, messages/3, @@ -39,8 +42,24 @@ a_repondu_a_message/2, possede_message/2, - % utiles : + % ip : + ip_table/0, + ban/2, + deban/1, + est_banni/1, can_register/1, + + % trolls : + trolls/0, + trolls/1, + put_troll/2, + mod_troll/2, + del_troll/1, + + % versions : + update_version/1, + + % utiles : resultat_transaction/1 ]). @@ -88,6 +107,7 @@ create_tables() -> ]), mnesia:create_table(troll, [ {attributes, record_info(fields, troll)}, + {index, [date_post]}, {disc_copies, [node()]} ]). @@ -105,10 +125,12 @@ reset() -> mnesia:clear_table(counter), mnesia:clear_table(user), mnesia:clear_table(reponse_minichat), - mnesia:clear_table(minichat), + mnesia:clear_table(minichat), + mnesia:clear_table(troll), + mnesia:clear_table(ip_table), % crée l'utilisateur root mnesia:transaction(fun() -> - User = #user{id = 0, pseudo = "Sys", login = "Sys", date_creation = now(), date_derniere_connexion = now()}, + User = #user{id = 0, pseudo = "Sys", login = "Sys", date_creation = now(), date_derniere_connexion = now(), ek_master = true}, mnesia:write(User), User end). @@ -151,7 +173,7 @@ set_profile(Cookie, Login, Password, Pseudo, Email, Css, Nick_format, Page_princ % TODO : pourquoi ne pas tester avec la valeur "undefined" plutôt qu'avec "is_list" ? % TODO : validation plus strict des données (pas de page négative dans les conv par exemple) login = if is_list(Login) -> Login; true -> User#user.login end, - password = if is_list(Password) -> Password; true -> User#user.password end, + password = if is_list(Password) andalso Password =/= [] -> Password; true -> User#user.password end, pseudo = if is_list(Pseudo) -> Pseudo; true -> User#user.pseudo end, email = if is_list(Email) -> Email; true -> User#user.email end, css = if is_list(Css) -> Css; true -> User#user.css end, @@ -172,10 +194,10 @@ set_profile(Cookie, Login, Password, Pseudo, Email, Css, Nick_format, Page_princ % Met à jour la date de la dernière connexion d'un utilisateur à maintenant -update_date_derniere_connexion(UserId) -> +update_date_derniere_connexion(User_id) -> mnesia:transaction( fun() -> - case mnesia:wread({user, UserId}) of + case mnesia:wread({user, User_id}) of [User] -> mnesia:write(User#user{date_derniere_connexion = now()}); _ -> @@ -183,7 +205,21 @@ update_date_derniere_connexion(UserId) -> end end ). - + + +% Met à jour l'ip d'un user +update_ip(User_id, IP) -> + mnesia:transaction( + fun() -> + case mnesia:wread({user, User_id}) of + [User] -> + mnesia:write(User#user{last_ip = IP}); + _ -> + mnesia:abort("update_ip: User inconnu") + end + end + ). + % Met à jour le pseudo du user update_pseudo_user(UserId, Pseudo) -> @@ -244,6 +280,19 @@ user_by_login(Login) -> )). +toggle_ek_master(User_id) -> + resultat_transaction(mnesia:transaction( + fun() -> + Users = qlc:e(qlc:q([E || E <- mnesia:table(user), E#user.id =:= User_id])), + case Users of + [User] -> + mnesia:write(User#user{ek_master = not User#user.ek_master}); + _ -> erreur + end + end + )). + + user_by_login_password(Login, Password) -> resultat_transaction(mnesia:transaction( fun() -> @@ -267,12 +316,6 @@ user_by_mess(Id) -> end )). - -% Renvoie l'utilisateur root -root() -> - {ok, User} = user_by_id(0), - User. - % Ajoute un message. Repond_A est une liste d'id auquel le message répond % retourne soit l'id du message soit erreur. @@ -288,7 +331,6 @@ nouveau_message(Mess, Auteur_id, Repond_A) -> if Nb_id_trouve =/= length(Repond_A) -> throw("Un ou plusieurs messages introuvable"); true -> ok end, - Id = nouvel_id(minichat), % compare les dernière Delta = delta_date_ms(Auteur#user.date_derniere_connexion, now()), Nouvel_indice_flood = Auteur#user.indice_flood + if Delta =< ?DUREE_SPAM -> 2; true -> -1 end, @@ -298,14 +340,13 @@ nouveau_message(Mess, Auteur_id, Repond_A) -> }, % est-ce que l'auteur à trop floodé ? if Auteur#user.indice_flood =/= ?INDICE_SPAM_MAX, Auteur_maj#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM -> - Root = root(), mnesia:write(Auteur#user{indice_flood = Auteur_maj#user.indice_flood}), - mnesia:write(#minichat{id=Id, auteur_id=Root#user.id, date=now(), pseudo=Root#user.pseudo, contenu=Auteur#user.pseudo ++ "(" ++ Auteur#user.login ++ ") est bloqué pour " ++ integer_to_list(trunc(?DUREE_BLOCAGE_SPAM / 1000)) ++ " secondes pour cause de flood.."}), - Id; + 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."); Auteur#user.indice_flood =:= ?INDICE_SPAM_MAX, Delta =< ?DUREE_BLOCAGE_SPAM -> erreur; true -> mnesia:write(Auteur_maj), + Id = nouvel_id(minichat), inserer_reponses(Id, Repond_A), mnesia:write(#minichat{id=Id, auteur_id=Auteur#user.id, date=now(), pseudo=Auteur#user.pseudo, contenu=Mess}), Id @@ -320,6 +361,18 @@ inserer_reponses(Id_repondant, [Id_mess | Reste]) -> inserer_reponses(_, []) -> ok. + +% Permet de créer un message système. +% Renvoie l'id du message système +nouveau_message_sys(Mess) -> + {ok, Root} = user_by_id(0), + mnesia:transaction( + fun() -> + Id = nouvel_id(minichat), + mnesia:write(#minichat{id=Id, auteur_id=0, date=now(), pseudo=Root#user.pseudo, contenu=Mess}) + end + ). + % Renvoie N messages se trouvant sur la première page messages(N) -> @@ -364,7 +417,6 @@ messages_by_ids(Ids) -> fun() -> % TODO : optimisations ? serait-ce du O(n) ? Query = qlc:q([E || E <- qlc:keysort(2, mnesia:table(minichat), [{order, ascending}]), lists:any(fun(Id) -> Id =:= E#minichat.id end, Ids)]), - %io:format("~s~n", [qlc:info(Query)]), qlc:e(Query) end )). @@ -440,10 +492,73 @@ possede_message(Id_user, Id_mess) -> _ -> false end. + +ip_table() -> + resultat_transaction(mnesia:transaction( + fun() -> + qlc:e(qlc:q([IP || IP <- mnesia:table(ip_table)])) + end + )). + + +% Bannie une ip pour un certain temps (en minute). +ban(IP, Duration) -> + mnesia:transaction( + fun() -> + case mnesia:wread({ip_table, IP}) of + [IP_tuple] -> + mnesia:write(IP_tuple#ip_table{ban = now(), ban_duration = Duration}); + _ -> + mnesia:write(#ip_table{ip = IP, ban = now(), ban_duration = Duration}) + end + end + ). + + +% Débanni une ip +deban(IP) -> + ban(IP, 0). + +% Renvoie soit {true, Temps} où Temps est le temps en minutes pendant lequel le user est encore banni +% ou false. +est_banni(User_id) -> + resultat_transaction(mnesia:transaction( + fun() -> + case qlc:e(qlc:q([ + {IP#ip_table.ban, IP#ip_table.ban_duration} || + U <- mnesia:table(user), + U#user.id =:= User_id, + IP <- mnesia:table(ip_table), + IP#ip_table.ip =:= U#user.last_ip + ])) of + [{Ban, Ban_duration}] -> + Echeance = date_plus_minutes(Ban, Ban_duration), + Now = now(), + if Echeance < Now -> % l'échéance est passée + false; + true -> + {true, trunc(delta_date_ms(Echeance, Now) / 1000 / 60)} + end; + _ -> + false + end + end + )). + + +% Ban est une date tel que retourner par now(). +% Ban_duration est un temps en minutes. +% retourne une date. +date_plus_minutes(Ban, Ban_duration) -> + Duration_sec = Ban_duration * 60, + {MegaSec, Sec, MicroSec} = Ban, + {MegaSec + if Sec + Duration_sec >= 1000000 -> 1; true -> 0 end,(Sec + Duration_sec) rem 1000000, MicroSec}. + + % Si deux enregistrements consequtifs de la même IP sont fait en moins d'une seconde alors % ip_table.nb_try_register est incrémenté de 1 sinon il est décrémenté de 1 (jusqu'a 0). -% Si ip_table.nb_try_register vaut 5 alors l'ip ne peux plus s'enregistrer pour une heure. +% Si ip_table.nb_try_register vaut 5 alors l'ip ne peux plus s'enregistrer pour une heure. can_register(IP) -> resultat_transaction(mnesia:transaction( fun() -> @@ -467,6 +582,130 @@ can_register(IP) -> end )). + +% Renvoie tous les trolls. +trolls() -> + resultat_transaction(mnesia:transaction( + fun() -> + qlc:e(qlc:q([T || T <- qlc:keysort(2, mnesia:table(troll))])) + end + )). + +% Renvoie les trolls manquants posté après Last_id. +% Si pas de trolls alors attend un événement tel qu'un ajout, une modification ou une suppression. +% renvoie : +% {mod, Troll} +% ou {add, [Trolls]} +% ou {del, Troll_id} +% ou timeout +trolls(Last_id) -> + case mnesia:subscribe({table, troll, detailed}) of + {error, E} = E -> + E; + _ -> + case resultat_transaction(mnesia:transaction( + fun() -> + qlc:e(qlc:q([T || T <- qlc:keysort(2, mnesia:table(troll)), T#troll.id > Last_id, T#troll.date_post =:= undefined])) + end + )) of + [] -> % pas de trolls + attend_evenement_troll(); + Trolls -> + {add, Trolls} + end + end. + +attend_evenement_troll() -> + % s'il n'y a pas de trolls que l'utilisateur n'a pas connaissance alors on attend un événement + receive + {mnesia_table_event, {write, troll, Troll, [_Old_troll | _], _}} -> + {mod, Troll}; + {mnesia_table_event, {write, troll, Troll, [], _}} -> + {add, [Troll]}; + {mnesia_table_event, {delete, troll, {troll, Id}, _, _}} -> + {del, Id}; + _ -> + attend_evenement_troll() + after 1000 * 60 * 60 -> + timeout + end. + + + % Renvoie l'id du nouveau troll + % ou max_troll_reached_per_user si le nombre de troll posté par l'utilisateur max a été atteind + % ou max_troll_reached si le nombre de troll posté max a été atteind + % ou user_unknown +put_troll(User_id, Content) -> + resultat_transaction(mnesia:transaction( + fun() -> + % control le nombre de troll déjà posté + Nb_troll_poste_par_user = length(qlc:e(qlc:q( + [ + E#troll.id || E <- mnesia:table(troll), + E#troll.id_user =:= User_id, + E#troll.date_post =:= undefined + ] + ))), + Nb_troll_poste_total = length(qlc:e(qlc:q( + [ + E#troll.id || E <- mnesia:table(troll), + E#troll.date_post =:= undefined + ] + ))), + User = user_by_id(User_id), + case User of + {ok, _} -> + if Nb_troll_poste_par_user >= ?NB_MAX_TROLL_WAITING_BY_USER -> + max_troll_reached_per_user; + Nb_troll_poste_total >= ?NB_MAX_TROLL_WAITING -> + max_troll_reached; + true -> + Id = nouvel_id(minichat), + mnesia:write(#troll{id = Id, id_user = User_id, date_create = now(), content = Content}), + Id + end; + _ -> + user_unknown + end + end + )). + + +% renvoie ok | erreur +mod_troll(Troll_id, Content) -> + mnesia:transaction( + fun() -> + case mnesia:wread({troll, Troll_id}) of + [Troll = #troll{date_post = undefined}] -> + mnesia:write(Troll#troll{content = Content}); + _ -> + mnesia:abort("mod_troll: Troll inconnu ou déjà posté") + end + end + ). + + +del_troll(Troll_id) -> + mnesia:transaction( + fun() -> + case mnesia:wread({troll, Troll_id}) of + [#troll{date_post = undefined}] -> + mnesia:delete({troll, Troll_id}); + _ -> + mnesia:abort("mod_troll: Troll inconnu ou déjà posté") + end + end + ). + + +update_version(1) -> + mnesia:transform_table( + ip_table, + fun() -> null end, + record_info(fields, ip_table), + ip_table + ). + % Renvoie le résultat d'une transaction (en décomposant le tuple fournit) resultat_transaction({_, T}) -> @@ -479,6 +718,7 @@ delta_date_ms(D1, D2) -> % Bizarre, cette fonction n'existe pas dans la stdlib. +% Pas utilisé mais bon ca me fait de la peine de l'enlever. ceiling(X) -> T = trunc(X), case (X - T) of