#!/usr/bin/env python3
"""Publish a committed revision only to the private ZeroGPU Space created here."""
import json
from fnmatch import fnmatchcase
from pathlib import Path
import subprocess
from huggingface_hub import HfApi, get_token
from deploy import ALLOWLIST

ROOT = Path(__file__).resolve().parents[1]
state = json.loads((ROOT / "deployment.json").read_text())
repo = state["repo_id"]
token = get_token()
if not token:
    raise SystemExit("Existing authentication required")
if subprocess.check_output(["git", "status", "--porcelain"], cwd=ROOT, text=True).strip():
    raise SystemExit("Commit the source before publishing")
api = HfApi(token=token)
info = api.space_info(repo)
hardware = info.runtime.hardware or info.runtime.requested_hardware
if not info.private or str(hardware) != "zero-a10g":
    raise SystemExit("Expected the existing private ZeroGPU deployment")
source = subprocess.check_output(["git", "rev-parse", "HEAD"], cwd=ROOT, text=True).strip()
commit = api.upload_folder(repo_id=repo, repo_type="space", folder_path=str(ROOT),
    allow_patterns=ALLOWLIST, parent_commit=info.sha,
    commit_message=f"Publish Qwen 2.1 Image source {source[:12]}")
state.update(hub_commit=commit.oid, latest_source_commit=source)
(ROOT / "deployment.json").write_text(json.dumps(state, indent=2)+"\n")
expected = {p.relative_to(ROOT).as_posix() for p in ROOT.rglob("*") if p.is_file()
            and any(fnmatchcase(p.relative_to(ROOT).as_posix(), pattern) for pattern in ALLOWLIST)}
remote = set(api.list_repo_files(repo, repo_type="space", revision=commit.oid))
if expected - remote:
    raise SystemExit("Hub omitted allowlisted files; inspect .gitignore before continuing: "
                     + ", ".join(sorted(expected - remote)))
print(json.dumps({"space": repo, "hub_commit": commit.oid, "source_commit": source}))