#pragma once

#include "esphome/components/i2c/i2c.h"
#include "esphome/components/light/light_output.h"
#include "esphome/components/output/float_output.h"
#include "esphome/core/component.h"
#include "esphome/core/log.h"

namespace esphome {
namespace c_monochromatic {

enum State { Off, Auto, On };

class MonochromaticLightOutput : public light::LightOutput,
                                 public i2c::I2CDevice,
                                 public Component {
public:
  void set_output(output::FloatOutput *upper, output::FloatOutput *lower) {
    upper_ = upper;
    lower_ = lower;
    state_ = Off;
    bright_ = 0.0;
  }
  light::LightTraits get_traits() override {
    auto traits = light::LightTraits();
    traits.set_supported_color_modes({light::ColorMode::BRIGHTNESS});
    return traits;
  }
  void write_state(light::LightState *state) override {
    if (this->state_ == Off) {
      this->light_state_ = state;
      state->current_values_as_brightness(&(this->bright_));
      this->upper_->set_level(this->bright_);
      this->lower_->set_level(this->bright_);
    }
  }

  void set_state(State s) {
    this->state_ = s;
    switch (s) {
    case (On):
      /* this -> light_state_->turn_on(); */
      this->upper_->set_level(this->bright_);
      this->lower_->set_level(this->bright_);
      break;
    case (Off):
      /* this -> light_state_->turn_off(); */
      this->upper_->set_level(0.0);
      this->lower_->set_level(0.0);
      break;
    case (Auto):
      break;
    }
    /* this -> write_state(this -> light_state_); */
  }

  void loop() {
    uint8_t data = 10;
    this->read_register(0x0, &data, 0x1);
    /* ESP_LOGI(TAG, "READ value: %d", data); */
    if (data != 0) {
      if (data & 1) {
        this->set_state(Off);
        ESP_LOGI("TEST", "Set to Off", data);
        /* } else if ( data & 2 ){ */
        /*     this->set_state(Auto); */
        /* ESP_LOGI(TAG, "Set to Auto", data); */
      } else if (data & 4) {
        this->set_state(On);
        ESP_LOGI("TEST", "Set to On", data);
      }
    }
    /* if(this->state_ == Auto) { */
    /*     if(data & 8){ */
    /*         // TODO: turn on led */
    /*         /\* ESP_LOGI(TAG, "Auto On"); *\/ */
    /*         this->upper_->set_level(this->b_upper_); */
    /*         this->lower_->set_level(this->b_lower_); */
    /*     } */
    /*     else{ */
    /*         /\* ESP_LOGI(TAG, "Auto Off"); *\/ */
    /*         this->upper_->set_level(0.0); */
    /*         this->lower_->set_level(0.0); */
    /*     } */
    /* } */
  }

protected:
  State state_;
  float bright_;
  light::LightState *light_state_;
  output::FloatOutput *upper_;
  output::FloatOutput *lower_;
};

} // namespace c_monochromatic
} // namespace esphome