"""Setup mocks for the Plugwise integration tests."""

from __future__ import annotations

from collections.abc import Generator
import json
from typing import Any
from unittest.mock import AsyncMock, MagicMock, patch

from packaging.version import Version
import pytest

from homeassistant.components.plugwise.const import DOMAIN
from homeassistant.const import (
    CONF_HOST,
    CONF_MAC,
    CONF_PASSWORD,
    CONF_PORT,
    CONF_USERNAME,
)
from homeassistant.core import HomeAssistant

from tests.common import MockConfigEntry, load_fixture


def _read_json(environment: str, call: str) -> dict[str, Any]:
    """Undecode the json data."""
    fixture = load_fixture(f"plugwise/{environment}/{call}.json")
    return json.loads(fixture)


@pytest.fixture
def cooling_present(request: pytest.FixtureRequest) -> str:
    """Pass the cooling_present boolean.

    Used with fixtures that require parametrization of the cooling capability.
    """
    return request.param


@pytest.fixture
def chosen_env(request: pytest.FixtureRequest) -> str:
    """Pass the chosen_env string.

    Used with fixtures that require parametrization of the user-data fixture.
    """
    return request.param


@pytest.fixture
def gateway_id(request: pytest.FixtureRequest) -> str:
    """Pass the gateway_id string.

    Used with fixtures that require parametrization of the gateway_id.
    """
    return request.param


@pytest.fixture
def heater_id(request: pytest.FixtureRequest) -> str:
    """Pass the heater_idstring.

    Used with fixtures that require parametrization of the heater_id.
    """
    return request.param


@pytest.fixture
def reboot(request: pytest.FixtureRequest) -> str:
    """Pass the reboot boolean.

    Used with fixtures that require parametrization of the reboot capability.
    """
    return request.param


@pytest.fixture
def mock_config_entry() -> MockConfigEntry:
    """Return the default mocked config entry."""
    return MockConfigEntry(
        title="My Plugwise",
        domain=DOMAIN,
        data={
            CONF_HOST: "127.0.0.1",
            CONF_MAC: "AA:BB:CC:DD:EE:FF",
            CONF_PASSWORD: "test-password",
            CONF_PORT: 80,
            CONF_USERNAME: "smile",
        },
        unique_id="smile98765",
    )


@pytest.fixture
def mock_setup_entry() -> Generator[AsyncMock]:
    """Mock setting up a config entry."""
    with patch(
        "homeassistant.components.plugwise.async_setup_entry", return_value=True
    ) as mock_setup:
        yield mock_setup


@pytest.fixture
def mock_smile_config_flow() -> Generator[MagicMock]:
    """Return a mocked Smile client."""
    with patch(
        "homeassistant.components.plugwise.config_flow.Smile",
        autospec=True,
    ) as smile_mock:
        smile = smile_mock.return_value

        smile.connect.return_value = Version("4.3.2")
        smile.smile_hostname = "smile12345"
        smile.smile_model = "Test Model"
        smile.smile_model_id = "Test Model ID"
        smile.smile_name = "Test Smile Name"
        smile.smile_version = "4.3.2"

        yield smile


@pytest.fixture
def mock_smile_adam() -> Generator[MagicMock]:
    """Create a Mock Adam environment for testing exceptions."""
    chosen_env = "m_adam_multiple_devices_per_zone"
    data = _read_json(chosen_env, "data")
    with (
        patch(
            "homeassistant.components.plugwise.coordinator.Smile", autospec=True
        ) as smile_mock,
        patch(
            "homeassistant.components.plugwise.config_flow.Smile",
            new=smile_mock,
        ),
    ):
        smile = smile_mock.return_value

        smile.async_update.return_value = data
        smile.cooling_present = False
        smile.connect.return_value = Version("3.0.15")
        smile.gateway_id = "fe799307f1624099878210aa0b9f1475"
        smile.heater_id = "90986d591dcd426cae3ec3e8111ff730"
        smile.reboot = True
        smile.smile_hostname = "smile98765"
        smile.smile_model = "Gateway"
        smile.smile_model_id = "smile_open_therm"
        smile.smile_name = "Adam"
        smile.smile_type = "thermostat"
        smile.smile_version = "3.0.15"

        yield smile


@pytest.fixture
def mock_smile_adam_heat_cool(
    chosen_env: str, cooling_present: bool
) -> Generator[MagicMock]:
    """Create a special base Mock Adam type for testing with different datasets."""
    data = _read_json(chosen_env, "data")
    with patch(
        "homeassistant.components.plugwise.coordinator.Smile", autospec=True
    ) as smile_mock:
        smile = smile_mock.return_value

        smile.async_update.return_value = data
        smile.connect.return_value = Version("3.6.4")
        smile.cooling_present = cooling_present
        smile.gateway_id = "da224107914542988a88561b4452b0f6"
        smile.heater_id = "056ee145a816487eaa69243c3280f8bf"
        smile.reboot = True
        smile.smile_hostname = "smile98765"
        smile.smile_model = "Gateway"
        smile.smile_model_id = "smile_open_therm"
        smile.smile_name = "Adam"
        smile.smile_type = "thermostat"
        smile.smile_version = "3.6.4"

        yield smile


@pytest.fixture
def mock_smile_adam_jip() -> Generator[MagicMock]:
    """Create a Mock adam-jip type for testing exceptions."""
    chosen_env = "m_adam_jip"
    data = _read_json(chosen_env, "data")
    with patch(
        "homeassistant.components.plugwise.coordinator.Smile", autospec=True
    ) as smile_mock:
        smile = smile_mock.return_value

        smile.async_update.return_value = data
        smile.connect.return_value = Version("3.2.8")
        smile.cooling_present = False
        smile.gateway_id = "b5c2386c6f6342669e50fe49dd05b188"
        smile.heater_id = "e4684553153b44afbef2200885f379dc"
        smile.reboot = True
        smile.smile_hostname = "smile98765"
        smile.smile_model = "Gateway"
        smile.smile_model_id = "smile_open_therm"
        smile.smile_name = "Adam"
        smile.smile_type = "thermostat"
        smile.smile_version = "3.2.8"

        yield smile


@pytest.fixture
def mock_smile_anna(chosen_env: str, cooling_present: bool) -> Generator[MagicMock]:
    """Create a Mock Anna type for testing."""
    data = _read_json(chosen_env, "data")
    with patch(
        "homeassistant.components.plugwise.coordinator.Smile", autospec=True
    ) as smile_mock:
        smile = smile_mock.return_value

        smile.async_update.return_value = data
        smile.connect.return_value = Version("4.0.15")
        smile.cooling_present = cooling_present
        smile.gateway_id = "015ae9ea3f964e668e490fa39da3870b"
        smile.heater_id = "1cbf783bb11e4a7c8a6843dee3a86927"
        smile.reboot = True
        smile.smile_hostname = "smile98765"
        smile.smile_model = "Gateway"
        smile.smile_model_id = "smile_thermo"
        smile.smile_name = "Smile Anna"
        smile.smile_type = "thermostat"
        smile.smile_version = "4.0.15"

        yield smile


@pytest.fixture
def mock_smile_p1(chosen_env: str, gateway_id: str) -> Generator[MagicMock]:
    """Create a base Mock P1 type for testing with different datasets and gateway-ids."""
    data = _read_json(chosen_env, "data")
    with patch(
        "homeassistant.components.plugwise.coordinator.Smile", autospec=True
    ) as smile_mock:
        smile = smile_mock.return_value

        smile.async_update.return_value = data
        smile.connect.return_value = Version("4.4.2")
        smile.gateway_id = gateway_id
        smile.heater_id = None
        smile.reboot = True
        smile.smile_hostname = "smile98765"
        smile.smile_model = "Gateway"
        smile.smile_model_id = "smile"
        smile.smile_name = "Smile P1"
        smile.smile_type = "power"
        smile.smile_version = "4.4.2"

        yield smile


@pytest.fixture
def mock_smile_legacy_anna() -> Generator[MagicMock]:
    """Create a Mock legacy Anna environment for testing exceptions."""
    chosen_env = "legacy_anna"
    data = _read_json(chosen_env, "data")
    with patch(
        "homeassistant.components.plugwise.coordinator.Smile", autospec=True
    ) as smile_mock:
        smile = smile_mock.return_value

        smile.async_update.return_value = data
        smile.connect.return_value = Version("1.8.22")
        smile.gateway_id = "0000aaaa0000aaaa0000aaaa0000aa00"
        smile.heater_id = "04e4cbfe7f4340f090f85ec3b9e6a950"
        smile.reboot = False
        smile.smile_hostname = "smile98765"
        smile.smile_model = "Gateway"
        smile.smile_model_id = None
        smile.smile_name = "Smile Anna"
        smile.smile_type = "thermostat"
        smile.smile_version = "1.8.22"

        yield smile


@pytest.fixture
def mock_stretch() -> Generator[MagicMock]:
    """Create a Mock Stretch environment for testing exceptions."""
    chosen_env = "stretch_v31"
    data = _read_json(chosen_env, "data")
    with patch(
        "homeassistant.components.plugwise.coordinator.Smile", autospec=True
    ) as smile_mock:
        smile = smile_mock.return_value

        smile.async_update.return_value = data
        smile.connect.return_value = Version("3.1.11")
        smile.gateway_id = "259882df3c05415b99c2d962534ce820"
        smile.heater_id = None
        smile.reboot = False
        smile.smile_hostname = "stretch98765"
        smile.smile_model = "Gateway"
        smile.smile_model_id = None
        smile.smile_name = "Stretch"
        smile.smile_type = "stretch"
        smile.smile_version = "3.1.11"

        yield smile


@pytest.fixture
async def init_integration(
    hass: HomeAssistant, mock_config_entry: MockConfigEntry
) -> MockConfigEntry:
    """Set up the Plugwise integration for testing."""
    mock_config_entry.add_to_hass(hass)

    await hass.config_entries.async_setup(mock_config_entry.entry_id)
    await hass.async_block_till_done()

    return mock_config_entry