hacs-connectd/custom_components/connectd/config_flow.py
2025-12-15 11:08:23 -06:00

71 lines
2.4 KiB
Python

"""config flow for connectd integration."""
from __future__ import annotations
import logging
import aiohttp
import voluptuous as vol
from homeassistant import config_entries
from homeassistant.const import CONF_HOST, CONF_PORT
from homeassistant.data_entry_flow import FlowResult
_LOGGER = logging.getLogger(__name__)
DOMAIN = "connectd"
DEFAULT_PORT = 8099
class ConnectdConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
"""handle a config flow for connectd."""
VERSION = 1
async def async_step_user(
self, user_input: dict | None = None
) -> FlowResult:
"""handle the initial step."""
errors = {}
if user_input is not None:
host = user_input[CONF_HOST]
port = user_input.get(CONF_PORT, DEFAULT_PORT)
# test connection
try:
timeout = aiohttp.ClientTimeout(total=10)
async with aiohttp.ClientSession(timeout=timeout) as session:
url = f"http://{host}:{port}/api/health"
async with session.get(url) as resp:
if resp.status == 200:
# connection works
await self.async_set_unique_id(f"{host}:{port}")
self._abort_if_unique_id_configured()
return self.async_create_entry(
title=f"connectd ({host})",
data={
"host": host,
"port": port,
},
)
else:
_LOGGER.error("connectd api returned status %s", resp.status)
errors["base"] = "cannot_connect"
except aiohttp.ClientError as err:
_LOGGER.error("connectd connection error: %s", err)
errors["base"] = "cannot_connect"
except Exception as err:
_LOGGER.exception("connectd unexpected error: %s", err)
errors["base"] = "unknown"
return self.async_show_form(
step_id="user",
data_schema=vol.Schema(
{
vol.Required(CONF_HOST, default="192.168.1.8"): str,
vol.Optional(CONF_PORT, default=DEFAULT_PORT): int,
}
),
errors=errors,
)