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] 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();
             });