diff --git a/BackupVaultWarden/backup-vaultwarden.sh b/BackupVaultWarden/backup-vaultwarden.sh index 8947f9b..8d226bf 100644 --- a/BackupVaultWarden/backup-vaultwarden.sh +++ b/BackupVaultWarden/backup-vaultwarden.sh @@ -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")" diff --git a/RecetteScripts/backup-bdd-recette.sh b/RecetteScripts/backup-bdd-recette.sh index 03a1a15..7809cad 100644 --- a/RecetteScripts/backup-bdd-recette.sh +++ b/RecetteScripts/backup-bdd-recette.sh @@ -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 </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 diff --git a/RecetteScripts/check-statut-recette.sh b/RecetteScripts/check-statut-recette.sh index 380a30f..23b1d65 100644 --- a/RecetteScripts/check-statut-recette.sh +++ b/RecetteScripts/check-statut-recette.sh @@ -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 diff --git a/RecetteScripts/rebuild-bdd-recette.sh b/RecetteScripts/rebuild-bdd-recette.sh new file mode 100644 index 0000000..4e61a95 --- /dev/null +++ b/RecetteScripts/rebuild-bdd-recette.sh @@ -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" \ No newline at end of file