From a04807415386a6dcbfcebe75a3e55099d515be9d Mon Sep 17 00:00:00 2001
From: Yang Luo <hsluoyz@qq.com>
Date: Fri, 23 Jun 2023 16:52:34 +0800
Subject: [PATCH] Add ListFiles()

---
 util/path.go | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/util/path.go b/util/path.go
index 5801b78..734a9ab 100644
--- a/util/path.go
+++ b/util/path.go
@@ -1,6 +1,9 @@
 package util
 
-import "os"
+import (
+	"io/ioutil"
+	"os"
+)
 
 func FileExist(path string) bool {
 	if _, err := os.Stat(path); os.IsNotExist(err) {
@@ -8,3 +11,20 @@ func FileExist(path string) bool {
 	}
 	return true
 }
+
+func ListFiles(path string) []string {
+	res := []string{}
+
+	files, err := ioutil.ReadDir(path)
+	if err != nil {
+		panic(err)
+	}
+
+	for _, f := range files {
+		if !f.IsDir() {
+			res = append(res, f.Name())
+		}
+	}
+
+	return res
+}