ADD Bannissement (sans le slap)
[euphorik.git] / modules / erl / euphorik_protocole.erl
index 1f5faf3..dc29bf5 100755 (executable)
@@ -6,11 +6,12 @@
 -module(euphorik_protocole).
 -export([
    register/2,
-   login/1,
+   login/2,
    logout/1,
    profile/1,
    wait_event/1,
    put_message/1,
+   ban/1,
    erreur/1
 ]).\r
 
@@ -28,6 +29,7 @@ register([{login, Login}, {password, Password}], IP) ->
                erreur("Login déjà existant");
             _ ->
                User = euphorik_bd:nouveau_user(Login, Password, generer_cookie()),
+               euphorik_bd:update_ip(User#user.id, IP),
                json_reponse_login_ok(User)
          end;
       true ->
@@ -38,6 +40,7 @@ register([], IP) ->
    Can_register = euphorik_bd:can_register(IP),
    if Can_register ->
          User = euphorik_bd:nouveau_user("<nick>", generer_cookie()),
+         euphorik_bd:update_ip(User#user.id, IP),
          json_reponse_login_ok(User);
       true ->
          erreur_register_flood()
@@ -48,16 +51,17 @@ erreur_register_flood() ->
    
 \r
 % Un utilisateur se logge (avec un couple {login, mot de passe})
-login([{login, Login}, {password, Password}]) ->
-   loginUser(euphorik_bd:user_by_login_password(Login, Password));
+login([{login, Login}, {password, Password}], IP) ->
+   loginUser(euphorik_bd:user_by_login_password(Login, Password), IP);
 % Un utilisateur se logge (avec un cookie)
-login([{cookie, Cookie}]) ->
-   loginUser(euphorik_bd:user_by_cookie(Cookie)).
+login([{cookie, Cookie}], IP) ->
+   loginUser(euphorik_bd:user_by_cookie(Cookie), IP).
    
-loginUser({ok, User}) ->
+loginUser({ok, User}, IP) ->
+   euphorik_bd:update_ip(User#user.id, IP),
    euphorik_bd:update_date_derniere_connexion(User#user.id),
    json_reponse_login_ok(User);
-loginUser(_) ->
+loginUser(_, _) ->
    % ajoute un délais d'attente (TODO : un autre moyen plus élégant ?)
    receive after 1000 ->
       erreur("Erreur login")
@@ -146,32 +150,28 @@ wait_event(Data) ->
                            Est_proprietaire = User =/= inconnu andalso User#user.id =:= Mess#minichat.auteur_id,
                            A_repondu_a_message = User =/= inconnu andalso euphorik_bd:a_repondu_a_message(User#user.id, Mess#minichat.id),
                            Est_une_reponse_a_user = User =/= inconnu andalso euphorik_bd:est_une_reponse_a_user(User#user.id, Mess#minichat.id),
-                           User_mess =
-                              if Mess#minichat.auteur_id =:= 0 ->
-                                    inconnu;
-                                 true ->
-                                    {ok, U2} = euphorik_bd:user_by_id(Mess#minichat.auteur_id),
-                                    U2
-                              end,
-                              {struct, [
-                                 {id, Mess#minichat.id},
-                                 {date, format_date(Mess#minichat.date)},
-                                 {system, Mess#minichat.auteur_id =:= 0},
-                                 {owner, Est_proprietaire},
-                                 {answered, A_repondu_a_message},
-                                 {is_a_reply, Est_une_reponse_a_user},
-                                 {nick, Mess#minichat.pseudo},
-                                 {login, if User_mess =:= inconnu -> Mess#minichat.pseudo; true -> User_mess#user.login end},
-                                 {content, Mess#minichat.contenu},
-                                 {answer_to, {array, lists:map(
-                                    fun(Id_mess) ->                   
-                                       {ok, M} = euphorik_bd:message_by_id(Id_mess),
-                                       {ok, User_reponse} = euphorik_bd:user_by_mess(M#minichat.id),
-                                       {struct, [{id, M#minichat.id}, {nick, M#minichat.pseudo}, {login, User_reponse#user.login}]}
-                                    end,
-                                    Repond_a
-                                 )}}
-                              ]}
+                           {ok, User_mess } = euphorik_bd:user_by_id(Mess#minichat.auteur_id),
+                           {struct, [
+                              {id, Mess#minichat.id},
+                              {user_id, User_mess#user.id},
+                              {date, format_date(Mess#minichat.date)},
+                              {system, Mess#minichat.auteur_id =:= 0},
+                              {owner, Est_proprietaire},
+                              {answered, A_repondu_a_message},
+                              {is_a_reply, Est_une_reponse_a_user},
+                              {nick, Mess#minichat.pseudo},
+                              {login, User_mess#user.login},
+                              {content, Mess#minichat.contenu},
+                              {answer_to, {array, lists:map(
+                                 fun(Id_mess) ->                   
+                                    {ok, M} = euphorik_bd:message_by_id(Id_mess),
+                                    {ok, User_reponse} = euphorik_bd:user_by_mess(M#minichat.id),
+                                    {struct, [{id, M#minichat.id}, {nick, M#minichat.pseudo}, {login, User_reponse#user.login}]}
+                                 end,
+                                 Repond_a
+                              )}},
+                              {ek_master, User_mess#user.ek_master}
+                           ]}
                         end,
                         Conv
                      )
@@ -200,23 +200,73 @@ put_message(
 ) ->
    case euphorik_bd:user_by_cookie(Cookie) of
       {ok, User} ->
-         Strip_content = string:strip(Content),
-         if (Strip_content =:= []) ->
-               erreur("Message vide");
-            true ->
-               % TODO : non-atomique (update_pseudo+nouveau_message)
-               euphorik_bd:update_pseudo_user(User#user.id, Nick),
-               case euphorik_bd:nouveau_message(Strip_content, User#user.id, Answer_to) of
-                     erreur -> erreur("Impossible d'ajouter un nouveau message");
-                  _ ->
-                     json_reponse_ok()
+         case euphorik_bd:est_banni(User#user.id) of
+            {true, Temps_restant} ->
+               erreur("Vous êtes banni pour encore " ++ format_minutes(Temps_restant));
+            _ ->
+               Strip_content = string:strip(Content),
+               if Strip_content =:= [] ->
+                     erreur("Message vide");
+                  true ->
+                     % TODO : non-atomique (update_pseudo+nouveau_message)
+                     euphorik_bd:update_pseudo_user(User#user.id, Nick),
+                     case euphorik_bd:nouveau_message(Strip_content, User#user.id, Answer_to) of
+                           erreur -> erreur("Impossible d'ajouter un nouveau message");
+                        _ ->
+                           json_reponse_ok()
+                     end
                end
          end;
-      _ ->
-         erreur("Utilisateur inconnu")
-      end.
+   _ ->
+      erreur("Utilisateur inconnu")
+   end.
+   
+   
+% Formatage de minutes.
+% par exemple : "1min", "45min", "1h23min", "1jour 2h34min"
+format_minutes(Min) ->
+   Jours = Min div (60 * 24),
+   Heures = Min rem (60 * 24) div 60,
+   Minutes = Min rem (60),
+   if Jours =/= 0 -> integer_to_list(Jours) ++ "Jour" ++ if Jours > 1 -> "s"; true -> "" end ++ " "; true -> "" end ++
+   if Heures =/= 0 -> integer_to_list(Heures) ++ "h"; true -> "" end ++
+   lists:flatten(io_lib:format(if Jours =:= 0, Heures =:= 0 -> "~w"; true -> "~2.2.0w" end, [Minutes])) ++ "min".
 
 
+% bannissement d'un utilisateur (son ip est bannie)
+ban(
+   [
+      {cookie, Cookie},
+      {duration, Duration},
+      {user_id, User_id}
+   ]) ->
+      % controle que l'utilisateur est un admin
+      case euphorik_bd:user_by_cookie(Cookie) of
+         {ok, User1 = #user{ek_master = true}} ->
+            case euphorik_bd:user_by_id(User_id) of
+               {ok, User1} ->
+                  erreur("Il n'est pas possible de s'auto bannir");
+               {ok, User2 = #user{ek_master = false}} ->
+                  euphorik_bd:ban(User2#user.last_ip, Duration),
+                  euphorik_bd:nouveau_message_sys(lists:flatten(io_lib:format("~s ~s est ~s pour ~s",
+                     [
+                        User2#user.pseudo,
+                        if User2#user.login =:= [] -> ""; true -> "(" ++ User2#user.login ++ ")" end,
+                        if Duration =< 15 -> "kické"; true -> "banni" end,
+                        format_minutes(Duration)
+                     ]
+                  ))),
+                  json_reponse_ok();
+               {ok, _} ->
+                  erreur("L'utilisateur est lui même un ekMaster");
+               _ ->
+                  erreur("Utilisateur à bannir inconnu")
+            end;
+         _ ->
+            erreur("Utilisateur inconnu ou non ek master")
+      end.
+      
+
 % Construit une erreur
 erreur(Message) ->
    {
@@ -247,7 +297,7 @@ format_date(Date) ->
       end ++\r
       io_lib:format("~2.10.0B:~2.10.0B:~2.10.0B", [Heure, Minute, Seconde])
    ).
-   
+
 
 json_reponse_ok() ->
    {struct, [{reply, "ok"}]}.
@@ -280,6 +330,7 @@ json_reponse_login_ok(User) ->
                   User#user.conversations
                )
             }
-         }
+         },
+      {ek_master, User#user.ek_master}
       ]
    }.