Upload ai_studio_code 4.sh

ai_studio_code 4.sh

@@ -0,0 +1,108 @@
+#!/bin/bash
+
+# === 配置 ===
+WEBDAV_URL=${WEBDAV_URL%/}
+FULL_WEBDAV_URL="${WEBDAV_URL}"
+TEMP_DIR="/tmp/backup_temp"
+# 我们要备份的目标文件名
+TARGET_FILENAME="accounts.json"
+# 远程保存的名字
+REMOTE_FILENAME="antigravity_accounts.json"
+
+# 日志函数
+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"
+}
+
+# === 检查环境变量 ===
+if [[ -z "$WEBDAV_URL" ]]; then
+    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
+fi
+
+# === 1. 恢复备份 ===
+download_backup() {
+    log "正在检查远程备份..."
+    REMOTE_FILE="$FULL_WEBDAV_URL/$REMOTE_FILENAME"
+    
+    # 默认恢复路径 (如果还没生成，我们先恢复到 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")
+    
+    if [[ "$HTTP_CODE" == "200" ]]; then
+        log "发现远程备份，正在恢复到 $RESTORE_PATH"
+        mv "$TEMP_DIR/restored.json" "$RESTORE_PATH"
+        
+        # 为了保险，也在根目录放一份
+        cp "$RESTORE_PATH" "/app/$TARGET_FILENAME"
+    else
+        log "未找到远程备份 (HTTP $HTTP_CODE)，跳过恢复"
+    fi
+}
+
+# === 2. 循环备份 ===
+loop_sync() {
+    mkdir -p $TEMP_DIR
+    
+    while true; do
+        # 每 30 秒检查一次，反应快一点
+        sleep 30
+        
+        # 动态寻找文件位置
+        CURRENT_FILE_PATH=$(find_target_path)
+        
+        if [ -n "$CURRENT_FILE_PATH" ]; then
+            # 检查文件大小，忽略空文件
+            FILE_SIZE=$(wc -c < "$CURRENT_FILE_PATH")
+            if [ "$FILE_SIZE" -lt 5 ]; then
+                 # log "文件存在但为空，跳过..."
+                 continue
+            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