#pragma once

#include <cmath>
#include "esphome/core/component.h"
#include "esphome/core/automation.h"
#include "light_output.h"

namespace esphome {
namespace yeelight {
namespace bs2 {
    
class BrightnessTrigger : public Trigger<float> {
public:
    explicit BrightnessTrigger(YeelightBS2LightOutput *parent) {
        parent->add_on_state_callback(
            [this](light::LightColorValues values, std::string light_mode) {
                auto new_brightness = values.get_brightness();
                if (values.get_state() == 0) {
                    new_brightness = 0.0f;
                }
                new_brightness = roundf(new_brightness * 100.0f) / 100.0f;
                if (last_brightness_ != new_brightness) {
                    trigger(new_brightness);
                    last_brightness_ = new_brightness;
                }
            }
        );
    }
protected:
    float last_brightness_ = -1.0f;
};

class LightModeTrigger : public Trigger<std::string> {
public:
    explicit LightModeTrigger(YeelightBS2LightOutput *parent) {
        parent->add_on_state_callback(
            [this](light::LightColorValues values, std::string light_mode) {
                if (last_light_mode_ != light_mode) {
                    trigger(light_mode);
                    last_light_mode_ = light_mode;
                }
            }
        );
    }
protected:
    std::string last_light_mode_ = LIGHT_MODE_UNKNOWN;
};

} // namespace yeelight_bs2
} // namespace yeelight
} // namespace bs2