Marek Miklewicz
212 lines
7.0 KiB
Bash
Executable File
212 lines
7.0 KiB
Bash
Executable File
#!/bin/bash
|
||
# =============================================================================
|
||
# restore_all.sh
|
||
# Przywracanie wszystkich baz PostgreSQL z katalogu backupu
|
||
# wykonanego przez postgres_backup.sh
|
||
# =============================================================================
|
||
|
||
CREDENTIALS_FILE="/usr/local/directadmin/conf/postgresql.conf"
|
||
|
||
# Kolory tylko na terminalu
|
||
if [[ -t 1 ]]; then
|
||
RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'
|
||
BOLD='\033[1m'; CYAN='\033[0;36m'; NC='\033[0m'
|
||
else
|
||
RED=''; GREEN=''; YELLOW=''; BOLD=''; CYAN=''; NC=''
|
||
fi
|
||
|
||
ok() { echo -e "${GREEN}[OK]${NC} $*"; }
|
||
info() { echo -e " $*"; }
|
||
warn() { echo -e "${YELLOW}[WARN]${NC} $*"; }
|
||
die() { echo -e "${RED}[BŁĄD]${NC} $*" >&2; exit 1; }
|
||
sekcja() { echo -e "\n${BOLD}${CYAN}--- $* ---${NC}\n"; }
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Sposób użycia
|
||
# ---------------------------------------------------------------------------
|
||
usage() {
|
||
cat <<EOF
|
||
|
||
Użycie:
|
||
$(basename "$0") <katalog_backupu> [--restore-users]
|
||
|
||
Opis:
|
||
Przywraca wszystkie bazy danych z podanego katalogu backupu.
|
||
Pliki w katalogu powinny być wykonane przez postgres_backup.sh.
|
||
|
||
Flagi:
|
||
--restore-users Przed przywracaniem baz odtworzy użytkowników, role
|
||
i ich hasła z pliku _globals.sql[.gz] (jeśli istnieje).
|
||
WYMAGANE do pełnego disaster recovery.
|
||
|
||
Kolejność przywracania:
|
||
1. _globals.sql[.gz] – użytkownicy i role (tylko z --restore-users)
|
||
2. *.sql[.gz] – bazy danych (alfabetycznie, z pominięciem _globals)
|
||
|
||
Przykłady:
|
||
$(basename "$0") /home/admin/postgres_backup/19-02-2026
|
||
$(basename "$0") /home/admin/postgres_backup/19-02-2026 --restore-users
|
||
|
||
Uwaga:
|
||
Każda przywracana baza zostanie usunięta i odtworzona od nowa.
|
||
Plik poświadczeń: ${CREDENTIALS_FILE}
|
||
|
||
EOF
|
||
}
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Parsowanie argumentów
|
||
# ---------------------------------------------------------------------------
|
||
BACKUP_DIR=""
|
||
RESTORE_USERS=false
|
||
|
||
for ARG in "$@"; do
|
||
case "$ARG" in
|
||
--restore-users) RESTORE_USERS=true ;;
|
||
--help|-h) usage; exit 0 ;;
|
||
-*) die "Nieznana flaga: ${ARG}. Użyj --help aby zobaczyć pomoc." ;;
|
||
*)
|
||
[[ -z "$BACKUP_DIR" ]] || die "Podano więcej niż jeden katalog. Użyj --help aby zobaczyć pomoc."
|
||
BACKUP_DIR="$ARG"
|
||
;;
|
||
esac
|
||
done
|
||
|
||
if [[ -z "$BACKUP_DIR" ]]; then
|
||
usage
|
||
exit 0
|
||
fi
|
||
|
||
[[ -d "$BACKUP_DIR" ]] || die "Katalog nie istnieje: ${BACKUP_DIR}"
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Poświadczenia
|
||
# ---------------------------------------------------------------------------
|
||
[[ -f "$CREDENTIALS_FILE" ]] \
|
||
|| die "Brak pliku poświadczeń: ${CREDENTIALS_FILE}"
|
||
|
||
_perms=$(stat -c '%a' "$CREDENTIALS_FILE")
|
||
[[ "$_perms" == "600" || "$_perms" == "400" ]] \
|
||
|| die "Zbyt otwarte uprawnienia na ${CREDENTIALS_FILE} (${_perms}). Wymagane: 600 lub 400."
|
||
|
||
_line=$(head -1 "$CREDENTIALS_FILE")
|
||
PGHOST=$(echo "$_line" | cut -d: -f1)
|
||
PGPORT=$(echo "$_line" | cut -d: -f2)
|
||
PGUSER=$(echo "$_line" | cut -d: -f4)
|
||
|
||
[[ -n "$PGHOST" && -n "$PGPORT" && -n "$PGUSER" ]] \
|
||
|| die "Nieprawidłowy format pliku poświadczeń. Oczekiwany: host:port:database:user:password"
|
||
|
||
export PGPASSFILE="$CREDENTIALS_FILE"
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Funkcja przywracająca jeden plik
|
||
# ---------------------------------------------------------------------------
|
||
restore_file() {
|
||
local file="$1"
|
||
local init_db="$2"
|
||
|
||
case "$file" in
|
||
*.sql.gz)
|
||
gunzip -c "$file" \
|
||
| psql -h "$PGHOST" -p "$PGPORT" -U "$PGUSER" \
|
||
--no-password -d "$init_db"
|
||
return ${PIPESTATUS[1]}
|
||
;;
|
||
*.sql)
|
||
psql -h "$PGHOST" -p "$PGPORT" -U "$PGUSER" \
|
||
--no-password -d "$init_db" -f "$file"
|
||
return $?
|
||
;;
|
||
esac
|
||
}
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Start
|
||
# ---------------------------------------------------------------------------
|
||
echo ""
|
||
info "Katalog: ${BACKUP_DIR}"
|
||
info "Serwer: ${PGHOST}:${PGPORT} Użytkownik: ${PGUSER}"
|
||
info "Restore users: ${RESTORE_USERS}"
|
||
echo ""
|
||
|
||
ERRORS=0
|
||
RESTORED=0
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 1. Globals (użytkownicy, role) – tylko z flagą --restore-users
|
||
# ---------------------------------------------------------------------------
|
||
if [[ "$RESTORE_USERS" == "true" ]]; then
|
||
sekcja "Przywracanie użytkowników i ról (_globals)"
|
||
|
||
GLOBALS_FILE=""
|
||
for _f in "${BACKUP_DIR}/_globals.sql.gz" "${BACKUP_DIR}/_globals.sql"; do
|
||
[[ -f "$_f" ]] && GLOBALS_FILE="$_f" && break
|
||
done
|
||
|
||
if [[ -z "$GLOBALS_FILE" ]]; then
|
||
warn "Nie znaleziono pliku _globals.sql[.gz] w ${BACKUP_DIR} – pomijam użytkowników."
|
||
else
|
||
info "Plik: ${GLOBALS_FILE##*/}"
|
||
# Globals przywracamy do postgres; DROP/CREATE ROLE może generować ostrzeżenia
|
||
# jeśli role już istnieją – to normalne, kontynuujemy
|
||
if restore_file "$GLOBALS_FILE" "postgres"; then
|
||
ok "Użytkownicy i role przywrócone."
|
||
RESTORED=$((RESTORED + 1))
|
||
else
|
||
warn "Przywracanie globals zakończone z ostrzeżeniami – sprawdź output powyżej."
|
||
ERRORS=$((ERRORS + 1))
|
||
fi
|
||
fi
|
||
fi
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# 2. Bazy danych (pliki *.sql i *.sql.gz z pominięciem _globals.*)
|
||
# ---------------------------------------------------------------------------
|
||
sekcja "Przywracanie baz danych"
|
||
|
||
# Zbierz pliki, posortuj alfabetycznie
|
||
mapfile -t DB_FILES < <(
|
||
find "$BACKUP_DIR" -maxdepth 1 -type f \( -name '*.sql' -o -name '*.sql.gz' \) \
|
||
! -name '_globals.*' \
|
||
| sort
|
||
)
|
||
|
||
if [[ ${#DB_FILES[@]} -eq 0 ]]; then
|
||
warn "Nie znaleziono plików .sql ani .sql.gz w katalogu ${BACKUP_DIR} (poza _globals)."
|
||
exit 0
|
||
fi
|
||
|
||
for FILE in "${DB_FILES[@]}"; do
|
||
DB_NAME=$(basename "$FILE" | sed 's/\.sql\.gz$//; s/\.sql$//')
|
||
|
||
# Baza 'postgres' nie może być przywracana przez siebie (DROP DATABASE blokuje aktywne połączenie)
|
||
if [[ "$DB_NAME" == "postgres" ]]; then
|
||
INIT_DB="template1"
|
||
else
|
||
INIT_DB="postgres"
|
||
fi
|
||
|
||
info "→ ${DB_NAME} (${FILE##*/})"
|
||
|
||
if restore_file "$FILE" "$INIT_DB"; then
|
||
ok "${DB_NAME}: przywrócona."
|
||
RESTORED=$((RESTORED + 1))
|
||
else
|
||
warn "${DB_NAME}: przywracanie zakończone z błędem lub ostrzeżeniami."
|
||
ERRORS=$((ERRORS + 1))
|
||
fi
|
||
|
||
echo ""
|
||
done
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# Podsumowanie
|
||
# ---------------------------------------------------------------------------
|
||
sekcja "Podsumowanie"
|
||
info "Przywrócono: ${RESTORED} | Błędy: ${ERRORS}"
|
||
echo ""
|
||
|
||
[[ "$ERRORS" -eq 0 ]] || exit 1
|
||
exit 0
|