Add back replace-types

Per #864
vektra · Jan 20, 2025 · 1d437a6 · 1d437a6
1 parent 85c4718
commit 1d437a6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/internal/config.go b/internal/config.go
@@ -418,6 +418,7 @@ type Config struct {
 	MockName           *string        `koanf:"mockname"`
 	PkgName            *string        `koanf:"pkgname"`
 	Recursive          *bool          `koanf:"recursive"`
+	ReplaceType        []string       `koanf:"replace-type"`
 	Template           *string        `koanf:"template"`
 	TemplateData       map[string]any `koanf:"template-data"`
 	UnrollVariadic     *bool          `koanf:"unroll-variadic"`

diff --git a/internal/template_generator.go b/internal/template_generator.go
@@ -192,6 +192,7 @@ func (g *TemplateGenerator) methodData(ctx context.Context, method *types.Func)
 			log.Debug().Str("import", imprt.Path()).Str("import-qualifier", imprt.Qualifier()).Msg("existing imports")
 		}
 
+		// TODO: ReplaceTypes logic should go here in the AddVar.
 		params[j] = template.ParamData{
 			Var:      methodScope.AddVar(param, ""),
 			Variadic: signature.Variadic() && j == signature.Params().Len()-1,
@@ -287,6 +288,7 @@ func (g *TemplateGenerator) Generate(
 					Str("method-name", method.Name).
 					Logger().
 					WithContext(ctx))
+			method.Scope.ReplaceTypes()
 		}
 
 		ifaceLog.Debug().Str("template-data", fmt.Sprintf("%v", ifaceMock.Config.TemplateData)).Msg("printing template data")