Merge pull request 'feat/392-script-reconstruction-bdd' (#11) from feat/392-script-reconstruction-bdd into develop
Reviewed-on: #11
This commit was merged in pull request #11.
This commit is contained in:
@@ -5,7 +5,7 @@ set -euo pipefail
|
||||
# Chemins fixes du script
|
||||
#######################################
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
ENV_FILE="/home/matt/vaultwarden/Malio-ops/BackupVaultWarden/.env"
|
||||
ENV_FILE="${SCRIPT_DIR}/.env"
|
||||
LOG_FILE="/var/log/vaultwarden_backup.log"
|
||||
|
||||
mkdir -p "$(dirname "$LOG_FILE")"
|
||||
|
||||
@@ -92,8 +92,23 @@ exec > >(tee -a "$LOG_FILE") 2>&1
|
||||
|
||||
log() { echo "---- $(date +'%Y-%m-%d %H:%M:%S') ---- $*"; }
|
||||
|
||||
require_cmd() {
|
||||
command -v "$1" >/dev/null 2>&1
|
||||
}
|
||||
|
||||
export PGPASSWORD
|
||||
|
||||
#######################################
|
||||
# Vérification dépendances minimales
|
||||
#######################################
|
||||
|
||||
for cmd in ssh scp curl jq pg_dump pg_dumpall; do
|
||||
require_cmd "$cmd" || {
|
||||
echo "ERROR: commande manquante : $cmd" >&2
|
||||
exit 1
|
||||
}
|
||||
done
|
||||
|
||||
#######################################
|
||||
# Configuration Discord
|
||||
#######################################
|
||||
@@ -122,11 +137,14 @@ discord_send() {
|
||||
#######################################
|
||||
|
||||
discord_msg_global_ok() {
|
||||
local msg="**BACKUP BDD ${ENV_NAME} 🟢**\n"
|
||||
msg+="Name: ${BACKUP_DIR_NAME}\n"
|
||||
msg+="Dumps transfer: ✅\n"
|
||||
msg+="Users transfer: ✅"
|
||||
|
||||
local msg
|
||||
msg="$(cat <<EOF
|
||||
**BACKUP BDD ${ENV_NAME} 🟢**
|
||||
Name: ${BACKUP_DIR_NAME}
|
||||
Dumps transfer: ✅
|
||||
Users transfer: ✅
|
||||
EOF
|
||||
)"
|
||||
discord_send "$msg"
|
||||
}
|
||||
|
||||
@@ -135,9 +153,12 @@ discord_msg_global_ok() {
|
||||
#######################################
|
||||
|
||||
discord_msg_users_ok_simple() {
|
||||
local msg="**BACKUP BDD ${ENV_NAME} 🟢**\n"
|
||||
msg+="Users backup validé"
|
||||
|
||||
local msg
|
||||
msg="$(cat <<EOF
|
||||
**BACKUP BDD ${ENV_NAME} 🟢**
|
||||
Users backup validé
|
||||
EOF
|
||||
)"
|
||||
discord_send "$msg"
|
||||
}
|
||||
|
||||
@@ -150,12 +171,25 @@ discord_msg_users_error() {
|
||||
export_disp=$([[ -n "$export_ok" ]] && echo "✅" || echo "❌")
|
||||
transfer_disp=$([[ -n "$transfer_ok" ]] && echo "✅" || echo "❌")
|
||||
|
||||
local msg="**${DISCORD_PING} BACKUP BDD ${ENV_NAME} 🔴**\n"
|
||||
msg+="Name: ${BACKUP_DIR_NAME}\n"
|
||||
msg+="Users export: ${export_disp}\n"
|
||||
msg+="Users transfer: ${transfer_disp}"
|
||||
|
||||
[[ -n "$details" ]] && msg+="\nDetails: ${details}"
|
||||
local msg
|
||||
if [[ -n "$details" ]]; then
|
||||
msg="$(cat <<EOF
|
||||
**${DISCORD_PING} BACKUP BDD ${ENV_NAME} 🔴**
|
||||
Name: ${BACKUP_DIR_NAME}
|
||||
Users export: ${export_disp}
|
||||
Users transfer: ${transfer_disp}
|
||||
Details: ${details}
|
||||
EOF
|
||||
)"
|
||||
else
|
||||
msg="$(cat <<EOF
|
||||
**${DISCORD_PING} BACKUP BDD ${ENV_NAME} 🔴**
|
||||
Name: ${BACKUP_DIR_NAME}
|
||||
Users export: ${export_disp}
|
||||
Users transfer: ${transfer_disp}
|
||||
EOF
|
||||
)"
|
||||
fi
|
||||
|
||||
discord_send "$msg"
|
||||
}
|
||||
@@ -166,9 +200,12 @@ discord_msg_users_error() {
|
||||
|
||||
discord_msg_db_ok_simple() {
|
||||
local db="$1"
|
||||
local msg="**BACKUP BDD ${ENV_NAME} 🟢**\n"
|
||||
msg+="Backup validé : ${db}"
|
||||
|
||||
local msg
|
||||
msg="$(cat <<EOF
|
||||
**BACKUP BDD ${ENV_NAME} 🟢**
|
||||
Backup validé : ${db}
|
||||
EOF
|
||||
)"
|
||||
discord_send "$msg"
|
||||
}
|
||||
|
||||
@@ -182,13 +219,27 @@ discord_msg_db_error() {
|
||||
dump_disp=$([[ -n "$dump_ok" ]] && echo "✅" || echo "❌")
|
||||
transfer_disp=$([[ -n "$transfer_ok" ]] && echo "✅" || echo "❌")
|
||||
|
||||
local msg="**${DISCORD_PING} BACKUP BDD ${ENV_NAME} 🔴**\n"
|
||||
msg+="Name: ${BACKUP_DIR_NAME}\n"
|
||||
msg+="Database: ${db}\n"
|
||||
msg+="Dump: ${dump_disp}\n"
|
||||
msg+="Transfer: ${transfer_disp}"
|
||||
|
||||
[[ -n "$details" ]] && msg+="\nDetails: ${details}"
|
||||
local msg
|
||||
if [[ -n "$details" ]]; then
|
||||
msg="$(cat <<EOF
|
||||
**${DISCORD_PING} BACKUP BDD ${ENV_NAME} 🔴**
|
||||
Name: ${BACKUP_DIR_NAME}
|
||||
Database: ${db}
|
||||
Dump: ${dump_disp}
|
||||
Transfer: ${transfer_disp}
|
||||
Details: ${details}
|
||||
EOF
|
||||
)"
|
||||
else
|
||||
msg="$(cat <<EOF
|
||||
**${DISCORD_PING} BACKUP BDD ${ENV_NAME} 🔴**
|
||||
Name: ${BACKUP_DIR_NAME}
|
||||
Database: ${db}
|
||||
Dump: ${dump_disp}
|
||||
Transfer: ${transfer_disp}
|
||||
EOF
|
||||
)"
|
||||
fi
|
||||
|
||||
discord_send "$msg"
|
||||
}
|
||||
@@ -220,7 +271,7 @@ if ! mkdir "$LOCK_DIR" 2>/dev/null; then
|
||||
exit 1
|
||||
fi
|
||||
|
||||
trap 'rm -rf "$LOCK_DIR"' EXIT
|
||||
trap 'rm -rf "$LOCK_DIR" "$TMP_DIR"' EXIT
|
||||
|
||||
#######################################
|
||||
# Préparation du dossier distant
|
||||
@@ -240,13 +291,18 @@ fi
|
||||
# Export des rôles PostgreSQL
|
||||
#######################################
|
||||
|
||||
ROLES_FILE="${TMP_DIR}/user_${TS}.dump"
|
||||
ROLES_FILE="${TMP_DIR}/user_${TS}.sql"
|
||||
|
||||
set +e
|
||||
|
||||
psql -h "$PGHOST" -p "$PGPORT" -U "$PGUSER" -d postgres -Atq <<'SQL' > "$ROLES_FILE"
|
||||
SELECT rolname FROM pg_roles WHERE rolname !~ '^pg_';
|
||||
SQL
|
||||
log "Export des rôles PostgreSQL"
|
||||
|
||||
pg_dumpall \
|
||||
-h "$PGHOST" \
|
||||
-p "$PGPORT" \
|
||||
-U "$PGUSER" \
|
||||
--globals-only \
|
||||
> "$ROLES_FILE"
|
||||
|
||||
RET=$?
|
||||
|
||||
@@ -254,18 +310,24 @@ if [[ $RET -ne 0 ]]; then
|
||||
USERS_OK=
|
||||
USERS_EXPORT_OK=
|
||||
USERS_DETAILS="roles export failed"
|
||||
else
|
||||
log "Export des rôles OK : $ROLES_FILE"
|
||||
fi
|
||||
|
||||
scp "${SSH_OPTS[@]}" "$ROLES_FILE" "$IA_SSH:${REMOTE_DIR}/user/"
|
||||
RET=$?
|
||||
if [[ -n "${USERS_EXPORT_OK:-}" ]]; then
|
||||
scp "${SSH_OPTS[@]}" "$ROLES_FILE" "$IA_SSH:${REMOTE_DIR}/user/"
|
||||
RET=$?
|
||||
|
||||
if [[ $RET -ne 0 ]]; then
|
||||
USERS_OK=
|
||||
USERS_TRANSFER_OK=
|
||||
if [[ -n "$USERS_DETAILS" ]]; then
|
||||
USERS_DETAILS+=" | roles transfer failed"
|
||||
if [[ $RET -ne 0 ]]; then
|
||||
USERS_OK=
|
||||
USERS_TRANSFER_OK=
|
||||
if [[ -n "$USERS_DETAILS" ]]; then
|
||||
USERS_DETAILS+=" | roles transfer failed"
|
||||
else
|
||||
USERS_DETAILS="roles transfer failed"
|
||||
fi
|
||||
else
|
||||
USERS_DETAILS="roles transfer failed"
|
||||
log "Transfert des rôles OK"
|
||||
fi
|
||||
fi
|
||||
|
||||
@@ -317,11 +379,13 @@ log "Starting remote rotation: delete backups older than ${RETENTION_DAYS} days"
|
||||
|
||||
set +e
|
||||
|
||||
ssh "${SSH_OPTS[@]}" "$IA_SSH" "find '${REMOTE_DIR}/user' -type f -name 'user_*.dump' -mtime +${RETENTION_DAYS} -delete"
|
||||
ssh "${SSH_OPTS[@]}" "$IA_SSH" "find '${REMOTE_DIR}/user' -type f -name 'user_*.sql' -mtime +${RETENTION_DAYS} -delete"
|
||||
RET=$?
|
||||
|
||||
if [[ $RET -ne 0 ]]; then
|
||||
log "ERROR: remote rotation failed for users"
|
||||
else
|
||||
log "Remote rotation OK for users"
|
||||
fi
|
||||
|
||||
for DB in "${DBS_ARRAY[@]}"; do
|
||||
|
||||
@@ -9,11 +9,11 @@ set -uo pipefail
|
||||
#
|
||||
# Fonctionnement global :
|
||||
# 1. charge la configuration depuis le fichier .env ;
|
||||
# 2. vérifie que le DNS du site est résolu ;
|
||||
# 2. vérifie le DNS de chaque application ;
|
||||
# 3. effectue une requête HTTP avec curl ;
|
||||
# 4. analyse le code HTTP retourné ;
|
||||
# 5. écrit le résultat dans un fichier de log local ;
|
||||
# 6. envoie une notification Discord avec l’état du service.
|
||||
# 4. écrit le résultat dans un fichier de log local ;
|
||||
# 5. construit un message récapitulatif unique ;
|
||||
# 6. envoie une seule notification Discord avec tous les statuts.
|
||||
###############################################################################
|
||||
|
||||
#######################################
|
||||
@@ -68,34 +68,12 @@ LOG_FILE="${LOG_DIR}/app_health_$(date +'%Y-%m-%d').log"
|
||||
DISCORD_WEBHOOK_URL="${DISCORD_WEBHOOK_URL:-}"
|
||||
DISCORD_PING="${DISCORD_PING:-@here}"
|
||||
|
||||
discord_ping() {
|
||||
local site="$1"
|
||||
local status="$2"
|
||||
local detail="$3"
|
||||
#######################################
|
||||
# Variables globales de synthèse
|
||||
#######################################
|
||||
|
||||
[[ -z "${DISCORD_WEBHOOK_URL:-}" ]] && return 0
|
||||
|
||||
local color icon ping_prefix=""
|
||||
if [[ "$status" == "OK" ]]; then
|
||||
color="🟢"
|
||||
icon="✅"
|
||||
else
|
||||
color="🔴"
|
||||
icon="❌"
|
||||
ping_prefix="${DISCORD_PING} "
|
||||
fi
|
||||
|
||||
local msg="**${ping_prefix}CHECK APP ${ENV_NAME} $color**\n"
|
||||
msg+="Application: ${site}\n"
|
||||
msg+="Details: ${detail}"
|
||||
|
||||
local payload
|
||||
payload="$(jq -n --arg content "$msg" '{content: $content}')"
|
||||
|
||||
curl -fsS -H "Content-Type: application/json" \
|
||||
-d "$payload" \
|
||||
"$DISCORD_WEBHOOK_URL" >/dev/null || true
|
||||
}
|
||||
SUMMARY_LINES=()
|
||||
FAILURES=0
|
||||
|
||||
#######################################
|
||||
# Logging
|
||||
@@ -104,8 +82,6 @@ discord_ping() {
|
||||
log_line() {
|
||||
printf "%s | %s | %s | %s\n" \
|
||||
"$(date +'%Y-%m-%d %H:%M:%S')" "$1" "$2" "$3" | tee -a "$LOG_FILE"
|
||||
|
||||
discord_ping "$2" "$1" "$3"
|
||||
}
|
||||
|
||||
#######################################
|
||||
@@ -116,22 +92,71 @@ dns_ok() {
|
||||
getent hosts "$1" >/dev/null 2>&1
|
||||
}
|
||||
|
||||
#######################################
|
||||
# Ajout au résumé Discord
|
||||
#######################################
|
||||
|
||||
add_summary_line() {
|
||||
local site="$1"
|
||||
local status="$2"
|
||||
local detail="$3"
|
||||
|
||||
local icon
|
||||
if [[ "$status" == "OK" ]]; then
|
||||
icon="✅"
|
||||
else
|
||||
icon="❌"
|
||||
fi
|
||||
|
||||
SUMMARY_LINES+=("${icon} ${site} : ${detail}")
|
||||
}
|
||||
|
||||
#######################################
|
||||
# Envoi du message Discord récapitulatif
|
||||
#######################################
|
||||
|
||||
send_discord_summary() {
|
||||
[[ -z "${DISCORD_WEBHOOK_URL:-}" ]] && return 0
|
||||
|
||||
local header_icon ping_prefix=""
|
||||
if [[ "$FAILURES" -eq 0 ]]; then
|
||||
header_icon="🟢"
|
||||
else
|
||||
header_icon="🔴"
|
||||
ping_prefix="${DISCORD_PING} "
|
||||
fi
|
||||
|
||||
local msg="**${ping_prefix}CHECK APP ${ENV_NAME} ${header_icon}**"
|
||||
|
||||
local line
|
||||
for line in "${SUMMARY_LINES[@]}"; do
|
||||
msg+=$'\n'"${line}"
|
||||
done
|
||||
|
||||
local payload
|
||||
payload="$(jq -n --arg content "$msg" '{content: $content}')"
|
||||
|
||||
curl -fsS -H "Content-Type: application/json" \
|
||||
-d "$payload" \
|
||||
"$DISCORD_WEBHOOK_URL" >/dev/null || true
|
||||
}
|
||||
|
||||
#######################################
|
||||
# Check application
|
||||
#######################################
|
||||
|
||||
check_site() {
|
||||
|
||||
local host="$1"
|
||||
local url="${SCHEME}://${host}/"
|
||||
|
||||
if ! dns_ok "$host"; then
|
||||
log_line "DOWN" "$host" "Résolution impossible (getent hosts)"
|
||||
add_summary_line "$host" "DOWN" "DOWN - DNS"
|
||||
return 1
|
||||
fi
|
||||
|
||||
local http_code curl_exit stderr
|
||||
|
||||
local http_code curl_exit err
|
||||
local stderr
|
||||
stderr="$(mktemp)"
|
||||
|
||||
http_code="$(
|
||||
@@ -141,31 +166,33 @@ check_site() {
|
||||
--max-time "$MAX_TIME" \
|
||||
"$url" 2>"$stderr"
|
||||
)"
|
||||
|
||||
curl_exit=$?
|
||||
|
||||
if [ $curl_exit -ne 0 ]; then
|
||||
local err
|
||||
if [[ "$curl_exit" -ne 0 ]]; then
|
||||
err="$(head -n 1 "$stderr" | tr -d '\r')"
|
||||
rm -f "$stderr"
|
||||
|
||||
log_line "DOWN" "$host" "curl exit=$curl_exit : ${err:-"(aucun)"}"
|
||||
add_summary_line "$host" "DOWN" "DOWN - curl"
|
||||
return 1
|
||||
fi
|
||||
|
||||
rm -f "$stderr"
|
||||
|
||||
if [[ "$http_code" =~ ^[0-9]{3}$ ]]; then
|
||||
if [ "$http_code" -ge 200 ] && [ "$http_code" -le 399 ]; then
|
||||
if [[ "$http_code" -ge 200 && "$http_code" -le 399 ]]; then
|
||||
log_line "OK" "$host" "HTTP $http_code"
|
||||
add_summary_line "$host" "OK" "OK"
|
||||
return 0
|
||||
fi
|
||||
|
||||
log_line "DOWN" "$host" "HTTP $http_code (erreur appli)"
|
||||
add_summary_line "$host" "DOWN" "DOWN - HTTP $http_code"
|
||||
return 1
|
||||
fi
|
||||
|
||||
log_line "DOWN" "$host" "Code HTTP inattendu: $http_code"
|
||||
add_summary_line "$host" "DOWN" "DOWN - code HTTP invalide"
|
||||
return 1
|
||||
}
|
||||
|
||||
@@ -174,7 +201,6 @@ check_site() {
|
||||
#######################################
|
||||
|
||||
main() {
|
||||
|
||||
local failures=0
|
||||
|
||||
for site in "${SITES[@]}"; do
|
||||
@@ -183,7 +209,10 @@ main() {
|
||||
fi
|
||||
done
|
||||
|
||||
if [ "$failures" -gt 0 ]; then
|
||||
FAILURES="$failures"
|
||||
send_discord_summary
|
||||
|
||||
if [[ "$failures" -gt 0 ]]; then
|
||||
exit 2
|
||||
fi
|
||||
|
||||
|
||||
447
RecetteScripts/rebuild-bdd-recette.sh
Normal file
447
RecetteScripts/rebuild-bdd-recette.sh
Normal file
@@ -0,0 +1,447 @@
|
||||
#!/usr/bin/env bash
|
||||
set -euo pipefail
|
||||
|
||||
###############################################################################
|
||||
# rebuild-bdd-recette.sh
|
||||
#
|
||||
# Script de reconstruction d'une base PostgreSQL à partir d'un dump distant.
|
||||
#
|
||||
# Fonctionnement global :
|
||||
# 1. charge la configuration depuis le fichier .env ;
|
||||
# 2. prépare les chemins, logs et options SSH ;
|
||||
# 3. installe PostgreSQL si absent ;
|
||||
# 4. démarre PostgreSQL si nécessaire ;
|
||||
# 5. crée le rôle PGUSER uniquement si PostgreSQL vient d'être installé ;
|
||||
# 6. propose à l'utilisateur de choisir une base à reconstruire ;
|
||||
# 7. teste la connexion SSH au serveur distant ;
|
||||
# 8. recherche le dernier dump distant de la base choisie ;
|
||||
# 9. recherche le dernier fichier SQL des rôles dans le dossier "user" ;
|
||||
# 10. télécharge les fichiers nécessaires ;
|
||||
# 11. restaure les rôles via psql (avec filtrage des rôles sensibles) ;
|
||||
# 12. supprime puis recrée la base cible ;
|
||||
# 13. restaure la base choisie via pg_restore ;
|
||||
# 14. envoie une notification Discord si tout s'est bien passé.
|
||||
###############################################################################
|
||||
|
||||
###############################################################################
|
||||
# Chemins fixes du script
|
||||
###############################################################################
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
ENV_FILE="${SCRIPT_DIR}/.env"
|
||||
|
||||
###############################################################################
|
||||
# Vérification du fichier .env
|
||||
###############################################################################
|
||||
if [[ ! -f "$ENV_FILE" ]]; then
|
||||
echo "ERROR: fichier .env introuvable : $ENV_FILE" >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
###############################################################################
|
||||
# Chargement du .env
|
||||
###############################################################################
|
||||
set -a
|
||||
# shellcheck disable=SC1090
|
||||
source "$ENV_FILE"
|
||||
set +a
|
||||
|
||||
###############################################################################
|
||||
# Variables obligatoires
|
||||
###############################################################################
|
||||
: "${ENV_NAME:?Variable ENV_NAME manquante}"
|
||||
: "${PGHOST:?Variable PGHOST manquante}"
|
||||
: "${PGPORT:?Variable PGPORT manquante}"
|
||||
: "${PGUSER:?Variable PGUSER manquante}"
|
||||
: "${PGPASSWORD:?Variable PGPASSWORD manquante}"
|
||||
: "${DBS:?Variable DBS manquante}"
|
||||
: "${BACKUP_REMOTE_USER:?Variable BACKUP_REMOTE_USER manquante}"
|
||||
: "${BACKUP_REMOTE_HOST:?Variable BACKUP_REMOTE_HOST manquante}"
|
||||
: "${BACKUP_REMOTE_DIR:?Variable BACKUP_REMOTE_DIR manquante}"
|
||||
: "${SSH_KEY:?Variable SSH_KEY manquante}"
|
||||
: "${BACKUP_LOG_DIR:?Variable BACKUP_LOG_DIR manquante}"
|
||||
|
||||
###############################################################################
|
||||
# Variables optionnelles
|
||||
###############################################################################
|
||||
LOCAL_RESTORE_DIR="${LOCAL_RESTORE_DIR:-${SCRIPT_DIR}/restore_tmp}"
|
||||
REMOTE_ROLES_DIR_NAME="${REMOTE_ROLES_DIR_NAME:-user}"
|
||||
SSH_CONNECT_TIMEOUT="${SSH_CONNECT_TIMEOUT:-8}"
|
||||
DISCORD_WEBHOOK_URL="${DISCORD_WEBHOOK_URL:-}"
|
||||
|
||||
###############################################################################
|
||||
# Préparation des dossiers locaux
|
||||
###############################################################################
|
||||
mkdir -p "$BACKUP_LOG_DIR" || {
|
||||
echo "ERROR: impossible de créer le dossier de logs : $BACKUP_LOG_DIR" >&2
|
||||
exit 1
|
||||
}
|
||||
|
||||
mkdir -p "$LOCAL_RESTORE_DIR" || {
|
||||
echo "ERROR: impossible de créer le dossier local de restauration : $LOCAL_RESTORE_DIR" >&2
|
||||
exit 1
|
||||
}
|
||||
|
||||
TIMESTAMP="$(date '+%Y-%m-%d_%H-%M-%S')"
|
||||
LOG_FILE="${BACKUP_LOG_DIR}/restore_${ENV_NAME,,}_${TIMESTAMP}.log"
|
||||
|
||||
touch "$LOG_FILE" || {
|
||||
echo "ERROR: impossible d'écrire dans le fichier de log : $LOG_FILE" >&2
|
||||
exit 1
|
||||
}
|
||||
|
||||
###############################################################################
|
||||
# Fonctions utilitaires
|
||||
###############################################################################
|
||||
log() {
|
||||
echo "[$(date '+%Y-%m-%d %H:%M:%S')] $*" | tee -a "$LOG_FILE"
|
||||
}
|
||||
|
||||
fail() {
|
||||
log "ERROR: $*"
|
||||
exit 1
|
||||
}
|
||||
|
||||
cleanup() {
|
||||
rm -f \
|
||||
"${LOCAL_DB_DUMP_FILE:-}" \
|
||||
"${LOCAL_ROLES_FILE:-}" \
|
||||
"${FILTERED_ROLES_FILE:-}" \
|
||||
"${ROLES_CREATE_LIST:-}" \
|
||||
"${ROLES_APPLY_FILE:-}"
|
||||
}
|
||||
trap cleanup EXIT
|
||||
|
||||
require_cmd() {
|
||||
command -v "$1" >/dev/null 2>&1
|
||||
}
|
||||
|
||||
###############################################################################
|
||||
# Envoi Discord
|
||||
#
|
||||
# Envoi simple d'un message texte via webhook Discord.
|
||||
# Si WEBHOOK_URL n'est pas défini, on ignore silencieusement l'envoi.
|
||||
###############################################################################
|
||||
send_discord_message() {
|
||||
local message="$1"
|
||||
local payload=""
|
||||
|
||||
[[ -n "$DISCORD_WEBHOOK_URL" ]] || {
|
||||
log "WEBHOOK_URL non défini : notification Discord ignorée."
|
||||
return 0
|
||||
}
|
||||
|
||||
if ! require_cmd curl; then
|
||||
log "curl absent : notification Discord ignorée."
|
||||
return 0
|
||||
fi
|
||||
|
||||
payload="$(python3 -c 'import json,sys; print(json.dumps({"content": sys.argv[1]}))' "$message")" || {
|
||||
log "Impossible de construire le payload JSON Discord."
|
||||
return 0
|
||||
}
|
||||
|
||||
curl -sS -X POST "$DISCORD_WEBHOOK_URL" \
|
||||
-H "Content-Type: application/json" \
|
||||
-d "$payload" \
|
||||
>/dev/null || log "Échec d'envoi de la notification Discord."
|
||||
}
|
||||
|
||||
###############################################################################
|
||||
# Vérifications de base
|
||||
###############################################################################
|
||||
[[ -f "$SSH_KEY" ]] || fail "clé SSH introuvable : $SSH_KEY"
|
||||
[[ -r "$SSH_KEY" ]] || fail "clé SSH non lisible : $SSH_KEY"
|
||||
|
||||
export PGPASSWORD
|
||||
|
||||
SSH_OPTS=(
|
||||
-i "$SSH_KEY"
|
||||
-o IdentitiesOnly=yes
|
||||
-o BatchMode=yes
|
||||
-o ConnectTimeout="$SSH_CONNECT_TIMEOUT"
|
||||
-o StrictHostKeyChecking=accept-new
|
||||
)
|
||||
|
||||
REMOTE_SSH="${BACKUP_REMOTE_USER}@${BACKUP_REMOTE_HOST}"
|
||||
|
||||
###############################################################################
|
||||
# Installation PostgreSQL si absent
|
||||
#
|
||||
# Le rôle PGUSER est créé uniquement si PostgreSQL vient d'être installé.
|
||||
###############################################################################
|
||||
POSTGRES_INSTALLED=false
|
||||
|
||||
if ! require_cmd psql || ! require_cmd pg_restore || ! require_cmd createdb || ! require_cmd dropdb; then
|
||||
log "PostgreSQL absent : installation en cours..."
|
||||
|
||||
sudo apt update >>"$LOG_FILE" 2>&1 || fail "échec de apt update"
|
||||
sudo apt install -y postgresql postgresql-client postgresql-contrib \
|
||||
>>"$LOG_FILE" 2>&1 || fail "échec de l'installation de PostgreSQL"
|
||||
|
||||
POSTGRES_INSTALLED=true
|
||||
log "Installation PostgreSQL terminée."
|
||||
else
|
||||
log "PostgreSQL déjà installé."
|
||||
fi
|
||||
|
||||
###############################################################################
|
||||
# Démarrage PostgreSQL
|
||||
###############################################################################
|
||||
if ! sudo systemctl is-active --quiet postgresql; then
|
||||
log "Démarrage du service PostgreSQL..."
|
||||
sudo systemctl start postgresql >>"$LOG_FILE" 2>&1 || fail "impossible de démarrer PostgreSQL"
|
||||
else
|
||||
log "Service PostgreSQL déjà actif."
|
||||
fi
|
||||
|
||||
###############################################################################
|
||||
# Attente disponibilité PostgreSQL
|
||||
###############################################################################
|
||||
log "Vérification de la disponibilité de PostgreSQL..."
|
||||
for _ in {1..20}; do
|
||||
if sudo -u postgres psql -d postgres -c "SELECT 1;" >/dev/null 2>&1; then
|
||||
log "PostgreSQL répond correctement."
|
||||
break
|
||||
fi
|
||||
sleep 1
|
||||
done
|
||||
|
||||
if ! sudo -u postgres psql -d postgres -c "SELECT 1;" >/dev/null 2>&1; then
|
||||
fail "PostgreSQL ne répond pas correctement"
|
||||
fi
|
||||
|
||||
###############################################################################
|
||||
# Création du rôle PGUSER uniquement si PostgreSQL vient d'être installé
|
||||
###############################################################################
|
||||
if [[ "$POSTGRES_INSTALLED" == "true" ]]; then
|
||||
log "Création du rôle PostgreSQL ${PGUSER} suite à une installation neuve..."
|
||||
|
||||
sudo -u postgres psql -d postgres -c \
|
||||
"CREATE ROLE \"${PGUSER}\" WITH LOGIN SUPERUSER CREATEDB CREATEROLE PASSWORD '${PGPASSWORD}';" \
|
||||
>>"$LOG_FILE" 2>&1 || fail "échec de création du rôle ${PGUSER}"
|
||||
|
||||
log "Rôle PostgreSQL ${PGUSER} créé."
|
||||
fi
|
||||
|
||||
###############################################################################
|
||||
# Affichage des bases disponibles
|
||||
###############################################################################
|
||||
read -r -a DBS_ARRAY <<< "$DBS"
|
||||
|
||||
if [[ "${#DBS_ARRAY[@]}" -eq 0 ]]; then
|
||||
fail "aucune base définie dans DBS"
|
||||
fi
|
||||
|
||||
echo "Bases disponibles dans le .env :"
|
||||
for i in "${!DBS_ARRAY[@]}"; do
|
||||
printf ' %d) %s\n' "$((i + 1))" "${DBS_ARRAY[$i]}"
|
||||
done
|
||||
|
||||
echo
|
||||
read -r -p "Voulez-vous utiliser une base de cette liste ? (oui/non) : " USE_LIST
|
||||
|
||||
DB=""
|
||||
|
||||
if [[ "${USE_LIST,,}" == "oui" || "${USE_LIST,,}" == "o" ]]; then
|
||||
read -r -p "Sélectionnez le numéro de la base à restaurer : " DB_INDEX
|
||||
|
||||
[[ "$DB_INDEX" =~ ^[0-9]+$ ]] || fail "numéro invalide"
|
||||
(( DB_INDEX >= 1 && DB_INDEX <= ${#DBS_ARRAY[@]} )) || fail "numéro hors plage"
|
||||
|
||||
DB="${DBS_ARRAY[$((DB_INDEX - 1))]}"
|
||||
else
|
||||
read -r -p "Nom exact de la base à restaurer : " DB
|
||||
[[ -n "$DB" ]] || fail "nom de base vide"
|
||||
fi
|
||||
|
||||
log "Environnement : $ENV_NAME"
|
||||
log "Base cible sélectionnée : $DB"
|
||||
|
||||
###############################################################################
|
||||
# Test de connexion SSH
|
||||
###############################################################################
|
||||
log "Test de connexion SSH vers ${REMOTE_SSH}..."
|
||||
|
||||
SSH_TEST_OUTPUT=""
|
||||
if ! SSH_TEST_OUTPUT="$(ssh "${SSH_OPTS[@]}" "$REMOTE_SSH" "exit 0" 2>&1)"; then
|
||||
echo "$SSH_TEST_OUTPUT" | tee -a "$LOG_FILE" >&2
|
||||
fail "connexion SSH impossible vers ${REMOTE_SSH}"
|
||||
fi
|
||||
|
||||
###############################################################################
|
||||
# Définition des chemins distants
|
||||
###############################################################################
|
||||
REMOTE_DB_DIR="${BACKUP_REMOTE_DIR}/${DB}"
|
||||
REMOTE_ROLES_DIR="${BACKUP_REMOTE_DIR}/${REMOTE_ROLES_DIR_NAME}"
|
||||
|
||||
log "Recherche du dernier dump distant pour ${DB} dans : ${REMOTE_DB_DIR}"
|
||||
log "Recherche du dernier fichier de rôles dans : ${REMOTE_ROLES_DIR}"
|
||||
|
||||
###############################################################################
|
||||
# Recherche du dernier dump de base
|
||||
###############################################################################
|
||||
LAST_REMOTE_DB_DUMP="$(
|
||||
ssh "${SSH_OPTS[@]}" "$REMOTE_SSH" \
|
||||
"find '${REMOTE_DB_DIR}' -maxdepth 1 -type f -name '${DB}_*.dump' | LC_ALL=C sort | tail -n 1"
|
||||
)"
|
||||
|
||||
if [[ -z "$LAST_REMOTE_DB_DUMP" ]]; then
|
||||
fail "aucun dump trouvé pour la base ${DB} dans ${REMOTE_DB_DIR}"
|
||||
fi
|
||||
|
||||
log "Dernier dump distant sélectionné : ${LAST_REMOTE_DB_DUMP}"
|
||||
|
||||
###############################################################################
|
||||
# Recherche du dernier fichier SQL des rôles
|
||||
###############################################################################
|
||||
LAST_REMOTE_ROLES_FILE="$(
|
||||
ssh "${SSH_OPTS[@]}" "$REMOTE_SSH" \
|
||||
"find '${REMOTE_ROLES_DIR}' -maxdepth 1 -type f -name 'user_*.sql' | LC_ALL=C sort | tail -n 1"
|
||||
)"
|
||||
|
||||
if [[ -n "$LAST_REMOTE_ROLES_FILE" ]]; then
|
||||
log "Dernier fichier des rôles sélectionné : ${LAST_REMOTE_ROLES_FILE}"
|
||||
else
|
||||
log "Aucun fichier des rôles trouvé sur le serveur distant."
|
||||
fi
|
||||
|
||||
###############################################################################
|
||||
# Téléchargement du dump principal
|
||||
###############################################################################
|
||||
LOCAL_DB_DUMP_FILE="${LOCAL_RESTORE_DIR}/$(basename "$LAST_REMOTE_DB_DUMP")"
|
||||
LOCAL_ROLES_FILE=""
|
||||
|
||||
log "Téléchargement du dump..."
|
||||
scp "${SSH_OPTS[@]}" "${REMOTE_SSH}:${LAST_REMOTE_DB_DUMP}" "$LOCAL_DB_DUMP_FILE" \
|
||||
>>"$LOG_FILE" 2>&1 || fail "échec du téléchargement du dump principal"
|
||||
|
||||
###############################################################################
|
||||
# Téléchargement du fichier des rôles si présent
|
||||
###############################################################################
|
||||
if [[ -n "$LAST_REMOTE_ROLES_FILE" ]]; then
|
||||
LOCAL_ROLES_FILE="${LOCAL_RESTORE_DIR}/$(basename "$LAST_REMOTE_ROLES_FILE")"
|
||||
|
||||
log "Téléchargement du fichier des rôles..."
|
||||
scp "${SSH_OPTS[@]}" "${REMOTE_SSH}:${LAST_REMOTE_ROLES_FILE}" "$LOCAL_ROLES_FILE" \
|
||||
>>"$LOG_FILE" 2>&1 || fail "échec du téléchargement du fichier des rôles"
|
||||
else
|
||||
log "La restauration des rôles sera ignorée."
|
||||
fi
|
||||
|
||||
###############################################################################
|
||||
# Test de connexion PostgreSQL locale avec PGUSER
|
||||
###############################################################################
|
||||
if ! psql -h "$PGHOST" -p "$PGPORT" -U "$PGUSER" -d postgres -c "SELECT 1;" \
|
||||
>>"$LOG_FILE" 2>&1; then
|
||||
fail "connexion PostgreSQL locale impossible avec PGUSER=${PGUSER}"
|
||||
fi
|
||||
|
||||
###############################################################################
|
||||
# Demande d'écrasement si la base existe déjà
|
||||
###############################################################################
|
||||
DB_EXISTS="$(
|
||||
psql -h "$PGHOST" -p "$PGPORT" -U "$PGUSER" -d postgres -tAc \
|
||||
"SELECT 1 FROM pg_database WHERE datname='${DB}'" 2>>"$LOG_FILE" || true
|
||||
)"
|
||||
|
||||
if [[ "$DB_EXISTS" == "1" ]]; then
|
||||
read -r -p "La base '${DB}' existe déjà. Voulez-vous l'écraser ? (oui/non) : " CONFIRM_OVERWRITE
|
||||
if [[ "${CONFIRM_OVERWRITE,,}" != "oui" && "${CONFIRM_OVERWRITE,,}" != "o" ]]; then
|
||||
fail "restauration annulée par l'utilisateur"
|
||||
fi
|
||||
|
||||
log "Suppression de la base existante : ${DB}"
|
||||
dropdb -h "$PGHOST" -p "$PGPORT" -U "$PGUSER" --if-exists "$DB" \
|
||||
>>"$LOG_FILE" 2>&1 || fail "échec de suppression de la base ${DB}"
|
||||
fi
|
||||
|
||||
###############################################################################
|
||||
# Restauration des rôles
|
||||
###############################################################################
|
||||
if [[ -n "$LOCAL_ROLES_FILE" ]]; then
|
||||
log "Restauration des rôles depuis : ${LOCAL_ROLES_FILE}"
|
||||
|
||||
FILTERED_ROLES_FILE="${LOCAL_RESTORE_DIR}/filtered_$(basename "$LOCAL_ROLES_FILE")"
|
||||
ROLES_CREATE_LIST="${LOCAL_RESTORE_DIR}/roles_to_create_$(basename "$LOCAL_ROLES_FILE")"
|
||||
ROLES_APPLY_FILE="${LOCAL_RESTORE_DIR}/roles_apply_$(basename "$LOCAL_ROLES_FILE")"
|
||||
|
||||
grep -viE '^(CREATE ROLE|ALTER ROLE) (backup_liot|postgres)\b' "$LOCAL_ROLES_FILE" \
|
||||
> "$FILTERED_ROLES_FILE" || true
|
||||
|
||||
log "Fichier des rôles filtré généré : ${FILTERED_ROLES_FILE}"
|
||||
|
||||
sed -nE 's/^CREATE ROLE "?([^" ;]+)"?;$/\1/p' "$FILTERED_ROLES_FILE" \
|
||||
> "$ROLES_CREATE_LIST" || true
|
||||
|
||||
if [[ -s "$ROLES_CREATE_LIST" ]]; then
|
||||
while IFS= read -r role_name; do
|
||||
[[ -z "$role_name" ]] && continue
|
||||
|
||||
ROLE_EXISTS="$(
|
||||
psql -h "$PGHOST" -p "$PGPORT" -U "$PGUSER" -d postgres -tAc \
|
||||
"SELECT 1 FROM pg_roles WHERE rolname='${role_name}'" 2>>"$LOG_FILE" || true
|
||||
)"
|
||||
|
||||
if [[ "$ROLE_EXISTS" != "1" ]]; then
|
||||
log "Création du rôle manquant : ${role_name}"
|
||||
psql -v ON_ERROR_STOP=1 \
|
||||
-h "$PGHOST" -p "$PGPORT" -U "$PGUSER" -d postgres \
|
||||
-c "CREATE ROLE \"${role_name}\";" \
|
||||
>>"$LOG_FILE" 2>&1 || fail "échec de création du rôle ${role_name}"
|
||||
else
|
||||
log "Rôle déjà présent, création ignorée : ${role_name}"
|
||||
fi
|
||||
done < "$ROLES_CREATE_LIST"
|
||||
fi
|
||||
|
||||
grep -viE '^CREATE ROLE ' "$FILTERED_ROLES_FILE" > "$ROLES_APPLY_FILE" || true
|
||||
|
||||
log "Application des ALTER ROLE / privilèges / memberships..."
|
||||
psql \
|
||||
-v ON_ERROR_STOP=1 \
|
||||
-h "$PGHOST" \
|
||||
-p "$PGPORT" \
|
||||
-U "$PGUSER" \
|
||||
-d postgres \
|
||||
-f "$ROLES_APPLY_FILE" \
|
||||
>>"$LOG_FILE" 2>&1 || fail "échec de restauration des rôles via psql"
|
||||
else
|
||||
log "Aucune restauration des rôles effectuée."
|
||||
fi
|
||||
|
||||
###############################################################################
|
||||
# Création de la base
|
||||
###############################################################################
|
||||
log "Création de la base : ${DB}"
|
||||
createdb -h "$PGHOST" -p "$PGPORT" -U "$PGUSER" "$DB" \
|
||||
>>"$LOG_FILE" 2>&1 || fail "échec de création de la base ${DB}"
|
||||
|
||||
###############################################################################
|
||||
# Restauration de la base principale
|
||||
###############################################################################
|
||||
log "Restauration de la base ${DB}..."
|
||||
pg_restore \
|
||||
-h "$PGHOST" \
|
||||
-p "$PGPORT" \
|
||||
-U "$PGUSER" \
|
||||
-d "$DB" \
|
||||
--clean \
|
||||
--if-exists \
|
||||
--no-owner \
|
||||
--no-privileges \
|
||||
"$LOCAL_DB_DUMP_FILE" \
|
||||
>>"$LOG_FILE" 2>&1 || fail "échec de restauration de la base ${DB}"
|
||||
|
||||
###############################################################################
|
||||
# Fin
|
||||
###############################################################################
|
||||
log "Restauration terminée avec succès pour la base : ${DB}"
|
||||
log "Fichier de log : ${LOG_FILE}"
|
||||
|
||||
SUCCESS_MESSAGE="✅ REBUILD BDD ${ENV_NAME}
|
||||
Base restaurée : ${DB}
|
||||
Hôte PostgreSQL : ${PGHOST}:${PGPORT}
|
||||
Dump utilisé : $(basename "$LAST_REMOTE_DB_DUMP")
|
||||
Log : ${LOG_FILE}"
|
||||
|
||||
send_discord_message "$SUCCESS_MESSAGE"
|
||||
Reference in New Issue
Block a user