from __future__ import annotations

import unittest

from seat_domain.intent import parse_local_intent


class LocalIntentParserTests(unittest.TestCase):
    @staticmethod
    def context(level: int = 0) -> dict:
        return {
            "last_target": "driver" if level else None,
            "devices": {
                "driver": {
                    "ventilation": {"state": "on" if level else "off", "level": level},
                    "heating": {"state": "off", "level": 0},
                },
                "passenger": {
                    "ventilation": {"state": "off", "level": 0},
                    "heating": {"state": "off", "level": 0},
                },
            },
        }

    def test_parses_driver_ventilation_on(self) -> None:
        result = parse_local_intent("打开主驾座椅通风")

        self.assertEqual(
            result["actions"],
            [
                {
                    "target": "driver",
                    "function": "ventilation",
                    "state": "on",
                    "requested_level": None,
                }
            ],
        )
        self.assertEqual(result["ambiguities"], [])

    def test_expands_front_seats(self) -> None:
        result = parse_local_intent("主副驾座椅通风都打开")

        self.assertEqual({item["target"] for item in result["actions"]}, {"driver", "passenger"})
        self.assertTrue(all(item["state"] == "on" for item in result["actions"]))

    def test_keeps_requested_level_for_safety_validation(self) -> None:
        result = parse_local_intent("主驾座椅加热开到三档")

        self.assertEqual(result["actions"][0]["requested_level"], 3)

    def test_requires_target(self) -> None:
        result = parse_local_intent("打开座椅通风")

        self.assertEqual(result["actions"], [])
        self.assertEqual(result["ambiguities"][0]["field"], "target")

    def test_maps_mild_hot_and_stuffy_feedback_to_level_one(self) -> None:
        result = parse_local_intent("主驾有点热，有点闷", self.context())

        self.assertEqual(
            result["actions"],
            [
                {
                    "target": "driver",
                    "function": "ventilation",
                    "state": "on",
                    "requested_level": 1,
                }
            ],
        )
        self.assertIn("热、闷", "".join(result["decision_summary"]))

    def test_increments_active_level_for_continuing_discomfort(self) -> None:
        result = parse_local_intent("还有点闷", self.context(level=1))

        self.assertEqual(result["actions"][0]["requested_level"], 2)
        self.assertEqual(result["actions"][0]["target"], "driver")
        self.assertIn("提升到2档", "".join(result["decision_summary"]))

    def test_cold_feedback_reduces_active_ventilation_before_heating(self) -> None:
        result = parse_local_intent("我有点冷", self.context(level=1))

        self.assertEqual(
            result["actions"],
            [
                {
                    "target": "driver",
                    "function": "ventilation",
                    "state": "off",
                    "requested_level": None,
                }
            ],
        )
        self.assertIn("不立即开启加热", "".join(result["decision_summary"]))

    def test_airflow_cause_targets_ventilation_not_heating(self) -> None:
        context = self.context()
        context["devices"]["passenger"]["ventilation"] = {"state": "on", "level": 2}
        result = parse_local_intent("副驾吹的我有点冷", context)

        self.assertEqual(
            result["actions"],
            [
                {
                    "target": "passenger",
                    "function": "ventilation",
                    "state": "on",
                    "requested_level": 1,
                }
            ],
        )
        self.assertNotIn("heating", {action["function"] for action in result["actions"]})
        self.assertIn("致冷来源", "".join(result["decision_summary"]))

    def test_cold_and_stuffy_are_independent_comfort_dimensions(self) -> None:
        result = parse_local_intent("主驾有点冷、有点闷", self.context())

        self.assertEqual(
            {
                (action["function"], action["state"], action["requested_level"])
                for action in result["actions"]
            },
            {("heating", "on", 1), ("ventilation", "on", 1)},
        )
        self.assertEqual(result["ambiguities"], [])
        self.assertIn("不将其判定为语义冲突", "".join(result["decision_summary"]))

    def test_vehicle_scope_does_not_inherit_a_seat_target(self) -> None:
        result = parse_local_intent("车上太闷了", self.context(level=1))

        self.assertEqual(result["actions"], [])
        self.assertEqual(result["ambiguities"][0]["field"], "domain")
        self.assertEqual(result["detected_domains"], ["climate"])
        self.assertEqual(result["scope"], "vehicle")
        self.assertIn("整车舱范围", "".join(result["decision_summary"]))


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