- Improved FS include

ayushsharma82 · Aug 30, 2023 · 19d2e46 · 19d2e46
1 parent 8c656e2
commit 19d2e46
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/src/ElegantOTA.h b/src/ElegantOTA.h
@@ -23,6 +23,10 @@ _____ _                        _    ___ _____  _
 #include "stdlib_noniso.h"
 #include "elop.h"
 
+#ifndef ELEGANTOTA_USE_SPIFFS
+  #define ELEGANTOTA_USE_SPIFFS 0
+#endif
+
 #ifndef ELEGANTOTA_USE_ASYNC_WEBSERVER
   #define ELEGANTOTA_USE_ASYNC_WEBSERVER 0
 #endif
@@ -42,7 +46,11 @@ _____ _                        _    ___ _____  _
 #endif
 
 #if defined(ESP8266)
-  #include "FS.h"
+  #if ELEGANTOTA_USE_SPIFFS == 1
+    #include "FS.h"
+  #else
+    #include "LittleFS.h"
+  #endif
   #include "Updater.h"
   #include "StreamString.h"
   #if ELEGANTOTA_USE_ASYNC_WEBSERVER == 1
@@ -57,7 +65,11 @@ _____ _                        _    ___ _____  _
   #endif
   #define HARDWARE "ESP8266"
 #elif defined(ESP32)
-  #include "SPIFFS.h"
+  #if ELEGANTOTA_USE_SPIFFS == 1
+    #include "SPIFFS.h"
+  #else
+    #include "LittleFS.h"
+  #endif
   #include "Update.h"
   #include "StreamString.h"
   #if ELEGANTOTA_USE_ASYNC_WEBSERVER == 1