import esphome.codegen as cg
import esphome.config_validation as cv
from esphome.components import uart
from esphome.const import CONF_ID


CODEOWNERS = ["@sqozz"]
DEPENDENCIES = ["uart"]
AUTO_LOAD = ["binary_sensor", "sensor", "text_sensor"]

tw7100_ns = cg.esphome_ns.namespace("tw7100")
tw7100 = tw7100_ns.class_("tw7100Component", cg.PollingComponent, uart.UARTDevice)

CONF_TW7100_ID = "tw7100_id"
CONFIG_SCHEMA = cv.Schema({
    cv.GenerateID(): cv.declare_id(tw7100),
}).extend(cv.polling_component_schema("60s")).extend(uart.UART_DEVICE_SCHEMA)

FINAL_VALIDATE_SCHEMA = uart.final_validate_device_schema(
    "tw7100", baud_rate=9600, require_rx=True, require_tx=True
)


async def to_code(config):
  var = cg.new_Pvariable(config[CONF_ID])
  await cg.register_component(var, config)
  await uart.register_uart_device(var, config)