find yml files also when specified -d flag

instrumenta · Aug 22, 2019 · d0d7e5a · d0d7e5a
1 parent ecd0802
commit d0d7e5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.go b/main.go
@@ -166,7 +166,7 @@ func aggregateFiles(args []string) ([]string, error) {
 			if err != nil {
 				return err
 			}
-			if !info.IsDir() && strings.HasSuffix(info.Name(), ".yaml") {
+			if !info.IsDir() && (strings.HasSuffix(info.Name(), ".yaml") || strings.HasSuffix(info.Name(), ".yml") {
 				files = append(files, path)
 			}
 			return nil