Create sync_data.sh

sync_data.sh

@@ -0,0 +1,132 @@
+#!/bin/bash
+
+echo "sync_data.sh script started at $(date)"
+
+# 检查 WebDAV 凭据
+if [[ -z "$WEBDAV_URL" ]] || [[ -z "$WEBDAV_USERNAME" ]] || [[ -z "$WEBDAV_PASSWORD" ]]; then
+    echo "Starting without backup functionality - missing WEBDAV_URL, WEBDAV_USERNAME, or WEBDAV_PASSWORD"
+    exec /opt/cloudreve/cloudreve
+    exit 0
+fi
+
+# 设置 WebDAV 备份路径
+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
+
+# 激活虚拟环境
+source /opt/venv/bin/activate
+
+# 设置 Cloudreve 数据目录
+CLOUDREVE_DATA_DIR="/opt/cloudreve/data"
+
+# 下载和恢复备份函数
+restore_backup() {
+    echo "开始从 WebDAV 下载最新备份..."
+    python3 -c "
+import sys
+import os
+import tarfile
+import requests
+import shutil
+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('cloudreve_backup_')]
+if not backups:
+    print('没有找到备份文件')
+    sys.exit()
+latest_backup = sorted(backups)[-1]
+print(f'最新备份文件：{latest_backup}')
+backup_path = f'/tmp/{latest_backup}'
+try:
+    client.download_file(f'/{latest_backup}', backup_path)
+    print(f'成功下载备份文件到 {backup_path}')
+    if os.path.exists(backup_path):
+        # 删除现有数据目录
+        if os.path.exists('$CLOUDREVE_DATA_DIR'):
+            shutil.rmtree('$CLOUDREVE_DATA_DIR')
+        os.makedirs('$CLOUDREVE_DATA_DIR', exist_ok=True)
+        # 解压备份
+        with tarfile.open(backup_path, 'r:gz') as tar:
+            tar.extractall('$CLOUDREVE_DATA_DIR')
+        print(f'成功从 {latest_backup} 恢复备份')
+    else:
+        print('下载的备份文件不存在')
+except Exception as e:
+    print(f'恢复备份失败：{e}')
+    sys.exit(1)
+"
+}
+
+# 启动时下载最新备份
+echo "Downloading latest backup from WebDAV..."
+restore_backup
+
+# 同步数据函数
+sync_data() {
+    while true; do
+        echo "Starting sync process at $(date)"
+
+        if [ -d "$CLOUDREVE_DATA_DIR" ]; then
+            timestamp=$(date +%Y%m%d_%H%M%S)
+            backup_file="cloudreve_backup_${timestamp}.tar.gz"
+
+            # 创建备份
+            tar -czf "/tmp/${backup_file}" -C "$CLOUDREVE_DATA_DIR" .
+
+            # 上传备份到 WebDAV
+            curl -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -T "/tmp/${backup_file}" "$FULL_WEBDAV_URL/${backup_file}"
+            if [ $? -eq 0 ]; then
+                echo "Successfully uploaded ${backup_file} to WebDAV"
+            else
+                echo "Failed to upload ${backup_file} to WebDAV"
+            fi
+
+            # 清理旧备份 (保留最新的 5 个)
+            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('cloudreve_backup_')]
+backups.sort()
+if len(backups) > 5:
+    to_delete = len(backups) - 5
+    for file in backups[:to_delete]:
+        try:
+            client.remove(f'/{file}')
+            print(f'Successfully deleted {file}.')
+        except Exception as e:
+            print(f'删除 {file} 失败: {e}')
+else:
+    print('Only {} backups found, no need to clean.'.format(len(backups)))
+" 2>&1
+
+            rm -f "/tmp/${backup_file}"
+        else
+            echo "Data directory does not exist yet, waiting for next sync..."
+        fi
+
+        SYNC_INTERVAL=${SYNC_INTERVAL:-600}
+        echo "Next sync in ${SYNC_INTERVAL} seconds..."
+        sleep $SYNC_INTERVAL
+    done
+}
+
+# 启动同步进程
+sync_data &
+
+# 启动 Cloudreve
+sleep 30
+exec /opt/cloudreve/cloudreve -c /opt/cloudreve/cloudreve.ini