import esphome.codegen as cg
import esphome.config_validation as cv
from esphome.components import sensor
from esphome.const import CONF_TYPE, DEVICE_CLASS_EMPTY, UNIT_PERCENT, ICON_LIGHTBULB
from . import APDS9960, CONF_APDS9960_ID

DEPENDENCIES = ["apds9960"]

TYPES = {
    "CLEAR": "set_clear_channel",
    "RED": "set_red_channel",
    "GREEN": "set_green_channel",
    "BLUE": "set_blue_channel",
    "PROXIMITY": "set_proximity",
}

CONFIG_SCHEMA = sensor.sensor_schema(
    UNIT_PERCENT, ICON_LIGHTBULB, 1, DEVICE_CLASS_EMPTY
).extend(
    {
        cv.Required(CONF_TYPE): cv.one_of(*TYPES, upper=True),
        cv.GenerateID(CONF_APDS9960_ID): cv.use_id(APDS9960),
    }
)


def to_code(config):
    hub = yield cg.get_variable(config[CONF_APDS9960_ID])
    var = yield sensor.new_sensor(config)
    func = getattr(hub, TYPES[config[CONF_TYPE]])
    cg.add(func(var))