Fix bug that struct is not imported correctly when file generate anot…

…her directory
k-yomo · Aug 18, 2020 · dba9518 · dba9518
1 parent 1cdeb40
commit dba9518
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 26 deletions.
diff --git a/factorytpl.go b/factorytpl.go
diff --git a/generate.go b/generate.go
@@ -2,6 +2,7 @@ package fixtory
 
 import (
 	"bytes"
+	"fmt"
 	"github.com/k-yomo/fixtory/pkg/astutil"
 	"go/ast"
 	"go/format"
@@ -41,11 +42,17 @@ func Generate(targetDir string, types []string, pkgName string, newWriter func()
 				fieldNames = append(fieldNames, field.Names[0].Name)
 			}
 			tpl := template.Must(template.New("factory").Funcs(template.FuncMap{"ToLower": strings.ToLower}).Parse(factoryTpl))
+			st := spec.Name.Name
+			if pkgName != "" {
+				st = fmt.Sprintf("%s.%s", walker.Pkg.Name, st)
+			}
 			params := struct {
 				StructName string
+				Struct string
 				FieldNames []string
 			}{
 				StructName: spec.Name.Name,
+				Struct: st,
 				FieldNames: fieldNames,
 			}
 			if err := tpl.Execute(body, params); err != nil {