Update Dockerfile

Dockerfile

@@ -1,18 +1,17 @@
 FROM lampon/omnibox:latest
 
-# 安装 Python3、pip 和 huggingface_hub（用于上传备份）
+# 安装 Python3、pip 和 huggingface_hub
 RUN apt-get update && apt-get install -y python3 python3-pip socat && \
     pip3 install --no-cache-dir huggingface_hub && \
     rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-# 创建备份脚本
+# 创建备份脚本（使用 heredoc 避免解析问题）
 RUN cat > /usr/local/bin/backup.sh << 'EOF'
 #!/bin/bash
 set -e
 
-# 环境变量检查
 if [ -z "$HF_DATASET" ] || [ -z "$HF_TOKEN" ]; then
     echo "❌ 缺少 HF_DATASET 或 HF_TOKEN 环境变量，跳过备份"
     exit 0
@@ -20,11 +19,8 @@ fi
 
 echo "$(date): 📦 开始备份 omnibox 数据..."
 
-# 创建临时打包目录
 mkdir -p /tmp/omnibox_backup
 
-# 复制需要备份的数据（根据你的实际数据路径修改）
-# 注意：omnibox 的数据通常位于 /app/data 目录下（包含数据库、订阅、配置等）
 if [ -d /app/data ]; then
     cp -r /app/data /tmp/omnibox_backup/
     echo "✅ 已复制 /app/data"
@@ -32,20 +28,14 @@ else
     echo "⚠️ 未找到 /app/data 目录，跳过数据复制"
 fi
 
-# 你也可以备份其他目录，例如 /app/config、/app/db 等
-# cp -r /app/config /tmp/omnibox_backup/ 2>/dev/null || true
-
-# 打包
 cd /tmp
 tar -czf omnibox_backup.tar.gz omnibox_backup
 BACKUP_FILE="/tmp/omnibox_backup.tar.gz"
 
-# 生成带时间戳的备份文件名（避免覆盖）
 TIMESTAMP=$(date +%Y%m%d_%H%M%S)
 REMOTE_PATH="omnibox_backup_${TIMESTAMP}.tar.gz"
 
-# 使用 Python 上传到 Hugging Face Dataset
-python3 <<EOF
+python3 <<PYEOF
 from huggingface_hub import HfApi
 import os
 
@@ -66,24 +56,23 @@ try:
     print(f"✅ 备份上传成功: {remote_path}")
 except Exception as e:
     print(f"❌ 备份上传失败: {e}")
-EOF
+PYEOF
 
-# 清理临时文件
 rm -rf /tmp/omnibox_backup /tmp/omnibox_backup.tar.gz
 echo "$(date): 🎉 备份流程完成"
 EOF
 
 RUN chmod +x /usr/local/bin/backup.sh
 
-# 修改启动脚本：先尝试恢复最新备份，再启动服务，同时启动定时备份
+# 创建启动脚本
 RUN cat > /usr/local/bin/start.sh << 'EOF'
 #!/bin/bash
 set -e
 
-# 1. 尝试恢复最新备份（如果有）
+# 恢复备份
 if [ -n "$HF_DATASET" ] && [ -n "$HF_TOKEN" ]; then
     echo "🔄 检查是否存在可恢复的备份..."
-    python3 <<EOF
+    python3 <<PYEOF
 from huggingface_hub import HfApi, hf_hub_download
 import os, tarfile, glob, re
 
@@ -92,13 +81,11 @@ token = os.environ.get("HF_TOKEN")
 api = HfApi()
 
 try:
-    # 获取数据集中的所有备份文件
     files = api.list_repo_files(repo_id=repo_id, repo_type="dataset", token=token)
     backup_files = [f for f in files if f.startswith("omnibox_backup_") and f.endswith(".tar.gz")]
     if not backup_files:
         print("📭 未找到历史备份文件")
     else:
-        # 按时间戳排序，取最新的一个
         backup_files.sort(reverse=True)
         latest = backup_files[0]
         print(f"📥 发现最新备份: {latest}，正在下载...")
@@ -109,31 +96,28 @@ try:
             token=token,
             local_dir="/tmp/restore"
         )
-        # 解压到 /app
         with tarfile.open(path, "r:gz") as tar:
             tar.extractall(path="/tmp/restore_data")
-        # 将恢复的数据复制到 /app/data（根据你的实际路径调整）
         if os.path.exists("/tmp/restore_data/omnibox_backup/data"):
             rm -rf /app/data
             cp -r /tmp/restore_data/omnibox_backup/data /app/
             echo "✅ 数据恢复成功"
         else:
             echo "⚠️ 备份文件中未找到 /data 目录，跳过恢复"
-        # 清理
         rm -rf /tmp/restore /tmp/restore_data
 except Exception as e:
     print(f"⚠️ 恢复过程出现异常: {e}")
-EOF
+PYEOF
 fi
 
-# 2. 启动原服务（后台运行）
+# 启动主服务
 echo "🚀 启动 Omnibox 主服务..."
 ./main &
 MAIN_PID=$!
 socat TCP-LISTEN:7860,fork TCP:127.0.0.1:7023 &
 SOCAT_PID=$!
 
-# 3. 启动定时备份（每30分钟）
+# 定时备份
 echo "⏰ 启动自动备份（每30分钟）..."
 (
     while true; do
@@ -143,10 +127,8 @@ echo "⏰ 启动自动备份（每30分钟）..."
 ) &
 BACKUP_PID=$!
 
-# 4. 容器退出时执行一次备份
 trap 'echo "🛑 容器停止，执行最后一次备份..."; /usr/local/bin/backup.sh; kill $MAIN_PID $SOCAT_PID $BACKUP_PID 2>/dev/null' EXIT
 
-# 5. 保持前台进程
 wait $MAIN_PID
 EOF