from __future__ import annotations

import os
import subprocess
import tempfile
import unittest
from pathlib import Path

REPO_ROOT = Path(__file__).resolve().parents[1]
SCRIPT = REPO_ROOT / "skills" / "builder" / "scripts" / "copy_artifact.sh"


def git(cwd: Path, *args: str) -> None:
    result = subprocess.run(
        ["git", *args],
        cwd=cwd,
        text=True,
        capture_output=True,
        check=False,
        env={
            **os.environ,
            "GIT_AUTHOR_NAME": "Test",
            "GIT_AUTHOR_EMAIL": "test@example.com",
            "GIT_COMMITTER_NAME": "Test",
            "GIT_COMMITTER_EMAIL": "test@example.com",
        },
    )
    if result.returncode != 0:
        raise AssertionError(f"git {args} failed: {result.stderr}")


class CopyArtifactTests(unittest.TestCase):
    def test_local_copy_and_alias(self) -> None:
        with tempfile.TemporaryDirectory() as temp:
            root = Path(temp)
            git(root, "init", "-b", "main")
            git(root, "config", "user.email", "test@example.com")
            git(root, "config", "user.name", "Test")
            src = root / "app.apk"
            src.write_bytes(b"apk")
            git(root, "add", "app.apk")
            git(root, "commit", "-m", "apk")
            dest = root / "out"
            result = subprocess.run(
                [str(SCRIPT), "-d", str(dest), "-a", "latest.apk", str(src)],
                cwd=root,
                capture_output=True,
                text=True,
                check=False,
            )
            self.assertEqual(result.returncode, 0, result.stdout + result.stderr)
            self.assertTrue((dest / "app.apk").is_file())
            self.assertTrue((dest / "latest.apk").is_file())
            self.assertEqual((dest / "latest.apk").read_bytes(), b"apk")
            self.assertIn("copy_artifact: 1 file(s)", result.stdout)

    def test_dirty_tree_requires_allow(self) -> None:
        with tempfile.TemporaryDirectory() as temp:
            root = Path(temp)
            git(root, "init", "-b", "main")
            git(root, "config", "user.email", "test@example.com")
            git(root, "config", "user.name", "Test")
            src = root / "app.apk"
            src.write_bytes(b"apk")
            git(root, "add", "app.apk")
            git(root, "commit", "-m", "apk")
            src.write_bytes(b"changed")
            dest = root / "out"
            blocked = subprocess.run(
                [str(SCRIPT), "-d", str(dest), str(src)],
                cwd=root,
                capture_output=True,
                text=True,
                check=False,
            )
            self.assertEqual(blocked.returncode, 3, blocked.stderr)
            allowed = subprocess.run(
                [str(SCRIPT), "-d", str(dest), str(src)],
                cwd=root,
                capture_output=True,
                text=True,
                check=False,
                env={**os.environ, "ALLOW_UNCOMMITTED": "1"},
            )
            self.assertEqual(allowed.returncode, 0, allowed.stdout + allowed.stderr)

    def test_dry_run_docker_dest_prints_docker_cp(self) -> None:
        with tempfile.TemporaryDirectory() as temp:
            root = Path(temp)
            git(root, "init", "-b", "main")
            git(root, "config", "user.email", "test@example.com")
            git(root, "config", "user.name", "Test")
            src = root / "app.apk"
            src.write_bytes(b"apk")
            git(root, "add", "app.apk")
            git(root, "commit", "-m", "apk")
            result = subprocess.run(
                [
                    str(SCRIPT),
                    "--dry-run",
                    "-d",
                    "file-server:/srv",
                    "-a",
                    "latest.apk",
                    str(src),
                ],
                cwd=root,
                capture_output=True,
                text=True,
                check=False,
            )
            self.assertEqual(result.returncode, 0, result.stdout + result.stderr)
            self.assertIn("docker cp", result.stdout)
            self.assertIn("file-server:/srv/app.apk", result.stdout)
            self.assertIn("file-server:/srv/latest.apk", result.stdout)


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