from __future__ import annotations

import subprocess
import sys
import tempfile
import unittest
from pathlib import Path

from deployment.hf_space import (
    create_space_commands,
    hackathon_space_commands,
    has_space_remote,
    missing_required_files,
    readme_has_space_metadata,
    space_build_status,
)


class HuggingFaceSpaceDeployTest(unittest.TestCase):
    def test_creates_manual_space_commands(self) -> None:
        commands = create_space_commands("demo-user", "demo-space")

        self.assertEqual(commands[0], "hf auth login")
        self.assertIn(
            "repo create demo-user/demo-space --type space --space-sdk gradio",
            commands[1],
        )
        self.assertIn("https://huggingface.co/spaces/demo-user/demo-space", commands[2])
        self.assertEqual(commands[3], "git push space main")
        self.assertIn("app.py", commands[4])

    def test_requires_hf_user(self) -> None:
        with self.assertRaises(ValueError):
            create_space_commands("")

    def test_detects_space_remote(self) -> None:
        remotes = (
            "origin\thttps://github.com/Ckal/codex.git (fetch)\n"
            "space-workbench\thttps://huggingface.co/spaces/u/s (push)"
        )

        self.assertTrue(has_space_remote(remotes, "space-workbench"))

    def test_hackathon_space_commands_define_two_remotes(self) -> None:
        commands = hackathon_space_commands()

        self.assertIn("workbench", commands)
        self.assertIn("plant", commands)
        self.assertIn("space-workbench", commands["workbench"][2])
        self.assertIn("space-plant", commands["plant"][2])
        self.assertIn("plant_space_app.py", commands["plant"][4])

    def test_validates_space_build_shape(self) -> None:
        with tempfile.TemporaryDirectory() as tmp:
            root = Path(tmp)
            (root / "config").mkdir()
            for path in [
                "app.py",
                "plant_space_app.py",
                "requirements.txt",
                "config/models.yaml",
                "config/training.yaml",
            ]:
                (root / path).write_text("", encoding="utf-8")
            (root / "README.md").write_text(
                "---\ntitle: Demo\nsdk: gradio\napp_file: app.py\n---\n",
                encoding="utf-8",
            )

            status = space_build_status(root, "space\thttps://huggingface.co/spaces/u/s (push)")

            self.assertEqual(missing_required_files(root), [])
            self.assertTrue(readme_has_space_metadata(root))
            self.assertTrue(status.required_files_present)
            self.assertTrue(status.has_space_remote)

    def test_plan_script_runs_from_script_path(self) -> None:
        result = subprocess.run(  # noqa: S603
            [sys.executable, "scripts/plan_hf_space.py", "--user", "demo-user"],
            check=True,
            capture_output=True,
            text=True,
        )

        self.assertIn("Commands to run manually", result.stdout)
        self.assertIn("hf auth login", result.stdout)


if __name__ == "__main__":
    unittest.main()