UTILS: Add ScanFilesInDir() method which will scan for files in a dir…

…ectory.
Cian911 · Dec 28, 2021 · de4b56c · de4b56c
1 parent 5f9b377
commit de4b56c
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/utils/utils.go b/utils/utils.go
@@ -33,16 +33,18 @@ func ValidateFileExt(ext string) bool {
 	return true
 }
 
-func ScanFilesInDir(path string) (map[string]string, error) {
+func ScanFilesInDir(path string) (map[string]bool, error) {
 	files, err := ioutil.ReadDir(path)
 	if err != nil {
 		return nil, err
 	}
 
-	fileList := map[string]string{}
-	/* for _, file := range files { */
-	/*      */
-	/* } */
+	// "sample.txt" -> false
+	// "folder1" -> true
+	fileList := make(map[string]bool)
+	for _, file := range files {
+		fileList[file.Name()] = file.IsDir()
+	}
 
 	return fileList, nil
 }