Fix #755: calculate skip number in GetModelMethod (#782)

* fix(#755): fix generate model

* test(gen-model): add test

* test(gen-model): fix

internal/generate/query.go

@@ -179,9 +179,9 @@ func (b *QueryStructMeta) ReviseDIYMethod() error {
 	if tableName == nil {
 		methods = append(methods, parser.DefaultMethodTableName(b.ModelStructName))
 	} else {
-		//e.g. return "@@table" => return TableNameUser
+		// e.g. return "@@table" => return TableNameUser
 		tableName.Body = strings.ReplaceAll(tableName.Body, "\"@@table\"", "TableName"+b.ModelStructName)
-		//e.g. return "t_@@table" => return "t_user"
+		// e.g. return "t_@@table" => return "t_user"
 		tableName.Body = strings.ReplaceAll(tableName.Body, "@@table", b.TableName)
 	}
 	b.ModelMethods = methods
@@ -194,7 +194,7 @@ func (b *QueryStructMeta) ReviseDIYMethod() error {
 
 func (b *QueryStructMeta) addMethodFromAddMethodOpt(methods ...interface{}) *QueryStructMeta {
 	for _, method := range methods {
-		modelMethods, err := parser.GetModelMethod(method, 5)
+		modelMethods, err := parser.GetModelMethod(method)
 		if err != nil {
 			panic("add diy method err:" + err.Error())
 		}

internal/parser/export.go

@@ -74,7 +74,7 @@ func fileExists(path string) bool {
 }
 
 // GetModelMethod get diy methods
-func GetModelMethod(v interface{}, skip int) (method *DIYMethods, err error) {
+func GetModelMethod(v interface{}) (method *DIYMethods, err error) {
 	method = new(DIYMethods)
 
 	// get diy method info by input value, must input a function or a struct
@@ -98,7 +98,15 @@ func GetModelMethod(v interface{}, skip int) (method *DIYMethods, err error) {
 	// if struct in main file
 	ctx := build.Default
 	if method.pkgPath == "main" {
-		_, file, _, _ := runtime.Caller(skip)
+		var skip int
+		var file string
+		for {
+			_, file, _, _ = runtime.Caller(skip)
+			if !(strings.Contains(file, "gorm/gen/generator.go") || strings.Contains(file, "gorm/gen/internal")) || file == "" {
+				break
+			}
+			skip++
+		}
 		p, err = ctx.ImportDir(filepath.Dir(file), build.ImportComment)
 	} else {
 		p, err = ctx.Import(method.pkgPath, "", build.ImportComment)