"""Sensors: температура помещения, код ошибки, состояние связи."""
from __future__ import annotations

from typing import Optional

from homeassistant.components.sensor import (
    SensorDeviceClass,
    SensorEntity,
    SensorStateClass,
)
from homeassistant.const import EntityCategory, UnitOfTemperature

from pyfglair import Prop, State
from pyfglair.templates import template_info

from .coordinator import FglairCoordinator, FglairRuntime
from .entity import FglairEntity


class FglairRoomTemperature(FglairEntity, SensorEntity):
    _attr_translation_key = "room_temperature"
    _attr_device_class = SensorDeviceClass.TEMPERATURE
    _attr_state_class = SensorStateClass.MEASUREMENT
    _attr_native_unit_of_measurement = UnitOfTemperature.CELSIUS
    _attr_suggested_display_precision = 1

    def __init__(self, coordinator: FglairCoordinator) -> None:
        super().__init__(coordinator, "display_temperature", "room_temperature")

    @property
    def native_value(self) -> Optional[float]:
        display = self.display_value(Prop.DISPLAY_TEMPERATURE)
        return display / 10 if display is not None else None


class FglairErrorCode(FglairEntity, SensorEntity):
    _attr_translation_key = "error_code"
    _attr_entity_category = EntityCategory.DIAGNOSTIC

    def __init__(self, coordinator: FglairCoordinator) -> None:
        super().__init__(coordinator, "error_code", "error_code")

    @property
    def native_value(self) -> Optional[int]:
        return self.raw_value(Prop.ERROR_CODE)


class FglairConnectionState(FglairEntity, SensorEntity):
    _attr_translation_key = "connection_state"
    _attr_device_class = SensorDeviceClass.ENUM
    _attr_entity_category = EntityCategory.DIAGNOSTIC
    _attr_options = [state.name.lower() for state in State]

    def __init__(self, coordinator: FglairCoordinator) -> None:
        super().__init__(coordinator, "connection_state", "connection_state")

    @property
    def available(self) -> bool:
        # Диагностика связи должна быть видна и при offline.
        return True

    @property
    def native_value(self) -> str:
        return self.coordinator.data.state.name.lower()


async def async_setup_entry(hass, entry, async_add_entities) -> None:
    runtime: FglairRuntime = entry.runtime_data
    coordinator = runtime.coordinator
    template = runtime.client.template
    props = {info.prop for info in template_info(template)}

    entities = [FglairConnectionState(coordinator)]
    if Prop.DISPLAY_TEMPERATURE in props:
        entities.append(FglairRoomTemperature(coordinator))
    if Prop.ERROR_CODE in props:
        entities.append(FglairErrorCode(coordinator))
    async_add_entities(entities)