"""Publish the Studio with a verified, committed LeRobot runtime bundle."""

import argparse
import hashlib
import json
import os
import subprocess
import sys
import tarfile
import tempfile
from pathlib import Path

from huggingface_hub import HfApi


def main():
    """Package committed code and publish the corresponding Studio catalog."""
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--repo-id", default="finerobotics/annotation-studio")
    parser.add_argument("--pipeline-ref", default="HEAD")
    parser.add_argument("--package-only", action="store_true")
    args = parser.parse_args()
    folder = Path(__file__).resolve().parent
    root = folder.parents[1]
    if subprocess.check_output(["git", "diff", "HEAD", "--", "src", "pyproject.toml"], cwd=root):
        raise RuntimeError("Commit or separately package tracked pipeline edits before publishing.")
    commit = subprocess.check_output(["git", "rev-parse", args.pipeline_ref], cwd=root, text=True).strip()
    archive = folder / "pipeline-source.tar.gz"
    subprocess.run(
        [
            "git",
            "archive",
            "--format=tar.gz",
            f"--output={archive}",
            commit,
            "src",
            "pyproject.toml",
            "LICENSE",
        ],
        cwd=root,
        check=True,
    )
    (folder / "pipeline-version.json").write_text(
        json.dumps(
            {
                "git_commit": commit,
                "sha256": hashlib.sha256(archive.read_bytes()).hexdigest(),
                "preset": "abc_phases_schema_with_original_window_recovery",
                "preset_date": "2026-09-25",
                "engine": "lerobot.annotations.runtime",
                "framework_pr": "https://github.com/huggingface/lerobot/pull/4892",
                "human_video_pr": "https://github.com/huggingface/lerobot/pull/4809",
            },
            indent=2,
        )
        + "\n"
    )
    with tempfile.TemporaryDirectory(prefix="studio-catalog-") as temporary:
        with tarfile.open(archive) as source:
            source.extractall(temporary, filter="data")
        catalog = json.loads(
            subprocess.check_output(
                [
                    sys.executable,
                    "-c",
                    "import json; from hand_actions import public_catalog; print(json.dumps(public_catalog()))",
                ],
                cwd=temporary,
                env={
                    **os.environ,
                    "PYTHONPATH": os.pathsep.join([str(Path(temporary) / "src"), str(folder)]),
                },
                text=True,
            )
        )
    catalog["lerobot_revision"] = commit
    (folder / "hand-pipeline-catalog.json").write_text(json.dumps(catalog, indent=2) + "\n")
    if args.package_only:
        print(f"Packaged pipeline {commit}")
        return
    api = HfApi()
    repo = args.repo_id
    api.create_repo(repo, repo_type="space", space_sdk="docker", private=False, exist_ok=True)
    result = api.upload_folder(
        repo_id=repo,
        repo_type="space",
        folder_path=folder,
        ignore_patterns=[
            "__pycache__/**",
            ".pytest_cache/**",
            ".ruff_cache/**",
            "*.pyc",
            "frontend/node_modules/**",
            "frontend/dist/**",
            ".cache/**",
            ".git/**",
        ],
        delete_patterns=["app.py", "frontend/src/HandSetupView.tsx", "frontend/src/hands.css"],
        commit_message="Use native EgoHand/SOMA hand actions with DROID-W and MoGe on HF Jobs",
    )
    print(f"https://huggingface.co/spaces/{repo}\nRevision: {result.oid}")


if __name__ == "__main__":
    main()