Update sync_data.sh

sync_data.sh

@@ -1,128 +1,131 @@
 #!/bin/bash
 
-mkdir -p ./data
+# 检查环境变量
+if [[ -z "$WEBDAV_URL" ]] || [[ -z "$WEBDAV_USERNAME" ]] || [[ -z "$WEBDAV_PASSWORD" ]]; then
+	echo "缺少 WEBDAV_URL、WEBDAV_USERNAME 或 WEBDAV_PASSWORD，启动时将不包含备份功能"
+	exit 0
+fi
+
+# 设置备份路径
+WEBDAV_BACKUP_PATH=${WEBDAV_BACKUP_PATH:-""}
+FULL_WEBDAV_URL="${WEBDAV_URL}"
+if [ -n "$WEBDAV_BACKUP_PATH" ]; then
+	FULL_WEBDAV_URL="${WEBDAV_URL}/${WEBDAV_BACKUP_PATH}"
+fi
 
-# 生成校验和文件
-generate_sum() {
-    local file=$1
-    local sum_file=$2
-    sha256sum "$file" > "$sum_file"
+# 下载最新备份并恢复
+restore_backup() {
+	echo "开始从 WebDAV 下载最新备份..."
+	python3 -c "
+import sys
+import os
+import tarfile
+import requests
+from webdav3.client import Client
+import shutil
+options = {
+	'webdav_hostname': '$FULL_WEBDAV_URL',
+	'webdav_login': '$WEBDAV_USERNAME',
+	'webdav_password': '$WEBDAV_PASSWORD'
+}
+client = Client(options)
+backups = [file for file in client.list() if file.endswith('.tar.gz') and file.startswith('webui_backup_')]
+if not backups:
+	print('没有找到备份文件')
+	sys.exit()
+latest_backup = sorted(backups)[-1]
+print(f'最新备份文件：{latest_backup}')
+with requests.get(f'$FULL_WEBDAV_URL/{latest_backup}', auth=('$WEBDAV_USERNAME', '$WEBDAV_PASSWORD'), stream=True) as r:
+	if r.status_code == 200:
+		with open(f'/tmp/{latest_backup}', 'wb') as f:
+			for chunk in r.iter_content(chunk_size=8192):
+				f.write(chunk)
+		print(f'成功下载备份文件到 /tmp/{latest_backup}')
+		if os.path.exists(f'/tmp/{latest_backup}'):
+			# 解压备份文件到临时目录
+			temp_dir = '/tmp/restore'
+			os.makedirs(temp_dir, exist_ok=True)
+			tar = tarfile.open(f'/tmp/{latest_backup}', 'r:gz')
+			tar.extractall(temp_dir)
+			tar.close()
+			# 查找并移动 webui.db 文件
+			for root, dirs, files in os.walk(temp_dir):
+				if 'webui.db' in files:
+					db_path = os.path.join(root, 'webui.db')
+					os.makedirs('./data', exist_ok=True)
+					os.replace(db_path, './data/webui.db')
+					print(f'成功从 {latest_backup} 恢复备份')
+					break
+			else:
+				print('备份文件中未找到 webui.db')
+			# 删除临时目录
+			try:
+				shutil.rmtree(temp_dir)
+			except Exception as e:
+				print(f'删除临时目录时出错：{e}')
+			os.remove(f'/tmp/{latest_backup}')
+		else:
+			print('下载的备份文件不存在')
+	else:
+		print(f'下载备份失败：{r.status_code}')
+"
 }
 
-# 优先从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
+# 首次启动时下载最新备份
+echo "正在从 WebDAV 下载最新备份..."
+restore_backup
 
 # 同步函数
 sync_data() {
-    while true; do
-        echo "开始同步..."
-        HOUR=$(date +%H)
-        
-        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 "文件未发生变化，跳过同步"
-                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
+	while true; do
+		echo "在 $(date) 开始同步进程"
+
+		if [ -f "./data/webui.db" ]; then
+			timestamp=$(date +%Y%m%d_%H%M%S)
+			backup_file="webui_backup_${timestamp}.tar.gz"
+
+			# 打包数据库文件
+			tar -czf "/tmp/${backup_file}" ./data/webui.db
+
+			# 上传新备份到WebDAV
+			curl -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -T "/tmp/${backup_file}" "$FULL_WEBDAV_URL/${backup_file}"
+			if [ $? -eq 0 ]; then
+				echo "成功将 ${backup_file} 上传至 WebDAV"
+			else
+				echo "上传 ${backup_file} 至 WebDAV 失败"
+			fi
+
+			# 清理旧备份文件
+			python3 -c "
+import sys
+from webdav3.client import Client
+options = {
+	'webdav_hostname': '$FULL_WEBDAV_URL',
+	'webdav_login': '$WEBDAV_USERNAME',
+	'webdav_password': '$WEBDAV_PASSWORD'
+}
+client = Client(options)
+backups = [file for file in client.list() if file.endswith('.tar.gz') and file.startswith('webui_backup_')]
+backups.sort()
+if len(backups) > 5:
+	to_delete = len(backups) - 5
+	for file in backups[:to_delete]:
+		client.clean(file)
+		print(f'成功删除 {file}。')
+else:
+	print('仅找到 {} 个备份，无需清理。'.format(len(backups)))
+" 2>&1
+
+			rm -f "/tmp/${backup_file}"
+		else
+			echo "数据库文件尚不存在，等待下次同步..."
+		fi
+
+		SYNC_INTERVAL=${SYNC_INTERVAL:-600}
+		echo "下次同步将在 ${SYNC_INTERVAL} 秒后进行..."
+		sleep $SYNC_INTERVAL
+	done
 }
 
-# 启动同步进程
+# 后台启动同步进程
 sync_data &