mirror of https://github.com/rwxrob/bonzai
Add util.File* functions
parent
566730b4c1
commit
107a091650
@ -0,0 +1,20 @@
|
||||
package util
|
||||
|
||||
import (
|
||||
"os"
|
||||
|
||||
"github.com/rwxrob/bonzai/filter"
|
||||
)
|
||||
|
||||
func Files(dir string) []string {
|
||||
files := []string{}
|
||||
finfo, _ := os.ReadDir(dir)
|
||||
for _, f := range finfo {
|
||||
files = append(files, f.Name())
|
||||
}
|
||||
return files
|
||||
}
|
||||
|
||||
func FilesWith(dir, pre string) []string {
|
||||
return filter.HasPrefix(Files(dir), pre)
|
||||
}
|
@ -0,0 +1,23 @@
|
||||
package util_test
|
||||
|
||||
import (
|
||||
"github.com/rwxrob/bonzai/filter"
|
||||
"github.com/rwxrob/bonzai/util"
|
||||
)
|
||||
|
||||
func ExampleFiles() {
|
||||
filter.Println(util.Files("testdata/files"))
|
||||
// Output:
|
||||
// bar
|
||||
// blah
|
||||
// foo
|
||||
// other
|
||||
// some
|
||||
}
|
||||
|
||||
func ExampleFilesWith() {
|
||||
filter.Println(util.FilesWith("testdata/files", "b"))
|
||||
// Output:
|
||||
// bar
|
||||
// blah
|
||||
}
|
Loading…
Reference in New Issue