entgql: add missing directive's argument (ent#567)

netfira · May 9, 2024 · 5585216 · 5585216
1 parent 2a81b09
commit 5585216
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 8 deletions.
diff --git a/entgql/internal/todo/ent.graphql b/entgql/internal/todo/ent.graphql
@@ -1,5 +1,5 @@
-directive @goField(forceResolver: Boolean, name: String) on FIELD_DEFINITION | INPUT_FIELD_DEFINITION
-directive @goModel(model: String, models: [String!]) on OBJECT | INPUT_OBJECT | SCALAR | ENUM | INTERFACE | UNION
+directive @goField(forceResolver: Boolean, name: String, omittable: Boolean) on FIELD_DEFINITION | INPUT_FIELD_DEFINITION
+directive @goModel(model: String, models: [String!], forceGenerate: Boolean) on OBJECT | INPUT_OBJECT | SCALAR | ENUM | INTERFACE | UNION
 type BillProduct implements Node {
   id: ID!
   name: String!

diff --git a/entgql/internal/todogotype/generated.go b/entgql/internal/todogotype/generated.go
diff --git a/entgql/internal/todopulid/generated.go b/entgql/internal/todopulid/generated.go
diff --git a/entgql/internal/todouuid/generated.go b/entgql/internal/todouuid/generated.go
diff --git a/entgql/schema.go b/entgql/schema.go
@@ -60,6 +60,10 @@ var (
 					Name: "models",
 					Type: ast.ListType(ast.NonNullNamedType("String", nil), nil),
 				},
+				{
+					Name: "forceGenerate",
+					Type: ast.NamedType("Boolean", nil),
+				},
 			},
 			Locations: []ast.DirectiveLocation{
 				ast.LocationObject,
@@ -82,6 +86,10 @@ var (
 					Name: "name",
 					Type: ast.NamedType("String", nil),
 				},
+				{
+					Name: "omittable",
+					Type: ast.NamedType("Boolean", nil),
+				},
 			},
 			Locations: []ast.DirectiveLocation{
 				ast.LocationFieldDefinition,