diff --git a/pkg/fsutils/findyaml.go b/pkg/fsutils/findyaml.go index 3121ebc..61fd44c 100644 --- a/pkg/fsutils/findyaml.go +++ b/pkg/fsutils/findyaml.go @@ -6,19 +6,22 @@ import ( "strings" ) +func isYaml(info os.FileInfo) bool { + return !info.IsDir() && (strings.HasSuffix(strings.ToLower(info.Name()), ".yaml") || strings.HasSuffix(strings.ToLower(info.Name()), ".yml")) +} + // FindYamlInDir will find yaml files in folder dir, and send their filenames in batches // of size batchSize to channel fileBatches -func FindYamlInDir(dir string, fileBatches chan<- string) error { - err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error { +func FindYamlInDir(dir string, files chan<- string) error { + return filepath.Walk(dir, func(path string, info os.FileInfo, err error) error { if err != nil { return err } - if !info.IsDir() && (strings.HasSuffix(info.Name(), ".yaml") || strings.HasSuffix(info.Name(), ".yml")) { - fileBatches <- path + if isYaml(info) { + files <- path } + return nil }) - - return err }