select.h

Go to the documentation of this file.

#pragma once
 
#include "esphome/core/component.h"
#include "esphome/core/entity_base.h"
#include "esphome/core/helpers.h"
#include "esphome/core/string_ref.h"
#include "select_call.h"
#include "select_traits.h"
 
namespace esphome::select {
 
#define LOG_SELECT(prefix, type, obj) \
  if (log_entity_header(TAG, prefix, LOG_STR_LITERAL(type), (obj))) { \
    LOG_ENTITY_ICON(TAG, prefix, *(obj)); \
  }
 
#define SUB_SELECT(name) \
 protected: \
  select::Select *name##_select_{nullptr}; \
\
 public: \
  void set_##name##_select(select::Select *select) { this->name##_select_ = select; }
 
class Select : public EntityBase {
 public:
  SelectTraits traits;
 
  Select() = default;
  ~Select() = default;
 
  void publish_state(const std::string &state) { this->publish_state(state.c_str()); }
  void publish_state(const char *state);
  void publish_state(size_t index);
 
  StringRef current_option() const {
    return this->has_state() ? StringRef(this->option_at(this->active_index_)) : StringRef();
  }
 
  SelectCall make_call() { return SelectCall(this); }
 
  bool has_option(const std::string &option) const { return this->index_of(option).has_value(); }
  bool has_option(const char *option) const { return this->index_of(option).has_value(); }
 
  bool has_index(size_t index) const { return index < this->size(); }
 
  size_t size() const { return this->traits.get_options().size(); }
 
  optional<size_t> index_of(const char *option, size_t len) const;
  optional<size_t> index_of(const std::string &option) const { return this->index_of(option.data(), option.size()); }
  optional<size_t> index_of(const char *option) const { return this->index_of(option, strlen(option)); }
 
  optional<size_t> active_index() const;
 
  optional<std::string> at(size_t index) const;
 
  const char *option_at(size_t index) const;
 
  template<typename F> void add_on_state_callback(F &&callback) {
    this->state_callback_.add(std::forward<F>(callback));
  }
 
 protected:
  friend class SelectCall;
 
  size_t active_index_{0};
 
  virtual void control(size_t index) { this->control(this->option_at(index)); }
 
  virtual void control(const std::string &value) {
    auto index = this->index_of(value);
    if (index.has_value()) {
      this->control(index.value());
    }
  }
 
  LazyCallbackManager<void(size_t)> state_callback_;
};
 
template<auto Fn> class ApplySelect final : public Select {
 public:
  // Not "= default": placement new would zero-fill .bss that is already zero.
  ApplySelect() {}
 
 protected:
  void control(size_t index) override { Fn(this, index); }
};
 
}  // namespace esphome::select