% coding: utf-8
% Copyright 2008 Grégory Burri
%
% This file is part of Euphorik.
%
% Euphorik is free software: you can redistribute it and/or modify
% it under the terms of the GNU General Public License as published by
% the Free Software Foundation, either version 3 of the License, or
% (at your option) any later version.
%
% Euphorik is distributed in the hope that it will be useful,
% but WITHOUT ANY WARRANTY; without even the implied warranty of
% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
% GNU General Public License for more details.
%
% You should have received a copy of the GNU General Public License
% along with Euphorik.  If not, see <http://www.gnu.org/licenses/>.
% 
% Module mettant à disposition tout un tas de fonction pour l'administration de la base de données euphorik comme :
%  - Création de la BD
%  - Mise à jour de la BD
%  - Backup et restauration
%  - Copie de la BD à partir d'un autre noeud
%
% @author G.Burri

-module(euphorik_bd_admin).
-export([
   version_bd/0,
   create/0,
   connect/0,
   connect/1,
   reset/0,
   update/0,
   
   backup_text/1,
   restore_text/1,
   
   toggle_ek_master/1,
   print_users/0,
   print_users/1,
   print_user/1
]).
-import(qlc, [e/2, q/1, cursor/2]).
-include("../include/euphorik_bd.hrl").
-include("../include/euphorik_defines.hrl").
-include_lib("stdlib/include/qlc.hrl").


% Renvoie la version courante de la BD.
version_bd() ->
   euphorik_bd:resultat_transaction(mnesia:transaction(
      fun() ->
         mnesia:read({proprietes, version})
      end
   )).
   

% Instructions pour créer une nouvelle base : 
% $erl -sname yaws -mnesia dir '"/projets/euphorik/BD"'
% voir doc/installation.txt
% >l(euphorik_bd).
% >euphorik_bd:create().
create() ->
   mnesia:stop(),
   mnesia:delete_schema([node()]),
   mnesia:create_schema([node()]), % nécessaire pour les tables sur disc
   mnesia:start(),
   create_tables(),
   reset().
   
create_tables() ->
   mnesia:create_table(counter, [
      {attributes, record_info(fields, counter)},
      {disc_copies, [node()]}
   ]),
   mnesia:create_table(proprietes, [
      {attributes, record_info(fields, proprietes)},
      {disc_copies, [node()]}
   ]),
   mnesia:create_table(minichat, [
      {attributes, record_info(fields, minichat)},
      {disc_copies, [node()]}
   ]),
   mnesia:create_table(reponse_minichat, [
      {type, bag},
      {attributes, record_info(fields, reponse_minichat)},
      {disc_copies, [node()]}
   ]),
   mnesia:create_table(user, [
      {attributes, record_info(fields, user)},
      {disc_copies, [node()]}
   ]),
   mnesia:create_table(ip_table, [
      {attributes, record_info(fields, ip_table)},
      {disc_copies, [node()]}
   ]),
   mnesia:create_table(troll, [
      {attributes, record_info(fields, troll)},
      {disc_copies, [node()]}
   ]),
   creer_indexes().
   
   
% mis à part car lors de la reprise de données avec load_textfile les indexes ne sont pas recréés
creer_indexes() ->
   mnesia:add_table_index(minichat, auteur_id),
   mnesia:add_table_index(minichat, troll_id),
   mnesia:add_table_index(reponse_minichat, cible),
   mnesia:add_table_index(user, cookie),
   mnesia:add_table_index(user, login),
   mnesia:add_table_index(troll, date_post).
   
   
% Connexion à la base de données de yaws sur overnux
connect() ->
   connect(yaws@flynux).
connect(Node) ->
   mnesia:start(),
   mnesia:change_config(extra_db_nodes, [Node]).


% Efface toutes les données de la base de données.
reset() ->
   mnesia:clear_table(counter),
   mnesia:clear_table(proprietes),
   mnesia:clear_table(user),
   mnesia:clear_table(reponse_minichat),
   mnesia:clear_table(minichat),
   mnesia:clear_table(troll),
   mnesia:clear_table(ip_table),
   % crée l'utilisateur root
   mnesia:transaction(fun() ->
      mnesia:write(#proprietes{nom = version, valeur = ?VERSION_BD}),
      User = #user{id = 0, profile = #profile{pseudo = "Sys"}, login = "Sys", date_creation = now(), date_derniere_connexion = now(), ek_master = true},
      mnesia:write(User),
      User
   end).
   

% Met à jour la bd, compare ?VERSION_BD avec la version dans la table 'proprietes'
% et exécute les patchs nécessaires.
update() ->
   case mnesia:dirty_read({proprietes, version}) of
      [#proprietes{valeur = Version}] ->
         update(Version);
      _ ->
         erreur
   end.
   

% Mise à jour de la BD.
% attention : il est nécessaire de se trouver dans une transaction.
update(?VERSION_BD) -> fini;
update(Version) ->
   case mnesia:backup(fichier_backup(Version)) of
      ok ->
         case patch(Version) of
            ok ->
               mnesia:dirty_write(#proprietes{nom = version, valeur = Version + 1}),
               update(Version + 1);
            Erreur ->
               Erreur
         end;
      {error, Raison} -> {error, lists:flatten(io_lib:format("Erreur de création du backup de la version ~w : ~w", [Version, Raison]))}
   end.
   
   
% Applique une modification de la BD pour passer d'une version à la suivante.
% crée un backup avant l'application du patch
% dans BD/backups nommé "backup<num>" où <num> et le numéro de la version.
% 1 -> 2
patch(1) ->
   % traitement des users   
   mnesia:transform_table(
      user,
      fun({user, Id, Cookie, Pseudo, Login, Password, Email, Date_creation, Date_derniere_connexion, Css, Nick_format, View_times, View_tooltips, Indice_flood, _Page_principale, Conversations, Ek_master, Last_ip}) ->
            {user, Id, Cookie, Login, Password, {profile, Pseudo, Email, patch1_transforme_css(Css), Nick_format, View_times, View_tooltips, light, reverse, lists:map(fun({R, _}) -> {R, false} end, Conversations)}, Date_creation, Date_derniere_connexion, Indice_flood, Ek_master, Last_ip}
      end,
      record_info(fields, user),
      user
   ),
   mnesia:transform_table(
      minichat,
      fun({minichat, Id, Auteur_id, Date, Pseudo, Contenu, Troll_id}) ->
            {minichat, Id, Auteur_id, Date, Pseudo, Contenu, Troll_id, Id}
      end,
      record_info(fields, minichat),
      minichat
   ),
   case mnesia:transaction(
      fun() ->
         % met à jour les enfants des racines
         % idéalement : utiliser un cursor mais je crois qu'il n'est pas possible de faire des modifs en itérant en même temps avec un cursor, a voir..
         Messages = e(q([M || M <- mnesia:table(minichat), euphorik_bd:parents(M#minichat.id) =:= []]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]),
         lists:foreach(fun(M) -> patch_1_traiter_message(M, M#minichat.id) end, Messages)
      end
   ) of
      {aborted, Raison} -> {erreur, Raison};
      {atomic, _} -> ok
   end.
   

% Prend un chemin vers la feuille de style de type "css/1/euphorik.css"
% et renvoie "styles/1/euphorik.css"
patch1_transforme_css("css" ++ Reste) ->
   "styles" ++ Reste;
patch1_transforme_css(F) ->
   F.

   
patch_1_traiter_message(M, Racine) ->
   % met à jour la racine de chaque message qui répond à M
   lists:foreach(
      fun(M2) ->
         mnesia:write(M2#minichat{racine_id = Racine}),
         patch_1_traiter_message(M2, Racine)
      end,
      euphorik_bd:enfants(M#minichat.id)
   ).

% crée un backup dont le nom est fournit dans le repertoire backups qui se trouve dans le repertoire de la BD.
%backup(Nom) ->
%   mnesia:backup(mnesia:system_info(directory) ++ "/backups/" ++ Nom).
   

% Reviens à une version précédente de la base de données
% (les données insérées durant les versions plus récentes sont perdues)
%restore(N) ->
%   mnesia:restore(fichier_backup(N), [{default_op, recreate_tables}]).


% Renvoie le fichier (avec le chemin) correspondant à la version Version, par exemple : "/var/euphorik/BD/backups/backup1"
fichier_backup(Version) -> 
   mnesia:system_info(directory) ++ "/backups/" ++ if is_integer(Version) -> "backup" ++ integer_to_list(Version); true -> Version end.


backup_text(_) -> todo.
restore_text(File) -> 
   mnesia:stop(),
   mnesia:delete_schema([node()]),
   mnesia:start(),
   case mnesia:load_textfile(File) of
      {atomic, ok} ->
         update(),
         creer_indexes();
      Erreur ->
         Erreur
   end.


toggle_ek_master(User_id) ->
   euphorik_bd:resultat_transaction(mnesia:transaction(
      fun() ->
         Users = e(q([E || E <- mnesia:table(user), E#user.id =:= User_id]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]),
         case Users of
            [User] ->
               mnesia:write(User#user{ek_master = not User#user.ek_master});
            _ -> erreur
         end
      end
   )).
   

% Affiche N user trié par leur date de dernière connexion.
% Opt est une liste d'option d'affichage :
%  * ekmaster : n'affiche que les admins
print_users(N, Opt) ->
   AfficheQueLesEkMaster = lists:any(fun(O) -> O =:= ekmaster end, Opt),
   euphorik_bd:resultat_transaction(mnesia:transaction(fun() ->
      C = cursor(
         qlc:keysort(
            #user.date_derniere_connexion, 
            if AfficheQueLesEkMaster ->
               q([E || E <- mnesia:table(user), E#user.ek_master =:= true]);
            true ->
               q([E || E <- mnesia:table(user)])
            end,
            [{order, descending}]
         ),
         [{tmpdir, ?KEY_SORT_TEMP_DIR}]
      ),
      Users = qlc:next_answers(C, N),
      lists:foreach(
         fun(U) ->
            print_user(U)
         end,
         Users
      ),
      qlc:delete_cursor(C)
   end)).
   
   
% Affiche tous les users.
print_users(Opt) ->
   print_users(all_remaining, Opt).

% Affiche tous les users.
print_users() ->
   print_users(all_remaining, []).
   
print_user(User) when is_record(User, user) ->
   #user{id = Id, profile = #profile{pseudo = Pseudo}, login = Login, ek_master = Ek_master, date_derniere_connexion = Date, last_ip = IP} = User,
   {{Annee, Mois, Jour}, {Heure, Min, _}} = calendar:now_to_local_time(Date),
   io:format(
      % id        pseudo     (login)        IP  Jour  Mois   Année  Heure Minute
      "~4w : ~10.10..s(~10.10..s) ~s ~2w.~2.2.0w.~w - ~2wh~2.2.0w~n",
      [  
         Id,
         if Ek_master -> "*"; true -> "" end ++ Pseudo,
         Login,
         euphorik_common:serialize_ip(IP),
         Jour, Mois, Annee, Heure, Min
      ]
   );
% Affichage d'un user en fonction de son login
print_user(Login) when is_list(Login) ->
   case euphorik_bd:user_by_login(Login) of
      {ok, User} ->
         print_user(User);
      _ ->
         {erreur, "Login pas trouvé : " ++ Login}
   end;
% Affichage d'un user en fonction de son id
print_user(Id) when is_integer(Id) -> 
   case euphorik_bd:user_by_id(Id) of 
      {ok, User} ->
         print_user(User);
      _ ->
         {erreur, "Id pas trouvé : " ++ integer_to_list(Id)}
   end.