"""Isolated, public, read-only office air quality server (ESP32 ESP-IDF)."""

import esphome.codegen as cg
import esphome.config_validation as cv
from esphome.components import sensor, socket
from esphome.const import CONF_ID, CONF_PORT, Framework

DEPENDENCIES = ["esp32", "network", "mdns"]
AUTO_LOAD = ["web_server_base", "socket"]

ns = cg.esphome_ns.namespace("air_quality_web")
AirQualityWeb = ns.class_("AirQualityWeb", cg.PollingComponent)


def validate_port(value):
    value = cv.port(value)
    if value in (0, 80, 32768):
        raise cv.Invalid("Choose a nonzero port other than 80 or 32768 (captive portal/control socket)")
    return value


CONFIG_SCHEMA = cv.All(
    cv.Schema(
        {
            cv.GenerateID(): cv.declare_id(AirQualityWeb),
            cv.Optional(CONF_PORT, default=8080): validate_port,
            cv.Optional("mdns_hostname"): cv.hostname,
            cv.Required("co2"): cv.use_id(sensor.Sensor),
            cv.Required("temperature"): cv.use_id(sensor.Sensor),
            cv.Required("humidity"): cv.use_id(sensor.Sensor),
        }
    ).extend(cv.COMPONENT_SCHEMA),
    cv.only_on_esp32,
    cv.only_with_framework(Framework.ESP_IDF),
    socket.consume_sockets(1, "air_quality_web", socket.SocketType.TCP_LISTEN),
    socket.consume_sockets(3, "air_quality_web", socket.SocketType.TCP),
    socket.consume_sockets(2, "air_quality_web", socket.SocketType.UDP),
)


async def to_code(config):
    var = cg.new_Pvariable(config[CONF_ID])
    await cg.register_component(var, config)
    cg.add(var.set_port(config[CONF_PORT]))
    if "mdns_hostname" in config:
        cg.add(var.set_mdns_hostname(config["mdns_hostname"]))
    for key in ("co2", "temperature", "humidity"):
        ref = await cg.get_variable(config[key])
        cg.add(getattr(var, f"set_{key}_sensor")(ref))