import esphome.codegen as cg
import esphome.config_validation as cv
from esphome.components import sensor
from esphome.const import (
    CONF_ID, CONF_CHANNEL, STATE_CLASS_MEASUREMENT, DEVICE_CLASS_VOLTAGE, UNIT_VOLT, ICON_FLASH
)
from . import ADS131M08Hub, ads131m08_ns

AUTO_LOAD = [
    "sensor",
]

CONF_ADS131M08_ID = "ads131m08_id"

ADS131M08Sensor = ads131m08_ns.class_("ADS131M08Sensor", sensor.Sensor, cg.Component)

CONFIG_SCHEMA = sensor.sensor_schema(
    unit_of_measurement=UNIT_VOLT,
    device_class=DEVICE_CLASS_VOLTAGE,
    icon=ICON_FLASH,
    state_class=STATE_CLASS_MEASUREMENT,
).extend({
    cv.GenerateID(): cv.declare_id(ADS131M08Sensor),
    cv.Required(CONF_ADS131M08_ID): cv.use_id(ADS131M08Hub),
    cv.Required(CONF_CHANNEL): cv.int_range(min=0, max=7),
}).extend(cv.COMPONENT_SCHEMA)

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

    hub = await cg.get_variable(config[CONF_ADS131M08_ID])
    cg.add(hub.register_sensor(config[CONF_CHANNEL], var))