Create sync_data.sh

sync_data.sh

@@ -0,0 +1,149 @@
+#!/bin/sh
+
+# 检查环境变量
+if [ -z "$WEBDAV_URL" ] || [ -z "$WEBDAV_USERNAME" ] || [ -z "$WEBDAV_PASSWORD" ]; then
+    echo "Starting without sync functionality - missing WEBDAV_URL, WEBDAV_USERNAME, or WEBDAV_PASSWORD"
+    exit 0
+fi
+
+# 设置备份路径
+WEBDAV_BACKUP_PATH=${WEBDAV_BACKUP_PATH:-"cloudreve_backups"}
+FULL_WEBDAV_URL="${WEBDAV_URL}/${WEBDAV_BACKUP_PATH}"
+
+# 创建 WebDAV 备份路径（如果不存在） - 尝试使用 curl 创建目录
+curl -f -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -X MKCOL "$FULL_WEBDAV_URL"
+
+# 下载最新备份并恢复
+restore_backup() {
+    echo "开始从 WebDAV 下载最新备份..."
+    python3 -c "
+import sys
+import os
+import tarfile
+import requests
+import shutil
+import subprocess
+from webdav3.client import Client
+
+# 设置 WebDAV 客户端
+options = {
+    'webdav_hostname': os.environ.get('FULL_WEBDAV_URL'),
+    'webdav_login': os.environ.get('WEBDAV_USERNAME'),
+    'webdav_password': os.environ.get('WEBDAV_PASSWORD')
+}
+client = Client(options)
+
+# 列出备份文件
+try:
+    backups = [file for file in client.list('${WEBDAV_BACKUP_PATH}/') if file.endswith('.tar.gz') and file.startswith('cloudreve_backup_')]
+except Exception as e:
+    print(f'连接 WebDAV 服务器出错: {e}')
+    sys.exit(1)
+
+if not backups:
+    print('没有找到备份文件，跳过恢复步骤。')
+    sys.exit(0)
+
+# 获取最新备份文件
+latest_backup = sorted(backups)[-1]
+print(f'最新备份文件：{latest_backup}')
+
+# 下载最新备份
+download_url = os.path.join(os.environ.get('FULL_WEBDAV_URL'), latest_backup)
+try:
+    with requests.get(download_url, auth=(os.environ.get('WEBDAV_USERNAME'), os.environ.get('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}')
+
+            # 检查数据库文件是否被占用 (Cloudreve 默认使用 cloudreve.db)
+            try:
+                pids = subprocess.check_output(['lsof', '-t', '/opt/cloudreve/data/cloudreve.db']).decode().strip()
+                if pids:
+                    print('数据库文件被占用，无法恢复备份。')
+                    sys.exit(1)
+            except subprocess.CalledProcessError:
+                # 没有进程占用数据库文件
+                pass
+
+            # 解压备份文件
+            with tarfile.open(f'/tmp/{latest_backup}', 'r:gz') as tar:
+                tar.extractall('/opt/cloudreve/')
+            print(f'成功从 {latest_backup} 恢复备份')
+        else:
+            print(f'下载备份失败：{r.status_code}')
+except Exception as e:
+    print(f'恢复备份过程中出错: {e}')
+"
+}
+
+# 首次启动时下载最新备份
+echo "Downloading latest backup from WebDAV..."
+restore_backup
+
+# 同步函数
+sync_data() {
+    while true; do
+        echo "Starting sync process at $(date)"
+
+        # 定义备份文件名
+        timestamp=$(date +%Y%m%d_%H%M%S)
+        backup_file="cloudreve_backup_${timestamp}.tar.gz"
+
+        # 备份数据库和配置文件
+        cd /opt/cloudreve
+        tar -czf "/tmp/${backup_file}" data/cloudreve.db cloudreve.ini
+
+        # 上传新备份到 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
+
+        # 清理旧备份文件
+        python3 -c "
+import sys
+import os
+from webdav3.client import Client
+
+options = {
+    'webdav_hostname': os.environ.get('FULL_WEBDAV_URL'),
+    'webdav_login': os.environ.get('WEBDAV_USERNAME'),
+    'webdav_password': os.environ.get('WEBDAV_PASSWORD')
+}
+client = Client(options)
+
+backup_path = os.environ.get('WEBDAV_BACKUP_PATH', 'cloudreve_backups')
+full_backup_path = f'/{backup_path}/' if backup_path else '/'
+
+try:
+    backups = [file.name for file in client.list(full_backup_path) if file.name.endswith('.tar.gz') and file.name.startswith('cloudreve_backup_')]
+    backups.sort()
+    if len(backups) > 5:
+        to_delete = len(backups) - 5
+        for file_name in backups[:to_delete]:
+            try:
+                client.delete(os.path.join(full_backup_path, file_name))
+                print(f'Successfully deleted {file_name}.')
+            except Exception as e:
+                print(f'删除 {file_name} 失败: {e}')
+    else:
+        print('Only {} backups found, no need to clean.'.format(len(backups)))
+except Exception as e:
+    print(f'清理旧备份过程中出错: {e}')
+" 2>&1
+
+        rm -f "/tmp/${backup_file}"
+
+        SYNC_INTERVAL=${SYNC_INTERVAL:-3600} # 默认每小时同步一次
+        echo "Next sync in ${SYNC_INTERVAL} seconds..."
+        sleep $SYNC_INTERVAL
+    done
+}
+
+# 启动同步进程
+sync_data &