mirror of
https://github.com/esphome/esphome.git
synced 2024-12-03 12:14:13 +01:00
82 lines
2.1 KiB
C++
82 lines
2.1 KiB
C++
#pragma once
|
|
|
|
#include <DNSServer.h>
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/core/helpers.h"
|
|
#include "esphome/core/preferences.h"
|
|
#include "esphome/components/web_server_base/web_server_base.h"
|
|
|
|
namespace esphome {
|
|
|
|
namespace captive_portal {
|
|
|
|
struct CaptivePortalSettings {
|
|
char ssid[33];
|
|
char password[65];
|
|
} PACKED; // NOLINT
|
|
|
|
class CaptivePortal : public AsyncWebHandler, public Component {
|
|
public:
|
|
CaptivePortal(web_server_base::WebServerBase *base);
|
|
void setup() override;
|
|
void dump_config() override;
|
|
void loop() override {
|
|
if (this->dns_server_ != nullptr)
|
|
this->dns_server_->processNextRequest();
|
|
}
|
|
float get_setup_priority() const override;
|
|
void start();
|
|
bool is_active() const { return this->active_; }
|
|
void end() {
|
|
this->active_ = false;
|
|
this->base_->deinit();
|
|
this->dns_server_->stop();
|
|
delete this->dns_server_;
|
|
}
|
|
|
|
bool canHandle(AsyncWebServerRequest *request) override {
|
|
if (!this->active_)
|
|
return false;
|
|
|
|
if (request->method() == HTTP_GET) {
|
|
if (request->url() == "/")
|
|
return true;
|
|
if (request->url() == "/stylesheet.css")
|
|
return true;
|
|
if (request->url() == "/wifi-strength-1.svg")
|
|
return true;
|
|
if (request->url() == "/wifi-strength-2.svg")
|
|
return true;
|
|
if (request->url() == "/wifi-strength-3.svg")
|
|
return true;
|
|
if (request->url() == "/wifi-strength-4.svg")
|
|
return true;
|
|
if (request->url() == "/lock.svg")
|
|
return true;
|
|
if (request->url() == "/wifisave")
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void handle_index(AsyncWebServerRequest *request);
|
|
|
|
void handle_wifisave(AsyncWebServerRequest *request);
|
|
|
|
void handleRequest(AsyncWebServerRequest *req) override;
|
|
|
|
protected:
|
|
void override_sta_(const std::string &ssid, const std::string &password);
|
|
|
|
web_server_base::WebServerBase *base_;
|
|
bool initialized_{false};
|
|
bool active_{false};
|
|
ESPPreferenceObject pref_;
|
|
DNSServer *dns_server_{nullptr};
|
|
};
|
|
|
|
extern CaptivePortal *global_captive_portal;
|
|
|
|
} // namespace captive_portal
|
|
} // namespace esphome
|