mirror of
https://github.com/esphome/esphome.git
synced 2025-11-14 22:05:54 +00:00
76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
from esphome.components import esp32_ble_tracker, esp32_ble_client
|
|
import esphome.config_validation as cv
|
|
import esphome.codegen as cg
|
|
from esphome.const import CONF_ACTIVE, CONF_ID
|
|
|
|
AUTO_LOAD = ["esp32_ble_client", "esp32_ble_tracker"]
|
|
DEPENDENCIES = ["api", "esp32"]
|
|
CODEOWNERS = ["@jesserockz"]
|
|
|
|
CONF_CONNECTIONS = "connections"
|
|
MAX_CONNECTIONS = 3
|
|
|
|
bluetooth_proxy_ns = cg.esphome_ns.namespace("bluetooth_proxy")
|
|
|
|
BluetoothProxy = bluetooth_proxy_ns.class_(
|
|
"BluetoothProxy", esp32_ble_tracker.ESPBTDeviceListener, cg.Component
|
|
)
|
|
BluetoothConnection = bluetooth_proxy_ns.class_(
|
|
"BluetoothConnection", esp32_ble_client.BLEClientBase
|
|
)
|
|
|
|
CONNECTION_SCHEMA = esp32_ble_tracker.ESP_BLE_DEVICE_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(BluetoothConnection),
|
|
}
|
|
).extend(cv.COMPONENT_SCHEMA)
|
|
|
|
|
|
def validate_connections(config):
|
|
if CONF_CONNECTIONS in config:
|
|
if not config[CONF_ACTIVE]:
|
|
raise cv.Invalid(
|
|
"Connections can only be used if the proxy is set to active"
|
|
)
|
|
else:
|
|
if config[CONF_ACTIVE]:
|
|
conf = config.copy()
|
|
conf[CONF_CONNECTIONS] = [
|
|
CONNECTION_SCHEMA({}) for _ in range(MAX_CONNECTIONS)
|
|
]
|
|
return conf
|
|
return config
|
|
|
|
|
|
CONFIG_SCHEMA = cv.All(
|
|
cv.Schema(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(BluetoothProxy),
|
|
cv.Optional(CONF_ACTIVE, default=False): cv.boolean,
|
|
cv.Optional(CONF_CONNECTIONS): cv.All(
|
|
cv.ensure_list(CONNECTION_SCHEMA),
|
|
cv.Length(min=1, max=MAX_CONNECTIONS),
|
|
),
|
|
}
|
|
)
|
|
.extend(esp32_ble_tracker.ESP_BLE_DEVICE_SCHEMA)
|
|
.extend(cv.COMPONENT_SCHEMA),
|
|
validate_connections,
|
|
)
|
|
|
|
|
|
async def to_code(config):
|
|
var = cg.new_Pvariable(config[CONF_ID])
|
|
await cg.register_component(var, config)
|
|
|
|
cg.add(var.set_active(config[CONF_ACTIVE]))
|
|
await esp32_ble_tracker.register_ble_device(var, config)
|
|
|
|
for connection_conf in config.get(CONF_CONNECTIONS, []):
|
|
connection_var = cg.new_Pvariable(connection_conf[CONF_ID])
|
|
await cg.register_component(connection_var, connection_conf)
|
|
cg.add(var.register_connection(connection_var))
|
|
await esp32_ble_tracker.register_client(connection_var, connection_conf)
|
|
|
|
cg.add_define("USE_BLUETOOTH_PROXY")
|