"""Config flow for Frontier Silicon Media Player integration.""" from __future__ import annotations import logging from typing import Any from urllib.parse import urlparse from afsapi import AFSAPI, ConnectionError as FSConnectionError, InvalidPinException import voluptuous as vol from homeassistant import config_entries from homeassistant.components import ssdp from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PORT from homeassistant.data_entry_flow import FlowResult from .const import ( CONF_PIN, CONF_WEBFSAPI_URL, DEFAULT_PIN, DEFAULT_PORT, DOMAIN, SSDP_ATTR_SPEAKER_NAME, ) _LOGGER = logging.getLogger(__name__) STEP_USER_DATA_SCHEMA = vol.Schema( { vol.Required(CONF_HOST): str, vol.Required(CONF_PORT, default=DEFAULT_PORT): int, } ) STEP_DEVICE_CONFIG_DATA_SCHEMA = vol.Schema( { vol.Required( CONF_PIN, default=DEFAULT_PIN, ): str, } ) def hostname_from_url(url: str) -> str: """Return the hostname from a url.""" return str(urlparse(url).hostname) class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): """Handle a config flow for Frontier Silicon Media Player.""" VERSION = 1 _name: str _webfsapi_url: str async def async_step_import(self, import_info: dict[str, Any]) -> FlowResult: """Handle the import of legacy configuration.yaml entries.""" device_url = f"http://{import_info[CONF_HOST]}:{import_info[CONF_PORT]}/device" try: webfsapi_url = await AFSAPI.get_webfsapi_endpoint(device_url) except FSConnectionError: return self.async_abort(reason="cannot_connect") except Exception as exception: # pylint: disable=broad-except _LOGGER.exception(exception) return self.async_abort(reason="unknown") try: afsapi = AFSAPI(webfsapi_url, import_info[CONF_PIN]) unique_id = await afsapi.get_radio_id() except FSConnectionError: return self.async_abort(reason="cannot_connect") except InvalidPinException: return self.async_abort(reason="invalid_auth") except Exception as exception: # pylint: disable=broad-except _LOGGER.exception(exception) return self.async_abort(reason="unknown") await self.async_set_unique_id(unique_id, raise_on_progress=False) self._abort_if_unique_id_configured() return self.async_create_entry( title=import_info[CONF_NAME] or "Radio", data={ CONF_WEBFSAPI_URL: webfsapi_url, CONF_PIN: import_info[CONF_PIN], }, ) async def async_step_user( self, user_input: dict[str, Any] | None = None ) -> FlowResult: """Handle the initial step of manual configuration.""" errors = {} if user_input: device_url = ( f"http://{user_input[CONF_HOST]}:{user_input[CONF_PORT]}/device" ) try: self._webfsapi_url = await AFSAPI.get_webfsapi_endpoint(device_url) except FSConnectionError: errors["base"] = "cannot_connect" except Exception as exception: # pylint: disable=broad-except _LOGGER.exception(exception) errors["base"] = "unknown" else: return await self._async_step_device_config_if_needed() data_schema = self.add_suggested_values_to_schema( STEP_USER_DATA_SCHEMA, user_input ) return self.async_show_form( step_id="user", data_schema=data_schema, errors=errors ) async def async_step_ssdp(self, discovery_info: ssdp.SsdpServiceInfo) -> FlowResult: """Process entity discovered via SSDP.""" device_url = discovery_info.ssdp_location if device_url is None: return self.async_abort(reason="cannot_connect") device_hostname = hostname_from_url(device_url) for entry in self._async_current_entries(include_ignore=False): if device_hostname == hostname_from_url(entry.data[CONF_WEBFSAPI_URL]): return self.async_abort(reason="already_configured") speaker_name = discovery_info.ssdp_headers.get(SSDP_ATTR_SPEAKER_NAME) self.context["title_placeholders"] = {"name": speaker_name} try: self._webfsapi_url = await AFSAPI.get_webfsapi_endpoint(device_url) except FSConnectionError: return self.async_abort(reason="cannot_connect") except Exception as exception: # pylint: disable=broad-except _LOGGER.debug(exception) return self.async_abort(reason="unknown") try: # try to login with default pin afsapi = AFSAPI(self._webfsapi_url, DEFAULT_PIN) unique_id = await afsapi.get_radio_id() except InvalidPinException: return self.async_abort(reason="invalid_auth") await self.async_set_unique_id(unique_id) self._abort_if_unique_id_configured( updates={CONF_WEBFSAPI_URL: self._webfsapi_url}, reload_on_update=True ) self._name = await afsapi.get_friendly_name() return await self.async_step_confirm() async def _async_step_device_config_if_needed(self) -> FlowResult: """Most users will not have changed the default PIN on their radio. We try to use this default PIN, and only if this fails ask for it via `async_step_device_config` """ try: # try to login with default pin afsapi = AFSAPI(self._webfsapi_url, DEFAULT_PIN) self._name = await afsapi.get_friendly_name() except InvalidPinException: # Ask for a PIN return await self.async_step_device_config() self.context["title_placeholders"] = {"name": self._name} unique_id = await afsapi.get_radio_id() await self.async_set_unique_id(unique_id) self._abort_if_unique_id_configured() return await self._async_create_entry() async def async_step_confirm( self, user_input: dict[str, Any] | None = None ) -> FlowResult: """Allow the user to confirm adding the device. Used when the default PIN could successfully be used.""" if user_input is not None: return await self._async_create_entry() self._set_confirm_only() return self.async_show_form(step_id="confirm") async def async_step_device_config( self, user_input: dict[str, Any] | None = None ) -> FlowResult: """Handle device configuration step. We ask for the PIN in this step. """ if user_input is None: return self.async_show_form( step_id="device_config", data_schema=STEP_DEVICE_CONFIG_DATA_SCHEMA ) errors = {} try: afsapi = AFSAPI(self._webfsapi_url, user_input[CONF_PIN]) self._name = await afsapi.get_friendly_name() except FSConnectionError: errors["base"] = "cannot_connect" except InvalidPinException: errors["base"] = "invalid_auth" except Exception as exception: # pylint: disable=broad-except _LOGGER.exception(exception) errors["base"] = "unknown" else: unique_id = await afsapi.get_radio_id() await self.async_set_unique_id(unique_id, raise_on_progress=False) self._abort_if_unique_id_configured() return await self._async_create_entry(user_input[CONF_PIN]) data_schema = self.add_suggested_values_to_schema( STEP_DEVICE_CONFIG_DATA_SCHEMA, user_input ) return self.async_show_form( step_id="device_config", data_schema=data_schema, errors=errors, ) async def _async_create_entry(self, pin: str | None = None): """Create the entry.""" return self.async_create_entry( title=self._name, data={CONF_WEBFSAPI_URL: self._webfsapi_url, CONF_PIN: pin or DEFAULT_PIN}, )