mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-04-23 04:37:15 +00:00
c2deabb672
* Support dynamic ingress port * Allow to remeber ports * Add tests * Fix schema * Cleanup handling / speed * Fix port
157 lines
4.2 KiB
Python
157 lines
4.2 KiB
Python
"""Validate functions."""
|
|
import re
|
|
import uuid
|
|
|
|
import voluptuous as vol
|
|
|
|
from .const import (
|
|
ATTR_ACCESS_TOKEN,
|
|
ATTR_ADDONS_CUSTOM_LIST,
|
|
ATTR_BOOT,
|
|
ATTR_CHANNEL,
|
|
ATTR_HASSIO,
|
|
ATTR_HASSOS,
|
|
ATTR_HASSOS_CLI,
|
|
ATTR_HOMEASSISTANT,
|
|
ATTR_IMAGE,
|
|
ATTR_LAST_BOOT,
|
|
ATTR_LAST_VERSION,
|
|
ATTR_PASSWORD,
|
|
ATTR_PORT,
|
|
ATTR_PORTS,
|
|
ATTR_REFRESH_TOKEN,
|
|
ATTR_SESSION,
|
|
ATTR_SSL,
|
|
ATTR_TIMEZONE,
|
|
ATTR_UUID,
|
|
ATTR_WAIT_BOOT,
|
|
ATTR_WATCHDOG,
|
|
CHANNEL_BETA,
|
|
CHANNEL_DEV,
|
|
CHANNEL_STABLE,
|
|
)
|
|
from .utils.validate import validate_timezone
|
|
|
|
RE_REPOSITORY = re.compile(r"^(?P<url>[^#]+)(?:#(?P<branch>[\w\-]+))?$")
|
|
|
|
NETWORK_PORT = vol.All(vol.Coerce(int), vol.Range(min=1, max=65535))
|
|
WAIT_BOOT = vol.All(vol.Coerce(int), vol.Range(min=1, max=60))
|
|
DOCKER_IMAGE = vol.Match(r"^[\w{}]+/[\-\w{}]+$")
|
|
ALSA_DEVICE = vol.Maybe(vol.Match(r"\d+,\d+"))
|
|
CHANNELS = vol.In([CHANNEL_STABLE, CHANNEL_BETA, CHANNEL_DEV])
|
|
UUID_MATCH = vol.Match(r"^[0-9a-f]{32}$")
|
|
SHA256 = vol.Match(r"^[0-9a-f]{64}$")
|
|
TOKEN = vol.Match(r"^[0-9a-f]{32,256}$")
|
|
|
|
|
|
def validate_repository(repository):
|
|
"""Validate a valid repository."""
|
|
data = RE_REPOSITORY.match(repository)
|
|
if not data:
|
|
raise vol.Invalid("No valid repository format!")
|
|
|
|
# Validate URL
|
|
# pylint: disable=no-value-for-parameter
|
|
vol.Url()(data.group("url"))
|
|
|
|
return repository
|
|
|
|
|
|
# pylint: disable=no-value-for-parameter
|
|
REPOSITORIES = vol.All([validate_repository], vol.Unique())
|
|
|
|
|
|
# pylint: disable=inconsistent-return-statements
|
|
def convert_to_docker_ports(data):
|
|
"""Convert data into Docker port list."""
|
|
# dynamic ports
|
|
if data is None:
|
|
return None
|
|
|
|
# single port
|
|
if isinstance(data, int):
|
|
return NETWORK_PORT(data)
|
|
|
|
# port list
|
|
if isinstance(data, list) and len(data) > 2:
|
|
return vol.Schema([NETWORK_PORT])(data)
|
|
|
|
# ip port mapping
|
|
if isinstance(data, list) and len(data) == 2:
|
|
return (vol.Coerce(str)(data[0]), NETWORK_PORT(data[1]))
|
|
|
|
raise vol.Invalid("Can't validate Docker host settings")
|
|
|
|
|
|
DOCKER_PORTS = vol.Schema(
|
|
{
|
|
vol.All(
|
|
vol.Coerce(str), vol.Match(r"^\d+(?:/tcp|/udp)?$")
|
|
): convert_to_docker_ports
|
|
}
|
|
)
|
|
|
|
|
|
# pylint: disable=no-value-for-parameter
|
|
SCHEMA_HASS_CONFIG = vol.Schema(
|
|
{
|
|
vol.Optional(ATTR_UUID, default=lambda: uuid.uuid4().hex): UUID_MATCH,
|
|
vol.Optional(ATTR_ACCESS_TOKEN): TOKEN,
|
|
vol.Optional(ATTR_BOOT, default=True): vol.Boolean(),
|
|
vol.Inclusive(ATTR_IMAGE, "custom_hass"): DOCKER_IMAGE,
|
|
vol.Inclusive(ATTR_LAST_VERSION, "custom_hass"): vol.Coerce(str),
|
|
vol.Optional(ATTR_PORT, default=8123): NETWORK_PORT,
|
|
vol.Optional(ATTR_PASSWORD): vol.Maybe(vol.Coerce(str)),
|
|
vol.Optional(ATTR_REFRESH_TOKEN): vol.Maybe(vol.Coerce(str)),
|
|
vol.Optional(ATTR_SSL, default=False): vol.Boolean(),
|
|
vol.Optional(ATTR_WATCHDOG, default=True): vol.Boolean(),
|
|
vol.Optional(ATTR_WAIT_BOOT, default=600): vol.All(
|
|
vol.Coerce(int), vol.Range(min=60)
|
|
),
|
|
},
|
|
extra=vol.REMOVE_EXTRA,
|
|
)
|
|
|
|
|
|
SCHEMA_UPDATER_CONFIG = vol.Schema(
|
|
{
|
|
vol.Optional(ATTR_CHANNEL, default=CHANNEL_STABLE): CHANNELS,
|
|
vol.Optional(ATTR_HOMEASSISTANT): vol.Coerce(str),
|
|
vol.Optional(ATTR_HASSIO): vol.Coerce(str),
|
|
vol.Optional(ATTR_HASSOS): vol.Coerce(str),
|
|
vol.Optional(ATTR_HASSOS_CLI): vol.Coerce(str),
|
|
},
|
|
extra=vol.REMOVE_EXTRA,
|
|
)
|
|
|
|
|
|
# pylint: disable=no-value-for-parameter
|
|
SCHEMA_HASSIO_CONFIG = vol.Schema(
|
|
{
|
|
vol.Optional(ATTR_TIMEZONE, default="UTC"): validate_timezone,
|
|
vol.Optional(ATTR_LAST_BOOT): vol.Coerce(str),
|
|
vol.Optional(
|
|
ATTR_ADDONS_CUSTOM_LIST,
|
|
default=["https://github.com/hassio-addons/repository"],
|
|
): REPOSITORIES,
|
|
vol.Optional(ATTR_WAIT_BOOT, default=5): WAIT_BOOT,
|
|
},
|
|
extra=vol.REMOVE_EXTRA,
|
|
)
|
|
|
|
|
|
SCHEMA_AUTH_CONFIG = vol.Schema({SHA256: SHA256})
|
|
|
|
|
|
SCHEMA_INGRESS_CONFIG = vol.Schema(
|
|
{
|
|
vol.Required(ATTR_SESSION, default=dict): vol.Schema(
|
|
{TOKEN: vol.Coerce(float)}
|
|
),
|
|
vol.Required(ATTR_PORTS, default=dict): vol.Schema(
|
|
{vol.Coerce(str): NETWORK_PORT}
|
|
),
|
|
},
|
|
extra=vol.REMOVE_EXTRA,
|
|
)
|