2 % Copyright 2008 Grégory Burri
4 % This file is part of Euphorik.
6 % Euphorik is free software: you can redistribute it and/or modify
7 % it under the terms of the GNU General Public License as published by
8 % the Free Software Foundation, either version 3 of the License, or
9 % (at your option) any later version.
11 % Euphorik is distributed in the hope that it will be useful,
12 % but WITHOUT ANY WARRANTY; without even the implied warranty of
13 % MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 % GNU General Public License for more details.
16 % You should have received a copy of the GNU General Public License
17 % along with Euphorik. If not, see <http://www.gnu.org/licenses/>.
19 % Module mettant à disposition tout un tas de fonction pour l'administration de la base de données euphorik comme :
21 % - Mise à jour de la BD
22 % - Backup et restauration
23 % - Copie de la BD à partir d'un autre noeud
28 -module(euphorik_bd_admin
).
46 -import(qlc
, [e
/2, q
/1, cursor
/2]).
47 -include("../include/euphorik_bd.hrl").
48 -include("../include/euphorik_defines.hrl").
49 -include_lib("stdlib/include/qlc.hrl").
52 % Renvoie la version courante de la BD.
54 euphorik_bd:resultat_transaction(mnesia:transaction(
56 mnesia:read({proprietes
, version
})
61 % Instructions pour créer une nouvelle base :
62 % $erl -sname yaws -mnesia dir '"/projets/euphorik/BD"'
63 % voir doc/installation.txt
65 % >euphorik_bd:create().
68 mnesia:delete_schema([node()]),
69 mnesia:create_schema([node()]), % nécessaire pour les tables sur disc
75 mnesia:create_table(counter
, [
76 {attributes
, record_info(fields
, counter
)},
77 {disc_copies
, [node()]}
79 mnesia:create_table(proprietes
, [
80 {attributes
, record_info(fields
, proprietes
)},
81 {disc_copies
, [node()]}
83 mnesia:create_table(texte
, [
84 {attributes
, record_info(fields
, texte
)},
85 {disc_copies
, [node()]}
87 mnesia:create_table(minichat
, [
88 {attributes
, record_info(fields
, minichat
)},
89 {disc_copies
, [node()]}
91 mnesia:create_table(reponse_minichat
, [
93 {attributes
, record_info(fields
, reponse_minichat
)},
94 {disc_copies
, [node()]}
96 mnesia:create_table(user
, [
97 {attributes
, record_info(fields
, user
)},
98 {disc_copies
, [node()]}
100 mnesia:create_table(ip_table
, [
101 {attributes
, record_info(fields
, ip_table
)},
102 {disc_copies
, [node()]}
104 mnesia:create_table(troll
, [
105 {attributes
, record_info(fields
, troll
)},
106 {disc_copies
, [node()]}
111 % mis à part car lors de la reprise de données avec load_textfile les indexes ne sont pas recréés
113 % commence par supprimer les anciens indexes
114 lists:foreach(fun(T
) ->
115 lists:foreach(fun(P
) ->
116 mnesia:del_table_index(T
, P
)
118 mnesia:table_info(T
, index
)
123 mnesia:add_table_index(minichat
, auteur_id
),
124 mnesia:add_table_index(reponse_minichat
, cible
),
125 mnesia:add_table_index(user
, cookie
),
126 mnesia:add_table_index(user
, login
),
127 mnesia:add_table_index(troll
, date_post
),
128 mnesia:add_table_index(troll
, id_minichat
).
131 % Connexion à la base de données de yaws sur overnux
133 connect(yaws@flynux
).
136 mnesia:change_config(extra_db_nodes
, [Node
]).
139 % Efface toutes les données de la base de données.
141 mnesia:clear_table(counter
),
142 mnesia:clear_table(proprietes
),
143 mnesia:clear_table(texte
),
144 mnesia:clear_table(user
),
145 mnesia:clear_table(reponse_minichat
),
146 mnesia:clear_table(minichat
),
147 mnesia:clear_table(troll
),
148 mnesia:clear_table(ip_table
),
149 % crée l'utilisateur root
150 mnesia:transaction(fun() ->
151 mnesia:write(#proprietes
{nom
= version
, valeur
= ?VERSION_BD
}),
152 User
= #user
{id
= 0, profile
= #profile
{pseudo
= "Sys"}, login
= "Sys", date_creation
= now(), date_derniere_connexion
= now(), ek_master
= true
},
160 mnesia:transaction(fun() ->
161 mnesia:write(#texte
{ id
= 10, fr
= "Login déjà existant"}),
162 mnesia:write(#texte
{ id
= 20, fr
= "Trop de register (flood)"}),
163 mnesia:write(#texte
{ id
= 30, fr
= "Couple login/pass introuvable"}),
164 mnesia:write(#texte
{ id
= 40, fr
= "Authentification impossible par cookie"}),
165 mnesia:write(#texte
{ id
= 50, fr
= "Impossible de mettre à jour le profile"}),
166 mnesia:write(#texte
{ id
= 60, fr
= "timeout"}),
167 mnesia:write(#texte
{ id
= 70, fr
= "Page inconnue"}),
168 mnesia:write(#texte
{ id
= 80, fr
= "Vous êtes banni pour encore ~s"}),
169 mnesia:write(#texte
{ id
= 90, fr
= "Message vide"}),
170 mnesia:write(#texte
{ id
= 100, fr
= "Impossible d'ajouter un nouveau message. Raison : ~s"}),
171 mnesia:write(#texte
{ id
= 110, fr
= "Utilisateur inconnu"}),
172 mnesia:write(#texte
{ id
= 120, fr
= "Il n'est pas possible de s'auto bannir"}),
173 mnesia:write(#texte
{ id
= 130, fr
= "L'utilisateur est lui même un ekMaster"}),
174 mnesia:write(#texte
{ id
= 140, fr
= "Utilisateur à bannir inconnu"}),
175 mnesia:write(#texte
{ id
= 150, fr
= "Utilisateur inconnu ou non ek master"}),
176 mnesia:write(#texte
{ id
= 160, fr
= "Utilisateur à slaper inconnu"}),
177 mnesia:write(#texte
{ id
= 170, fr
= "Utilisateur inconnu ou non ek master"}),
178 mnesia:write(#texte
{ id
= 180, fr
= "Le nombre de troll maximum par utilisateur est atteint : ~w "}),
179 mnesia:write(#texte
{ id
= 190, fr
= "Le nombre de troll maximum en attente est atteint : ~w "}),
180 mnesia:write(#texte
{ id
= 200, fr
= "Seul les ekMaster peuvent proposer des trolls"}),
181 mnesia:write(#texte
{ id
= 210, fr
= "Vous ne posséder pas ce troll"}),
182 mnesia:write(#texte
{ id
= 220, fr
= "Seul les ekMaster peuvent proposer des trolls"}),
183 mnesia:write(#texte
{ id
= 230, fr
= "Seul les ekMaster peuvent connaitre la liste des ips bannies"})
187 % Met à jour la bd, compare ?VERSION_BD avec la version dans la table 'proprietes'
188 % et exécute les patchs nécessaires.
190 case mnesia:dirty_read({proprietes
, version
}) of
191 [#proprietes
{valeur
= Version
}] ->
198 % Mise à jour de la BD.
199 % attention : il est nécessaire de se trouver dans une transaction.
200 update(?VERSION_BD
) -> fini
;
202 case mnesia:backup(fichier_backup(Version
)) of
204 case patch(Version
) of
206 mnesia:dirty_write(#proprietes
{nom
= version
, valeur
= Version
+ 1}),
211 {error
, Raison
} -> {error
, lists:flatten(io_lib:format("Erreur de création du backup de la version ~w : ~w", [Version
, Raison
]))}
215 % Applique une modification de la BD pour passer d'une version à la suivante.
216 % crée un backup avant l'application du patch
217 % dans BD/backups nommé "backup<num>" où <num> et le numéro de la version.
220 % Prend un chemin vers la feuille de style de type "css/1/euphorik.css"
221 % et renvoie "styles/1/euphorik.css"
222 Transforme_css
= fun("css" ++ Reste
) ->
226 Traiter_message
= fun(M
, Racine
) ->
227 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
228 % met à jour la racine de chaque message qui répond à M
231 mnesia:write(M2#minichat
{racine_id
= Racine
}),
234 euphorik_bd:enfants(M#minichat
.id
)
239 mnesia:create_table(texte
, [
240 {attributes
, record_info(fields
, texte
)},
241 {disc_copies
, [node()]}
244 % traitement des users
245 mnesia:transform_table(
247 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
}) ->
248 {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
}
250 record_info(fields
, user
)
252 mnesia:transform_table(
254 fun({minichat
, Id
, Auteur_id
, Date
, Pseudo
, Contenu
, Troll_id
}) ->
255 {minichat
, Id
, Auteur_id
, Date
, Pseudo
, Contenu
, Troll_id
, Id
}
257 record_info(fields
, minichat
)
259 case mnesia:transaction(
261 % met à jour les enfants des racines
262 % 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..
263 Messages
= e(q([M
|| M
<- mnesia:table(minichat
), euphorik_bd:parents(M#minichat
.id
) =:= []]), [{tmpdir
, ?KEY_SORT_TEMP_DIR
}]),
264 lists:foreach(fun(M
) -> Traiter_message(M
, M#minichat
.id
) end, Messages
)
267 {aborted
, Raison
} -> {erreur
, Raison
};
272 mnesia:transform_table(
274 fun({troll
, Id_troll
, Id_user
, Date_create
, Date_post
, Content
}) ->
275 % recherche le message associé s'il existe
276 Id_minichat
= case e(q([M
|| M
<- mnesia:table(minichat
), element(7, M
) =:= Id_troll
]), [{tmpdir
, ?KEY_SORT_TEMP_DIR
}]) of
277 [M
] -> element(2, M
);
280 {troll
, Id_troll
, Id_user
, Id_minichat
, Date_create
, Date_post
, Content
}
282 record_info(fields
, troll
)
284 %mnesia:del_table_index(minichat, troll_id),
285 mnesia:transform_table(
287 fun({minichat
, Id
, Auteur_id
, Date
, Pseudo
, Contenu
, _Troll_id
, Racine_id
}) ->
288 {minichat
, Id
, Auteur_id
, Date
, Pseudo
, Contenu
, Racine_id
, normal
}
290 record_info(fields
, minichat
)
292 mnesia:transaction(fun() ->
293 % comble les trous entre les id non-contigues
294 lists:foreach(fun(Id
) ->
295 case mnesia:read({minichat
, Id
}) of
297 {ok
, #user
{profile
= Profile
}} = euphorik_bd:user_by_id(0),
298 mnesia:write(#minichat
{id
= Id
, auteur_id
= 0, date = undefined
, pseudo
= Profile#profile
.pseudo
, contenu
= "Comblement...", racine_id
= Id
, status
= deleted
});
302 lists:seq(1, mnesia:table_info(minichat
, size))
304 % la table troll utilise maintenant son index et pas celui de la table minichat (correction d'un vieux bug)
305 mnesia:write(#counter
{key
= troll
, value
= mnesia:table_info(minichat
, size)})
307 creer_indexes(). % uniquement pour l'indice sur id_minichat de la table troll
310 % Renvoie le dossier dans lequel les backups sont effectué, ce dossier doit être en écriture.
312 mnesia:system_info(directory
) ++ "/backups/".
315 % Renvoie le fichier (avec le chemin) correspondant à la version Version, par exemple : "/var/euphorik/BD/backups/backup1"
316 fichier_backup(Version
) when is_integer(Version
) ->
317 dossier_backups() ++ "backup" ++ integer_to_list(Version
).
320 % crée un backup dont le nom est fournit dans le repertoire backups qui se trouve dans le repertoire de la BD.
322 mnesia:backup(dossier_backups() ++ Fichier
).
325 % Restaure un backup de la BD.
326 restore(Fichier
) when is_list(Fichier
) ->
327 mnesia:restore(dossier_backups() ++ Fichier
, []);
328 % Reviens à une version précédente de la base de données.
329 % (les données insérées durant les versions plus récentes sont perdues).
330 restore(Version
) when is_integer(Version
) ->
331 mnesia:restore(fichier_backup(Version
), []). % [{default_op, recreate_tables}]).
334 % Change le nom du noeud d'un backup.
335 % provient d'ici : http://www.erlang.org/doc/apps/mnesia/Mnesia_chap7.html#6.9
336 % From : l'ancien nom
337 % To : le nouveau nom
338 % Source : le nom du fichier de backup
339 % Target : le nom du fichier du nouveau backup
340 change_node_name(From
, To
, Source
, Target
) ->
342 fun(Node
) when Node
== From
-> To
;
343 (Node
) when Node
== To
-> throw({error
, already_exists
});
347 fun({schema
, db_nodes
, Nodes
}, Acc
) ->
348 {[{schema
, db_nodes
, lists:map(Switch
,Nodes
)}], Acc
};
349 ({schema
, version
, Version
}, Acc
) ->
350 {[{schema
, version
, Version
}], Acc
};
351 ({schema
, cookie
, Cookie
}, Acc
) ->
352 {[{schema
, cookie
, Cookie
}], Acc
};
353 ({schema
, Tab
, CreateList
}, Acc
) ->
354 Keys
= [ram_copies
, disc_copies
, disc_only_copies
],
357 case lists:member(Key
, Keys
) of
358 true
-> {Key
, lists:map(Switch
, Val
)};
362 {[{schema
, Tab
, lists:map(OptSwitch
, CreateList
)}], Acc
};
366 mnesia:traverse_backup(Source
, Target
, Convert
, switched
).
370 %~ backup_text(File) ->
371 %~ mnesia:dump_to_textfile(File).
372 %~ restore_text(File) ->
374 %~ mnesia:delete_schema([node()]),
377 %~ case mnesia:load_textfile(File) of
386 toggle_ek_master(User_id
) ->
387 euphorik_bd:resultat_transaction(mnesia:transaction(
389 Users
= e(q([E
|| E
<- mnesia:table(user
), E#user
.id
=:= User_id
]), [{tmpdir
, ?KEY_SORT_TEMP_DIR
}]),
392 mnesia:write(User#user
{ek_master
= not User#user
.ek_master
});
399 % Affiche N user trié par leur date de dernière connexion.
400 % Opt est une liste d'option d'affichage :
401 % * ekmaster : n'affiche que les admins
402 print_users(N
, Opt
) ->
403 AfficheQueLesEkMaster
= lists:any(fun(O
) -> O
=:= ekmaster
end, Opt
),
404 euphorik_bd:resultat_transaction(mnesia:transaction(fun() ->
407 #user
.date_derniere_connexion
,
408 if AfficheQueLesEkMaster
->
409 q([E
|| E
<- mnesia:table(user
), E#user
.ek_master
=:= true
]);
411 q([E
|| E
<- mnesia:table(user
)])
413 [{order
, descending
}]
415 [{tmpdir
, ?KEY_SORT_TEMP_DIR
}]
417 Users
= qlc:next_answers(C
, N
),
428 % Affiche tous les users.
430 print_users(all_remaining
, Opt
).
432 % Affiche tous les users.
434 print_users(all_remaining
, []).
436 print_user(User
) when is_record(User
, user
) ->
437 #user
{id
= Id
, profile
= #profile
{pseudo
= Pseudo
}, login
= Login
, ek_master
= Ek_master
, date_derniere_connexion
= Date
, last_ip
= IP
} = User
,
438 {{Annee
, Mois
, Jour
}, {Heure
, Min
, _
}} = calendar:now_to_local_time(Date
),
440 % id pseudo (login) IP Jour Mois Année Heure Minute
441 "~4w : ~10.10..s(~10.10..s) ~s ~2w.~2.2.0w.~w - ~2wh~2.2.0w~n",
444 if Ek_master
-> "*"; true
-> "" end ++ Pseudo
,
446 euphorik_common:serialize_ip(IP
),
447 Jour
, Mois
, Annee
, Heure
, Min
450 % Affichage d'un user en fonction de son login
451 print_user(Login
) when is_list(Login
) ->
452 case euphorik_bd:user_by_login(Login
) of
456 {erreur
, "Login pas trouvé : " ++ Login
}
458 % Affichage d'un user en fonction de son id
459 print_user(Id
) when is_integer(Id
) ->
460 case euphorik_bd:user_by_id(Id
) of
464 {erreur
, "Id pas trouvé : " ++ integer_to_list(Id
)}