#pragma once

#include "common.h"

namespace esphome {
namespace yeelight {
namespace bs2 {

class ColorNightLight : public GPIOOutputs {
public:
    bool set_light_color_values(light::LightColorValues v) {
        // This class can handle the GPIO outputs for the night light mode.
        //
        // At the lowest brightness setting, switch to night light mode.  In
        // the Yeelight integration in Home Assistant, this feature is
        // exposed trough a separate switch. I have found that the switch is
        // both confusing and made me run into issues when automating the
        // lights.
        //
        // I don't simply check for a brightness at or below 0.01 (1%),
        // because the lowest brightness setting from Home Assistant turns
        // up as 0.011765 in here (which is 3/255).
        if (v.get_brightness() >= 0.012f) {
            return false;
        }

        values = v;

        // This night light mode is activated when white light is selected.
        // Based on measurements using the original device firmware, so it
        // matches the night light of the original firmware.
        if (v.get_white() > 0) {
            red = 0.968f;
            green = 0.968f;
            blue = 0.972f;
            white = 0.0f;
        }
        // In RGB mode, the selected color is used to give the night light a
        // specific color, instead of the default. This is a nice extra for
        // this firmware, as the original firmware does not support it.
        else {
            red = 0.9997f - v.get_red() * (0.9997f - 0.9680f);
            green = 0.9997f - v.get_green() * (0.9997f - 0.9680f);
            auto blue_on = 0.9720f + (0.9640f - 0.9720f) *  (1.0f - (v.get_red() + v.get_green())/2.0f);
            blue = 0.9997f - v.get_blue() * (0.9997f - blue_on);
            white = 0.0f;
        }

        return true;
    }
};

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