Create scripts/restore.sh

scripts/restore.sh

@@ -0,0 +1,63 @@
+#!/bin/bash
+set -e
+
+echo "[Restore] Data restoration script started."
+
+# --- 变量与暂存区定义 ---
+LIVE_DATA_DIR="/app/server/storage"
+LIVE_DB_FILE="/app/server/anythingllm.db"
+STAGE_DIR="/tmp/restore_stage"
+RESTORE_SOURCE="" # 将用于标记最终从哪里恢复
+
+# --- 步骤 1: 强制清理线上旧数据和暂存区 ---
+echo "[Restore] Forcing a clean slate..."
+rm -rf "${LIVE_DATA_DIR}" "${LIVE_DB_FILE}" "${STAGE_DIR}"
+mkdir -p "${STAGE_DIR}"
+
+# --- 步骤 2: 尝试从 WebDAV 恢复到暂存区 ---
+if [ -n "${WEBDAV_URL}" ]; then
+    echo "[Restore] Attempting to restore from WebDAV to staging area..."
+    if rclone copy "webdav:anythingllm_backup/" "${STAGE_DIR}/" --max-depth 20; then
+        # 检查关键文件是否存在，验证恢复是否有效
+        if [ -f "${STAGE_DIR}/anythingllm.db" ]; then
+            echo "[Restore] WebDAV restore to staging area successful."
+            RESTORE_SOURCE="${STAGE_DIR}"
+        else
+            echo "[Restore] WebDAV backup seems incomplete. Ignoring."
+        fi
+    else
+        echo "[Restore] WebDAV restore failed."
+    fi
+fi
+
+# --- 步骤 3: 如果 WebDAV 未成功，尝试从 GitHub 恢复 ---
+if [ -z "${RESTORE_SOURCE}" ] && [ -n "${GITHUB_REPO}" ]; then
+    echo "[Restore] Attempting to restore from GitHub to staging area..."
+    if git clone https://${GITHUB_TOKEN}@github.com/${GITHUB_REPO##*//github.com/} "${STAGE_DIR}"; then
+        if [ -f "${STAGE_DIR}/anythingllm.db" ]; then
+            echo "[Restore] GitHub restore to staging area successful."
+            RESTORE_SOURCE="${STAGE_DIR}"
+        else
+            echo "[Restore] GitHub backup seems incomplete. Ignoring."
+        fi
+    else
+        echo "[Restore] GitHub restore failed."
+    fi
+fi
+
+# --- 步骤 4: 从选定的恢复源部署到线上位置 ---
+if [ -n "${RESTORE_SOURCE}" ]; then
+    echo "[Restore] Deploying data from ${RESTORE_SOURCE} to live location..."
+    # 确保线上目录存在
+    mkdir -p "${LIVE_DATA_DIR}"
+    # 使用 rsync 安全地移动文件
+    rsync -a "${RESTORE_SOURCE}/storage/" "${LIVE_DATA_DIR}/"
+    mv "${RESTORE_SOURCE}/anythingllm.db" "${LIVE_DB_FILE}"
+    echo "[Restore] Deployment successful."
+else
+    echo "[Restore] No valid remote backups found to restore from. Starting fresh."
+fi
+
+# --- 步骤 5: 清理暂存区 ---
+echo "[Restore] Cleaning up staging area..."
+rm -rf "${STAGE_DIR}"