fix: [go] fix import equal source issue

pkg/infrastructure/ast/cocago/cocago_builder.go

@@ -110,6 +110,17 @@ func BuildMethodCall(codeFunc *CodeFunction, item ast.Stmt, fields []CodeField,
 		call := BuildCallFromExpr(it.Call, fields, imports, packageName, localVars)
 		codeFunc.FunctionCalls = append(codeFunc.FunctionCalls, call)
 	case *ast.AssignStmt:
+		vars := BuildLocalVars(it)
+
+		localVars = vars
+	default:
+		fmt.Fprintf(output, "methodCall %s\n", reflect.TypeOf(it))
+	}
+
+	return localVars
+}
+
+func BuildLocalVars(it *ast.AssignStmt) []CodeProperty {
 	var vars []CodeProperty
 	for _, lh := range it.Lhs {
 		var left string
@@ -128,13 +139,7 @@ func BuildMethodCall(codeFunc *CodeFunction, item ast.Stmt, fields []CodeField,
 			vars = append(vars, property)
 		}
 	}
-
-		localVars = vars
-	default:
-		fmt.Fprintf(output, "methodCall %s\n", reflect.TypeOf(it))
-	}
-
-	return localVars
+	return vars
 }
 
 func BuildMethodCallExprStmt(it *ast.ExprStmt, codeFunc *CodeFunction, fields []CodeField, imports []CodeImport, currentPackage string, localVars []CodeProperty) {
@@ -151,7 +156,6 @@ func BuildCallFromExpr(expr *ast.CallExpr, fields []CodeField, imports []CodeImp
 	_, selector, selName := BuildExpr(expr.Fun.(ast.Expr))
 	target := ParseTarget(selector, fields, localVars)
 	packageName := getPackageName(target, imports)
-	//fmt.Println(packageName)
 	if packageName == "" {
 		packageName = currentPackage
 	}
@@ -185,6 +189,12 @@ func getPackageName(target string, imports []CodeImport) string {
 			}
 		}
 	}
+
+	for _, imp := range imports {
+		if imp.Source == target {
+			return target
+		}
+	}
 	return packageName
 }
 
@@ -200,5 +210,6 @@ func ParseTarget(selector string, fields []CodeField, localVars []CodeProperty)
 			return field.TypeType
 		}
 	}
-	return ""
+
+	return selector
 }

pkg/infrastructure/ast/cocago/cocago_parser_test.go

@@ -147,5 +147,6 @@ func main() {
 	fmt.Println(calls)
 	g.Expect(calls[0].Package).To(Equal("sync"))
 	g.Expect(calls[0].Type).To(Equal("sync.Mutex"))
+	g.Expect(calls[2].Package).To(Equal("fmt"))
 	g.Expect(len(calls)).To(Equal(3))
 }

pkg/infrastructure/ast/cocago/testdata/node_infos/hello_world.json

@@ -23,7 +23,7 @@
             {
               "MethodName": "Println",
               "NodeName": "fmt",
-              "Package": "main",
+              "Package": "fmt",
               "Parameters": [
                 {
                   "Modifiers": null,
@@ -40,7 +40,7 @@
                 "StopLine": 0,
                 "StopLinePosition": 0
               },
-              "Type": ""
+              "Type": "fmt"
             }
           ],
           "InnerFunctions": null,

pkg/infrastructure/ast/cocago/testdata/node_infos/struct_with_func_decl.json

@@ -15,7 +15,7 @@
             {
               "MethodName": "Println",
               "NodeName": "fmt",
-              "Package": "testdata",
+              "Package": "fmt",
               "Parameters": [
                 {
                   "Modifiers": null,
@@ -32,7 +32,7 @@
                 "StopLine": 0,
                 "StopLinePosition": 0
               },
-              "Type": ""
+              "Type": "fmt"
             }
           ],
           "InnerFunctions": null,