Update Dockerfile

Dockerfile

@@ -127,12 +127,39 @@ def restore():
         print("Restore skip: RESTORE_SKIP=all, skipping all restore.")
         return
 
-    # 解析跳過列表（用於在解壓時逐條目跳過，不是解壓後刪除）
-    skip_set = _parse_skip_list("RESTORE_SKIP")
+    # ── 方案 B：以 Dataset 中的 initialized.flag 作為觸發條件 ─────────────────
+    # 邏輯：
+    #   - Dataset 中無 initialized.flag  → 首次部署 → 執行 restore，完成後寫入標記
+    #   - Dataset 中有 initialized.flag  → 普通 restart → 跳過 restore
+    #   - FORCE_RESTORE=true             → 無視標記，強制執行 restore（用於 factory rebuild）
+    #
+    # Factory rebuild 場景的兩種觸發方式（任選其一）：
+    #   1. 在 HF Space Settings 中設置環境變量 FORCE_RESTORE=true，重建後再刪除此變量
+    #   2. 直接在 HF Dataset 網頁上刪除 initialized.flag 文件，再點 Restart
+    INIT_FLAG = "initialized.flag"
+    force_restore = os.getenv("FORCE_RESTORE", "").strip().lower() in ("true", "1", "yes")
 
     try:
         all_files = list(api.list_repo_files(repo_id=repo_id, repo_type="dataset", token=token))
+    except Exception as e:
+        print(f"Restore Error (list_repo_files): {e}")
+        return
+
+    flag_exists = INIT_FLAG in all_files
+
+    if flag_exists and not force_restore:
+        print("Restore skip: initialized.flag found in Dataset, this is a normal restart.")
+        return
+
+    if force_restore:
+        print("Restore: FORCE_RESTORE=true, ignoring initialized.flag.")
+    else:
+        print("Restore: initialized.flag not found, this is first deploy or rebuild.")
 
+    # 解析跳過列表（用於在解壓時逐條目跳過，不是解壓後刪除）
+    skip_set = _parse_skip_list("RESTORE_SKIP")
+
+    try:
         # ── 從 tar.gz 備份恢復 ───────────────────────────────────────────────
         # 備份範圍：/root/.openclaw 下所有文件及文件夾（保持完整目錄結構）
         # tar 包內 arcname 直接對應 .openclaw/ 下的相對路徑，解壓目標為 /root/.openclaw/
@@ -157,11 +184,24 @@ def restore():
                 print(f"Success: Restored from {name}")
                 break
 
+        # ── 寫入 initialized.flag 到 Dataset ────────────────────────────────
+        # 無論是否找到備份，只要走到這裡就說明應當完成初始化流程，寫入標記
+        import io
+        flag_content = f"initialized at container startup\n".encode()
+        api.upload_file(
+            path_or_fileobj=io.BytesIO(flag_content),
+            path_in_repo=INIT_FLAG,
+            repo_id=repo_id,
+            repo_type="dataset",
+            token=token,
+            commit_message="Set initialized.flag",
+        )
+        print(f"initialized.flag written to Dataset.")
+
     except Exception as e:
         print(f"Restore Error: {e}")
 
 
-
 # ── backup() ──────────────────────────────────────────────────────────────────
 
 def backup():
@@ -259,23 +299,12 @@ RUN cat <<'EOF' > /usr/local/bin/start-openclaw
 set -e
 
 # ── restore() 觸發條件判斷 ──────────────────────────────────────────────────
-# HF Space 磁盤是 ephemeral 的，每次容器重啟（首次部署、factory rebuild、普通 restart）
-# /root/.openclaw 都從空狀態開始。平台本身無法區分三種重啟場景。
-# 注意：start-openclaw-code-server 在調用本腳本之前會執行
-#   mkdir -p /root/.openclaw/logs
-# 因此不能以目錄是否存在或為空作為判定依據。
-#
-# 改用 workspace/USER.md 作為判定標誌：
-#   - 該文件由本腳本在 restore() 之後的配置寫入步驟生成
-#   - mkdir -p /root/.openclaw/logs 不會創建它
-#   - 首次部署 / 任何形式的容器重建：文件不存在 → 執行 restore()
-#   - 運行中再次調用本腳本（文件已存在）           → 跳過 restore()
-if [ ! -f /root/.openclaw/workspace/USER.md ]; then
-    echo "=== First deploy / rebuild detected: running restore() ==="
-    python3 /usr/local/bin/sync.py restore
-else
-    echo "=== USER.md exists, skipping restore() ==="
-fi
+# 判定邏輯已移入 sync.py restore() 函數內部（方案 B）：
+#   - Dataset 中無 initialized.flag  → 首次部署 → 執行 restore，完成後寫入標記
+#   - Dataset 中有 initialized.flag  → 普通 restart → 跳過 restore
+#   - FORCE_RESTORE=true             → 強制執行 restore（用於 factory rebuild）
+# 此處直接調用，觸發與否由 restore() 自行決定。
+python3 /usr/local/bin/sync.py restore
 
 mkdir -p /root/.openclaw/sessions