"""SMLIGHT SLZB device integration.""" from pysmlight import Api2 from homeassistant.const import CONF_HOST, Platform from homeassistant.core import HomeAssistant from homeassistant.helpers import config_validation as cv from homeassistant.helpers.aiohttp_client import async_get_clientsession from homeassistant.helpers.typing import ConfigType from .bluetooth import async_setup_ble_scanner from .const import DOMAIN from .coordinator import ( SmConfigEntry, SmDataUpdateCoordinator, SmFirmwareUpdateCoordinator, SmlightData, ) from .services import async_setup_services CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN) PLATFORMS: list[Platform] = [ Platform.BINARY_SENSOR, Platform.BUTTON, Platform.INFRARED, Platform.LIGHT, Platform.SENSOR, Platform.SWITCH, Platform.UPDATE, ] async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool: """Set up the SMLIGHT services.""" async_setup_services(hass) return True async def async_setup_entry(hass: HomeAssistant, entry: SmConfigEntry) -> bool: """Set up SMLIGHT from a config entry.""" client = Api2(host=entry.data[CONF_HOST], session=async_get_clientsession(hass)) data_coordinator = SmDataUpdateCoordinator(hass, entry, client) firmware_coordinator = SmFirmwareUpdateCoordinator(hass, entry, client) await data_coordinator.async_config_entry_first_refresh() await firmware_coordinator.async_config_entry_first_refresh() info = data_coordinator.data.info if info.legacy_api < 2: entry.async_create_background_task( hass, client.sse.client(), "smlight-sse-client" ) if info.ble is not None and ( unload_callback := await async_setup_ble_scanner(hass, entry, client, info) ): entry.async_on_unload(unload_callback) entry.runtime_data = SmlightData( data=data_coordinator, firmware=firmware_coordinator, ) await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS) return True async def async_unload_entry(hass: HomeAssistant, entry: SmConfigEntry) -> bool: """Unload SMLIGHT config entry.""" return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)