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