import esphome.codegen as cg
import esphome.config_validation as cv
from esphome.components import select
from esphome.const import CONF_POWER, ENTITY_CATEGORY_CONFIG, CONF_ID

from .. import tw7100, tw7100_ns, CONF_TW7100_ID

DEPENDENCIES = ["tw7100"]
CODEOWNERS = ["@sqozz"]

tw7100Select = tw7100_ns.class_("tw7100Select", select.Select, cg.Component)

CONFIG_SCHEMA = cv.Schema(
    {
        cv.GenerateID(): cv.use_id(tw7100),
        cv.Optional("SOURCE"): select.select_schema(
            tw7100Select
        ),
        cv.Optional("LUMINANCE"): select.select_schema(
            tw7100Select
        ),
    }
)


async def to_code(config):
    parent = await cg.get_variable(config[CONF_ID])

    if "SOURCE" in config:
        select_ = await select.new_select(config["SOURCE"], options=[])
        cg.add(select_.set_cmd("SOURCE"))
        cg.add(select_.set_tw7100_parent(parent))
        cg.add(parent.set_set_source(select_))

    if "LUMINANCE" in config:
        select_ = await select.new_select(config["LUMINANCE"], options=[])
        cg.add(select_.set_cmd("LUMINANCE"))
        cg.add(select_.set_tw7100_parent(parent))
        cg.add(parent.set_set_luminance(select_))