From 159674bbbafa544ef07c08654a34840230b609b1 Mon Sep 17 00:00:00 2001
From: David Gauchard <gauchard@laas.fr>
Date: Wed, 21 Feb 2018 14:50:42 +0100
Subject: [PATCH] allow user loop while in AP mode

---
 WiFiManager.cpp | 4 ++++
 WiFiManager.h   | 3 +++
 2 files changed, 7 insertions(+)

diff --git a/WiFiManager.cpp b/WiFiManager.cpp
index 1e6d805..a0d5948 100644
--- a/WiFiManager.cpp
+++ b/WiFiManager.cpp
@@ -252,6 +252,10 @@ boolean  WiFiManager::startConfigPortal(char const *apName, char const *apPasswo
         break;
       }
     }
+
+    if (_loop)
+      _loop();
+
     yield();
   }
 
diff --git a/WiFiManager.h b/WiFiManager.h
index 3da163e..7a18462 100644
--- a/WiFiManager.h
+++ b/WiFiManager.h
@@ -118,6 +118,8 @@ class WiFiManager
     //if this is true, remove duplicated Access Points - defaut true
     void          setRemoveDuplicateAPs(boolean removeDuplicates);
 
+    void          setUserLoopInAP (void (*loop)()) { _loop = loop; }
+
   private:
     std::unique_ptr<DNSServer>        dnsServer;
     std::unique_ptr<ESP8266WebServer> server;
@@ -137,6 +139,7 @@ class WiFiManager
     unsigned long _configPortalTimeout    = 0;
     unsigned long _connectTimeout         = 0;
     unsigned long _configPortalStart      = 0;
+    void          (*_loop)()              = 0;
 
     IPAddress     _ap_static_ip;
     IPAddress     _ap_static_gw;