feat: generate resolvers for inputs if fields are missing

99designs · Nov 30, 2020 · d4b43ad · d4b43ad
1 parent 0e12bfb
commit d4b43ad
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 5 deletions.
diff --git a/codegen/field.go b/codegen/field.go
@@ -452,6 +452,9 @@ func (f *Field) GoNameUnexported() string {
 }
 
 func (f *Field) ShortInvocation() string {
+	if f.Object.Kind == ast.InputObject {
+		return fmt.Sprintf("%s().%s(ctx, &it, data)", f.Object.Definition.Name, f.GoFieldName)
+	}
 	return fmt.Sprintf("%s().%s(%s)", f.Object.Definition.Name, f.GoFieldName, f.CallArgs())
 }
 
@@ -472,6 +475,13 @@ func (f *Field) ResolverType() string {
 }
 
 func (f *Field) ShortResolverDeclaration() string {
+	if f.Object.Kind == ast.InputObject {
+		return fmt.Sprintf("(ctx context.Context, obj %s, data %s) error",
+			templates.CurrentImports.LookupType(f.Object.Reference()),
+			templates.CurrentImports.LookupType(f.TypeReference.GO),
+		)
+	}
+
 	res := "(ctx context.Context"
 
 	if !f.Object.Root {

diff --git a/codegen/generated!.gotpl b/codegen/generated!.gotpl
@@ -35,6 +35,11 @@ type ResolverRoot interface {
 		{{$object.Name}}() {{$object.Name}}Resolver
 	{{ end }}
 {{- end }}
+{{- range $object := .Inputs -}}
+	{{ if $object.HasResolvers -}}
+		{{$object.Name}}() {{$object.Name}}Resolver
+	{{ end }}
+{{- end }}
 }
 
 type DirectiveRoot struct {
@@ -70,6 +75,18 @@ type ComplexityRoot struct {
 	{{- end }}
 {{- end }}
 
+{{ range $object := .Inputs -}}
+	{{ if $object.HasResolvers }}
+		type {{$object.Name}}Resolver interface {
+		{{ range $field := $object.Fields -}}
+			{{- if $field.IsResolver }}
+				{{- $field.GoFieldName}}{{ $field.ShortResolverDeclaration }}
+			{{- end }}
+		{{ end }}
+		}
+	{{- end }}
+{{- end }}
+
 type executableSchema struct {
 	resolvers  ResolverRoot
 	directives DirectiveRoot

diff --git a/codegen/input.gotpl b/codegen/input.gotpl
@@ -26,20 +26,38 @@
 						return it, graphql.ErrorOnPath(ctx, err)
 					}
 					if data, ok := tmp.({{ $field.TypeReference.GO | ref }}) ; ok {
-						it.{{$field.GoFieldName}} = data
+						{{- if $field.IsResolver }}
+							if err = ec.resolvers.{{ $field.ShortInvocation }}; err != nil {
+								return it, err
+							}
+						{{- else }}
+							it.{{$field.GoFieldName}} = data
+						{{- end }}
 					{{- if $field.TypeReference.IsNilable }}
+						{{- if not $field.IsResolver }}
 						} else if tmp == nil {
 							it.{{$field.GoFieldName}} = nil
+						{{- end }}
 					{{- end }}
 					} else {
 						err := fmt.Errorf(`unexpected type %T from directive, should be {{ $field.TypeReference.GO }}`, tmp)
 						return it, graphql.ErrorOnPath(ctx, err)
 					}
 				{{- else }}
-					it.{{$field.GoFieldName}}, err = ec.{{ $field.TypeReference.UnmarshalFunc }}(ctx, v)
-					if err != nil {
-						return it, err
-					}
+					{{- if $field.IsResolver }}
+						data, err := ec.{{ $field.TypeReference.UnmarshalFunc }}(ctx, v)
+						if err != nil {
+							return it, err
+						}
+						if err = ec.resolvers.{{ $field.ShortInvocation }}; err != nil {
+							return it, err
+						}
+					{{- else }}
+						it.{{$field.GoFieldName}}, err = ec.{{ $field.TypeReference.UnmarshalFunc }}(ctx, v)
+						if err != nil {
+							return it, err
+						}
+					{{- end }}
 				{{- end }}
 			{{- end }}
 			}