Support globbing input paths

actgardner · Apr 8, 2019 · b1b6826 · b1b6826
1 parent eaf7537
commit b1b6826
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/gogen-avro/config.go b/gogen-avro/config.go
@@ -4,6 +4,7 @@ import (
 	"flag"
 	"fmt"
 	"os"
+	"path/filepath"
 	"strings"
 )
 
@@ -56,6 +57,15 @@ func parseCmdLine() config {
 	}
 
 	cfg.targetDir = flag.Arg(0)
-	cfg.files = flag.Args()[1:]
+	cfg.files = make([]string, 0)
+
+	for _, glob := range flag.Args()[1:] {
+		files, err := filepath.Glob(glob)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "Error parsing input file as glob: %v", err)
+			os.Exit(1)
+		}
+		cfg.files = append(cfg.files, files...)
+	}
 	return cfg
 }
diff --git a/test/complex-arrays-multifile/generate.go b/test/complex-arrays-multifile/generate.go
@@ -1,3 +1,3 @@
 package avro
 
-//go:generate $GOPATH/bin/gogen-avro  .  parent.avsc child.avsc
+//go:generate $GOPATH/bin/gogen-avro  .  *.avsc