Update sync_data.sh

sync_data.sh

@@ -8,7 +8,7 @@ BACKUP_RETENTION=${BACKUP_RETENTION:-5}
 
 # 检查环境变量
 if [[ -z "$WEBDAV_URL" ]] || [[ -z "$WEBDAV_USERNAME" ]] || [[ -z "$WEBDAV_PASSWORD" ]]; then
-    echo "Warning: Backup functionality will not be enabled - missing WEBDAV_URL, WEBDAV_USERNAME, or WEBDAV_PASSWORD"
+    echo "警告：备份功能将不会启用 - 缺少 WEBDAV_URL, WEBDAV_USERNAME, 或 WEBDAV_PASSWORD"
     exit 0
 fi
 
@@ -19,18 +19,19 @@ if [ -n "$WEBDAV_BACKUP_PATH" ]; then
     FULL_WEBDAV_URL="${WEBDAV_URL}/${WEBDAV_BACKUP_PATH}"
 fi
 
-# 设置 Cloudreve 数据目录 (修改为正确的路径)
+# 设置 Cloudreve 数据目录 (根据 Dockerfile 修改)
 CLOUDREVE_DATA_DIR="/opt/cloudreve/data"
+CLOUDREVE_CONFIG_FILE="/opt/cloudreve/cloudreve.ini" # 假设这是配置文件路径
 
-# 激活虚拟环境
-if [ -f "$HOME/venv/bin/activate" ]; then
-    source "$HOME/venv/bin/activate"
+# 激活虚拟环境 (根据 Dockerfile 修改)
+if [ -f "/opt/venv/bin/activate" ]; then
+    source "/opt/venv/bin/activate"
 fi
 
 # 下载最新备份并恢复
 restore_backup() {
-    echo "Starting to download the latest backup from WebDAV..."
-    WEBDAV_URL="$FULL_WEBDAV_URL" WEBDAV_USERNAME="$WEBDAV_USERNAME" WEBDAV_PASSWORD="$WEBDAV_PASSWORD" CLOUDREVE_DATA_DIR="$CLOUDREVE_DATA_DIR" python3 -c "
+    echo "开始从 WebDAV 下载最新备份..."
+    WEBDAV_URL="$FULL_WEBDAV_URL" WEBDAV_USERNAME="$WEBDAV_USERNAME" WEBDAV_PASSWORD="$WEBDAV_PASSWORD" CLOUDREVE_DATA_DIR="$CLOUDREVE_DATA_DIR" CLOUDREVE_CONFIG_FILE="$CLOUDREVE_CONFIG_FILE" python3 -c "
 import sys
 import os
 import tarfile
@@ -42,6 +43,7 @@ WEBDAV_URL = os.environ.get('WEBDAV_URL')
 WEBDAV_USERNAME = os.environ.get('WEBDAV_USERNAME')
 WEBDAV_PASSWORD = os.environ.get('WEBDAV_PASSWORD')
 CLOUDREVE_DATA_DIR = os.environ.get('CLOUDREVE_DATA_DIR')
+CLOUDREVE_CONFIG_FILE = os.environ.get('CLOUDREVE_CONFIG_FILE') # 添加了配置文件
 
 options = {
     'webdav_hostname': WEBDAV_URL,
@@ -52,78 +54,88 @@ client = Client(options)
 try:
     files = client.list()
 except Exception as e:
-    print(f'Error listing WebDAV directory: {e}')
+    print(f'列出 WebDAV 目录时出错: {e}')
     sys.exit(1)
 
-backups = [file for file in files if file.endswith('.tar.gz') and file.startswith('alist_backup_')]
+backups = [file for file in files if file.endswith('.tar.gz') and file.startswith('cloudreve_backup_')]
 if not backups:
-    print('No backup files found on WebDAV.')
+    print('在 WebDAV 上没有找到 Cloudreve 备份文件。')
     sys.exit()
 
 latest_backup = sorted(backups)[-1]
 backup_file_path = f'/tmp/{latest_backup}'
-print(f'Latest backup file found: {latest_backup}')
+print(f'找到最新的备份文件：{latest_backup}')
 
 try:
     client.download_file(latest_backup, backup_file_path)
-    print(f'Successfully downloaded backup file to {backup_file_path}')
+    print(f'成功下载备份文件到 {backup_file_path}')
 except Exception as e:
-    print(f'Failed to download backup file: {e}')
+    print(f'下载备份文件失败：{e}')
     sys.exit(1)
 
 if os.path.exists(backup_file_path):
     if os.path.exists(CLOUDREVE_DATA_DIR):
-        print(f\"Deleting existing data directory: {CLOUDREVE_DATA_DIR}\")
+        print(f\"删除现有的数据目录：{CLOUDREVE_DATA_DIR}\")
         shutil.rmtree(CLOUDREVE_DATA_DIR)
     os.makedirs(CLOUDREVE_DATA_DIR, exist_ok=True)
-    print(f\"Extracting backup to: {CLOUDREVE_DATA_DIR}\")
+    print(f\"将备份解压到：{CLOUDREVE_DATA_DIR}\")
     try:
         with tarfile.open(backup_file_path, 'r:gz') as tar:
             tar.extractall(CLOUDREVE_DATA_DIR)
-        print(f'Successfully restored backup from {latest_backup}')
+        print(f'已成功从 {latest_backup} 恢复备份')
     except Exception as e:
-        print(f'Error extracting backup file: {e}')
+        print(f'解压备份文件时出错：{e}')
         sys.exit(1)
+
+    # 同时恢复配置文件
+    config_file_name = os.path.basename(CLOUDREVE_CONFIG_FILE)
+    extracted_config_path = os.path.join(CLOUDREVE_DATA_DIR, config_file_name)
+    if os.path.exists(extracted_config_path):
+        print(f"将恢复的配置文件移动到 {CLOUDREVE_CONFIG_FILE}")
+        shutil.move(extracted_config_path, CLOUDREVE_CONFIG_FILE)
+    else:
+        print("警告：备份中未找到配置文件。")
+
 else:
-    print('Downloaded backup file does not exist.')
+    print('下载的备份文件不存在。')
 "
 }
 
 # 首次启动时下载最新备份
-echo "Downloading latest backup from WebDAV..."
+echo "正在从 WebDAV 下载最新备份..."
 restore_backup
 
-# 等待30秒后启动程序
+# 等待一段时间以确保 Cloudreve 正常启动
 sleep 30
 
 # 启动程序 (根据您的日志，您运行的是 Cloudreve)
-./cloudreve -c /opt/cloudreve/conf/cloudreve.ini &
+./cloudreve -c "$CLOUDREVE_CONFIG_FILE" &
 
 # 同步函数
 sync_data() {
     while true; do
-        echo "Starting sync process at $(date)"
+        echo "开始同步进程于 $(date)"
 
         if [ ! -d "$CLOUDREVE_DATA_DIR" ]; then
             mkdir -p "$CLOUDREVE_DATA_DIR"
-            echo "Data directory created."
+            echo "数据目录已创建。"
         fi
 
         timestamp=$(date +%Y%m%d_%H%M%S)
-        backup_file="alist_backup_${timestamp}.tar.gz"
+        backup_file="cloudreve_backup_${timestamp}.tar.gz"
         backup_file_path="/tmp/${backup_file}"
 
-        echo "Creating backup archive: ${backup_file_path}"
-        # 压缩数据目录，排除可能存在的临时文件
-        tar -czf "${backup_file_path}" --exclude='*.tmp' -C "$CLOUDREVE_DATA_DIR" .
+        echo "创建备份归档：${backup_file_path}"
+        # 压缩数据目录和配置文件
+        tar -czf "${backup_file_path}" -C "$CLOUDREVE_DATA_DIR" . -C "$(dirname "$CLOUDREVE_CONFIG_FILE")" "$(basename "$CLOUDREVE_CONFIG_FILE")"
 
         if [ -f "${backup_file_path}" ]; then
-            echo "Uploading backup to WebDAV: ${FULL_WEBDAV_URL}/${backup_file}"
+            echo "正在上传备份到 WebDAV：${FULL_WEBDAV_URL}/${backup_file}"
             curl -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -T "${backup_file_path}" "${FULL_WEBDAV_URL}/${backup_file}"
             if [ $? -eq 0 ]; then
-                echo "Successfully uploaded ${backup_file} to WebDAV"
+                echo "成功上传 ${backup_file} 到 WebDAV"
             else
-                echo "Failed to upload ${backup_file} to WebDAV"
+                echo "上传 ${backup_file} 到 WebDAV 失败"
             fi
 
             # 清理旧备份文件
@@ -143,26 +155,27 @@ options = {
 }
 client = Client(options)
 try:
-    backups = sorted([file for file in client.list() if file.endswith('.tar.gz') and file.startswith('alist_backup_')])
+    backups = sorted([file for file in client.list() if file.endswith('.tar.gz') and file.startswith('cloudreve_backup_')])
     if len(backups) > BACKUP_RETENTION:
         num_to_delete = len(backups) - BACKUP_RETENTION
         for file in backups[:num_to_delete]:
             try:
                 client.clean(file)
-                print(f'Successfully deleted old backup: {file}')
+                print(f'成功删除旧备份：{file}')
             except Exception as e:
-                print(f'Error deleting {file}: {e}')
+                print(f'删除 {file} 时出错：{e}')
 except Exception as e:
-    print(f'Error managing backups: {e}')
+    print(f'管理备份时出错：{e}')
 " 2>&1
 
             rm -f "${backup_file_path}"
         else
-            echo "Failed to create backup archive."
+            echo "创建备份归档失败。"
         fi
 
+
         SYNC_INTERVAL=${SYNC_INTERVAL:-600}
-        echo "Next sync in ${SYNC_INTERVAL} seconds..."
+        echo "下一次同步将在 ${SYNC_INTERVAL} 秒后进行..."
         sleep "$SYNC_INTERVAL"
     done
 }