Update Dockerfile

Dockerfile

@@ -1,5 +1,4 @@
 FROM node:20-slim AS builder
-
 WORKDIR /app
 RUN apt-get update && apt-get install -y git python3 make g++ && rm -rf /var/lib/apt/lists/*
 RUN git clone --depth=1 https://github.com/decolua/9router.git .
@@ -7,26 +6,21 @@ RUN sed -i 's/sameSite: "lax"/sameSite: "none"/' src/app/api/auth/login/route.js
 RUN cat > next.config.mjs << 'NEXTEOF'
 const nextConfig = {
   output: "standalone",
-  serverExternalPackages: ["better-sqlite3", "sql.js", "node:sqlite", "bun:sqlite"],
-  images: { unoptimized: true },
-  env: {},
-  webpack: (config, { isServer }) => {
-    if (!isServer) config.resolve.fallback = { ...config.resolve.fallback, fs: false, path: false };
-    config.watchOptions = { ...config.watchOptions, ignored: /[\/](logs|\.next)[\/]/ };
+  serverExternalPackages: ["better-sqlite3","sql.js","node:sqlite","bun:sqlite"],
+  images:{unoptimized:true},env:{},
+  webpack:(config,{isServer})=>{
+    if(!isServer)config.resolve.fallback={...config.resolve.fallback,fs:false,path:false};
+    config.watchOptions={...config.watchOptions,ignored:/[\/](logs|\.next)[\/]/};
     return config;
   },
-  async headers() {
-    return [{ source: "/(.*)", headers: [{ key: "X-Frame-Options", value: "ALLOWALL" }, { key: "Content-Security-Policy", value: "frame-ancestors *" }] }];
-  },
-  async rewrites() {
-    return [
-      { source: "/v1/v1/:path*", destination: "/api/v1/:path*" },
-      { source: "/v1/v1", destination: "/api/v1" },
-      { source: "/codex/:path*", destination: "/api/v1/responses" },
-      { source: "/v1/:path*", destination: "/api/v1/:path*" },
-      { source: "/v1", destination: "/api/v1" }
-    ];
-  }
+  async headers(){return[{source:"/(.*)",headers:[{key:"X-Frame-Options",value:"ALLOWALL"},{key:"Content-Security-Policy",value:"frame-ancestors *"}]}];},
+  async rewrites(){return[
+    {source:"/v1/v1/:path*",destination:"/api/v1/:path*"},
+    {source:"/v1/v1",destination:"/api/v1"},
+    {source:"/codex/:path*",destination:"/api/v1/responses"},
+    {source:"/v1/:path*",destination:"/api/v1/:path*"},
+    {source:"/v1",destination:"/api/v1"}
+  ];}
 };
 export default nextConfig;
 NEXTEOF
@@ -34,7 +28,8 @@ RUN npm install && npm run build
 
 # ---- Runtime ----
 FROM node:20-slim
-RUN apt-get update && apt-get install -y curl && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y python3 python3-pip curl && rm -rf /var/lib/apt/lists/*
+RUN pip3 install huggingface_hub --break-system-packages --quiet
 
 WORKDIR /app
 COPY --from=builder /app/.next/standalone ./
@@ -49,59 +44,89 @@ ENV INITIAL_PASSWORD=123456
 ENV AUTH_COOKIE_SECURE=true
 ENV DB_PATH=/root/.9router/db/data.sqlite
 ENV HF_DATASET=gunnybd01/9router-db-backup
+ENV SYNC_INTERVAL=3600
 
 EXPOSE 7860
 
-RUN cat > /sync_db.sh << 'SYNCEOF'
-#!/bin/sh
-DB_FILE="${DB_PATH:-/root/.9router/db/data.sqlite}"
-TOKEN="$HF_TOKEN"
-DATASET="${HF_DATASET:-gunnybd01/9router-db-backup}"
-if [ -z "$TOKEN" ]; then echo "[sync] no HF_TOKEN, skip"; exit 0; fi
-if [ ! -f "$DB_FILE" ]; then echo "[sync] no DB file, skip"; exit 0; fi
-echo "[sync] Pushing DB at $(date)..."
-HTTP=$(curl -s -o /tmp/sr.txt -w "%{http_code}" -X PUT -H "Authorization: Bearer $TOKEN" -T "$DB_FILE" "https://huggingface.co/api/datasets/${DATASET}/upload/main/data.sqlite")
-echo "[sync] HTTP $HTTP: $(cat /tmp/sr.txt | head -c 100)"
-SYNCEOF
+# Python script: push DB lên HF Dataset
+RUN cat > /sync_db.py << 'PYEOF'
+import os, sys, time
+from huggingface_hub import HfApi
 
-RUN cat > /restore_db.sh << 'RESTOREEOF'
-#!/bin/sh
-DB_FILE="${DB_PATH:-/root/.9router/db/data.sqlite}"
-DB_DIR=$(dirname "$DB_FILE")
-TOKEN="$HF_TOKEN"
-DATASET="${HF_DATASET:-gunnybd01/9router-db-backup}"
-mkdir -p "$DB_DIR"
-if [ -z "$TOKEN" ]; then echo "[restore] no HF_TOKEN"; exit 0; fi
-echo "[restore] Pulling DB from $DATASET..."
-HTTP=$(curl -L -s -o /tmp/db.sqlite -w "%{http_code}" -H "Authorization: Bearer $TOKEN" "https://huggingface.co/datasets/${DATASET}/resolve/main/data.sqlite")
-if [ "$HTTP" = "200" ]; then
-  SZ=$(wc -c < /tmp/db.sqlite)
-  if [ "$SZ" -gt "4096" ]; then
-    cp /tmp/db.sqlite "$DB_FILE"
-    echo "[restore] OK - ${SZ} bytes restored"
-  else
-    echo "[restore] Download too small (${SZ}b), starting fresh"
-  fi
-else
-  echo "[restore] HTTP $HTTP - no backup yet, starting fresh"
-fi
-RESTOREEOF
+DB_FILE = os.environ.get("DB_PATH", "/root/.9router/db/data.sqlite")
+TOKEN   = os.environ.get("HF_TOKEN", "")
+DATASET = os.environ.get("HF_DATASET", "gunnybd01/9router-db-backup")
+
+if not TOKEN:
+    print("[sync] HF_TOKEN not set, skip"); sys.exit(0)
+if not os.path.exists(DB_FILE):
+    print(f"[sync] DB not found: {DB_FILE}"); sys.exit(0)
+
+sz = os.path.getsize(DB_FILE)
+print(f"[sync] Pushing DB ({sz} bytes) -> {DATASET} ...")
+try:
+    api = HfApi(token=TOKEN)
+    api.upload_file(
+        path_or_fileobj=DB_FILE,
+        path_in_repo="data.sqlite",
+        repo_id=DATASET,
+        repo_type="dataset",
+        commit_message=f"DB sync {time.strftime('%Y-%m-%d %H:%M')}",
+    )
+    print(f"[sync] OK at {time.strftime('%Y-%m-%d %H:%M:%S')}")
+except Exception as e:
+    print(f"[sync] ERROR: {e}")
+PYEOF
+
+# Python script: pull DB từ HF Dataset khi startup
+RUN cat > /restore_db.py << 'PYEOF2'
+import os, sys
+from huggingface_hub import hf_hub_download
+
+DB_FILE = os.environ.get("DB_PATH", "/root/.9router/db/data.sqlite")
+TOKEN   = os.environ.get("HF_TOKEN", "")
+DATASET = os.environ.get("HF_DATASET", "gunnybd01/9router-db-backup")
+
+os.makedirs(os.path.dirname(DB_FILE), exist_ok=True)
+
+if not TOKEN:
+    print("[restore] HF_TOKEN not set, starting fresh"); sys.exit(0)
+
+print(f"[restore] Pulling DB from {DATASET}...")
+try:
+    path = hf_hub_download(
+        repo_id=DATASET,
+        filename="data.sqlite",
+        repo_type="dataset",
+        token=TOKEN,
+        local_dir="/tmp/hf_restore",
+    )
+    sz = os.path.getsize(path)
+    if sz > 4096:
+        import shutil
+        shutil.copy2(path, DB_FILE)
+        print(f"[restore] OK - {sz} bytes restored from backup")
+    else:
+        print(f"[restore] Backup too small ({sz}b), starting fresh")
+except Exception as e:
+    print(f"[restore] No backup yet ({e}), starting fresh")
+PYEOF2
 
 RUN cat > /start.sh << 'STARTEOF'
 #!/bin/sh
-# 1. Restore DB from HF Dataset backup
-sh /restore_db.sh
-# 2. Start app
+# 1. Restore DB from backup
+python3 /restore_db.py
+# 2. Start Next.js
 node /app/server.js &
 APP_PID=$!
-# 3. Hourly sync in background
-(while true; do sleep 3600; sh /sync_db.sh; done) &
-# 4. Sync on shutdown (Space sleep)
-trap "sh /sync_db.sh; kill $APP_PID 2>/dev/null" TERM INT
+# 3. Sync loop (default 1h)
+INTERVAL="${SYNC_INTERVAL:-3600}"
+(while true; do sleep "$INTERVAL"; python3 /sync_db.py; done) &
+# 4. Sync on SIGTERM (Space sleeping)
+trap "echo "[trap] Syncing before sleep..."; python3 /sync_db.py; kill $APP_PID 2>/dev/null; exit 0" TERM INT
 # 5. Wait
 wait $APP_PID
 STARTEOF
 
-RUN chmod +x /sync_db.sh /restore_db.sh /start.sh
-
+RUN chmod +x /start.sh
 CMD ["/bin/sh", "/start.sh"]