#pragma once 

#include "../common.h"
#include "../front_panel_hal.h"
#include "esphome/components/binary_sensor/binary_sensor.h"

namespace esphome {
namespace yeelight {
namespace bs2 {

/**
 * This class implements a binary sensor for the buttons on the
 * Yeelight Bedside Lamp 2.
 */
class YeelightBS2Button : public binary_sensor::BinarySensor, public Component {
public:
    void set_front_panel_hal(FrontPanelHAL *front_panel) {
        front_panel_ = front_panel;
    }

    void set_part(int part) {
        part_ = part;
    }

    void setup() {
        ESP_LOGCONFIG(TAG, "Setting up binary_sensor ...");
        ESP_LOGCONFIG(TAG, "  Part id: %d", part_);

        front_panel_->add_on_event_callback(
            [this](EVENT ev) {
                // Filter events by part, when requested.
                ESP_LOGI(TAG, "FILTER event=%d, part=%d", ev, part_);
                if (part_ > 0) {
                    if ((ev & FLAG_PART_MASK) != (part_ << FLAG_PART_SHIFT)) {
                        return;
                    }
                }
                // Publish the new state, based on the touch/release status..
                auto on_or_off = (ev & FLAG_TYPE_MASK) == FLAG_TYPE_TOUCH;
                this->publish_state(on_or_off); 
            }
        );
    }

protected:
    FrontPanelHAL *front_panel_;
    EVENT part_;
};
    
} // namespace bs2
} // namespace yeelight
} // namespace esphome