2 % Ce module permet de gérer les données persistantes lié au minichat d'euphorik.ch
3 % Il permet d'ajouter des message, de demande les messages sur une page donnée, etc..
4 % Ce module utilise la base mnesia créée par le module euphorik_bd.
8 -module(euphorik_minichat
).
15 get_messages_depuis_ids
/1,
23 get_user_by_login_password
/2,
25 a_repondu_a_message
/2,
26 est_une_reponse_a_user
/2,
30 update_date_derniere_connexion
/1,
36 resultat_transaction
/1]).
38 -include("../include/euphorik_bd.hrl").
39 -include_lib("stdlib/include/qlc.hrl").
41 % Un message est considéré comme du spam s'il est posté 1 seconde ou moins après le dernier posté
42 -define(DUREE_SPAM
, 1000). % ms
44 % Lorsque l'indice de spam d'un utilisateur atteind cette valeur alors il ne peut plus poster pendant un moment
45 -define(INDICE_SPAM_MAX
, 6).
47 % Un utilisateur ayant trop spamé est bloqué pendant ce temps
48 -define(DUREE_BLOCAGE_SPAM
, 20000). % ms
51 % Connexion à la base de données de yaws sur overnux
54 mnesia:change_config(extra_db_nodes
, [yaws@Plop
]).
57 % Efface tous les users, minichat_reponse et minichat.
59 mnesia:clear_table(counter
),
60 mnesia:clear_table(user
),
61 mnesia:clear_table(reponse_minichat
),
62 mnesia:clear_table(minichat
),
63 % crée l'utilisateur root
64 mnesia:transaction(fun() ->
65 User
= #user
{id
= 0, pseudo
= "Sys", login
= "Sys", date_creation
= now(), date_derniere_connexion
= now()},
71 % Est-ce qu'un utilisateur existe en fonction de son cookie ?
72 % Renvoie {ok, User} ou erreur
73 get_user_by_cookie(Cookie
) ->
74 resultat_transaction(mnesia:transaction(
76 Users
= qlc:e(qlc:q([E
|| E
<- mnesia:table(user
), E#user
.cookie
=:= Cookie
])),
86 resultat_transaction(mnesia:transaction(
88 Users
= qlc:e(qlc:q([E
|| E
<- mnesia:table(user
), E#user
.id
=:= ID
])),
97 get_user_by_login(Login
) ->
98 resultat_transaction(mnesia:transaction(
100 Users
= qlc:e(qlc:q([E
|| E
<- mnesia:table(user
), E#user
.login
=:= Login
])),
102 [User
] -> {ok
, User
};
109 get_user_by_login_password(Login
, Password
) ->
110 resultat_transaction(mnesia:transaction(
112 Users
= qlc:e(qlc:q([E
|| E
<- mnesia:table(user
), E#user
.login
=:= Login
, E#user
.password
=:= Password
])),
114 [User
] -> {ok
, User
};
121 % Renvoie l'utilisateur root
123 {ok
, User
} = get_user_by_id(0),
127 % Est-ce que Id_user possède Id_mess ?
128 possede_message(Id_user
, Id_mess
) ->
129 case mnesia:transaction(
131 qlc:e(qlc:q([E#minichat
.auteur_id
|| E
<- mnesia:table(minichat
), E#minichat
.id
=:= Id_mess
]))
134 {atomic
, [Id_user
| []]} -> true
;
139 % Est-ce que Id_user à répondu au message Id_mess
140 a_repondu_a_message(Id_user
, Id_mess
) ->
141 case mnesia:transaction(
144 M#minichat
.auteur_id
|| M
<- mnesia:table(minichat
), R
<- mnesia:table(reponse_minichat
),
145 R#reponse_minichat
.cible
=:= Id_mess
, R#reponse_minichat
.repondant
=:= M#minichat
.id
, M#minichat
.auteur_id
=:= Id_user
146 ]), [{unique_all
, true
}])
149 {atomic
, [_
]} -> true
;
154 % Est-ce que le message Id_mess est une réponse d'une message de Id_user ?
155 est_une_reponse_a_user(Id_user
, Id_mess
) ->
156 case mnesia:transaction(
159 M#minichat
.auteur_id
|| M
<- mnesia:table(minichat
), R
<- mnesia:table(reponse_minichat
),
160 M#minichat
.auteur_id
=:= Id_user
, M#minichat
.id
=:= R#reponse_minichat
.cible
, R#reponse_minichat
.repondant
=:= Id_mess
161 ]), [{unique_all
, true
}])
164 {atomic
, [_
]} -> true
;
169 % Met à jour le pseudo du user
170 update_pseudo_user(UserId
, Pseudo
) ->
173 case mnesia:wread({user
, UserId
}) of
174 [User
] when User#user
.pseudo
=/= Pseudo
->
175 mnesia:write(User#user
{pseudo
= Pseudo
});
177 mnesia:abort("update_pseudo_user: User inconnu ou pseudo deja à jour")
183 % Mise à par Cookie les autres peuvent être undefined ce qui veut dire qu'ils ne seront pas modifié.
184 set_profile(Cookie
, Login
, Password
, Pseudo
, Email
, Css
) ->
185 resultat_transaction(mnesia:transaction(
187 case get_user_by_cookie(Cookie
) of
189 case get_user_by_login(Login
) of
190 {ok
, U
} when U#user
.id
=/= User#user
.id
->
193 User_modifie
= User#user
{
194 login
= if is_list(Login
) -> Login
; true
-> User#user
.login
end,
195 password
= if is_list(Password
) -> Password
; true
-> User#user
.password
end,
196 pseudo
= if is_list(Pseudo
) -> Pseudo
; true
-> User#user
.pseudo
end,
197 email
= if is_list(Email
) -> Email
; true
-> User#user
.email
end,
198 css
= if is_list(Css
) -> Css
; true
-> User#user
.css
end
200 mnesia:write(User_modifie
),
209 % Met à jour la date de la dernière connexion d'un utilisateur à maintenant
210 update_date_derniere_connexion(UserId
) ->
213 case mnesia:wread({user
, UserId
}) of
215 mnesia:write(User#user
{date_derniere_connexion
= now()});
217 mnesia:abort("update_date_derniere_connexion: User inconnu")
223 % Ajoute un nouveau user et le renvoie
224 nouveau_user(Pseudo
, Cookie
) ->
226 Id
= get_nouvel_id(user
),
227 User
= #user
{id
= Id
, cookie
= Cookie
, pseudo
= Pseudo
, date_creation
= now(), date_derniere_connexion
= now()},
231 resultat_transaction(mnesia:transaction(F
)).
234 % Ajoute un nouveau user et le renvoie
235 nouveau_user(Login
, Password
, Cookie
) ->
237 Id
= get_nouvel_id(user
),
238 User
= #user
{id
= Id
, cookie
= Cookie
, pseudo
= Login
, login
= Login
, password
= Password
, date_creation
= now(), date_derniere_connexion
= now()},
242 resultat_transaction(mnesia:transaction(F
)).
245 % Ajoute un message. Repond_A est une liste d'id auquel le message répond
246 % retourne soit l'id du message soit erreur.
247 nouveau_message(Mess
, Auteur_id
, Repond_A
) ->
248 % regarde si les id 'Repond_A' existent
250 Nb_id_trouve
= length(qlc:e(qlc:q([E#minichat
.id
|| E
<- mnesia:table(minichat
), lists:member(E#minichat
.id
, Repond_A
)]))),
251 % est-ce que l'auteur existe ?
252 Auteur
= case qlc:e(qlc:q([E
|| E
<- mnesia:table(user
), E#user
.id
=:= Auteur_id
])) of
254 _
-> throw("L'auteur du message est introuvable")
256 if Nb_id_trouve
=/= length(Repond_A
) -> throw("Un ou plusieurs messages introuvable");
259 Id
= get_nouvel_id(minichat
),
260 % compare les dernière
261 Delta
= delta_date_ms(Auteur#user
.date_derniere_connexion
, now()),
262 Nouvel_indice_flood
= Auteur#user
.indice_flood
+ if Delta
=< ?DUREE_SPAM
-> 2; true
-> -1 end,
263 Auteur_maj
= Auteur#user
{
264 indice_flood
= if Nouvel_indice_flood
> ?INDICE_SPAM_MAX
-> ?INDICE_SPAM_MAX
; Nouvel_indice_flood
< 0 -> 0; true
-> Nouvel_indice_flood
end,
265 date_derniere_connexion
= now()
267 % est-ce que l'auteur à trop floodé ?
268 if Auteur#user
.indice_flood
=/= ?INDICE_SPAM_MAX
, Auteur_maj#user
.indice_flood
=:= ?INDICE_SPAM_MAX
, Delta
=< ?DUREE_BLOCAGE_SPAM
->
270 mnesia:write(Auteur#user
{indice_flood
= Auteur_maj#user
.indice_flood
}),
271 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.."}),
273 Auteur#user
.indice_flood
=:= ?INDICE_SPAM_MAX
, Delta
=< ?DUREE_BLOCAGE_SPAM
->
276 mnesia:write(Auteur_maj
),
277 inserer_reponses(Id
, Repond_A
),
278 mnesia:write(#minichat
{id
=Id
, auteur_id
=Auteur#user
.id
, date=now(), pseudo
=Auteur#user
.pseudo
, contenu
=Mess
}),
282 resultat_transaction(mnesia:transaction(F
)).
285 % Retourne la difference entre deux timestamp (erlang:now()) en miliseconde
286 delta_date_ms(D1
, D2
) ->
287 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).
290 % Définit Id_repondant comme étant la réponse à Ids. Ids est une liste d'id.
291 inserer_reponses(Id_repondant
, [Id_mess
| Reste
]) ->
292 mnesia:write(#reponse_minichat
{repondant
= Id_repondant
, cible
= Id_mess
}),
293 inserer_reponses(Id_repondant
, Reste
);
294 inserer_reponses(_
, []) ->
298 % Renvoie un nouvel id pour une table donnée
299 get_nouvel_id(Table
) ->
300 mnesia:dirty_update_counter(counter
, Table
, 1).
303 % Renvoie les messages manquants pour la page P en sachant qu'il y a N message
304 % par page et que le dernier message que l'on possède est Id
305 messages(Id
, N
, P
) ->
306 lists:filter(fun (M
) -> M#minichat
.id
> Id
end, messages(N
, P
)).
309 % Renvoie N messages se trouvant sur la page P
312 C
= qlc:cursor(qlc:q([E
|| E
<- qlc:keysort(2, mnesia:table(minichat
), [{order
, descending
}])])),
313 if P
> 1 -> qlc:next_answers(C
, N
* (P
- 1));
316 R
= qlc:next_answers(C
, N
),
317 qlc:delete_cursor(C
),
320 resultat_transaction(mnesia:transaction(F
)).
323 % Renvoie N messages se trouvant sur la première page
328 % Renvoie une liste de message (voir #minichat de euphorik_bd.hrl) à partir d'une liste d'id (Ids).
329 get_messages_depuis_ids(Ids
) ->
330 resultat_transaction(mnesia:transaction(
332 % TODO : optimisations ? serait-ce du O(n) ?
333 Query
= qlc:q([E
|| E
<- qlc:keysort(2, mnesia:table(minichat
), [{order
, ascending
}]), lists:any(fun(Id
) -> Id
=:= E#minichat
.id
end, Ids
)]),
334 io:format("~s~n", [qlc:info(Query
)]),
340 % Renvoie le nombre de page total en fonction du nombre de message par page N
341 % TODO : ya pas un moyen moins pourri pour avoir le nombre de record d'une table ?
343 resultat_transaction(mnesia:transaction(fun() ->
344 ceiling(length(qlc:e(qlc:q([E#minichat
.id
|| E
<- mnesia:table(minichat
)]))) / N
)
348 % Renvoie le numéro de la page sur lequel se trouve le message M en sachant qu'il y a N messages par page.
350 resultat_transaction(mnesia:transaction(fun() ->
351 ceiling((length(qlc:e(qlc:q([E
|| E
<- mnesia:table(minichat
), E#minichat
.id
>= M
])))) / N
)
355 % Bizarre, cette fonction n'existe pas dans la stdlib.
359 Neg
when Neg
< 0 -> T
;
360 Pos
when Pos
> 0 -> T
+ 1;
365 % Renvoie tous les users.
367 resultat_transaction(mnesia:transaction(fun() ->
368 qlc:e(qlc:q([E
|| E
<- mnesia:table(user
)]))
372 % Renvoie les reponses (utilisé normalement uniquement pendant le debug).
375 qlc:e(qlc:q([E
|| E
<- mnesia:table(reponse_minichat
)]))
377 resultat_transaction(mnesia:transaction(F
)).
380 % Renvoie les messages auquel M_id répond.
382 resultat_transaction(mnesia:transaction(
385 [M
|| E
<- mnesia:table(reponse_minichat
),
386 M
<- mnesia:table(minichat
),
387 E#reponse_minichat
.repondant
=:= M_id
,
388 M#minichat
.id
=:= E#reponse_minichat
.cible
]))
393 % Renvoie le résultat d'une transaction (en décomposant le tuple fournit)
394 resultat_transaction({_
, T
}) ->