core/homeassistant/components/envertech_evt800/config_flow.py

"""Config flow for the ENVERTECH EVT800 integration."""

from typing import Any, override

from pyenvertechevt800 import EnvertechEVT800
import voluptuous as vol

from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
from homeassistant.const import CONF_IP_ADDRESS, CONF_PORT, CONF_TYPE
from homeassistant.helpers import config_validation as cv

from .const import DEFAULT_PORT, DOMAIN, TYPE_TCP_SERVER_MODE

SCHEMA_DEVICE = vol.Schema(
    {
        vol.Required(CONF_IP_ADDRESS): cv.string,
        vol.Required(CONF_PORT, default=DEFAULT_PORT): cv.port,
    }
)


class EnvertechFlowHandler(ConfigFlow, domain=DOMAIN):
    """Config flow for Envertech EVT800."""

    VERSION = 1
    MINOR_VERSION = 1

    @override
    async def async_step_user(
        self, user_input: dict[str, Any] | None = None
    ) -> ConfigFlowResult:
        """First step in config flow."""
        errors: dict[str, str] = {}
        if user_input is not None:
            ip_address = user_input[CONF_IP_ADDRESS]
            port = user_input[CONF_PORT]

            self._async_abort_entries_match(
                {
                    CONF_IP_ADDRESS: ip_address,
                    CONF_PORT: port,
                }
            )
            evt800 = EnvertechEVT800(ip_address, port)

            can_connect = await evt800.test_connection()

            if not can_connect:
                errors["base"] = "cannot_connect"

            if not errors:
                return self.async_create_entry(
                    title="Envertech EVT800",
                    data={CONF_TYPE: TYPE_TCP_SERVER_MODE, **user_input},
                )

        return self.async_show_form(
            step_id="user",
            data_schema=SCHEMA_DEVICE,
            errors=errors,
        )