#pragma once

#include "esphome/core/component.h"
#include "esphome/components/button/button.h"
#include "TrumaiNetBoxApp.h"

namespace esphome {
namespace truma_inetbox {

class TriggerDiscoveryButton : public button::Button, public Component {
 public:
  void set_parent(TrumaiNetBoxApp *parent) { this->parent_ = parent; }

 protected:
  void press_action() override;

  TrumaiNetBoxApp *parent_;
};

}  // namespace truma_inetbox
}  // namespace esphome