feat(rust): make pass for first function calls

Author: phodal

chapi-ast-rust/src/main/kotlin/chapi/ast/rustast/RustFullIdentListener.kt

@@ -27,8 +27,6 @@ class RustFullIdentListener(val fileName: String) : RustAstBaseListener(fileName
                 }
             }
         }
-
-        println("localVars: $localVars")
     }
 
     override fun enterCallExpression(ctx: RustParser.CallExpressionContext?) {
@@ -43,17 +41,25 @@ class RustFullIdentListener(val fileName: String) : RustAstBaseListener(fileName
     }
 
     override fun enterMethodCallExpression(ctx: RustParser.MethodCallExpressionContext?) {
-        val functionName = ctx?.expression()?.text
-        val split = functionName?.split("::")
+        val instanceVar = ctx?.expression()?.text
+        val nodeName = localVars.getOrDefault(instanceVar ?: "", instanceVar)
+
+        val functionName = lookupFunctionName(ctx?.pathExprSegment())
+
         // todo: handle method call
         functionInstance.FunctionCalls += CodeCall(
-            Package = split?.dropLast(1)?.joinToString("::") ?: "",
-            NodeName = split?.dropLast(1)?.joinToString("::") ?: "",
-            FunctionName = split?.last() ?: "",
+            Package = nodeName ?: "",
+            NodeName = nodeName ?: "",
+            OriginNodeName = instanceVar ?: "",
+            FunctionName = functionName,
             Parameters = buildParameters(ctx?.callParams()),
         )
     }
 
+    private fun lookupFunctionName(pathExprSegmentContext: RustParser.PathExprSegmentContext?): String {
+        return pathExprSegmentContext?.pathIdentSegment()?.identifier()?.text ?: ""
+    }
+
     private fun buildParameters(callParamsContext: RustParser.CallParamsContext?) = callParamsContext?.expression()?.map {
         CodeProperty(
             TypeValue = it.text,

chapi-ast-rust/src/test/kotlin/chapi/ast/rustast/RustFullIdentListenerTest.kt

@@ -201,8 +201,8 @@ class RustFullIdentListenerTest {
 
         val codeContainer = RustAnalyser().analysis(code, "test.rs")
         val codeDataStruct = codeContainer.DataStructures[0]
-        assertEquals(1, codeDataStruct.Functions.size)
         val functionCalls = codeDataStruct.Functions[0].FunctionCalls
+
         assertEquals(1, functionCalls.size)
         assertEquals("Point", functionCalls[0].NodeName)
         assertEquals("new", functionCalls[0].FunctionName)
@@ -221,8 +221,11 @@ class RustFullIdentListenerTest {
 
         val codeContainer = RustAnalyser().analysis(code, "test.rs")
         val codeDataStruct = codeContainer.DataStructures[0]
-        assertEquals(1, codeDataStruct.Functions.size)
         val functionCalls = codeDataStruct.Functions[0].FunctionCalls
         assertEquals(2, functionCalls.size)
+
+        assertEquals("crate::Point", functionCalls[1].NodeName)
+        assertEquals("print", functionCalls[1].FunctionName)
+        assertEquals("p", functionCalls[1].OriginNodeName)
     }
 }

chapi-domain/src/main/kotlin/chapi/domain/core/CodeCall.kt

@@ -7,6 +7,7 @@ enum class CallType(val calltype: String) {
     FIELD("field"),
     LAMBDA("lambda"),
     ARROW("arrow"),
+
     // for Kotlin,
     CREATOR("creator"),
     FUNCTION("function"),
@@ -22,17 +23,24 @@ enum class CallType(val calltype: String) {
 @Serializable
 data class CodeCall(
     var Package: String = "",
-    // for Java, it can be CreatorClass, lambda
-    // for TypeScript, can be anonymous function, arrow function
+    /**
+     * for Java, it can be CreatorClass, lambda
+     * for TypeScript, can be anonymous function, arrow function
+     */
     var Type: CallType = CallType.FUNCTION,
-    // for Class/DataStruct, it's ClassName
-    // for Function, it's empty
+    /**
+     * for Class/DataStruct, it's ClassName
+     * for Function, it's empty
+     */
     var NodeName: String = "",
     var FunctionName: String = "",
     var Parameters: List<CodeProperty> = listOf(),
     var Position: CodePosition = CodePosition(),
-    // like "v1.Group", the v1 will be the Receiver
-    // since 2.0.0-Beta.9
+    /**
+     * like "v1.Group", the v1 will be the OriginNodeName
+     *
+     * @since 2.0.0-Beta.9
+    */
     var OriginNodeName: String = "",
 ) {