Update sync_data.sh

sync_data.sh

@@ -1,128 +1,97 @@
 #!/bin/bash
 
+# 检查必要的环境变量
+if [ -z "$WEBDAV_URL" ] || [ -z "$WEBDAV_USERNAME" ] || [ -z "$WEBDAV_PASSWORD" ]; then
+    echo "缺少必要的环境变量 WEBDAV_URL、WEBDAV_USERNAME 或 WEBDAV_PASSWORD"
+    exit 1
+fi
+
+# 创建数据目录
 mkdir -p ./data
 
-# 生成校验和文件
-generate_sum() {
-    local file=$1
-    local sum_file=$2
-    sha256sum "$file" > "$sum_file"
+# 获取最新的带日期的数据库文件名
+get_latest_dated_file() {
+    curl -s --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL" | grep -o 'webui_[0-9]\{2\}_[0-9]\{2\}_[0-9]\{4\}.db' | sort -r | head -n 1
 }
 
-# 优先从WebDAV恢复数据
-if [ ! -z "$WEBDAV_URL" ] && [ ! -z "$WEBDAV_USERNAME" ] && [ ! -z "$WEBDAV_PASSWORD" ]; then
-    echo "尝试从WebDAV恢复数据..."
-    curl -L --fail --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/webui.db" -o "./data/webui.db" && {
-        echo "从WebDAV恢复数据成功"
-    } || {
-        if [ ! -z "$G_NAME" ] && [ ! -z "$G_TOKEN" ]; then
-            echo "从WebDAV恢复失败,尝试从GitHub恢复..."
-            REPO_URL="https://${G_TOKEN}@github.com/${G_NAME}.git"
-            git clone "$REPO_URL" ./data/temp && {
-                if [ -f ./data/temp/webui.db ]; then
-                    mv ./data/temp/webui.db ./data/webui.db
-                    echo "从GitHub仓库恢复成功"
-                    rm -rf ./data/temp
-                else
-                    echo "GitHub仓库中未找到webui.db"
-                    rm -rf ./data/temp
-                fi
-            }
-        else
-            echo "WebDAV恢复失败,且未配置GitHub"
-        fi
-    }
-else
-    echo "未配置WebDAV,跳过数据恢复"
-fi
+# 从文件名中提取日期
+get_date_from_filename() {
+    local filename=$1
+    echo "$filename" | grep -o '[0-9]\{2\}_[0-9]\{2\}' | head -n 1
+}
+
+# 从 WebDAV 下载webui.db文件
+echo "正在从 WebDAV 下载数据库文件..."
+curl -L --fail --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/webui.db" -o "./data/webui.db" || {
+    echo "下载失败，脚本退出"
+    exit 1
+}
+echo "下载成功"
 
-# 同步函数
+# 定义同步函数
 sync_data() {
+    # 首次等待同步间隔
+    SYNC_INTERVAL=${SYNC_INTERVAL:-7200}  # 默认间隔时间为 7200 秒
+    echo "初始下载完成，等待 ${SYNC_INTERVAL} 秒后开始同步..."
+    sleep $SYNC_INTERVAL
+
+    # 记录上次上传的日期
+    LAST_UPLOAD_DATE=""
+
     while true; do
-        echo "开始同步..."
-        HOUR=$(date +%H)
-        
+        echo "正在开始同步"
+        # 检查数据库文件是否存在
         if [ -f "./data/webui.db" ]; then
-            # 生成新的校验和文件
-            generate_sum "./data/webui.db" "./data/webui.db.sha256.new"
+            # 获取当前日期
+            CURRENT_DATE=$(date +'%m_%d')
             
-            # 检查文件是否变化
-            if [ ! -f "./data/webui.db.sha256" ] || ! cmp -s "./data/webui.db.sha256.new" "./data/webui.db.sha256"; then
-                echo "检测到文件变化，开始同步..."
-                mv "./data/webui.db.sha256.new" "./data/webui.db.sha256"
-                
-                # 同步到WebDAV
-                if [ ! -z "$WEBDAV_URL" ] && [ ! -z "$WEBDAV_USERNAME" ] && [ ! -z "$WEBDAV_PASSWORD" ]; then
-                    echo "同步到WebDAV..."
-                    
-                    # 上传数据文件
-                    curl -L -T "./data/webui.db" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/webui.db" && {
-                        echo "WebDAV更新成功"
-                        
-                        # 每日备份(包括WebDAV和GitHub)，在每天0点进行
-                        if [ "$HOUR" = "00" ]; then
-                            echo "开始每日备份..."
-                            
-                            # 获取前一天的日期
-                            YESTERDAY=$(date -d "yesterday" '+%Y%m%d')
-                            FILENAME_DAILY="webui_${YESTERDAY}.db"
-                            
-                            # WebDAV每日备份
-                            curl -L -T "./data/webui.db" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$FILENAME_DAILY" && {
-                                echo "WebDAV日期备份成功: $FILENAME_DAILY"
-                                
-                                # GitHub每日备份
-                                if [ ! -z "$G_NAME" ] && [ ! -z "$G_TOKEN" ]; then
-                                    echo "开始GitHub每日备份..."
-                                    REPO_URL="https://${G_TOKEN}@github.com/${G_NAME}.git"
-                                    git clone "$REPO_URL" ./data/temp || {
-                                        echo "GitHub克隆失败"
-                                        rm -rf ./data/temp
-                                    }
-                                    
-                                    if [ -d "./data/temp" ]; then
-                                        cd ./data/temp
-                                        git config user.name "AutoSync Bot"
-                                        git config user.email "[email protected]"
-                                        git checkout main || git checkout master
-                                        cp ../webui.db ./webui.db
-                                        
-                                        if [[ -n $(git status -s) ]]; then
-                                            git add webui.db
-                                            git commit -m "Auto sync webui.db for ${YESTERDAY}"
-                                            git push origin HEAD && {
-                                                echo "GitHub推送成功"
-                                            } || echo "GitHub推送失败"
-                                        else
-                                            echo "GitHub: 无数据变化"
-                                        fi
-                                        cd ../..
-                                        rm -rf ./data/temp
-                                    fi
-                                fi
-                            } || echo "WebDAV日期备份失败"
-                        fi
-                    } || {
-                        echo "WebDAV上传失败,重试..."
-                        sleep 10
-                        curl -L -T "./data/webui.db" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/webui.db" || {
-                            echo "WebDAV重试失败"
-                        }
-                    }
+            # 需要上传带日期的文件标志
+            NEED_DATED_UPLOAD=false
+
+            # 如果没有记录上次上传日期，从WebDAV获取最新的带日期文件
+            if [ -z "$LAST_UPLOAD_DATE" ]; then
+                LATEST_FILE=$(get_latest_dated_file)
+                if [ ! -z "$LATEST_FILE" ]; then
+                    LAST_UPLOAD_DATE=$(get_date_from_filename "$LATEST_FILE")
                 fi
-            else
-                echo "文件未发生变化，跳过同步"
-                rm -f "./data/webui.db.sha256.new"
             fi
+
+            # 如果上次上传日期与当前日期不同，需要上传带日期的文件
+            if [ "$LAST_UPLOAD_DATE" != "$CURRENT_DATE" ]; then
+                NEED_DATED_UPLOAD=true
+                LAST_UPLOAD_DATE=$CURRENT_DATE
+            fi
+
+            # 如果需要上传带日期的文件
+            if [ "$NEED_DATED_UPLOAD" = true ]; then
+                FILENAME="webui_${CURRENT_DATE}_$(date +'%H%M').db"
+                echo "上传带日期的文件到 WebDAV..."
+                curl -L -T "./data/webui.db" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$FILENAME" || {
+                    echo "带日期文件上传失败"
+                }
+            fi
+
+            # 始终更新主文件
+            echo "更新主文件到 WebDAV..."
+            curl -L -T "./data/webui.db" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/webui.db" && {
+                echo "WebDAV 更新主文件成功"
+            } || {
+                echo "WebDAV 更新主文件失败，尝试重试..."
+                sleep 10
+                curl -L -T "./data/webui.db" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/webui.db" || {
+                    echo "重试失败，放弃上传。"
+                }
+            }
         else
-            echo "未找到webui.db,跳过同步"
+            echo "未找到 webui.db 文件，跳过同步"
         fi
-        
-        echo "当前时间: $(date '+%Y-%m-%d %H:%M:%S')"
-        echo "下次同步: $(date -d '+5 minutes' '+%Y-%m-%d %H:%M:%S')"
-        sleep 300
+
+        # 等待下一次同步间隔
+        echo "当前时间 $(date '+%Y-%m-%d %H:%M:%S')"
+        echo "等待 ${SYNC_INTERVAL} 秒后进行下一次同步..."
+        sleep $SYNC_INTERVAL
     done
 }
 
-# 启动同步进程
+# 后台启动同步进程
 sync_data &