mirror of
https://github.com/esphome/esphome.git
synced 2024-09-30 04:28:42 +02:00
95 lines
3.1 KiB
Python
95 lines
3.1 KiB
Python
from __future__ import annotations
|
|
|
|
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome import helpers
|
|
from esphome.components.esp32 import add_idf_sdkconfig_option
|
|
from esphome.const import (
|
|
CONF_ENABLE_IPV6,
|
|
CONF_MIN_IPV6_ADDR_COUNT,
|
|
PLATFORM_ESP32,
|
|
PLATFORM_ESP8266,
|
|
PLATFORM_RP2040,
|
|
CONF_HOSTS,
|
|
CONF_HOSTSFILE,
|
|
CONF_IP_ADDRESS,
|
|
CONF_NAME,
|
|
)
|
|
from esphome.core import CORE
|
|
|
|
CODEOWNERS = ["@esphome/core"]
|
|
AUTO_LOAD = ["mdns"]
|
|
|
|
network_ns = cg.esphome_ns.namespace("network")
|
|
IPAddress = network_ns.class_("IPAddress")
|
|
Resolver = network_ns.class_("Resolver")
|
|
CONF_NETWORK_ID = "network_id"
|
|
|
|
CONFIG_SCHEMA = cv.Schema(
|
|
{
|
|
cv.SplitDefault(CONF_ENABLE_IPV6): cv.All(
|
|
cv.boolean, cv.only_on([PLATFORM_ESP32, PLATFORM_ESP8266, PLATFORM_RP2040])
|
|
),
|
|
cv.Optional(CONF_MIN_IPV6_ADDR_COUNT, default=0): cv.positive_int,
|
|
cv.GenerateID(CONF_NETWORK_ID): cv.declare_id(Resolver),
|
|
cv.Optional(CONF_HOSTSFILE): cv.file_,
|
|
cv.Optional(CONF_HOSTS): cv.ensure_list(
|
|
cv.Schema(
|
|
{
|
|
cv.Required(CONF_NAME): cv.string,
|
|
cv.Required(CONF_IP_ADDRESS): cv.string,
|
|
}
|
|
)
|
|
),
|
|
}
|
|
)
|
|
|
|
|
|
def parse_hosts_file(hosts_contents: str) -> list[tuple[str, IPAddress]]:
|
|
"""Parse a hosts file"""
|
|
hosts: list[tuple[str, IPAddress]] = []
|
|
for line in hosts_contents.splitlines():
|
|
if line.startswith("#"):
|
|
continue
|
|
split_line = line.split()
|
|
if len(split_line) < 2:
|
|
continue
|
|
ip_address: IPAddress = IPAddress(split_line[0])
|
|
hosts.extend([(host, ip_address) for host in split_line[1:]])
|
|
return hosts
|
|
|
|
|
|
async def to_code(config):
|
|
if CONF_ENABLE_IPV6 in config:
|
|
cg.add_define("USE_NETWORK_IPV6", config[CONF_ENABLE_IPV6])
|
|
cg.add_define(
|
|
"USE_NETWORK_MIN_IPV6_ADDR_COUNT", config[CONF_MIN_IPV6_ADDR_COUNT]
|
|
)
|
|
if CORE.using_esp_idf:
|
|
add_idf_sdkconfig_option("CONFIG_LWIP_IPV6", config[CONF_ENABLE_IPV6])
|
|
add_idf_sdkconfig_option(
|
|
"CONFIG_LWIP_IPV6_AUTOCONFIG", config[CONF_ENABLE_IPV6]
|
|
)
|
|
else:
|
|
if config[CONF_ENABLE_IPV6]:
|
|
cg.add_build_flag("-DCONFIG_LWIP_IPV6")
|
|
cg.add_build_flag("-DCONFIG_LWIP_IPV6_AUTOCONFIG")
|
|
if CORE.is_rp2040:
|
|
cg.add_build_flag("-DPIO_FRAMEWORK_ARDUINO_ENABLE_IPV6")
|
|
if CORE.is_esp8266:
|
|
cg.add_build_flag("-DPIO_FRAMEWORK_ARDUINO_LWIP2_IPV6_LOW_MEMORY")
|
|
hosts = []
|
|
if CONF_HOSTS in config:
|
|
hosts = [
|
|
(host[CONF_NAME], IPAddress(host[CONF_IP_ADDRESS]))
|
|
for host in config[CONF_HOSTS]
|
|
]
|
|
if CONF_HOSTSFILE in config:
|
|
hosts_contents = helpers.read_file(
|
|
CORE.relative_config_path(config[CONF_HOSTSFILE])
|
|
)
|
|
hosts.extend(parse_hosts_file(hosts_contents))
|
|
|
|
map_ = cg.std_ns.class_("multimap").template(cg.std_string, IPAddress)
|
|
cg.new_Pvariable(config[CONF_NETWORK_ID], map_(hosts))
|