From 1f5ae02390678ec2b7bfa7b2d234d299d6a3c34a Mon Sep 17 00:00:00 2001
From: eadmaster <925171+eadmaster@users.noreply.github.com>
Date: Fri, 26 Jul 2024 01:47:57 +0200
Subject: [PATCH] added optional file extension filter in the filepicker

---
 src/core/sd_functions.cpp        | 10 +++++-----
 src/core/sd_functions.h          |  4 ++--
 src/modules/others/TV-B-Gone.cpp |  3 +--
 src/modules/rf/rf.cpp            |  3 +--
 4 files changed, 9 insertions(+), 11 deletions(-)

diff --git a/src/core/sd_functions.cpp b/src/core/sd_functions.cpp
index 67e53d8f..d4aa796c 100644
--- a/src/core/sd_functions.cpp
+++ b/src/core/sd_functions.cpp
@@ -302,7 +302,7 @@ void sortList(String fileList[][3], int fileListCount) {
 ** Function name: sortList
 ** Description:   sort files for name
 ***************************************************************************************/
-void readFs(FS fs, String folder, String result[][3]) {
+void readFs(FS fs, String folder, String result[][3], String allowed_ext) {
 
     int allFilesCount = 0;
     while(allFilesCount<MAXFILES) {
@@ -331,7 +331,7 @@ void readFs(FS fs, String folder, String result[][3]) {
                 result[allFilesCount][2] = "file";
                 allFilesCount++;
               }
-            else {
+            else if(allowed_ext=="*" || ext==allowed_ext) {
               result[allFilesCount][0] = fileName.substring(fileName.lastIndexOf("/") + 1);
               result[allFilesCount][1] = file2.path();
               result[allFilesCount][2] = "file";
@@ -373,7 +373,7 @@ void readFs(FS fs, String folder, String result[][3]) {
 **  Function: loopSD
 **  Where you choose what to do with your SD Files
 **********************************************************************/
-String loopSD(FS &fs, bool filePicker) {
+String loopSD(FS &fs, bool filePicker, String allowed_ext) {
   String result = "";
   bool reload=false;
   bool redraw = true;
@@ -386,7 +386,7 @@ String loopSD(FS &fs, bool filePicker) {
   closeSdCard();
   setupSdCard();
 
-  readFs(fs, Folder, fileList);
+  readFs(fs, Folder, fileList, allowed_ext);
 
   for(int i=0; i<MAXFILES; i++) if(fileList[i][2]!="") maxFiles++; else break;
   while(1){
@@ -397,7 +397,7 @@ String loopSD(FS &fs, bool filePicker) {
         tft.fillScreen(BGCOLOR);
         tft.drawRoundRect(5,5,WIDTH-10,HEIGHT-10,5,FGCOLOR);
         index=0;
-        readFs(fs, Folder, fileList);
+        readFs(fs, Folder, fileList, allowed_ext);
         PreFolder = Folder;
         maxFiles=0;
         for(int i=0; i<MAXFILES; i++) if(fileList[i][2]!="") maxFiles++; else break;
diff --git a/src/core/sd_functions.h b/src/core/sd_functions.h
index b7ebc4c4..6d708a34 100644
--- a/src/core/sd_functions.h
+++ b/src/core/sd_functions.h
@@ -26,8 +26,8 @@ bool createFolder(FS fs, String path);
 
 String readLineFromFile(File myFile);
 
-void readFs(FS fs, String folder, String result[][3]);
+void readFs(FS fs, String folder, String result[][3], String allowed_ext = "*");
 
 void sortList(String fileList[][3], int fileListCount);
 
-String loopSD(FS &fs, bool filePicker = false);
\ No newline at end of file
+String loopSD(FS &fs, bool filePicker = false, String allowed_ext = "*");
diff --git a/src/modules/others/TV-B-Gone.cpp b/src/modules/others/TV-B-Gone.cpp
index 5a86f157..9be1b8d9 100644
--- a/src/modules/others/TV-B-Gone.cpp
+++ b/src/modules/others/TV-B-Gone.cpp
@@ -280,8 +280,7 @@ void otherIRcodes() {
     // no need to proceed, go back
   }
     
-  //filepath = loopSD(*fs, true, "IR");
-  filepath = loopSD(*fs, true);
+  filepath = loopSD(*fs, true, "IR");
   databaseFile = fs->open(filepath, FILE_READ);
   drawMainBorder();
   pinMode(IrTx, OUTPUT);
diff --git a/src/modules/rf/rf.cpp b/src/modules/rf/rf.cpp
index caa3c345..f9e7b328 100644
--- a/src/modules/rf/rf.cpp
+++ b/src/modules/rf/rf.cpp
@@ -393,8 +393,7 @@ void otherRFcodes() {
     // no need to proceed, go back
   }
   
-  //filepath = loopSD(*fs, true, "SUB");
-  filepath = loopSD(*fs, true);
+  filepath = loopSD(*fs, true, "SUB");
   databaseFile = fs->open(filepath, FILE_READ);
   drawMainBorder();