From 775d42562aee72519eb1e4ed2f199e2abfccace9 Mon Sep 17 00:00:00 2001
From: Marius Kintel <marius@kintel.net>
Date: Tue, 1 Feb 2022 23:50:04 -0500
Subject: [PATCH] Support new WiFiEvent_t enum in Arduino ESP32 2.0

---
 src/EspSimpleWifiHandler.cpp | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/EspSimpleWifiHandler.cpp b/src/EspSimpleWifiHandler.cpp
index fa0b0f9..b533a19 100644
--- a/src/EspSimpleWifiHandler.cpp
+++ b/src/EspSimpleWifiHandler.cpp
@@ -49,17 +49,25 @@ EspSimpleWifiHandler::EspSimpleWifiHandler(
     });
   #else // for ESP32
     WiFi.onEvent(
-      [this](WiFiEvent_t event, system_event_info_t info) {
+      [this](WiFiEvent_t event, WiFiEventInfo_t info) {
         this->_onWifiConnected();
       },
+#if ESP_ARDUINO_VERSION_MAJOR >= 2
+      WiFiEvent_t::ARDUINO_EVENT_WIFI_STA_GOT_IP
+#else
       WiFiEvent_t::SYSTEM_EVENT_STA_GOT_IP
+#endif
     );
 
     WiFi.onEvent(
-      [this](WiFiEvent_t event, system_event_info_t info) {
+      [this](WiFiEvent_t event, WiFiEventInfo_t info) {
         this->_onWifiDisconnected();
       },
+#if ESP_ARDUINO_VERSION_MAJOR >= 2
+      WiFiEvent_t::ARDUINO_EVENT_WIFI_STA_DISCONNECTED
+#else
       WiFiEvent_t::SYSTEM_EVENT_STA_DISCONNECTED
+#endif
     );
   #endif