network_state.cpp

Go to the documentation of this file.

#include "sensesp.h"
 
#include "network_state.h"
 
#include "esp_arduino_version.h"
 
#if ESP_ARDUINO_VERSION_MAJOR >= 3
#include <Network.h>
#include <WiFi.h>
#else
#include <WiFi.h>
#endif
 
namespace sensesp {
 
namespace {
 
void defer_emit(NetworkStateProducer* producer, NetworkState state) {
  event_loop()->onDelay(0, [producer, state]() { producer->emit(state); });
}
 
}  // namespace
 
NetworkStateProducer::NetworkStateProducer() {
  this->output_ = kNetworkNoConnection;
 
#if ESP_ARDUINO_VERSION_MAJOR >= 3
  // ----- Arduino-ESP32 3.x: unified Network event bus ---------------------
  wifi_got_ip_handle_ = Network.onEvent(
      [this](arduino_event_id_t, arduino_event_info_t) {
        ESP_LOGI("net_state", "WiFi station got IP %s",
                 WiFi.localIP().toString().c_str());
        defer_emit(this, kNetworkConnected);
      },
      ARDUINO_EVENT_WIFI_STA_GOT_IP);
 
  wifi_disconnected_handle_ = Network.onEvent(
      [this](arduino_event_id_t, arduino_event_info_t) {
        ESP_LOGI("net_state", "WiFi station disconnected");
        defer_emit(this, kNetworkDisconnected);
      },
      ARDUINO_EVENT_WIFI_STA_DISCONNECTED);
 
  wifi_ap_start_handle_ = Network.onEvent(
      [this](arduino_event_id_t, arduino_event_info_t) {
        ESP_LOGI("net_state", "WiFi soft-AP started, IP %s",
                 WiFi.softAPIP().toString().c_str());
        defer_emit(this, kNetworkAPMode);
      },
      ARDUINO_EVENT_WIFI_AP_START);
 
  wifi_ap_stop_handle_ = Network.onEvent(
      [this](arduino_event_id_t, arduino_event_info_t) {
        ESP_LOGI("net_state", "WiFi soft-AP stopped");
        // Stopping the soft-AP is not an outage while the station is up.
        if (!WiFi.isConnected()) {
          defer_emit(this, kNetworkDisconnected);
        }
      },
      ARDUINO_EVENT_WIFI_AP_STOP);
 
  // ----- Ethernet events (3.x only) --------------------------------------
  eth_got_ip_handle_ = Network.onEvent(
      [this](arduino_event_id_t, arduino_event_info_t) {
        ESP_LOGI("net_state", "Ethernet got IP");
        defer_emit(this, kNetworkConnected);
      },
      ARDUINO_EVENT_ETH_GOT_IP);
 
  eth_lost_ip_handle_ = Network.onEvent(
      [this](arduino_event_id_t, arduino_event_info_t) {
        ESP_LOGI("net_state", "Ethernet lost IP");
        defer_emit(this, kNetworkDisconnected);
      },
      ARDUINO_EVENT_ETH_LOST_IP);
 
  eth_disconnected_handle_ = Network.onEvent(
      [this](arduino_event_id_t, arduino_event_info_t) {
        ESP_LOGI("net_state", "Ethernet PHY disconnected");
        defer_emit(this, kNetworkDisconnected);
      },
      ARDUINO_EVENT_ETH_DISCONNECTED);
 
#else
  // ----- Arduino-ESP32 2.x: WiFi-only event bus --------------------------
  wifi_got_ip_handle_ = WiFi.onEvent(
      [this](WiFiEvent_t, WiFiEventInfo_t) {
        ESP_LOGI("net_state", "WiFi station got IP %s",
                 WiFi.localIP().toString().c_str());
        defer_emit(this, kNetworkConnected);
      },
      WiFiEvent_t::ARDUINO_EVENT_WIFI_STA_GOT_IP);
 
  wifi_disconnected_handle_ = WiFi.onEvent(
      [this](WiFiEvent_t, WiFiEventInfo_t) {
        ESP_LOGI("net_state", "WiFi station disconnected");
        defer_emit(this, kNetworkDisconnected);
      },
      WiFiEvent_t::ARDUINO_EVENT_WIFI_STA_DISCONNECTED);
 
  wifi_ap_start_handle_ = WiFi.onEvent(
      [this](WiFiEvent_t, WiFiEventInfo_t) {
        ESP_LOGI("net_state", "WiFi soft-AP started, IP %s",
                 WiFi.softAPIP().toString().c_str());
        defer_emit(this, kNetworkAPMode);
      },
      WiFiEvent_t::ARDUINO_EVENT_WIFI_AP_START);
 
  wifi_ap_stop_handle_ = WiFi.onEvent(
      [this](WiFiEvent_t, WiFiEventInfo_t) {
        ESP_LOGI("net_state", "WiFi soft-AP stopped");
        // Stopping the soft-AP is not an outage while the station is up.
        if (!WiFi.isConnected()) {
          defer_emit(this, kNetworkDisconnected);
        }
      },
      WiFiEvent_t::ARDUINO_EVENT_WIFI_AP_STOP);
 
  // Ethernet events not available on Arduino-ESP32 2.x; eth_*_handle_
  // members remain zero-initialized.
#endif
 
  // Re-emit the current (initial) state once the event loop is running so
  // late subscribers (added in SensESPApp::setup) see the latest value.
  event_loop()->onDelay(0, [this]() { this->emit(this->output_); });
}
 
NetworkStateProducer::~NetworkStateProducer() {
#if ESP_ARDUINO_VERSION_MAJOR >= 3
  Network.removeEvent(wifi_got_ip_handle_);
  Network.removeEvent(wifi_disconnected_handle_);
  Network.removeEvent(wifi_ap_start_handle_);
  Network.removeEvent(wifi_ap_stop_handle_);
  Network.removeEvent(eth_got_ip_handle_);
  Network.removeEvent(eth_lost_ip_handle_);
  Network.removeEvent(eth_disconnected_handle_);
#else
  WiFi.removeEvent(wifi_got_ip_handle_);
  WiFi.removeEvent(wifi_disconnected_handle_);
  WiFi.removeEvent(wifi_ap_start_handle_);
  WiFi.removeEvent(wifi_ap_stop_handle_);
#endif
}
 
}  // namespace sensesp