Use the new json2 module (from yaws)
[euphorik.git] / modules / erl / euphorik_bd_admin.erl
index 64ea4ac..0abd2bd 100644 (file)
@@ -15,7 +15,7 @@
 %
 % 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
    connect/1,
    reset/0,
    update/0,
-   
-   backup_text/1,
-   restore_text/1,
-   
+
+   backup/1,
+   restore/1,
+   change_node_name/4,
+
    toggle_ek_master/1,
    print_users/0,
    print_users/1,
@@ -55,9 +56,9 @@ version_bd() ->
          mnesia:read({proprietes, version})
       end
    )).
-   
 
-% Instructions pour créer une nouvelle base : 
+
+% Instructions pour créer une nouvelle base :
 % $erl -sname yaws -mnesia dir '"/projets/euphorik/BD"'
 % voir doc/installation.txt
 % >l(euphorik_bd).
@@ -69,7 +70,7 @@ create() ->
    mnesia:start(),
    create_tables(),
    reset().
-   
+
 create_tables() ->
    mnesia:create_table(counter, [
       {attributes, record_info(fields, counter)},
@@ -105,18 +106,28 @@ create_tables() ->
       {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() ->
+   % commence par supprimer les anciens indexes
+   lists:foreach(fun(T) ->
+         lists:foreach(fun(P) ->
+               mnesia:del_table_index(T, P)
+            end,
+            mnesia:table_info(T, index)
+         )
+      end,
+      ?TABLES
+   ),
    mnesia:add_table_index(minichat, auteur_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),\r
+   mnesia:add_table_index(troll, date_post),
    mnesia:add_table_index(troll, id_minichat).
-   
-   
+
+
 % Connexion à la base de données de yaws sur overnux
 connect() ->
    connect(yaws@flynux).
@@ -138,13 +149,13 @@ reset() ->
    % 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},
+      User = #user{id = 0, profile = #profile{pseudo = "Sys"}, login = "Sys", date_creation = erlang:timestamp(), date_derniere_connexion = erlang:timestamp(), ek_master = true},
       mnesia:write(User),
       User
    end),
    peupler_texte().
-   
-   
+
+
 peupler_texte() ->
    mnesia:transaction(fun() ->
       mnesia:write(#texte{ id = 10, fr = "Login déjà existant"}),
@@ -182,7 +193,7 @@ update() ->
       _ ->
          erreur
    end.
-   
+
 
 % Mise à jour de la BD.
 % attention : il est nécessaire de se trouver dans une transaction.
@@ -199,55 +210,51 @@ update(Version) ->
          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) ->\r
-   % Prend un chemin vers la feuille de style de type "css/1/euphorik.css"\r
-   % et renvoie "styles/1/euphorik.css"\r
-   Transforme_css = fun(F) ->\r
-      case F of\r
-         "css" ++ Reste -> "styles" ++ Reste;\r
-         _ -> F\r
-      end\r
-   end,\r
-   Traiter_message = fun(M, Racine) ->\r
-      F = fun(F, M2) -> % seul moyen à ma connaissance pour faire de la récursion dans une lambda fonction, voir : http://www.nabble.com/Auto-generated-functions-td15279499.html\r
-         % met à jour la racine de chaque message qui répond à M\r
-         lists:foreach(\r
-            fun(M3) ->\r
-               mnesia:write(M2#minichat{racine_id = Racine}),\r
-               F(F, M3)\r
-            end,\r
-            euphorik_bd:enfants(M#minichat.id)\r
-         )\r
-      end,\r
-      F(F, M, Racine)\r
+patch(1) ->
+   % Prend un chemin vers la feuille de style de type "css/1/euphorik.css"
+   % et renvoie "styles/1/euphorik.css"
+   Transforme_css = fun("css" ++ Reste) ->
+         "styles" ++ Reste;
+         (F) -> F
+   end,
+   Traiter_message = fun(M, Racine) ->
+      F = fun(F, M2) -> % seul moyen à ma connaissance pour faire de la récursion dans une lambda fonction, voir : http://www.nabble.com/Auto-generated-functions-td15279499.html
+         % met à jour la racine de chaque message qui répond à M
+         lists:foreach(
+            fun(M3) ->
+               mnesia:write(M2#minichat{racine_id = Racine}),
+               F(F, M3)
+            end,
+            euphorik_bd:enfants(M#minichat.id)
+         )
+      end,
+      F(F, M, Racine)
    end,
    mnesia:create_table(texte, [
       {attributes, record_info(fields, texte)},
       {disc_copies, [node()]}
    ]),
    peupler_texte(),
-   % traitement des users   
+   % 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, 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
+      record_info(fields, 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
+      record_info(fields, minichat)
    ),
    case mnesia:transaction(
       fun() ->
@@ -259,62 +266,121 @@ patch(1) ->
    ) of
       {aborted, Raison} -> {erreur, Raison};
       {atomic, _} -> ok
-   end;\r
-% 2 -> 3\r
-patch(2) ->\r
-   mnesia:transform_table(\r
-      troll,\r
-      fun({troll, Id_troll, Id_user, Date_create, Date_post, Content}) ->\r
-         % recherche le message associé s'il existe\r
-         Id_minichat = case mnesia:index_read(minichat, Id_troll, 7) of\r
-            [#minichat{id = Id}] -> Id;\r
-            _ -> undefined\r
-         end,\r
-         {troll, Id_troll, Id_user, Id_minichat, Date_create, Date_post, Content}\r
-      end,\r
-      record_info(fields, troll),\r
-      user\r
-   ),\r
-   mnesia:del_table_index(minichat, troll_id),\r
-   mnesia:transform_table(\r
-      minichat,\r
-      fun({minichat, Id, Auteur_id, Date, Pseudo, Contenu, _Troll_id, Racine_id}) ->\r
-         {minichat, Id, Auteur_id, Date, Pseudo, Contenu, Racine_id}\r
-      end,\r
-      record_info(fields, troll),\r
-      user\r
-   ),   \r
-   creer_indexes(). % uniquement pour l'indice sur id_minichat de la table troll\r
-   \r
+   end;
+% 2 -> 3
+patch(2) ->
+   mnesia:transform_table(
+      troll,
+      fun({troll, Id_troll, Id_user, Date_create, Date_post, Content}) ->
+         % recherche le message associé s'il existe
+         Id_minichat = case e(q([M || M <- mnesia:table(minichat), element(7, M) =:= Id_troll]), [{tmpdir, ?KEY_SORT_TEMP_DIR}]) of
+            [M] -> element(2, M);
+            _ -> undefined
+         end,
+         {troll, Id_troll, Id_user, Id_minichat, Date_create, Date_post, Content}
+      end,
+      record_info(fields, troll)
+   ),
+   %mnesia:del_table_index(minichat, troll_id),
+   mnesia:transform_table(
+      minichat,
+      fun({minichat, Id, Auteur_id, Date, Pseudo, Contenu, _Troll_id, Racine_id}) ->
+         {minichat, Id, Auteur_id, Date, Pseudo, Contenu, Racine_id, normal}
+      end,
+      record_info(fields, minichat)
+   ),
+   mnesia:transaction(fun() ->
+      % comble les trous entre les id non-contigues
+      lists:foreach(fun(Id) ->
+            case mnesia:read({minichat, Id}) of
+               [] ->
+                  {ok, #user{profile = Profile}} = euphorik_bd:user_by_id(0),
+                  mnesia:write(#minichat{id = Id, auteur_id = 0, date = undefined, pseudo = Profile#profile.pseudo, contenu = "Comblement...", racine_id = Id, status = deleted});
+               _ -> rien
+            end
+         end,
+         lists:seq(1, mnesia:table_info(minichat, size))
+      ),
+      % la table troll utilise maintenant son index et pas celui de la table minichat (correction d'un vieux bug)
+      mnesia:write(#counter{key = troll, value = mnesia:table_info(minichat, size)})
+   end),
+   creer_indexes(). % uniquement pour l'indice sur id_minichat de la table troll
 
-% 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 dossier dans lequel les backups sont effectué, ce dossier doit être en écriture.
+dossier_backups() ->
+   mnesia:system_info(directory) ++ "/backups/".
 
 
 % 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.
+fichier_backup(Version) when is_integer(Version) ->
+   dossier_backups() ++ "backup" ++ integer_to_list(Version).
 
 
-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.
+% crée un backup dont le nom est fournit dans le repertoire backups qui se trouve dans le repertoire de la BD.
+backup(Fichier) ->
+   mnesia:backup(dossier_backups() ++ Fichier).
+
+
+% Restaure un backup de la BD.
+restore(Fichier) when is_list(Fichier) ->
+   mnesia:restore(dossier_backups() ++ Fichier, []);
+% 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(Version) when is_integer(Version) ->
+   mnesia:restore(fichier_backup(Version), []). % [{default_op, recreate_tables}]).
+
+
+% Change le nom du noeud d'un backup.
+% provient d'ici : http://www.erlang.org/doc/apps/mnesia/Mnesia_chap7.html#6.9
+% From : l'ancien nom
+% To : le nouveau nom
+% Source : le nom du fichier de backup
+% Target : le nom du fichier du nouveau backup
+change_node_name(From, To, Source, Target) ->
+    Switch =
+        fun(Node) when Node == From -> To;
+           (Node) when Node == To -> throw({error, already_exists});
+           (Node) -> Node
+        end,
+    Convert =
+        fun({schema, db_nodes, Nodes}, Acc) ->
+                {[{schema, db_nodes, lists:map(Switch,Nodes)}], Acc};
+           ({schema, version, Version}, Acc) ->
+                {[{schema, version, Version}], Acc};
+           ({schema, cookie, Cookie}, Acc) ->
+                {[{schema, cookie, Cookie}], Acc};
+           ({schema, Tab, CreateList}, Acc) ->
+                Keys = [ram_copies, disc_copies, disc_only_copies],
+                OptSwitch =
+                    fun({Key, Val}) ->
+                            case lists:member(Key, Keys) of
+                                true -> {Key, lists:map(Switch, Val)};
+                                false-> {Key, Val}
+                            end
+                    end,
+                {[{schema, Tab, lists:map(OptSwitch, CreateList)}], Acc};
+           (Other, Acc) ->
+                {[Other], Acc}
+        end,
+    mnesia:traverse_backup(Source, Target, Convert, switched).
+
+
+% Obsolète
+%~ backup_text(File) ->
+   %~ mnesia:dump_to_textfile(File).
+%~ restore_text(File) ->
+   %~ mnesia:stop(),
+   %~ mnesia:delete_schema([node()]),
+   %~ mnesia:start(),
+   %~ create_tables(),
+   %~ case mnesia:load_textfile(File) of
+      %~ {atomic, ok} ->
+         %~ update(),
+         %~ creer_indexes();
+      %~ Erreur ->
+         %~ Erreur
+   %~ end.
 
 
 toggle_ek_master(User_id) ->
@@ -328,7 +394,7 @@ toggle_ek_master(User_id) ->
          end
       end
    )).
-   
+
 
 % Affiche N user trié par leur date de dernière connexion.
 % Opt est une liste d'option d'affichage :
@@ -338,7 +404,7 @@ print_users(N, Opt) ->
    euphorik_bd:resultat_transaction(mnesia:transaction(fun() ->
       C = cursor(
          qlc:keysort(
-            #user.date_derniere_connexion, 
+            #user.date_derniere_connexion,
             if AfficheQueLesEkMaster ->
                q([E || E <- mnesia:table(user), E#user.ek_master =:= true]);
             true ->
@@ -357,8 +423,8 @@ print_users(N, Opt) ->
       ),
       qlc:delete_cursor(C)
    end)).
-   
-   
+
+
 % Affiche tous les users.
 print_users(Opt) ->
    print_users(all_remaining, Opt).
@@ -366,14 +432,14 @@ print_users(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,
@@ -390,11 +456,10 @@ print_user(Login) when is_list(Login) ->
          {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 
+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.
-   
\ No newline at end of file