fix handling default value in input object

mununki · Apr 2, 2024 · 1aeac27 · 1aeac27
1 parent 5f17c77
commit 1aeac27
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 0 deletions.
diff --git a/lib/graphql.go b/lib/graphql.go
@@ -82,6 +82,7 @@ type Field struct {
 	Null         bool
 	IsList       bool
 	IsListNull   bool
+	DefaultValue *string
 	Directives   []*Directive
 	Descriptions *[]string
 	Comments     *[]string

diff --git a/lib/schema.go b/lib/schema.go
@@ -330,6 +330,13 @@ func (s *Schema) Parse(p *Parser) {
 					} else {
 						fd.Null = true
 					}
+
+					if p.lex.peek() == '=' {
+						p.lex.consumeToken(tokEqual)
+						tex, _ := p.lex.consumeIdentInclString(tokNumber)
+						te := tex.String()
+						fd.DefaultValue = &te
+					}
 				}
 
 				fd.Directives = p.parseDirectives()

diff --git a/lib/write.go b/lib/write.go
@@ -237,6 +237,9 @@ func (ms *MergedSchema) WriteSchema(s *Schema) string {
 			if p.IsList && !p.IsListNull {
 				ms.buf.WriteString("!")
 			}
+			if p.DefaultValue != nil {
+				ms.buf.WriteString(" = " + *p.DefaultValue)
+			}
 
 			ms.stitchDirectives(p.Directives)
 

diff --git a/test/default_value/generated.graphql b/test/default_value/generated.graphql
@@ -10,3 +10,6 @@ type Query {
 
 
 
+input User {
+    name: String! = "woonki"
+}
Original file line number	Diff line number	Diff line change
Expand Up		@@ -10,3 +10,6 @@ type Query {



		input User {
		name: String! = "woonki"
		}