Update sync.sh

sync.sh

@@ -1,108 +1,123 @@
 #!/bin/bash
 
-# === 配置 ===
+# ================= 配置 =================
 WEBDAV_URL=${WEBDAV_URL%/}
-FULL_WEBDAV_URL="${WEBDAV_URL}"
-TEMP_DIR="/tmp/backup_temp"
-# 我们要备份的目标文件名
+WEBDAV_BACKUP_PATH=${WEBDAV_BACKUP_PATH:-""}
+WEBDAV_BACKUP_PATH=${WEBDAV_BACKUP_PATH#/}
+WEBDAV_BACKUP_PATH=${WEBDAV_BACKUP_PATH%/}
+
+if [ -n "$WEBDAV_BACKUP_PATH" ]; then
+    FULL_WEBDAV_URL="${WEBDAV_URL}/${WEBDAV_BACKUP_PATH}"
+else
+    FULL_WEBDAV_URL="${WEBDAV_URL}"
+fi
+
+SYNC_INTERVAL=${SYNC_INTERVAL:-60}
 TARGET_FILENAME="accounts.json"
-# 远程保存的名字
 REMOTE_FILENAME="antigravity_accounts.json"
+TEMP_DIR="/tmp/backup_temp"
+
+# =======================================
 
-# 日志函数
 log() { echo "[Backup] [$(date '+%H:%M:%S')] $1"; }
 
-# === 核心函数：寻找文件位置 ===
-find_target_path() {
-    # 1. 先猜它在 data 目录下 (最常见的情况)
-    if [ -f "/app/data/$TARGET_FILENAME" ]; then
-        FOUND_PATH="/app/data/$TARGET_FILENAME"
-    # 2. 再猜它在根目录下
-    elif [ -f "/app/$TARGET_FILENAME" ]; then
-        FOUND_PATH="/app/$TARGET_FILENAME"
-    # 3. 实在找不到，用 find 命令全盘搜 (排除 node_modules)
-    else
-        FOUND_PATH=$(find /app -name "$TARGET_FILENAME" -type f -not -path "*/node_modules/*" | head -n 1)
-    fi
-    
-    echo "$FOUND_PATH"
-}
+# 确保临时目录存在
+mkdir -p "$TEMP_DIR"
 
-# === 检查环境变量 ===
+# 检查环境变量
 if [[ -z "$WEBDAV_URL" ]]; then
-    log "未设置 WEBDAV_URL，跳过备份功能"
+    log "未设置 WEBDAV_URL，跳过备份"
     exit 0
 fi
 
-# 安装 curl
 if ! command -v curl &> /dev/null; then
-    apk add --no-cache curl || apt-get update && apt-get install -y curl
+    apk add --no-cache curl || (apt-get update && apt-get install -y curl)
 fi
 
-# === 1. 恢复备份 ===
-download_backup() {
-    log "正在检查远程备份..."
+# === 寻找目标文件 ===
+find_target_path() {
+    # 优先找根目录 (这是 antigravity 最常见的默认位置)
+    if [ -f "/app/$TARGET_FILENAME" ]; then
+        echo "/app/$TARGET_FILENAME"
+    elif [ -f "/app/data/$TARGET_FILENAME" ]; then
+        echo "/app/data/$TARGET_FILENAME"
+    else
+        # 默认回落到根目录，方便恢复时使用
+        echo "/app/$TARGET_FILENAME"
+    fi
+}
+
+# === 功能1：恢复备份 ===
+restore() {
+    log "正在从 WebDAV 恢复备份..."
     REMOTE_FILE="$FULL_WEBDAV_URL/$REMOTE_FILENAME"
+    TEMP_FILE="$TEMP_DIR/restored.json"
     
-    # 默认恢复路径 (如果还没生成，我们先恢复到 data 目录，因为 Dockerfile 里创建了这个目录)
-    RESTORE_DIR="/app/data"
-    mkdir -p "$RESTORE_DIR"
-    RESTORE_PATH="$RESTORE_DIR/$TARGET_FILENAME"
-    
-    HTTP_CODE=$(curl -s -o "$TEMP_DIR/restored.json" -w "%{http_code}" -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$REMOTE_FILE")
+    # 尝试下载
+    HTTP_CODE=$(curl -s -o "$TEMP_FILE" -w "%{http_code}" -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$REMOTE_FILE")
     
     if [[ "$HTTP_CODE" == "200" ]]; then
-        log "发现远程备份，正在恢复到 $RESTORE_PATH"
-        mv "$TEMP_DIR/restored.json" "$RESTORE_PATH"
-        
-        # 为了保险，也在根目录放一份
-        cp "$RESTORE_PATH" "/app/$TARGET_FILENAME"
+        # 关键修复：检查文件是否真的下载下来了且不为空
+        if [ -s "$TEMP_FILE" ]; then
+            # 恢复到 /app/accounts.json (这是程序读取的默认位置)
+            DEST_PATH="/app/$TARGET_FILENAME"
+            
+            cp "$TEMP_FILE" "$DEST_PATH"
+            
+            # 同时复制到 data 目录，防止不同版本路径差异
+            mkdir -p /app/data
+            cp "$TEMP_FILE" "/app/data/$TARGET_FILENAME"
+            
+            log "✅ 备份已恢复到: $DEST_PATH"
+        else
+            log "❌ 下载返回200但文件为空，跳过恢复"
+        fi
     else
-        log "未找到远程备份 (HTTP $HTTP_CODE)，跳过恢复"
+        log "📭 未找到远程备份 (HTTP $HTTP_CODE)，将使用新配置启动"
     fi
 }
 
-# === 2. 循环备份 ===
-loop_sync() {
-    mkdir -p $TEMP_DIR
+# === 功能2：循环备份 ===
+loop() {
+    log "启动增量备份服务 (间隔: ${SYNC_INTERVAL}秒)"
+    
+    # 如果设置了子目录，尝试创建
+    if [ -n "$WEBDAV_BACKUP_PATH" ]; then
+        curl -s -o /dev/null -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -X MKCOL "$FULL_WEBDAV_URL/"
+    fi
     
     while true; do
-        # 每 30 秒检查一次，反应快一点
-        sleep 30
+        sleep $SYNC_INTERVAL
         
-        # 动态寻找文件位置
-        CURRENT_FILE_PATH=$(find_target_path)
+        # 动态查找文件
+        CURRENT_PATH=$(find /app -name "$TARGET_FILENAME" -type f -not -path "*/node_modules/*" | head -n 1)
         
-        if [ -n "$CURRENT_FILE_PATH" ]; then
-            # 检查文件大小，忽略空文件
-            FILE_SIZE=$(wc -c < "$CURRENT_FILE_PATH")
-            if [ "$FILE_SIZE" -lt 5 ]; then
-                 # log "文件存在但为空，跳过..."
-                 continue
+        if [ -n "$CURRENT_PATH" ]; then
+            FILE_SIZE=$(wc -c < "$CURRENT_PATH")
+            # 只有文件大于 5 字节才备份 (避免备份空的 [])
+            if [ "$FILE_SIZE" -gt 5 ]; then
+                log "正在备份: $CURRENT_PATH ($FILE_SIZE bytes)"
+                HTTP_CODE=$(curl -s -o /dev/null -w "%{http_code}" -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -T "$CURRENT_PATH" "$FULL_WEBDAV_URL/$REMOTE_FILENAME")
+                
+                if [[ "$HTTP_CODE" != "201" ]] && [[ "$HTTP_CODE" != "204" ]] && [[ "$HTTP_CODE" != "200" ]]; then
+                    log "备份上传失败: HTTP $HTTP_CODE"
+                fi
             fi
-
-            log "找到文件: $CURRENT_FILE_PATH (大小: $FILE_SIZE)"
-            
-            # 开始上传
-            REMOTE_FILE="$FULL_WEBDAV_URL/$REMOTE_FILENAME"
-            HTTP_CODE=$(curl -s -o /dev/null -w "%{http_code}" -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -T "$CURRENT_FILE_PATH" "$REMOTE_FILE")
-            
-            if [[ "$HTTP_CODE" == "201" ]] || [[ "$HTTP_CODE" == "204" ]] || [[ "$HTTP_CODE" == "200" ]]; then
-                log "备份成功！"
-            else
-                log "备份失败 (HTTP $HTTP_CODE)"
-            fi
-            
-        else
-            log "⚠️ 找不到 $TARGET_FILENAME"
-            log "--- 调试信息：当前 /app 目录下的文件结构 ---"
-            # 列出所有 JSON 文件，帮你看看到底生成在哪了
-            find /app -maxdepth 3 -name "*.json" -not -path "*/node_modules/*"
-            log "----------------------------------------"
         fi
     done
 }
 
-# === 主流程 ===
-download_backup
-loop_sync
+# === 入口判断 ===
+case "$1" in
+    "restore")
+        restore
+        ;;
+    "loop")
+        loop
+        ;;
+    *)
+        # 如果没有参数，默认先恢复再循环 (兼容旧用法，但不推荐，因为会阻塞)
+        restore
+        loop
+        ;;
+esac