Add config option to omit root objects from models

Ref: 99designs#2874
ghjm · Jan 9, 2024 · 99c2b0a · 99c2b0a
1 parent 15c05a7
commit 99c2b0a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/codegen/config/config.go b/codegen/config/config.go
@@ -38,6 +38,7 @@ type Config struct {
 	OmitComplexity                bool                       `yaml:"omit_complexity,omitempty"`
 	OmitGQLGenFileNotice          bool                       `yaml:"omit_gqlgen_file_notice,omitempty"`
 	OmitGQLGenVersionInFileNotice bool                       `yaml:"omit_gqlgen_version_in_file_notice,omitempty"`
+	OmitRootModels                bool                       `yaml:"omit_root_models,omitempty"`
 	StructFieldsAlwaysPointers    bool                       `yaml:"struct_fields_always_pointers,omitempty"`
 	ReturnPointersInUmarshalInput bool                       `yaml:"return_pointers_in_unmarshalinput,omitempty"`
 	ResolversAlwaysReturnPointers bool                       `yaml:"resolvers_always_return_pointers,omitempty"`

diff --git a/plugin/modelgen/models.go b/plugin/modelgen/models.go
@@ -139,10 +139,12 @@ func (m *Plugin) MutateConfig(cfg *config.Config) error {
 			b.Interfaces = append(b.Interfaces, it)
 		case ast.Object, ast.InputObject:
 			if cfg.IsRoot(schemaType) {
-				b.Models = append(b.Models, &Object{
-					Description: schemaType.Description,
-					Name:        schemaType.Name,
-				})
+				if !cfg.OmitRootModels {
+					b.Models = append(b.Models, &Object{
+						Description: schemaType.Description,
+						Name:        schemaType.Name,
+					})
+				}
 				continue
 			}