Upload ai_studio_code.sh

ai_studio_code.sh

@@ -0,0 +1,98 @@
+#!/bin/bash
+
+# ================= 配置区域 =================
+# 1. WebDAV URL 处理
+WEBDAV_URL=${WEBDAV_URL%/}
+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
+
+# 2. 同步间隔 (默认 10 分钟 = 600 秒)
+SYNC_INTERVAL=${SYNC_INTERVAL:-600}
+
+# 3. 文件名配置
+BACKUP_NAME="gcli2api_creds.tar.gz"
+TEMP_FILE="/tmp/$BACKUP_NAME"
+TARGET_DIR="creds"
+
+# ===========================================
+
+log() { echo "[Backup] [$(date '+%H:%M:%S')] $1"; }
+
+# 检查环境
+if [[ -z "$WEBDAV_URL" ]]; then
+    log "未设置 WEBDAV_URL，跳过备份功能"
+    exit 0
+fi
+
+# 确保目标目录存在
+mkdir -p "$TARGET_DIR"
+
+# === 功能1: 恢复备份 ===
+restore() {
+    log "正在检查远程备份..."
+    REMOTE_FILE="$FULL_WEBDAV_URL/$BACKUP_NAME"
+    
+    # 下载到临时文件
+    HTTP_CODE=$(curl -s -o "$TEMP_FILE" -w "%{http_code}" -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$REMOTE_FILE")
+    
+    if [[ "$HTTP_CODE" == "200" ]]; then
+        if [ -s "$TEMP_FILE" ]; then
+            log "发现备份，正在恢复到 $TARGET_DIR..."
+            # 解压覆盖
+            tar -xzf "$TEMP_FILE"
+            log "✅ 恢复完成"
+        else
+            log "❌ 下载文件为空，跳过"
+        fi
+    else
+        log "📭 未找到远程备份 (HTTP $HTTP_CODE)，使用新环境启动"
+    fi
+}
+
+# === 功能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
+        sleep $SYNC_INTERVAL
+        
+        # 检查 creds 目录是否有文件
+        if [ "$(ls -A $TARGET_DIR)" ]; then
+            # 打包 creds 目录
+            # 排除 config.toml 如果你不想备份它，但通常需要备份
+            tar -czf "$TEMP_FILE" "$TARGET_DIR"
+            
+            FILE_SIZE=$(wc -c < "$TEMP_FILE")
+            log "正在上传备份 (大小: $FILE_SIZE bytes)..."
+            
+            HTTP_CODE=$(curl -s -o /dev/null -w "%{http_code}" -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -T "$TEMP_FILE" "$FULL_WEBDAV_URL/$BACKUP_NAME")
+            
+            if [[ "$HTTP_CODE" == "201" ]] || [[ "$HTTP_CODE" == "204" ]] || [[ "$HTTP_CODE" == "200" ]]; then
+                log "备份成功"
+            else
+                log "备份失败 (HTTP $HTTP_CODE)"
+            fi
+        else
+            log "⚠️ $TARGET_DIR 目录为空，跳过备份"
+        fi
+    done
+}
+
+# === 入口 ===
+case "$1" in
+    "restore") restore ;;
+    "loop") loop ;;
+    *) restore; loop ;;
+esac