import copy
import json
import tempfile
import unittest
from pathlib import Path
from unittest import mock

import generate_relay_protocol as generator


class RelayProtocolGeneratorTest(unittest.TestCase):
    def setUp(self) -> None:
        self.spec = json.loads(generator.SPEC_PATH.read_text(encoding="utf-8"))

    def test_supported_pattern_renders_both_targets(self) -> None:
        dart_output = generator.dart_source(copy.deepcopy(self.spec))
        go_output = generator.go_source(copy.deepcopy(self.spec))

        self.assertIn(
            f"RegExp(r{generator.SUPPORTED_ID_PATTERN!r})",
            dart_output,
        )
        self.assertIn("func validRelayID(value string, maxLength int) bool", go_output)

    def test_changed_pattern_fails_before_writing_either_target(self) -> None:
        changed_spec = copy.deepcopy(self.spec)
        changed_spec["idPattern"] = r"^[A-Za-z0-9_.-]+$"

        for renderer in (generator.dart_source, generator.go_source):
            with self.subTest(renderer=renderer.__name__):
                with self.assertRaisesRegex(ValueError, "idPattern"):
                    renderer(changed_spec)

        with tempfile.TemporaryDirectory() as temporary_directory:
            root = Path(temporary_directory)
            spec_path = root / "relay_protocol.json"
            dart_path = root / "relay_protocol.g.dart"
            go_path = root / "relay_protocol_gen.go"
            spec_path.write_text(json.dumps(changed_spec), encoding="utf-8")
            dart_path.write_text("dart sentinel\n", encoding="utf-8")
            go_path.write_text("go sentinel\n", encoding="utf-8")

            with (
                mock.patch.object(generator, "SPEC_PATH", spec_path),
                mock.patch.object(generator, "DART_PATH", dart_path),
                mock.patch.object(generator, "GO_PATH", go_path),
            ):
                with self.assertRaisesRegex(ValueError, "idPattern"):
                    generator.main()

            self.assertEqual(dart_path.read_text(encoding="utf-8"), "dart sentinel\n")
            self.assertEqual(go_path.read_text(encoding="utf-8"), "go sentinel\n")

    def test_missing_pattern_is_rejected(self) -> None:
        spec = copy.deepcopy(self.spec)
        del spec["idPattern"]

        with self.assertRaisesRegex(ValueError, "idPattern is required"):
            generator.validated_id_pattern(spec)

    def test_non_string_pattern_is_rejected(self) -> None:
        for value in (None, 42, [generator.SUPPORTED_ID_PATTERN]):
            with self.subTest(value=value):
                spec = copy.deepcopy(self.spec)
                spec["idPattern"] = value

                with self.assertRaisesRegex(ValueError, "idPattern must be a string"):
                    generator.validated_id_pattern(spec)


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