104 lines
3.4 KiB
Python
104 lines
3.4 KiB
Python
"""Config flow to configure ZWaveMe integration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
from url_normalize import url_normalize
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.config_entries import ConfigFlow, ConfigFlowResult
|
|
from homeassistant.const import CONF_TOKEN, CONF_URL
|
|
from homeassistant.helpers.service_info.zeroconf import ZeroconfServiceInfo
|
|
|
|
from . import helpers
|
|
from .const import DOMAIN
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class ZWaveMeConfigFlow(ConfigFlow, domain=DOMAIN):
|
|
"""ZWaveMe integration config flow."""
|
|
|
|
def __init__(self) -> None:
|
|
"""Initialize flow."""
|
|
self.url: str | None = None
|
|
self.token: str | None = None
|
|
self.uuid: str | None = None
|
|
|
|
async def async_step_user(
|
|
self, user_input: dict[str, str] | None = None
|
|
) -> ConfigFlowResult:
|
|
"""Handle a flow initialized by the user or started with zeroconf."""
|
|
errors = {}
|
|
placeholders = {
|
|
"local_token": "/112f7a4a-0051-cc2b-3b61-1898181b9950",
|
|
"find_token": (
|
|
"0481effe8a5c6f757b455babb678dc0e764feae279/112f7a4a-0051"
|
|
"-cc2b-3b61-1898181b9950"
|
|
),
|
|
"local_url": "ws://192.168.1.39:8083",
|
|
"add_on_url": "ws://127.0.0.1:8083",
|
|
"find_url": "wss://find.z-wave.me",
|
|
"remote_url": "wss://87.250.250.242:8083",
|
|
}
|
|
if self.url is None:
|
|
schema = vol.Schema(
|
|
{
|
|
vol.Required(CONF_URL): str,
|
|
vol.Required(CONF_TOKEN): str,
|
|
}
|
|
)
|
|
else:
|
|
schema = vol.Schema(
|
|
{
|
|
vol.Required(CONF_TOKEN): str,
|
|
}
|
|
)
|
|
|
|
if user_input is not None:
|
|
if self.url is None:
|
|
self.url = user_input[CONF_URL]
|
|
|
|
self.token = user_input[CONF_TOKEN]
|
|
if not self.url.startswith(("ws://", "wss://")):
|
|
self.url = f"ws://{self.url}"
|
|
self.url = url_normalize(self.url, default_scheme="ws")
|
|
assert self.url
|
|
if self.uuid is None:
|
|
self.uuid = await helpers.get_uuid(self.url, self.token)
|
|
if self.uuid is not None:
|
|
await self.async_set_unique_id(self.uuid, raise_on_progress=False)
|
|
self._abort_if_unique_id_configured()
|
|
else:
|
|
errors["base"] = "no_valid_uuid_set"
|
|
|
|
if not errors:
|
|
return self.async_create_entry(
|
|
title=self.url,
|
|
data={CONF_URL: self.url, CONF_TOKEN: self.token},
|
|
)
|
|
|
|
return self.async_show_form(
|
|
step_id="user",
|
|
description_placeholders=placeholders,
|
|
data_schema=schema,
|
|
errors=errors,
|
|
)
|
|
|
|
async def async_step_zeroconf(
|
|
self, discovery_info: ZeroconfServiceInfo
|
|
) -> ConfigFlowResult:
|
|
"""Handle a discovered Z-Wave accessory - get url to pass into user step.
|
|
|
|
This flow is triggered by the discovery component.
|
|
"""
|
|
self.url = discovery_info.host
|
|
self.uuid = await helpers.get_uuid(self.url)
|
|
if self.uuid is None:
|
|
return self.async_abort(reason="no_valid_uuid_set")
|
|
|
|
await self.async_set_unique_id(self.uuid)
|
|
self._abort_if_unique_id_configured()
|
|
return await self.async_step_user()
|