feat(generate): add ability to generate multiple targeted packages

pkg/generate/command.go

@@ -15,6 +15,7 @@ const (
 
 var (
 	packageName            string
+	packageNames           []string
 	refetch                bool
 	includeIntegrationTest bool
 )
@@ -34,6 +35,7 @@ is up to date with your configured GraphQL API.
 	Run: func(cmd *cobra.Command, args []string) {
 		util.LogIfError(log.ErrorLevel, Generate(GeneratorOptions{
 			PackageName:            packageName,
+			PackageNames:           packageNames,
 			Refetch:                refetch,
 			IncludeIntegrationTest: includeIntegrationTest,
 		}))
@@ -42,6 +44,7 @@ is up to date with your configured GraphQL API.
 
 func init() {
 	Command.Flags().StringVarP(&packageName, "package", "p", "", "Go package name for the generated files")
+	Command.Flags().StringSliceVar(&packageNames, "packages", []string{}, "List of package names to generate. Example: --packages=package1,package2")
 
 	Command.Flags().StringP("schema", "s", fetch.DefaultSchemaCacheFile, "Schema file to read from")
 	util.LogIfError(log.ErrorLevel, viper.BindPFlag("schema_file", Command.Flags().Lookup("schema")))

pkg/generate/generate.go

@@ -18,6 +18,7 @@ import (
 
 type GeneratorOptions struct {
 	PackageName            string
+	PackageNames           []string
 	Refetch                bool
 	IncludeIntegrationTest bool
 }
@@ -76,6 +77,16 @@ func Generate(options GeneratorOptions) error {
 		return generateForPackage(options.PackageName, cfg, s, options.IncludeIntegrationTest)
 	}
 
+	if options.PackageNames != nil {
+		for _, packageName := range options.PackageNames {
+			if err := generateForPackage(packageName, cfg, s, options.IncludeIntegrationTest); err != nil {
+				return err
+			}
+		}
+
+		return nil
+	}
+
 	// Generate for all configured packages
 	for _, pkgConfig := range cfg.Packages {
 		if err := generateForPackage(pkgConfig.Name, cfg, s, options.IncludeIntegrationTest); err != nil {