#!/usr/bin/env bash
set -euo pipefail

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
cd "$ROOT_DIR"

if [[ "${1:-}" != "--force" || -z "${2:-}" ]]; then
  echo "用法: bash scripts/restore.sh --force <backup.sql|backup.sql.gz>" >&2
  exit 1
fi

BACKUP_FILE="$2"
if [[ ! -f "$BACKUP_FILE" ]]; then
  echo "备份文件不存在: $BACKUP_FILE" >&2
  exit 1
fi

if [[ -f ".env.local" ]]; then
  # shellcheck disable=SC1091
  source ".env.local"
fi
if [[ -f ".env" ]]; then
  # shellcheck disable=SC1091
  source ".env"
fi

DB_URL="${DATABASE_URL:-host=/var/run/postgresql dbname=llm_intelligence user=long sslmode=disable}"

echo "开始恢复到目标数据库..."
psql "$DB_URL" -v ON_ERROR_STOP=1 -c "DROP SCHEMA IF EXISTS public CASCADE; CREATE SCHEMA public;"

if [[ "$BACKUP_FILE" == *.gz ]]; then
  gzip -dc "$BACKUP_FILE" | psql "$DB_URL" -v ON_ERROR_STOP=1
else
  psql "$DB_URL" -v ON_ERROR_STOP=1 -f "$BACKUP_FILE"
fi

echo "恢复完成: $BACKUP_FILE"