import subprocess
import sys
import unittest
from unittest import mock

from tests.python.support.paths import add_repo_path, repo_path

add_repo_path("skills/dxf/scripts")

from dxf import cli as dxf


class DxfCliTests(unittest.TestCase):
    def test_requires_explicit_target(self) -> None:
        with self.assertRaises(SystemExit) as cm:
            dxf.main([])
        self.assertEqual(2, cm.exception.code)

    def test_passes_targets_in_order(self) -> None:
        with mock.patch.object(dxf, "generate_dxf_targets", return_value=0) as generate:
            self.assertEqual(0, dxf.main(["drawings/second.py", "drawings/first.py"]))

        generate.assert_called_once_with(["drawings/second.py", "drawings/first.py"], output=None, verbose=False)

    def test_passes_verbose_flag(self) -> None:
        with mock.patch.object(dxf, "generate_dxf_targets", return_value=0) as generate:
            self.assertEqual(0, dxf.main(["drawings/part.py", "--verbose"]))

        generate.assert_called_once_with(["drawings/part.py"], output=None, verbose=True)

    def test_passes_output_flag(self) -> None:
        with mock.patch.object(dxf, "generate_dxf_targets", return_value=0) as generate:
            self.assertEqual(0, dxf.main(["drawings/part.py", "-o", "DXF/part.dxf"]))

        generate.assert_called_once_with(["drawings/part.py"], output="DXF/part.dxf", verbose=False)

    def test_output_flag_rejects_multiple_targets(self) -> None:
        with self.assertRaises(SystemExit) as cm:
            dxf.main(["drawings/first.py", "drawings/second.py", "-o", "DXF/first.dxf"])
        self.assertEqual(2, cm.exception.code)

    def test_scripts_dxf_directory_invokes_cli(self) -> None:
        skill_root = repo_path("skills/dxf")
        result = subprocess.run(
            [sys.executable, "scripts/dxf", "--help"],
            cwd=skill_root,
            check=False,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            text=True,
        )
        self.assertEqual("", result.stderr)
        self.assertEqual(0, result.returncode)
        self.assertIn("usage: dxf", result.stdout)
        self.assertIn("--output", result.stdout)

    def test_cli_import_does_not_import_heavy_cad_modules(self) -> None:
        skill_root = repo_path("skills/dxf")
        code = (
            "import sys; sys.path.insert(0, 'scripts'); import dxf.cli; "
            "print('OCP.OCP' in sys.modules); "
            "print('cadpy.step_scene' in sys.modules)"
        )
        result = subprocess.run(
            [sys.executable, "-c", code],
            cwd=skill_root,
            check=False,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            text=True,
        )
        self.assertEqual("", result.stderr)
        self.assertEqual(0, result.returncode)
        self.assertEqual(["False", "False"], result.stdout.strip().splitlines())


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