Update sync_data.sh

sync_data.sh

@@ -1,139 +1,128 @@
 #!/bin/bash
-# 注意：确保脚本执行时系统中已安装 date、git、curl 等工具，并设置好 TZ 时区环境（可在每个 date 命令中临时指定时区）
 
-# 检查必要的环境变量
-if [ -z "$G_NAME" ] || [ -z "$G_TOKEN" ]; then
-    echo "缺少必要的环境变量 G_NAME 或 G_TOKEN"
-    exit 1
-fi
-
-# 解析仓库名和用户名
-IFS='/' read -r GITHUB_USER GITHUB_REPO <<< "$G_NAME"
+mkdir -p ./data
 
-# 构建 GitHub 仓库的克隆 URL，包含令牌
-REPO_URL="https://${G_TOKEN}@github.com/${G_NAME}.git"
-mkdir -p  ./data/github_data
-
-# 克隆仓库
-echo "正在克隆仓库……"
-git clone "$REPO_URL" ./data/github_data || {
-    echo "克隆失败，请检查 G_NAME 和 G_TOKEN 是否正确。"
-    exit 1
+# 生成校验和文件
+generate_sum() {
+    local file=$1
+    local sum_file=$2
+    sha256sum "$file" > "$sum_file"
 }
 
-if [ -f ./data/github_data/webui.db ]; then
-    cp ./data/github_data/webui.db ./data/webui.db
-    echo "从 GitHub 仓库中拉取成功"
+# 优先从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 "GitHub 仓库中未找到 webui.db，将在同步时推送"
+    echo "未配置WebDAV,跳过数据恢复"
 fi
 
-# 定义同步函数，按照北京时间 08:00～24:00（包含整点同步）的要求
+# 同步函数
 sync_data() {
     while true; do
-        # 使用 Asia/Shanghai 时区获取当前时间及其组成部分
-        CURRENT_TS=$(TZ=Asia/Shanghai date +%s)
-        CURRENT_DATE=$(TZ=Asia/Shanghai date '+%Y-%m-%d')
-        CURRENT_HOUR=$(TZ=Asia/Shanghai date +%H)  # 00~23
-        CURRENT_MIN=$(TZ=Asia/Shanghai date +%M)
-        CURRENT_SEC=$(TZ=Asia/Shanghai date +%S)
+        echo "开始同步..."
+        HOUR=$(date +%H)
         
-        # 计算下一次同步的目标时间戳（北京时间）
-        # 如果当前时间早于 08:00，则目标为今天 08:00
-        if [ "$CURRENT_HOUR" -lt 8 ]; then
-            TARGET_TS=$(TZ=Asia/Shanghai date -d "${CURRENT_DATE} 08:00:00" +%s)
-        # 如果在 08:00 至 22:59，则下一个整点在当日
-        elif [ "$CURRENT_HOUR" -ge 8 ] && [ "$CURRENT_HOUR" -lt 23 ]; then
-            # 如果正好在整点（秒与分都为 0）则立刻同步
-            if [ "$CURRENT_MIN" -eq 0 ] && [ "$CURRENT_SEC" -eq 0 ]; then
-                TARGET_TS=$CURRENT_TS
-            else
-                NEXT_HOUR=$((10#$CURRENT_HOUR + 1))
-                TARGET_TS=$(TZ=Asia/Shanghai date -d "${CURRENT_DATE} ${NEXT_HOUR}:00:00" +%s)
-            fi
-        # 如果当前时间处于 23:00~23:59，则下次目标为次日 00:00（也就是24:00同步）
-        else  # CURRENT_HOUR == 23
-            if [ "$CURRENT_MIN" -eq 0 ] && [ "$CURRENT_SEC" -eq 0 ]; then
-                TARGET_TS=$CURRENT_TS
-            else
-                TOMORROW=$(TZ=Asia/Shanghai date -d "tomorrow" '+%Y-%m-%d')
-                TARGET_TS=$(TZ=Asia/Shanghai date -d "${TOMORROW} 00:00:00" +%s)
-            fi
-        fi
-
-        # 计算等待时间（若正好同步时则 sleep_time 为 0）
-        SLEEP_TIME=$(( TARGET_TS - CURRENT_TS ))
-        if [ "$SLEEP_TIME" -gt 0 ]; then
-            echo "距离下一次同步还有 ${SLEEP_TIME} 秒（北京时间下次同步时间为 $(TZ=Asia/Shanghai date -d "@$TARGET_TS" '+%Y-%m-%d %H:%M:%S')）"
-            sleep "$SLEEP_TIME"
-        fi
-
-        # 同步时输出当前北京时间
-        CURRENT_TIME=$(TZ=Asia/Shanghai date '+%Y-%m-%d %H:%M:%S')
-        echo "当前时间 $CURRENT_TIME"
-
-        # ---- 开始同步流程 ----
-
-        # 1. 同步到 GitHub
-        echo "开始执行 GitHub 同步……"
-        cd ./data/github_data || { echo "切换目录失败"; exit 1; }
-        git config user.name "AutoSync Bot"
-        git config user.email "autosync@bot.com"
-
-        # 确保在 main 分支，如切换失败则尝试 master 分支
-        git checkout main 2>/dev/null || git checkout master
-
-        # 将最新数据库文件复制到仓库目录下
-        if [ -f "../webui.db" ]; then  
-            cp ../webui.db ./webui.db  
-        else  
-            echo "数据库尚未初始化"
-        fi 
-
-        # 检查是否有变化
-        if [[ -n $(git status -s) ]]; then
-            git add webui.db
-            git commit -m "Auto sync webui.db $(TZ=Asia/Shanghai date '+%Y-%m-%d %H:%M:%S')"
-            git push origin HEAD && {
-                echo "GitHub 推送成功"
-            } || {
-                echo "推送失败，等待重试..."
-                sleep 10
-                git push origin HEAD || {
-                    echo "重试失败，放弃推送到 GitHub。"
-                }
-            }
-        else
-            echo "GitHub：没有检测到数据库变化"
-        fi
-        # 返回主目录
-        cd ../..
-
-        # 2. 同步到 WebDAV（若环境变量配置完整）
-        if [ -z "$WEBDAV_URL" ] || [ -z "$WEBDAV_USERNAME" ] || [ -z "$WEBDAV_PASSWORD" ]; then
-            echo "WebDAV 环境变量缺失，跳过 WebDAV 同步。"
-        else
-            echo "开始执行 WebDAV 同步……"
-            FILENAME="webui_$(TZ=Asia/Shanghai date +'%m_%d').db"
-            if [ -f ./data/webui.db ]; then
-                curl -T ./data/webui.db --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$FILENAME" && {
-                    echo "WebDAV 上传成功"
-                } || {
-                    echo "WebDAV 上传失败，等待重试..."
-                    sleep 10
-                    curl -T ./data/webui.db --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$FILENAME" || {
-                        echo "重试失败，放弃 WebDAV 上传。"
+        if [ -f "./data/webui.db" ]; then
+            # 生成新的校验和文件
+            generate_sum "./data/webui.db" "./data/webui.db.sha256.new"
+            
+            # 检查文件是否变化
+            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 "autosync@bot.com"
+                                        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重试失败"
+                        }
                     }
-                }
+                fi
             else
-                echo "未找到 webui.db 文件，跳过 WebDAV 同步。"
+                echo "文件未发生变化，跳过同步"
+                rm -f "./data/webui.db.sha256.new"
             fi
+        else
+            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
     done
 }
 
-# 后台启动同步进程
+# 启动同步进程
 sync_data &