Upload 3 files

Dockerfile

@@ -0,0 +1,49 @@
+FROM nanyangzesi/tgdrive:latest
+
+RUN apk update && apk add --no-cache \
+    curl \
+    rclone \
+    dcron \
+    davfs2 \
+    ca-certificates \
+    python3 \
+    py3-pip \
+    tar \
+    bash
+
+RUN python3 -m venv /app/venv && \
+    /app/venv/bin/pip install --no-cache-dir webdavclient3
+
+WORKDIR /app
+
+RUN mkdir -p /app/db
+RUN mkdir -p /app/data
+RUN mkdir -p /app/backup
+RUN mkdir -p /app/logs
+RUN mkdir -p /app/config
+
+RUN chmod 777 /app/db
+RUN chmod 777 /app/data
+RUN chmod 777 /app/backup
+RUN chmod 777 /app/logs
+RUN chmod 777 /app/config
+
+COPY sync-script.sh /app/
+RUN chmod +x /app/sync-script.sh
+
+COPY entrypoint.sh /app/
+RUN chmod +x /app/entrypoint.sh
+
+EXPOSE 8085
+
+ENV WEBDAV_URL=""
+ENV WEBDAV_USERNAME=""
+ENV WEBDAV_PASSWORD=""
+ENV SYNC_INTERVAL="60"
+ENV PATH="/app/venv/bin:$PATH"
+ENV REMOTE_BACKUP_DIR="tgdrive_backup"
+ENV BACKUP_FILENAME="tgdrive_db_backup.tar.gz"
+ENV DB_PATH="/app/db"
+ENV BACKUP_PATH="/app/backup"
+
+ENTRYPOINT ["/app/entrypoint.sh"] 

entrypoint.sh

@@ -0,0 +1,62 @@
+#!/bin/bash
+set -e
+
+mkdir -p /app/logs 2>/dev/null || true
+
+log() {
+    echo "$(date '+%Y-%m-%d %H:%M:%S') - $1" | tee -a /app/logs/entrypoint.log 2>/dev/null || echo "$(date '+%Y-%m-%d %H:%M:%S') - $1"
+}
+
+log "容器启动中..."
+
+SYNC_INTERVAL="${SYNC_INTERVAL:-60}"
+log "设置同步间隔: ${SYNC_INTERVAL}分钟"
+
+DB_PATH="${DB_PATH:-/app/db}"
+BACKUP_PATH="${BACKUP_PATH:-/app/backup}"
+REMOTE_BACKUP_DIR="${REMOTE_BACKUP_DIR:-tgdrive_backup}"
+BACKUP_FILENAME="${BACKUP_FILENAME:-tgdrive_db_backup.tar.gz}"
+
+mkdir -p "$DB_PATH" 2>/dev/null || true
+mkdir -p "$BACKUP_PATH" 2>/dev/null || true
+
+log "数据库路径: ${DB_PATH}"
+log "备份路径: ${BACKUP_PATH}"
+log "远程备份目录: ${REMOTE_BACKUP_DIR}"
+log "备份文件名: ${BACKUP_FILENAME}"
+
+if [ -n "$WEBDAV_URL" ] && [ -n "$WEBDAV_USERNAME" ] && [ -n "$WEBDAV_PASSWORD" ]; then
+    log "检测到WebDAV配置，尝试初始同步..."
+    
+    /app/sync-script.sh restore || log "首次恢复尝试失败，可能是首次运行或WebDAV上没有备份"
+    
+    log "设置定时同步任务..."
+    while true; do
+        sleep $((SYNC_INTERVAL * 60))
+        /app/sync-script.sh sync >> /app/logs/sync_cron.log 2>&1
+    done &
+    
+    log "后台同步服务已启动，每${SYNC_INTERVAL}分钟执行一次"
+else
+    log "未检测到WebDAV配置，跳过同步设置"
+fi
+
+log "tgDrive 同步版本启动成功"
+
+trap_handler() {
+    log "接收到停止信号，执行备份..."
+    if [ -n "$WEBDAV_URL" ] && [ -n "$WEBDAV_USERNAME" ] && [ -n "$WEBDAV_PASSWORD" ]; then
+        /app/sync-script.sh backup
+        log "备份完成，容器即将停止"
+    else
+        log "未配置WebDAV，跳过备份"
+    fi
+    exit 0
+}
+
+trap trap_handler SIGTERM SIGINT
+
+log "启动tgDrive应用..."
+
+cd /app
+exec java -jar app.jar "$@" 

sync-script.sh

@@ -0,0 +1,200 @@
+#!/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})..."
+    
+    $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)
+
+# 检查远程目录是否存在
+if not client.check('$REMOTE_BACKUP_DIR'):
+    client.mkdir('$REMOTE_BACKUP_DIR')
+
+# 上传文件
+local_path = '${BACKUP_PATH}/${BACKUP_FILENAME}'
+remote_path = '$REMOTE_BACKUP_DIR/${BACKUP_FILENAME}'
+
+if os.path.exists(local_path):
+    try:
+        client.upload_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})"
+        echo "LAST_SYNC=$(date +%s)" > "$CONFIG_FILE"
+        return 0
+    else
+        log "上传到WebDAV失败"
+        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
+            else
+                tar -czf "${BACKUP_PATH}/db_before_restore_${timestamp}.tar.gz" -C $(dirname "$DB_PATH") $(basename "$DB_PATH")
+            fi
+            log "当前数据库已备份到 ${BACKUP_PATH}/db_before_restore_${timestamp}.tar.gz"
+        fi
+        
+        rm -rf "$DB_PATH"
+        mkdir -p "$DB_PATH"
+        
+        if [[ "$DB_PATH" == "/app/db" ]]; then
+            tar -xzf "${BACKUP_PATH}/${BACKUP_FILENAME}" -C /app
+        else
+            mkdir -p /tmp/tgdrive_restore
+            tar -xzf "${BACKUP_PATH}/${BACKUP_FILENAME}" -C /tmp/tgdrive_restore
+            cp -r /tmp/tgdrive_restore/db/* "$DB_PATH"/
+            rm -rf /tmp/tgdrive_restore
+        fi
+        
+        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