99 lines
3.2 KiB
Python
99 lines
3.2 KiB
Python
import esphome.codegen as cg
|
|
from esphome.components.esp32 import (
|
|
VARIANT_ESP32P4,
|
|
VARIANT_ESP32S2,
|
|
VARIANT_ESP32S3,
|
|
add_idf_component,
|
|
add_idf_sdkconfig_option,
|
|
idf_version,
|
|
only_on_variant,
|
|
)
|
|
import esphome.config_validation as cv
|
|
from esphome.const import CONF_DEVICES, CONF_ID
|
|
from esphome.core import CORE
|
|
from esphome.cpp_types import Component
|
|
from esphome.types import ConfigType
|
|
|
|
AUTO_LOAD = ["bytebuffer"]
|
|
CODEOWNERS = ["@clydebarrow"]
|
|
DEPENDENCIES = ["esp32"]
|
|
usb_host_ns = cg.esphome_ns.namespace("usb_host")
|
|
USBHost = usb_host_ns.class_("USBHost", Component)
|
|
USBClient = usb_host_ns.class_("USBClient", Component)
|
|
DOMAIN = "usb_host"
|
|
CONF_VID = "vid"
|
|
CONF_PID = "pid"
|
|
CONF_ENABLE_HUBS = "enable_hubs"
|
|
CONF_MAX_TRANSFER_REQUESTS = "max_transfer_requests"
|
|
CONF_MAX_PACKET_SIZE = "max_packet_size"
|
|
|
|
|
|
def usb_device_schema(cls=USBClient, vid: int = None, pid: int = None) -> cv.Schema:
|
|
schema = cv.COMPONENT_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(cls),
|
|
}
|
|
)
|
|
if vid:
|
|
schema = schema.extend({cv.Optional(CONF_VID, default=vid): cv.hex_uint16_t})
|
|
else:
|
|
schema = schema.extend({cv.Required(CONF_VID): cv.hex_uint16_t})
|
|
if pid:
|
|
schema = schema.extend({cv.Optional(CONF_PID, default=pid): cv.hex_uint16_t})
|
|
else:
|
|
schema = schema.extend({cv.Required(CONF_PID): cv.hex_uint16_t})
|
|
return schema
|
|
|
|
|
|
def _set_max_packet_size(config: dict) -> dict:
|
|
CORE.data.setdefault(DOMAIN, {})[CONF_MAX_PACKET_SIZE] = config[
|
|
CONF_MAX_PACKET_SIZE
|
|
]
|
|
return config
|
|
|
|
|
|
def get_max_packet_size() -> int:
|
|
return CORE.data.get(DOMAIN, {}).get(CONF_MAX_PACKET_SIZE, 64)
|
|
|
|
|
|
CONFIG_SCHEMA = cv.All(
|
|
cv.COMPONENT_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(USBHost),
|
|
cv.Optional(CONF_ENABLE_HUBS, default=False): cv.boolean,
|
|
cv.Optional(CONF_MAX_TRANSFER_REQUESTS, default=16): cv.int_range(
|
|
min=1, max=32
|
|
),
|
|
cv.Optional(CONF_MAX_PACKET_SIZE, default=64): cv.one_of(
|
|
64, 128, 256, 512, 1024, int=True
|
|
),
|
|
cv.Optional(CONF_DEVICES): cv.ensure_list(usb_device_schema()),
|
|
}
|
|
),
|
|
only_on_variant(supported=[VARIANT_ESP32P4, VARIANT_ESP32S2, VARIANT_ESP32S3]),
|
|
_set_max_packet_size,
|
|
)
|
|
|
|
|
|
async def register_usb_client(config):
|
|
var = cg.new_Pvariable(config[CONF_ID], config[CONF_VID], config[CONF_PID])
|
|
await cg.register_component(var, config)
|
|
return var
|
|
|
|
|
|
async def to_code(config: ConfigType) -> None:
|
|
# IDF 6.0 moved USB host to an external component
|
|
if idf_version() >= cv.Version(6, 0, 0):
|
|
add_idf_component(name="espressif/usb", ref="1.3.0")
|
|
add_idf_sdkconfig_option("CONFIG_USB_HOST_CONTROL_TRANSFER_MAX_SIZE", 1024)
|
|
if config.get(CONF_ENABLE_HUBS):
|
|
add_idf_sdkconfig_option("CONFIG_USB_HOST_HUBS_SUPPORTED", True)
|
|
|
|
cg.add_define("USB_HOST_MAX_REQUESTS", config[CONF_MAX_TRANSFER_REQUESTS])
|
|
cg.add_define("USB_HOST_MAX_PACKET_SIZE", config[CONF_MAX_PACKET_SIZE])
|
|
|
|
var = cg.new_Pvariable(config[CONF_ID])
|
|
await cg.register_component(var, config)
|
|
for device in config.get(CONF_DEVICES) or ():
|
|
await register_usb_client(device)
|