"""Runtime BLE device selector for a JK-BMS ESPHome bridge."""
import esphome.codegen as cg
import esphome.config_validation as cv
from esphome.components import ble_client, esp32_ble_tracker
from esphome.const import CONF_ID

DEPENDENCIES = ["esp32", "esp32_ble_tracker", "ble_client", "web_server"]

ns = cg.esphome_ns.namespace("jk_bms_selector")
JkBmsSelector = ns.class_(
    "JkBmsSelector",
    cg.Component,
    esp32_ble_tracker.ESPBTDeviceListener,
)

CONFIG_SCHEMA = cv.Schema(
    {
        cv.GenerateID(): cv.declare_id(JkBmsSelector),
        cv.Required("ble_client_id"): cv.use_id(ble_client.BLEClient),
    }
).extend(cv.COMPONENT_SCHEMA).extend(esp32_ble_tracker.ESP_BLE_DEVICE_SCHEMA)


async def to_code(config):
    var = cg.new_Pvariable(config[CONF_ID])
    await cg.register_component(var, config)
    client = await cg.get_variable(config["ble_client_id"])
    cg.add(var.set_ble_client(client))
    await esp32_ble_tracker.register_ble_device(var, config)