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

from .. import tw7100, tw7100_ns, CONF_TW7100_ID

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

tw7100Number = tw7100_ns.class_("tw7100Number", number.Number, cg.Component)

CONFIG_SCHEMA = cv.Schema(
    {
        cv.GenerateID(): cv.use_id(tw7100),
        cv.Optional("VOL"): number.number_schema(
            tw7100Number
        ),
    }
)


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

    if "VOL" in config:
        number_ = await number.new_number(config["VOL"], min_value=0, max_value=255, step=1)
        cg.add(number_.set_cmd("VOL"))
        cg.add(number_.set_tw7100_parent(parent))
        cg.add(parent.set_set_volume(number_))