Fix support for resolvers on input types in single file mode in the resolver plugin (#8)

Author: obs-gh-christiannester

codegen/testserver/followschema/resolver.go

@@ -537,6 +537,11 @@ func (r *wrappedSliceResolver) Get(ctx context.Context, obj WrappedSlice, idx in
 	panic("not implemented")
 }
 
+// OverrideFirstField is the resolver for the overrideFirstField field.
+func (r *fieldsOrderInputResolver) OverrideFirstField(ctx context.Context, obj *FieldsOrderInput, data *string) error {
+	panic("not implemented")
+}
+
 // BackedByInterface returns BackedByInterfaceResolver implementation.
 func (r *Resolver) BackedByInterface() BackedByInterfaceResolver {
 	return &backedByInterfaceResolver{r}
@@ -589,6 +594,9 @@ func (r *Resolver) WrappedMap() WrappedMapResolver { return &wrappedMapResolver{
 // WrappedSlice returns WrappedSliceResolver implementation.
 func (r *Resolver) WrappedSlice() WrappedSliceResolver { return &wrappedSliceResolver{r} }
 
+// FieldsOrderInput returns FieldsOrderInputResolver implementation.
+func (r *Resolver) FieldsOrderInput() FieldsOrderInputResolver { return &fieldsOrderInputResolver{r} }
+
 type backedByInterfaceResolver struct{ *Resolver }
 type deferModelResolver struct{ *Resolver }
 type errorsResolver struct{ *Resolver }
@@ -605,3 +613,4 @@ type subscriptionResolver struct{ *Resolver }
 type userResolver struct{ *Resolver }
 type wrappedMapResolver struct{ *Resolver }
 type wrappedSliceResolver struct{ *Resolver }
+type fieldsOrderInputResolver struct{ *Resolver }

codegen/testserver/singlefile/resolver.go

@@ -537,6 +537,11 @@ func (r *wrappedSliceResolver) Get(ctx context.Context, obj WrappedSlice, idx in
 	panic("not implemented")
 }
 
+// OverrideFirstField is the resolver for the overrideFirstField field.
+func (r *fieldsOrderInputResolver) OverrideFirstField(ctx context.Context, obj *FieldsOrderInput, data *string) error {
+	panic("not implemented")
+}
+
 // BackedByInterface returns BackedByInterfaceResolver implementation.
 func (r *Resolver) BackedByInterface() BackedByInterfaceResolver {
 	return &backedByInterfaceResolver{r}
@@ -589,6 +594,9 @@ func (r *Resolver) WrappedMap() WrappedMapResolver { return &wrappedMapResolver{
 // WrappedSlice returns WrappedSliceResolver implementation.
 func (r *Resolver) WrappedSlice() WrappedSliceResolver { return &wrappedSliceResolver{r} }
 
+// FieldsOrderInput returns FieldsOrderInputResolver implementation.
+func (r *Resolver) FieldsOrderInput() FieldsOrderInputResolver { return &fieldsOrderInputResolver{r} }
+
 type backedByInterfaceResolver struct{ *Resolver }
 type deferModelResolver struct{ *Resolver }
 type errorsResolver struct{ *Resolver }
@@ -605,3 +613,4 @@ type subscriptionResolver struct{ *Resolver }
 type userResolver struct{ *Resolver }
 type wrappedMapResolver struct{ *Resolver }
 type wrappedSliceResolver struct{ *Resolver }
+type fieldsOrderInputResolver struct{ *Resolver }

plugin/resolvergen/resolver.go

@@ -66,7 +66,7 @@ func (m *Plugin) generateSingleFile(data *codegen.Data) error {
 		return err
 	}
 
-	for _, o := range data.Objects {
+	for _, o := range append(data.Objects, data.Inputs...) {
 		if o.HasResolvers() {
 			caser := cases.Title(language.English, cases.NoLower)
 			rewriter.MarkStructCopied(templates.LcFirst(o.Name) + templates.UcFirst(data.Config.Resolver.Type))

plugin/resolvergen/resolver_test.go

@@ -26,6 +26,13 @@ func TestLayoutSingleFile(t *testing.T) {
 	require.NoError(t, err)
 
 	require.NoError(t, p.GenerateCode(data))
+
+	b, err := os.ReadFile("testdata/singlefile/out/resolver.go")
+	require.NoError(t, err)
+	source := string(b)
+	require.Contains(t, source, "func (r *CustomResolverType) Query() QueryResolver { return &queryCustomResolverType{r} }")
+	require.Contains(t, source, "func (r *testInputCustomResolverType) InputName(ctx context.Context, obj *TestInput, data string) error {")
+
 	assertNoErrors(t, "github.com/99designs/gqlgen/plugin/resolvergen/testdata/singlefile/out")
 }
 

plugin/resolvergen/testdata/filetemplate/gqlgen.yml

@@ -14,5 +14,7 @@ resolver:
 models:
   Resolver:
     model: github.com/99designs/gqlgen/plugin/resolvergen/testdata/singlefile/out.Resolver
+  TestInput:
+    model: github.com/99designs/gqlgen/plugin/resolvergen/testdata/singlefile/out.TestInput
 
 omit_gqlgen_version_in_file_notice: true

plugin/resolvergen/testdata/filetemplate/out/model.go

@@ -3,6 +3,7 @@ package customresolver
 import "context"
 
 type Resolver struct{}
+type TestInput struct{}
 
 type QueryResolver interface {
 	Resolver(ctx context.Context) (*Resolver, error)
@@ -11,3 +12,6 @@ type QueryResolver interface {
 type ResolverResolver interface {
 	Name(ctx context.Context, obj *Resolver) (string, error)
 }
+type TestInputResolver interface {
+	InputName(ctx context.Context, obj *TestInput, data string) error
+}

plugin/resolvergen/testdata/followschema/gqlgen.yml

@@ -13,5 +13,7 @@ resolver:
 models:
   Resolver:
     model: github.com/99designs/gqlgen/plugin/resolvergen/testdata/singlefile/out.Resolver
+  TestInput:
+    model: github.com/99designs/gqlgen/plugin/resolvergen/testdata/singlefile/out.TestInput
 
 omit_gqlgen_version_in_file_notice: true

plugin/resolvergen/testdata/followschema/out/model.go

@@ -3,6 +3,7 @@ package customresolver
 import "context"
 
 type Resolver struct{}
+type TestInput struct{}
 
 type QueryResolver interface {
 	Resolver(ctx context.Context) (*Resolver, error)
@@ -11,3 +12,6 @@ type QueryResolver interface {
 type ResolverResolver interface {
 	Name(ctx context.Context, obj *Resolver) (string, error)
 }
+type TestInputResolver interface {
+	InputName(ctx context.Context, obj *TestInput, data string) error
+}

plugin/resolvergen/testdata/omit_template_comment/gqlgen.yml

@@ -14,5 +14,7 @@ resolver:
 models:
   Resolver:
     model: github.com/99designs/gqlgen/plugin/resolvergen/testdata/singlefile/out.Resolver
+  TestInput:
+    model: github.com/99designs/gqlgen/plugin/resolvergen/testdata/singlefile/out.TestInput
 
 omit_gqlgen_version_in_file_notice: true