Try underlying types for named type custom methods

builder/builder.go

@@ -1,6 +1,8 @@
 package builder
 
 import (
+	"go/types"
+
 	"github.com/dave/jennifer/jen"
 	"github.com/jmattheis/goverter/namer"
 	"github.com/jmattheis/goverter/xtype"
@@ -40,6 +42,7 @@ type MethodContext struct {
 	TargetType   *xtype.Type
 	AutoMap      []string
 	Flags        ConversionFlags
+	HasExtend    func(types.Type, types.Type) bool
 	SeenNamed    map[string]struct{}
 
 	TargetVar *jen.Statement

builder/flags.go

@@ -8,6 +8,7 @@ const (
 	FlagMatchIgnoreCase
 	FlagIgnoreUnexported
 	FlagZeroValueOnPtrInconsistency
+	FlagTryUnderlyingTypeExtends
 )
 
 type ConversionFlags map[ConversionFlag]bool

builder/named_extend.go

@@ -0,0 +1,70 @@
+package builder
+
+import (
+	"github.com/dave/jennifer/jen"
+	"github.com/jmattheis/goverter/xtype"
+)
+
+// NamedExtendSource handles FlagTryUnderlyingTypeExtends
+type NamedExtendSource struct{}
+
+// Matches returns true, if the builder can create handle the given types.
+func (*NamedExtendSource) Matches(ctx *MethodContext, source, target *xtype.Type) bool {
+	if !ctx.Flags.Has(FlagTryUnderlyingTypeExtends) {
+		return false
+	}
+
+	sourceUnderlying, targetUnderlying := findUnderlyingExtendMapping(ctx, source, target)
+	return sourceUnderlying || targetUnderlying
+}
+
+// Build creates conversion source code for the given source and target type.
+func (*NamedExtendSource) Build(gen Generator, ctx *MethodContext, sourceID *xtype.JenID, source, target *xtype.Type) ([]jen.Code, *xtype.JenID, *Error) {
+	sourceUnderlying, targetUnderlying := findUnderlyingExtendMapping(ctx, source, target)
+
+	innerSource := source
+	innerTarget := target
+
+	if sourceUnderlying {
+		innerSource = xtype.TypeOf(source.NamedType.Underlying())
+		sourceID = xtype.OtherID(innerSource.TypeAsJen().Call(sourceID.Code))
+	}
+
+	if targetUnderlying {
+		innerTarget = xtype.TypeOf(target.NamedType.Underlying())
+	}
+
+	stmt, id, err := gen.Build(ctx, sourceID, innerSource, innerTarget, NoWrap)
+	if err != nil {
+		return nil, nil, err.Lift(&Path{
+			SourceID:   "*",
+			SourceType: innerSource.T.String(),
+			TargetID:   "*",
+			TargetType: innerTarget.T.String(),
+		})
+	}
+
+	if targetUnderlying {
+		id = xtype.OtherID(target.TypeAsJen().Call(id.Code))
+	}
+
+	return stmt, id, err
+}
+
+func findUnderlyingExtendMapping(ctx *MethodContext, source, target *xtype.Type) (underlyingSource bool, underlyingTarget bool) {
+	if source.Named {
+		if ctx.HasExtend(source.NamedType.Underlying(), target.NamedType) {
+			return true, false
+		}
+
+		if target.Named && ctx.HasExtend(source.NamedType.Underlying(), target.NamedType.Underlying()) {
+			return true, true
+		}
+	}
+
+	if target.Named && ctx.HasExtend(source.NamedType, target.NamedType.Underlying()) {
+		return false, true
+	}
+
+	return false, false
+}

comments/parse_docs.go

@@ -229,6 +229,9 @@ func parseConverterComment(comment string, config ConverterConfig) (ConverterCon
 			case "useZeroValueOnPointerInconsistency":
 				config.Flags.Set(builder.FlagZeroValueOnPtrInconsistency)
 				continue
+			case "tryUnderlyingNamedTypeExtends":
+				config.Flags.Set(builder.FlagTryUnderlyingTypeExtends)
+				continue
 			}
 			return config, fmt.Errorf("unknown %s comment: %s", prefix, line)
 		}
@@ -324,6 +327,9 @@ func parseMethodComment(comment string) (Method, error) {
 				}
 				m.Flags.Set(builder.FlagWrapErrors)
 				continue
+			case "tryUnderlyingNamedTypeExtends":
+				m.Flags.Set(builder.FlagTryUnderlyingTypeExtends)
+				continue
 			}
 			return m, fmt.Errorf("unknown %s comment: %s", prefix, line)
 		}

docs/conversion/misc.md

@@ -151,3 +151,7 @@ func (c *ConverterImpl) Convert(source example.Input) example.Output {
 ```
 
 <!-- tabs:end -->
+
+## Try underlying named types for custom methods
+
+TODO

generator/generate.go

@@ -24,6 +24,7 @@ type Config struct {
 
 // BuildSteps that'll used for generation.
 var BuildSteps = []builder.Builder{
+	&builder.NamedExtendSource{},
 	&builder.BasicTargetPointerRule{},
 	&builder.Pointer{},
 	&builder.SourcePointer{},

generator/generator.go

@@ -168,6 +168,15 @@ func (g *generator) appendToFile() {
 	}
 }
 
+func (g *generator) HasExtend(source, target types.Type) bool {
+	signature := xtype.Signature{Source: source.String(), Target: target.String()}
+	_, ok := g.extend[signature]
+	if !ok {
+		_, ok = g.lookup[signature]
+	}
+	return ok
+}
+
 func (g *generator) findOverlappingExplicitStructMethod(method *methodDefinition) (*methodDefinition, bool) {
 	source := method.Source
 	target := method.Target
@@ -211,6 +220,7 @@ func (g *generator) buildMethod(method *methodDefinition, errWrapper builder.Err
 		Fields:       method.Fields,
 		FieldsTarget: fieldsTarget,
 		SeenNamed:    map[string]struct{}{},
+		HasExtend:    g.HasExtend,
 		Flags:        method.Flags,
 		AutoMap:      method.AutoMap,
 		TargetType:   method.Target,
@@ -447,11 +457,5 @@ func (g *generator) Build(
 		return g.Build(ctx, sourceID, source, target, errWrapper)
 	}
 
-	for _, rule := range BuildSteps {
-		if rule.Matches(ctx, source, target) {
-			return rule.Build(g, ctx, sourceID, source, target)
-		}
-	}
-
-	return nil, nil, builder.NewError(fmt.Sprintf("TypeMismatch: Cannot convert %s to %s", source.T, target.T))
+	return g.buildNoLookup(ctx, sourceID, source, target)
 }

scenario/extend_named_underlying_both.yml

@@ -0,0 +1,34 @@
+input:
+    input.go: |
+        package structs
+
+        // goverter:converter
+        // goverter:extend ConvertUnderlying
+        type Converter interface {
+            // goverter:tryUnderlyingNamedTypeExtends
+            Convert(source Input) Output
+        }
+
+        func ConvertUnderlying(s int) string {
+            return ""
+        }
+
+        type InputID  int
+        type OutputID string
+
+        type Input struct  { ID InputID  }
+        type Output struct { ID OutputID }
+success: |
+    // Code generated by github.com/jmattheis/goverter, DO NOT EDIT.
+
+    package generated
+
+    import execution "github.com/jmattheis/goverter/execution"
+
+    type ConverterImpl struct{}
+
+    func (c *ConverterImpl) Convert(source execution.Input) execution.Output {
+    	var structsOutput execution.Output
+    	structsOutput.ID = execution.OutputID(execution.ConvertUnderlying(int(source.ID)))
+    	return structsOutput
+    }

scenario/extend_named_underlying_source.yml

@@ -0,0 +1,34 @@
+input:
+    input.go: |
+        package structs
+
+        // goverter:converter
+        // goverter:extend ConvertUnderlying
+        type Converter interface {
+            // goverter:tryUnderlyingNamedTypeExtends
+            Convert(source Input) Output
+        }
+
+        func ConvertUnderlying(s int) OutputID {
+            return ""
+        }
+
+        type InputID  int
+        type OutputID string
+
+        type Input struct  { ID InputID  }
+        type Output struct { ID OutputID }
+success: |
+    // Code generated by github.com/jmattheis/goverter, DO NOT EDIT.
+
+    package generated
+
+    import execution "github.com/jmattheis/goverter/execution"
+
+    type ConverterImpl struct{}
+
+    func (c *ConverterImpl) Convert(source execution.Input) execution.Output {
+    	var structsOutput execution.Output
+    	structsOutput.ID = execution.ConvertUnderlying(int(source.ID))
+    	return structsOutput
+    }

scenario/extend_named_underlying_target.yml

@@ -0,0 +1,34 @@
+input:
+    input.go: |
+        package structs
+
+        // goverter:converter
+        // goverter:extend ConvertUnderlying
+        type Converter interface {
+            // goverter:tryUnderlyingNamedTypeExtends
+            Convert(source Input) Output
+        }
+
+        func ConvertUnderlying(s InputID) string {
+            return ""
+        }
+
+        type InputID  int
+        type OutputID string
+
+        type Input struct  { ID InputID  }
+        type Output struct { ID OutputID }
+success: |
+    // Code generated by github.com/jmattheis/goverter, DO NOT EDIT.
+
+    package generated
+
+    import execution "github.com/jmattheis/goverter/execution"
+
+    type ConverterImpl struct{}
+
+    func (c *ConverterImpl) Convert(source execution.Input) execution.Output {
+    	var structsOutput execution.Output
+    	structsOutput.ID = execution.OutputID(execution.ConvertUnderlying(source.ID))
+    	return structsOutput
+    }