import esphome.codegen as cg
import esphome.config_validation as cv
from esphome.components import cover, remote_transmitter
from esphome.const import CONF_ID

DEPENDENCIES = ["mqtt", "remote_transmitter"]
AUTO_LOAD = ["mqtt"]

somfy_ns = cg.esphome_ns.namespace("somfy")
SomfyRTSCover = somfy_ns.class_("SomfyRTSCover", cover.Cover, cg.Component)

CONF_TRANSMITTER = "transmitter"
CONF_REMOTE_ID_BASE = "remote_id_base"
CONF_REMOTE_ID_OFFSET = "remote_id_offset"

CONFIG_SCHEMA = cover.COVER_SCHEMA.extend(
    {
        cv.GenerateID(): cv.declare_id(SomfyRTSCover),
        cv.Required(CONF_TRANSMITTER): cv.use_id(remote_transmitter.RemoteTransmitterComponent),
        cv.Required(CONF_REMOTE_ID_BASE): cv.uint32_t,
        cv.Required(CONF_REMOTE_ID_OFFSET): cv.uint32_t,
    }
).extend(cv.COMPONENT_SCHEMA)

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

    transmitter = await cg.get_variable(config[CONF_TRANSMITTER])
    cg.add(var.set_transmitter(transmitter))
    cg.add(var.set_remote_id(
        config[CONF_REMOTE_ID_BASE] + config[CONF_REMOTE_ID_OFFSET]
    ))