"""Publish envs/superbeam_env as a Docker Space (StarAtNyte1/superbeam-env) using the
Modal-held HF write token.  The token stays in memory; it is never printed or saved.

  modal run scripts/publish_env_space.py

Bundles the game files first (scripts/prepare_superbeam_env.py) and uploads server/Dockerfile
as the repo-root Dockerfile (Spaces requires it there).
"""
import os
import shutil
import subprocess
import sys
import tempfile
from pathlib import Path

import modal

REPO_ID = "StarAtNyte1/superbeam-env"
ROOT = Path(__file__).resolve().parent.parent
ENV_DIR = ROOT / "envs" / "superbeam_env"

app = modal.App("superbeam-env-publish")


@app.function(image=modal.Image.debian_slim(python_version="3.12"),
              secrets=[modal.Secret.from_name("steerability-submission-write")])
def get_write_token():
    token = os.environ.get("HUGGINGFACE_TOKEN") or os.environ.get("HF_TOKEN")
    if not token:
        raise RuntimeError("No HF write token in configured secret")
    return token


@app.local_entrypoint()
def main(private: bool = False, repo_type: str = "space"):
    """repo_type 'space' = Docker Space (needs HF PRO on free hardware); 'model' = a plain Hub repo
    holding the same files, run locally with uvicorn or `docker build`."""
    from huggingface_hub import HfApi

    subprocess.run([sys.executable, str(ROOT / "scripts" / "prepare_superbeam_env.py")], check=True,
                   stdout=subprocess.DEVNULL)
    token = get_write_token.remote()
    api = HfApi(token=token)
    if not api.repo_exists(repo_id=REPO_ID, repo_type=repo_type):
        kw = {"space_sdk": "docker"} if repo_type == "space" else {}
        api.create_repo(repo_id=REPO_ID, repo_type=repo_type, private=private, **kw)
    url = "https://huggingface.co/%s%s" % ("spaces/" if repo_type == "space" else "", REPO_ID)
    with tempfile.TemporaryDirectory() as tmp:
        stage = Path(tmp) / "env"
        shutil.copytree(ENV_DIR, stage, ignore=shutil.ignore_patterns("__pycache__", "*.pyc", ".venv*"))
        shutil.copy2(stage / "server" / "Dockerfile", stage / "Dockerfile")
        print(f"Uploading to {url}", flush=True)
        api.upload_folder(repo_id=REPO_ID, repo_type=repo_type, folder_path=str(stage),
                          commit_message="SUPERBEAM OpenEnv environment")
    print(f"Done: {url}", flush=True)