Create sync-script.sh

sync-script.sh

@@ -0,0 +1,199 @@
+#!/bin/bash
+set -e
+
+WEBDAV_URL="${WEBDAV_URL}"
+WEBDAV_USERNAME="${WEBDAV_USERNAME}"
+WEBDAV_PASSWORD="${WEBDAV_PASSWORD}"
+DB_PATH="${DB_PATH:-/app/db}" # 设置默认数据库路径
+BACKUP_PATH="${BACKUP_PATH:-/app/backup}" # 设置默认备份路径
+BACKUP_FILENAME="${BACKUP_FILENAME:-tgdrive_db_backup.tar.gz}" # 设置默认备份文件名
+REMOTE_BACKUP_DIR="${REMOTE_BACKUP_DIR:-tgdrive_backup}" # 设置默认远程备份目录
+CONFIG_FILE="/app/config/sync_config"
+PYTHON_CMD="/app/venv/bin/python3"
+
+mkdir -p ${BACKUP_PATH}
+mkdir -p /app/logs
+
+log() {
+    echo "$(date '+%Y-%m-%d %H:%M:%S') - $1" >> /app/logs/sync.log 2>/dev/null || true
+    echo "$(date '+%Y-%m-%d %H:%M:%S') - $1"
+}
+
+mkdir -p /app/logs 2>/dev/null || true
+touch /app/logs/sync.log 2>/dev/null || true
+
+if [ ! -f "$CONFIG_FILE" ]; then
+    log "创建初始配置文件"
+    mkdir -p /app/config 2>/dev/null || true
+    echo "LAST_SYNC=0" > "$CONFIG_FILE" 2>/dev/null || true
+fi
+
+if [ -f "$CONFIG_FILE" ]; then
+    source "$CONFIG_FILE" 2>/dev/null || true
+fi
+
+if [ -z "$WEBDAV_URL" ] || [ -z "$WEBDAV_USERNAME" ] || [ -z "$WEBDAV_PASSWORD" ]; then
+    log "错误: WebDAV配置不完整，请设置WEBDAV_URL, WEBDAV_USERNAME, WEBDAV_PASSWORD环境变量"
+    exit 1
+fi
+
+backup_db() {
+    log "开始备份数据库..."
+    if [ -d "$DB_PATH" ]; then
+        if [[ "$DB_PATH" == "/app/db" ]]; then
+            tar -czf "${BACKUP_PATH}/${BACKUP_FILENAME}" -C /app db
+        else
+            tar -czf "${BACKUP_PATH}/${BACKUP_FILENAME}" -C "$(dirname "$DB_PATH")" "$(basename "$DB_PATH")"
+        fi
+        log "数据库已备份到 ${BACKUP_PATH}/${BACKUP_FILENAME}"
+        return 0
+    else
+        log "警告: 数据库目录不存在"
+        return 1
+    fi
+}
+
+upload_to_webdav() {
+    log "开始上传备份到WebDAV (目标目录: ${REMOTE_BACKUP_DIR})..."
+
+    FULL_WEBDAV_URL="${WEBDAV_URL}/${REMOTE_BACKUP_DIR}"
+    local_path="${BACKUP_PATH}/${BACKUP_FILENAME}"
+    remote_file="${BACKUP_FILENAME}"
+
+    if [ -f "$local_path" ]; then
+        curl -u "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" -T "$local_path" "$FULL_WEBDAV_URL/$remote_file"
+        if [ $? -eq 0 ]; then
+            log "备份已成功上传到WebDAV (${REMOTE_BACKUP_DIR}/${BACKUP_FILENAME})"
+            echo "LAST_SYNC=$(date +%s)" > "$CONFIG_FILE"
+            return 0
+        else
+            log "上传到WebDAV失败"
+            return 1
+        fi
+    else
+        log "错误: 本地备份文件不存在"
+        return 1
+    fi
+}
+
+download_from_webdav() {
+    log "开始从WebDAV下载备份 (源目录: ${REMOTE_BACKUP_DIR})..."
+
+    $PYTHON_CMD -c "
+import webdav3.client as wc
+import os
+
+options = {
+    'webdav_hostname': '$WEBDAV_URL',
+    'webdav_login': '$WEBDAV_USERNAME',
+    'webdav_password': '$WEBDAV_PASSWORD',
+    'webdav_timeout': 30
+}
+
+client = wc.Client(options)
+
+# 检查远程文件是否存在
+remote_path = '$REMOTE_BACKUP_DIR/${BACKUP_FILENAME}'
+local_path = '${BACKUP_PATH}/${BACKUP_FILENAME}'
+
+if client.check(remote_path):
+    try:
+        client.download_sync(remote_path=remote_path, local_path=local_path)
+        print('下载成功')
+    except Exception as e:
+        print(f'下载失败: {e}')
+else:
+    print('远程备份文件不存在')
+"
+
+    if [ $? -eq 0 ]; then
+        log "备份已成功从WebDAV (${REMOTE_BACKUP_DIR}/${BACKUP_FILENAME}) 下载"
+        return 0
+    else
+        log "从WebDAV下载备份失败"
+        return 1
+    fi
+}
+
+restore_db() {
+    log "开始恢复数据库到 ${DB_PATH}..."
+
+    if [ -f "${BACKUP_PATH}/${BACKUP_FILENAME}" ]; then
+        if [ -d "$DB_PATH" ] && [ "$(ls -A $DB_PATH 2>/dev/null)" ]; then
+            local timestamp=$(date +%Y%m%d%H%M%S)
+            if [[ "$DB_PATH" == "/app/db" ]]; then
+                tar -czf "${BACKUP_PATH}/db_before_restore_${timestamp}.tar.gz" -C /app db 2>/dev/null || log "无法创建当前数据库备份，继续恢复..."
+            else
+                tar -czf "${BACKUP_PATH}/db_before_restore_${timestamp}.tar.gz" -C "$(dirname "$DB_PATH")" "$(basename "$DB_PATH")" 2>/dev/null || log "无法创建当前数据库备份，继续恢复..."
+            fi
+            log "尝试备份当前数据库到 ${BACKUP_PATH}/db_before_restore_${timestamp}.tar.gz"
+        fi
+
+        # 创建临时恢复目录
+        mkdir -p /tmp/tgdrive_restore 2>/dev/null || true
+
+        # 解压备份文件到临时目录
+        tar -xzf "${BACKUP_PATH}/${BACKUP_FILENAME}" -C /tmp/tgdrive_restore 2>/dev/null || {
+            log "解压备份文件失败，可能是格式不正确"
+            rm -rf /tmp/tgdrive_restore 2>/dev/null || true
+            return 1
+        }
+
+        # 确保数据库目录存在
+        mkdir -p "$DB_PATH" 2>/dev/null || true
+
+        # 清空数据库目录内容（逐个文件删除而不是删除整个目录）
+        find "$DB_PATH" -mindepth 1 -delete 2>/dev/null || log "无法清空数据库目录，将尝试直接覆盖..."
+
+        # 复制恢复的文件到数据库目录
+        if [ -d "/tmp/tgdrive_restore/db" ]; then
+            cp -rf /tmp/tgdrive_restore/db/* "$DB_PATH"/ 2>/dev/null || {
+                log "复制恢复文件失败，尝试使用rsync..."
+                which rsync >/dev/null 2>&1 && rsync -a /tmp/tgdrive_restore/db/ "$DB_PATH"/ 2>/dev/null
+            }
+        else
+            log "备份中找不到db目录，尝试直接复制临时目录内容..."
+            cp -rf /tmp/tgdrive_restore/* "$DB_PATH"/ 2>/dev/null || {
+                log "复制恢复文件失败，尝试使用rsync..."
+                which rsync >/dev/null 2>&1 && rsync -a /tmp/tgdrive_restore/ "$DB_PATH"/ 2>/dev/null
+            }
+        fi
+
+        rm -rf /tmp/tgdrive_restore 2>/dev/null || true
+
+        log "数据库已尝试恢复到 ${DB_PATH}"
+        return 0
+    else
+        log "错误: 备份文件不存在，无法恢复数据库"
+        return 1
+    fi
+}
+
+case "$1" in
+    backup)
+        backup_db && upload_to_webdav
+        ;;
+    restore)
+        download_from_webdav && restore_db
+        ;;
+    sync)
+        current_time=$(date +%s)
+        download_from_webdav
+
+        if { [ ! -d "$DB_PATH" ] || [ ! "$(ls -A $DB_PATH 2>/dev/null)" ]; } && [ -f "${BACKUP_PATH}/${BACKUP_FILENAME}" ]; then
+            log "检测到本地数据库不存在或为空，开始恢复..."
+            restore_db
+        fi
+
+        backup_db && upload_to_webdav
+        ;;
+    *)
+        echo "用法: $0 {backup|restore|sync}"
+        echo "  backup  - 备份数据库并上传到WebDAV"
+        echo "  restore - 从WebDAV下载备份并恢复数据库"
+        echo "  sync    - 同步数据库（双向）"
+        exit 1
+        ;;
+esac
+
+exit 0