mirror of
https://github.com/esphome/esphome.git
synced 2024-12-02 11:44:13 +01:00
113 lines
2.8 KiB
Python
113 lines
2.8 KiB
Python
from esphome.const import (
|
|
CONF_ID,
|
|
CONF_PORT,
|
|
CONF_PROTOCOL,
|
|
CONF_SERVICES,
|
|
CONF_SERVICE,
|
|
KEY_CORE,
|
|
KEY_FRAMEWORK_VERSION,
|
|
CONF_DISABLED,
|
|
)
|
|
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.core import CORE, coroutine_with_priority
|
|
from esphome.components.esp32 import add_idf_component
|
|
|
|
CODEOWNERS = ["@esphome/core"]
|
|
DEPENDENCIES = ["network"]
|
|
|
|
mdns_ns = cg.esphome_ns.namespace("mdns")
|
|
MDNSComponent = mdns_ns.class_("MDNSComponent", cg.Component)
|
|
MDNSTXTRecord = mdns_ns.struct("MDNSTXTRecord")
|
|
MDNSService = mdns_ns.struct("MDNSService")
|
|
|
|
|
|
def _remove_id_if_disabled(value):
|
|
value = value.copy()
|
|
if value[CONF_DISABLED]:
|
|
value.pop(CONF_ID)
|
|
return value
|
|
|
|
|
|
CONF_TXT = "txt"
|
|
|
|
SERVICE_SCHEMA = cv.Schema(
|
|
{
|
|
cv.Required(CONF_SERVICE): cv.string,
|
|
cv.Required(CONF_PROTOCOL): cv.string,
|
|
cv.Optional(CONF_PORT, default=0): cv.Any(0, cv.port),
|
|
cv.Optional(CONF_TXT, default={}): {cv.string: cv.string},
|
|
}
|
|
)
|
|
|
|
CONFIG_SCHEMA = cv.All(
|
|
cv.Schema(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(MDNSComponent),
|
|
cv.Optional(CONF_DISABLED, default=False): cv.boolean,
|
|
cv.Optional(CONF_SERVICES, default=[]): cv.ensure_list(SERVICE_SCHEMA),
|
|
}
|
|
),
|
|
_remove_id_if_disabled,
|
|
)
|
|
|
|
|
|
def mdns_txt_record(key: str, value: str):
|
|
return cg.StructInitializer(
|
|
MDNSTXTRecord,
|
|
("key", key),
|
|
("value", value),
|
|
)
|
|
|
|
|
|
def mdns_service(
|
|
service: str, proto: str, port: int, txt_records: list[dict[str, str]]
|
|
):
|
|
return cg.StructInitializer(
|
|
MDNSService,
|
|
("service_type", service),
|
|
("proto", proto),
|
|
("port", port),
|
|
("txt_records", txt_records),
|
|
)
|
|
|
|
|
|
@coroutine_with_priority(55.0)
|
|
async def to_code(config):
|
|
if CORE.using_arduino:
|
|
if CORE.is_esp32:
|
|
cg.add_library("ESPmDNS", None)
|
|
elif CORE.is_esp8266:
|
|
cg.add_library("ESP8266mDNS", None)
|
|
elif CORE.is_rp2040:
|
|
cg.add_library("LEAmDNS", None)
|
|
|
|
if CORE.using_esp_idf and CORE.data[KEY_CORE][KEY_FRAMEWORK_VERSION] >= cv.Version(
|
|
5, 0, 0
|
|
):
|
|
add_idf_component(
|
|
name="mdns",
|
|
repo="https://github.com/espressif/esp-protocols.git",
|
|
ref="mdns-v1.0.9",
|
|
path="components/mdns",
|
|
)
|
|
|
|
if config[CONF_DISABLED]:
|
|
return
|
|
|
|
cg.add_define("USE_MDNS")
|
|
|
|
var = cg.new_Pvariable(config[CONF_ID])
|
|
await cg.register_component(var, config)
|
|
|
|
for service in config[CONF_SERVICES]:
|
|
txt = [
|
|
mdns_txt_record(txt_key, txt_value)
|
|
for txt_key, txt_value in service[CONF_TXT].items()
|
|
]
|
|
|
|
exp = mdns_service(
|
|
service[CONF_SERVICE], service[CONF_PROTOCOL], service[CONF_PORT], txt
|
|
)
|
|
|
|
cg.add(var.add_extra_service(exp))
|