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 +}