Update sync_data.sh

sync_data.sh

@@ -20,7 +20,6 @@ restore_from_webdav() {
 
     # 获取WebDAV上所有备份文件列表
     echo "获取 WebDAV 备份文件列表..."
-
     # 尝试使用PROPFIND方法获取目录列表（WebDAV标准方法）
     RAW_RESPONSE=$(curl -s -X PROPFIND \
         --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" \
@@ -28,27 +27,17 @@ restore_from_webdav() {
         -H "Content-Type: text/xml" \
         -d '<?xml version="1.0" encoding="utf-8"?><propfind xmlns="DAV:"><prop><displayname/></prop></propfind>' \
         "$WEBDAV_URL/")
-
     # 如果PROPFIND失败，尝试简单的GET请求
     if [ -z "$RAW_RESPONSE" ] || echo "$RAW_RESPONSE" | grep -q "Method Not Allowed"; then
         RAW_RESPONSE=$(curl -s --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/")
     fi
-
-    # 多种匹配方式
-    # 方式1: 从XML响应中提取文件名
-    BACKUP_LIST1=$(echo "$RAW_RESPONSE" | grep -oE '<D:displayname[^>]*>[^<]*mcp_backup_[0-9]+_[0-9]+\.tar\.gz[^<]*</D:displayname>' | sed 's/<[^>]*>//g')
-    # 方式2: 从href属性中提取
-    BACKUP_LIST2=$(echo "$RAW_RESPONSE" | grep -oE 'href="[^"]*mcp_backup_[0-9]+_[0-9]+\.tar\.gz[^"]*"' | sed 's/.*href="[^"]*\///g' | sed 's/".*//g')
-    # 方式3: 简单文本匹配
-    BACKUP_LIST3=$(echo "$RAW_RESPONSE" | grep -oE 'mcp_backup_[0-9]+_[0-9]+\.tar\.gz')
-
-    # 合并所有结果并排序
-    BACKUP_LIST=$(printf "%s\n%s\n%s\n" "$BACKUP_LIST1" "$BACKUP_LIST2" "$BACKUP_LIST3" | grep -v '^$' | sort -u | sort -r)
+    # 多种匹配方式提取文件名
+    BACKUP_LIST=$(echo "$RAW_RESPONSE" | grep -oE 'mcp_backup_[0-9]+_[0-9]+\.tar\.gz' | sort -u | sort -r)
 
     if [ -z "$BACKUP_LIST" ]; then
-        echo "未找到任何备份文件"
+        echo "未找到任何备份文件。"
         rm -rf "$RESTORE_DIR"
-        return 1
+        return 1 # 返回1表示失败，这是关键
     fi
 
     echo "找到以下备份文件："
@@ -57,37 +46,16 @@ restore_from_webdav() {
         echo "  ... 等共 $(echo "$BACKUP_LIST" | wc -l) 个文件"
     fi
 
-    # 尝试下载最新的备份文件
     LATEST_BACKUP=$(echo "$BACKUP_LIST" | head -1)
     echo "尝试下载最新备份: $LATEST_BACKUP"
 
     if curl -f --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$LATEST_BACKUP" -o "$RESTORE_DIR/$LATEST_BACKUP" 2>/dev/null; then
         echo "下载备份成功: $LATEST_BACKUP"
-        
-        # 解压缩备份文件
         cd "$RESTORE_DIR"
         tar -xzf "$LATEST_BACKUP" && {
             echo "解压缩备份文件成功"
-            
-            # 恢复配置文件
-            if [ -d "config" ] && [ "$(ls -A config 2>/dev/null)" ]; then
-                echo "恢复 config 目录..."
-                mkdir -p /mcp-proxy-server/config
-                cp -r config/* /mcp-proxy-server/config/
-                echo "config 目录恢复成功"
-            else
-                echo "备份中没有 config 目录或目录为空"
-            fi
-            
-            if [ -d "tools" ] && [ "$(ls -A tools 2>/dev/null)" ]; then
-                echo "恢复 tools 目录..."
-                mkdir -p /tools
-                cp -r tools/* /tools/
-                echo "tools 目录恢复成功"
-            else
-                echo "备份中没有 tools 目录或目录为空"
-            fi
-            
+            [ -d "config" ] && [ "$(ls -A config 2>/dev/null)" ] && echo "恢复 config 目录..." && mkdir -p /mcp-proxy-server/config && cp -r config/* /mcp-proxy-server/config/
+            [ -d "tools" ] && [ "$(ls -A tools 2>/dev/null)" ] && echo "恢复 tools 目录..." && mkdir -p /tools && cp -r tools/* /tools/
             echo "从 WebDAV 恢复配置文件完成"
         } || {
             echo "解压缩备份文件失败"
@@ -97,9 +65,8 @@ restore_from_webdav() {
         echo "下载最新备份失败，将跳过恢复。"
     fi
 
-    # 清理临时目录
     rm -rf "$RESTORE_DIR"
-    return 0
+    return 0 # 返回0表示成功
 }
 
 # 定义备份到 WebDAV 的函数
@@ -111,32 +78,23 @@ backup_to_webdav() {
     mkdir -p "$BACKUP_DIR"
     
     # 准备要备份的目录
-    if [ -d "/mcp-proxy-server/config" ]; then
-        cp -r /mcp-proxy-server/config "$BACKUP_DIR/"
-    fi
-    if [ -d "/tools" ]; then
-        cp -r /tools "$BACKUP_DIR/"
-    fi
+    [ -d "/mcp-proxy-server/config" ] && cp -r /mcp-proxy-server/config "$BACKUP_DIR/"
+    [ -d "/tools" ] && cp -r /tools "$BACKUP_DIR/"
     
-    # 如果备份目录为空，则不执行备份
     if [ -z "$(ls -A $BACKUP_DIR 2>/dev/null)" ]; then
         echo "没有需要备份的文件，跳过此次备份。"
         rm -rf "$BACKUP_DIR"
         return
     fi
 
-    # 创建备份压缩包
     TIMESTAMP=$(date +%Y%m%d_%H%M%S)
     BACKUP_FILE="mcp_backup_${TIMESTAMP}.tar.gz"
     
     cd "$BACKUP_DIR"
     tar -czf "$BACKUP_FILE" *
     
-    # 上传到 WebDAV
     echo "正在上传备份文件: $BACKUP_FILE"
-    curl -T "$BACKUP_FILE" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$BACKUP_FILE"
-    
-    if [ $? -eq 0 ]; then
+    if curl -T "$BACKUP_FILE" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$BACKUP_FILE"; then
         echo "备份成功上传到 WebDAV。"
     else
         echo "备份上传失败。"
@@ -148,40 +106,41 @@ backup_to_webdav() {
     # 清理旧的备份 (可选, 保留最近的10个)
     MAX_BACKUPS=${MAX_BACKUPS:-10}
     echo "清理 WebDAV 上的旧备份，保留最新的 $MAX_BACKUPS 个..."
-    
-    # 重新获取备份列表
-    # 尝试使用PROPFIND方法获取目录列表
-    RAW_RESPONSE=$(curl -s -X PROPFIND --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -H "Depth: 1" -H "Content-Type: text/xml" -d '<?xml version="1.0" encoding="utf-8"?><propfind xmlns="DAV:"><prop><displayname/></prop></propfind>' "$WEBDAV_URL/")
-    # 如果PROPFIND失败，尝试简单的GET请求
-    if [ -z "$RAW_RESPONSE" ] || echo "$RAW_RESPONSE" | grep -q "Method Not Allowed"; then
-        RAW_RESPONSE=$(curl -s --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/")
-    fi
+    RAW_RESPONSE=$(curl -s --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/")
     BACKUP_LIST=$(echo "$RAW_RESPONSE" | grep -oE 'mcp_backup_[0-9]+_[0-9]+\.tar\.gz' | sort -r)
-    
     OLD_BACKUPS=$(echo "$BACKUP_LIST" | tail -n +$((MAX_BACKUPS + 1)))
     
     for old_backup in $OLD_BACKUPS; do
         echo "删除旧备份: $old_backup"
-        curl -X DELETE --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$old_backup"
+        curl -s -X DELETE --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$old_backup"
     done
 }
 
 
-# 脚本主逻辑
+# --- 脚本主逻辑 ---
 
-# 检查命令行参数
+# 处理 --restore-only 参数
 if [ "$1" = "--restore-only" ]; then
     restore_from_webdav
     exit 0
 fi
 
-# 启动时先恢复一次
-restore_from_webdav
+# 启动时先尝试恢复
+echo "脚本启动，首先尝试从 WebDAV 恢复..."
+if ! restore_from_webdav; then
+    # 如果恢复函数返回非0值 (即失败，通常是因为没找到备份)
+    echo "恢复失败或未找到备份文件。这可能是首次启动。"
+    # 等待一小段时间，给主程序生成初始配置的机会
+    INITIAL_BACKUP_DELAY=${INITIAL_BACKUP_DELAY:-30} 
+    echo "将在 ${INITIAL_BACKUP_DELAY} 秒后执行首次备份，以捕获初始配置文件..."
+    sleep $INITIAL_BACKUP_DELAY
+    backup_to_webdav
+fi
 
 # 定义同步（备份）间隔
 SYNC_INTERVAL=${SYNC_INTERVAL:-3600} # 默认1小时
 
-# 循环执行备份
+# 进入正常的循环备份流程
 while true; do
     echo "----------------------------------------"
     echo "下一次 WebDAV 备份将在 $SYNC_INTERVAL 秒后进行..."