Updates to WiFi logic to allow mode to work correctly.

espressif#1306

Author: sticilface

libraries/WiFi/src/WiFiGeneric.cpp

@@ -369,23 +369,32 @@ void WiFiGenericClass::persistent(bool persistent)
  */
 bool WiFiGenericClass::mode(wifi_mode_t m)
 {
+    if (!_esp_wifi_started) {
+        wifiLowLevelInit(); 
+    }
     wifi_mode_t cm = getMode();
     if(cm == WIFI_MODE_MAX){
         return false;
     }
     if(cm == m) {
         return true;
     }
+
+    if(m){
+        espWiFiStart();
+    } else {
+      return espWiFiStop();
+    }
+
     esp_err_t err;
     err = esp_wifi_set_mode(m);
     if(err){
         log_e("Could not set mode! %u", err);
         return false;
     }
-    if(m){
-        return espWiFiStart();
-    }
-    return espWiFiStop();
+
+    return true; 
+
 }
 
 /**
@@ -394,8 +403,12 @@ bool WiFiGenericClass::mode(wifi_mode_t m)
  */
 wifi_mode_t WiFiGenericClass::getMode()
 {
-    if(!wifiLowLevelInit()){
-        return WIFI_MODE_MAX;
+    // if(!wifiLowLevelInit()){
+    //     return WIFI_MODE_MAX;
+    // }
+    if (!_esp_wifi_started)
+    {
+        return WIFI_MODE_NULL; 
     }
     uint8_t mode;
     esp_wifi_get_mode((wifi_mode_t*)&mode);