From 7c4963fac29da4cf02107825733fc2fd91445388 Mon Sep 17 00:00:00 2001
From: Shawn A <spam@rumental.com>
Date: Wed, 21 Feb 2018 20:12:54 -0600
Subject: [PATCH 1/2] fix bug

---
 WiFiManager.cpp | 4 ++--
 WiFiManager.h   | 5 +++--
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/WiFiManager.cpp b/WiFiManager.cpp
index 256eb9e..43e2685 100644
--- a/WiFiManager.cpp
+++ b/WiFiManager.cpp
@@ -1995,8 +1995,8 @@ String WiFiManager::WiFi_SSID(){
   #endif
 }
 
+#ifdef ESP32
 void WiFiManager::WiFiEvent(WiFiEvent_t event,system_event_info_t info){
-  #ifdef ESP32
     // WiFiManager _WiFiManager;
     // Serial.print("WM: EVENT: ");Serial.println(event);
     if(event == SYSTEM_EVENT_STA_DISCONNECTED){
@@ -2011,8 +2011,8 @@ void WiFiManager::WiFiEvent(WiFiEvent_t event,system_event_info_t info){
       Serial.println("*WM: Event: SYSTEM_EVENT_STA_DISCONNECTED, reconnecting");
       WiFi.reconnect();
     }
-  #endif  
 }
+#endif
 
 void WiFiManager::WiFi_autoReconnect(){
   #ifdef ESP8266
diff --git a/WiFiManager.h b/WiFiManager.h
index df7c655..c6e2ee2 100644
--- a/WiFiManager.h
+++ b/WiFiManager.h
@@ -301,9 +301,10 @@ class WiFiManager
     void          WiFi_autoReconnect();
     String        WiFi_SSID();
 
+    #ifdef ESP32
     static void   WiFiEvent(WiFiEvent_t event, system_event_info_t info);
+    #endif
 
-    
     // output helpers
     String        getParamOut();
     String        getIpForm(String id, String title, String value);
@@ -338,7 +339,7 @@ class WiFiManager
 
     // debugging
     boolean       _debug              = true;
-    uint8_t       _debugLevel         = 1;
+    uint8_t       _debugLevel         = 2;
     Stream&     _debugPort; // debug output stream ref
     template <typename Generic>
     void        DEBUG_WM(Generic text);

From edbaf5acdd8e8c02b0f480c2dbeabba1dd06849c Mon Sep 17 00:00:00 2001
From: Shawn A <spam@rumental.com>
Date: Wed, 21 Feb 2018 20:52:24 -0600
Subject: [PATCH 2/2] finish up menu customization

---
 WiFiManager.cpp                                    | 14 ++++++++------
 .../OnDemandConfigPortal/OnDemandConfigPortal.ino  | 11 +++++++----
 2 files changed, 15 insertions(+), 10 deletions(-)

diff --git a/WiFiManager.cpp b/WiFiManager.cpp
index 43e2685..c6097ca 100644
--- a/WiFiManager.cpp
+++ b/WiFiManager.cpp
@@ -180,8 +180,6 @@ boolean WiFiManager::autoConnect() {
 boolean WiFiManager::autoConnect(char const *apName, char const *apPassword) {
   DEBUG_WM(F("AutoConnect"));
 
-  Serial.println(getMenuOut());
-
   // attempt to connect using saved settings, on fail fallback to AP config portal
   WiFi.enableSTA(true);
   _usermode = WIFI_STA;
@@ -738,9 +736,8 @@ String WiFiManager::getMenuOut(){
   String page;  
   int i;
   for(i=0;i<sizeof(_menuIds);i++){
-    if(i == MENU_PARAM){
-      if(_paramsCount == 0) continue;
-    }
+    if(_menuIds[i] == 255) continue;
+    if((i == MENU_PARAM) && (_paramsCount == 0)) continue; // no params set, omit params
     page += HTTP_PORTAL_MENU[_menuIds[i]];
   }
   return page;
@@ -1701,7 +1698,12 @@ bool  WiFiManager::setHostname(const char * hostname){
 }
 
 void WiFiManager::setMenu(uint8_t menu[]){
-  // _menuIds = menu;
+  int i;
+  int n = sizeof(menu);
+  for(i=0;i<sizeof(_menuIds);i++){
+    if(_menuIds[i] == MENU_PARAM) _paramsInWifi = false; // param auto flag
+    _menuIds[i] = i<n ? menu[i] : 255;
+  }
 }
 
 // GETTERS
diff --git a/examples/OnDemandConfigPortal/OnDemandConfigPortal.ino b/examples/OnDemandConfigPortal/OnDemandConfigPortal.ino
index 0b80f51..ee7b76c 100644
--- a/examples/OnDemandConfigPortal/OnDemandConfigPortal.ino
+++ b/examples/OnDemandConfigPortal/OnDemandConfigPortal.ino
@@ -6,10 +6,13 @@
 #define TRIGGER_PIN 0
 const char* modes[] = { "NULL", "STA", "AP", "STA+AP" };
 
+WiFiManager wifiManager;
+
 
 void setup() {
   // put your setup code here, to run once:
   Serial.begin(115200);
+  delay(3000);
   Serial.println("\n Starting");
 
   Serial.setDebugOutput(true);  
@@ -17,7 +20,6 @@ void setup() {
   WiFi.printDiag(Serial);
 
   //Local intialization. Once its business is done, there is no need to keep it around
-  WiFiManager wifiManager;
   //reset settings - for testing
   //wifiManager.resetSettings();
 
@@ -25,14 +27,16 @@ void setup() {
   //useful to make it all retry or go to sleep
   //in seconds
   wifiManager.setConfigPortalTimeout(180);
-  
+  // uint8_t menu[] = {wifiManager.MENU_WIFI,wifiManager.MENU_INFO,wifiManager.MENU_PARAM,wifiManager.MENU_CLOSE};
+  // wifiManager.setMenu(menu);
+
   //fetches ssid and pass and tries to connect
   //if it does not connect it starts an access point with the specified name
   //here  "AutoConnectAP"
   //and goes into a blocking loop awaiting configuration
   if(!wifiManager.autoConnect("AutoConnectAP","12345678")) {
     Serial.println("failed to connect and hit timeout");
-  } 
+  }
 
   pinMode(TRIGGER_PIN, INPUT);
 }
@@ -43,7 +47,6 @@ void loop() {
   if ( digitalRead(TRIGGER_PIN) == LOW ) {
     //WiFiManager
     //Local intialization. Once its business is done, there is no need to keep it around
-    WiFiManager wifiManager;
 
     //reset settings - for testing
     //wifiManager.resetSettings();