__init__.py

from __future__ import annotations
import pathlib
import yaml

from homeassistant.core import HomeAssistant
from homeassistant.config_entries import ConfigEntry

from .const import (
    DOMAIN,
    CONF_HOST,
    CONF_USERNAME,
    CONF_PASSWORD,
    CONF_SCAN_INTERVAL,
    DEFAULT_SCAN_INTERVAL,
    CONF_OUTDOOR_TEMP_OFFSET,
    DEFAULT_OUTDOOR_TEMP_OFFSET,
)
from .api import ST480Api
from .coordinator import ST480Coordinator

PLATFORMS = ["sensor"]


def _read_yaml_file(path: pathlib.Path):
    return yaml.safe_load(path.read_text(encoding="utf-8"))


async def _load_yaml(hass: HomeAssistant, filename: str):
    p = pathlib.Path(__file__).with_name(filename)
    return await hass.async_add_executor_job(_read_yaml_file, p)


async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
    host = entry.data[CONF_HOST]
    user = entry.data[CONF_USERNAME]
    pwd = entry.data[CONF_PASSWORD]
    interval = entry.data.get(CONF_SCAN_INTERVAL, DEFAULT_SCAN_INTERVAL)

    api = ST480Api(host, user, pwd)

    regs = await _load_yaml(hass, "registers.yaml")
    states_500 = await _load_yaml(hass, "states_500.yaml")

    # build request list from regs
    request_gts = []
    for r in regs:
        request_gts.append(r.get("source_gt") or r["gt"])
    # unique, keep order
    seen = set()
    request_gts = [x for x in request_gts if not (x in seen or seen.add(x))]

    coord = ST480Coordinator(hass, api, interval, request_gts)

    # store parsed config on coordinator
    coord.regs = regs
    coord.state_maps = {"states_500": states_500}

    # outdoor temp offset: options take priority over initial data
    coord.outdoor_temp_offset = entry.options.get(
        CONF_OUTDOOR_TEMP_OFFSET,
        entry.data.get(CONF_OUTDOOR_TEMP_OFFSET, DEFAULT_OUTDOOR_TEMP_OFFSET),
    )

    await coord.async_config_entry_first_refresh()
    hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coord

    await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)

    # reload coordinator when options change so offset takes effect immediately
    entry.async_on_unload(entry.add_update_listener(_async_update_listener))

    return True


async def _async_update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None:
    """Reload entry when options are updated."""
    await hass.config_entries.async_reload(entry.entry_id)


async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
    coord: ST480Coordinator = hass.data[DOMAIN].pop(entry.entry_id)
    await coord.async_shutdown()
    return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)