From 57c2f3183b6cad140cce571ed11748182b9fec9d Mon Sep 17 00:00:00 2001
From: Nicu Hodos <nicu@hodos.ro>
Date: Sat, 29 Jun 2024 15:43:10 +0200
Subject: [PATCH 1/2] fix wifi reconnection issue by using enable:
 onStationModeDisconnected is invoked every second and restartDelayed keeps
 getting postponed

---
 library.json | 2 +-
 src/wifi.h   | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/library.json b/library.json
index 0162e48..cc7b8cd 100644
--- a/library.json
+++ b/library.json
@@ -1,6 +1,6 @@
 {
     "name": "Wifi",
-    "version": "1.0.1",
+    "version": "1.0.2",
     "description": "Helper classes for handling wifi connectiviy and OTA",
     "repository":
     {
diff --git a/src/wifi.h b/src/wifi.h
index 1b54e3c..c15e6fe 100644
--- a/src/wifi.h
+++ b/src/wifi.h
@@ -33,7 +33,7 @@ namespace Wifi {
             currentPsk = WiFi.psk();
             tReconnect.enable();
             }
-            );
+        );
 
     void setup(Scheduler& ts, void(*onConnected)() = nullptr) {
         stationConnectedHandler = WiFi.onStationModeGotIP(
@@ -47,7 +47,7 @@ namespace Wifi {
         stationDisconnectedHandler = WiFi.onStationModeDisconnected(
             [](const WiFiEventStationModeDisconnected& e) {
                 Serial.println("Disconnected from network.");
-                tReconnect.restartDelayed();
+                tReconnect.enableIfNot();
             });
 
 

From da372ea077b75d08c97971278f2e9df759682963 Mon Sep 17 00:00:00 2001
From: Nicu Hodos <nicu@hodos.ro>
Date: Sat, 29 Jun 2024 17:25:11 +0200
Subject: [PATCH 2/2] add support for onDisconnected callback

---
 src/wifi.h | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/wifi.h b/src/wifi.h
index c15e6fe..b0643e9 100644
--- a/src/wifi.h
+++ b/src/wifi.h
@@ -35,7 +35,7 @@ namespace Wifi {
             }
         );
 
-    void setup(Scheduler& ts, void(*onConnected)() = nullptr) {
+    void setup(Scheduler& ts, void(*onConnected)() = nullptr, void(*onDisconnected)() = nullptr) {
         stationConnectedHandler = WiFi.onStationModeGotIP(
             [onConnected](const WiFiEventStationModeGotIP& e) {
                 Serial.println("Connected to network.");
@@ -45,9 +45,10 @@ namespace Wifi {
             });
 
         stationDisconnectedHandler = WiFi.onStationModeDisconnected(
-            [](const WiFiEventStationModeDisconnected& e) {
+            [onDisconnected](const WiFiEventStationModeDisconnected& e) {
                 Serial.println("Disconnected from network.");
                 tReconnect.enableIfNot();
+                if (onDisconnected) onDisconnected();
             });