web_server.h

Go to the documentation of this file.

#pragma once

#ifdef USE_ARDUINO

#include "list_entities.h"

#include "esphome/components/web_server_base/web_server_base.h"
#include "esphome/core/component.h"
#include "esphome/core/controller.h"

#include <vector>

namespace esphome {
namespace web_server {

struct UrlMatch {
  std::string domain;  
  std::string id;      
  std::string method;  
  bool valid;          
};

enum JsonDetail { DETAIL_ALL, DETAIL_STATE };

class WebServer : public Controller, public Component, public AsyncWebHandler {
 public:
  WebServer(web_server_base::WebServerBase *base) : base_(base), entities_iterator_(ListEntitiesIterator(this)) {}

  void set_css_url(const char *css_url);

  void set_css_include(const char *css_include);

  void set_js_url(const char *js_url);

  void set_js_include(const char *js_include);

  void set_include_internal(bool include_internal) { include_internal_ = include_internal; }
  void set_allow_ota(bool allow_ota) { this->allow_ota_ = allow_ota; }

  // ========== INTERNAL METHODS ==========
  // (In most use cases you won't need these)
  void setup() override;
  void loop() override;

  void dump_config() override;

  float get_setup_priority() const override;

  void handle_index_request(AsyncWebServerRequest *request);

#ifdef USE_WEBSERVER_CSS_INCLUDE
  void handle_css_request(AsyncWebServerRequest *request);
#endif

#ifdef USE_WEBSERVER_JS_INCLUDE
  void handle_js_request(AsyncWebServerRequest *request);
#endif

#ifdef USE_SENSOR
  void on_sensor_update(sensor::Sensor *obj, float state) override;
  void handle_sensor_request(AsyncWebServerRequest *request, const UrlMatch &match);

  std::string sensor_json(sensor::Sensor *obj, float value, JsonDetail start_config);
#endif

#ifdef USE_SWITCH
  void on_switch_update(switch_::Switch *obj, bool state) override;

  void handle_switch_request(AsyncWebServerRequest *request, const UrlMatch &match);

  std::string switch_json(switch_::Switch *obj, bool value, JsonDetail start_config);
#endif

#ifdef USE_BUTTON
  void handle_button_request(AsyncWebServerRequest *request, const UrlMatch &match);

  std::string button_json(button::Button *obj, JsonDetail start_config);
#endif

#ifdef USE_BINARY_SENSOR
  void on_binary_sensor_update(binary_sensor::BinarySensor *obj, bool state) override;

  void handle_binary_sensor_request(AsyncWebServerRequest *request, const UrlMatch &match);

  std::string binary_sensor_json(binary_sensor::BinarySensor *obj, bool value, JsonDetail start_config);
#endif

#ifdef USE_FAN
  void on_fan_update(fan::Fan *obj) override;

  void handle_fan_request(AsyncWebServerRequest *request, const UrlMatch &match);

  std::string fan_json(fan::Fan *obj, JsonDetail start_config);
#endif

#ifdef USE_LIGHT
  void on_light_update(light::LightState *obj) override;

  void handle_light_request(AsyncWebServerRequest *request, const UrlMatch &match);

  std::string light_json(light::LightState *obj, JsonDetail start_config);
#endif

#ifdef USE_TEXT_SENSOR
  void on_text_sensor_update(text_sensor::TextSensor *obj, const std::string &state) override;

  void handle_text_sensor_request(AsyncWebServerRequest *request, const UrlMatch &match);

  std::string text_sensor_json(text_sensor::TextSensor *obj, const std::string &value, JsonDetail start_config);
#endif

#ifdef USE_COVER
  void on_cover_update(cover::Cover *obj) override;

  void handle_cover_request(AsyncWebServerRequest *request, const UrlMatch &match);

  std::string cover_json(cover::Cover *obj, JsonDetail start_config);
#endif

#ifdef USE_NUMBER
  void on_number_update(number::Number *obj, float state) override;
  void handle_number_request(AsyncWebServerRequest *request, const UrlMatch &match);

  std::string number_json(number::Number *obj, float value, JsonDetail start_config);
#endif

#ifdef USE_SELECT
  void on_select_update(select::Select *obj, const std::string &state, size_t index) override;
  void handle_select_request(AsyncWebServerRequest *request, const UrlMatch &match);

  std::string select_json(select::Select *obj, const std::string &value, JsonDetail start_config);
#endif

#ifdef USE_CLIMATE
  void on_climate_update(climate::Climate *obj) override;
  void handle_climate_request(AsyncWebServerRequest *request, const UrlMatch &match);

  std::string climate_json(climate::Climate *obj, JsonDetail start_config);
#endif

#ifdef USE_LOCK
  void on_lock_update(lock::Lock *obj) override;

  void handle_lock_request(AsyncWebServerRequest *request, const UrlMatch &match);

  std::string lock_json(lock::Lock *obj, lock::LockState value, JsonDetail start_config);
#endif

  bool canHandle(AsyncWebServerRequest *request) override;
  void handleRequest(AsyncWebServerRequest *request) override;
  bool isRequestHandlerTrivial() override;

 protected:
  friend ListEntitiesIterator;
  web_server_base::WebServerBase *base_;
  AsyncEventSource events_{"/events"};
  ListEntitiesIterator entities_iterator_;
  const char *css_url_{nullptr};
  const char *css_include_{nullptr};
  const char *js_url_{nullptr};
  const char *js_include_{nullptr};
  bool include_internal_{false};
  bool allow_ota_{true};
};

}  // namespace web_server
}  // namespace esphome

#endif  // USE_ARDUINO