feat(context): add CppVariableContextBuilder

This commit adds a new file `CppVariableContextBuilder.kt` which implements the `VariableContextBuilder` interface. It provides the functionality to retrieve the variable context for a given `PsiElement` in C++ code. The `getVariableContext` function filters out `OCDeclaration` elements, extracts the declarators, and maps them to `VariableContext` objects. The first `VariableContext` object is returned as the result.

Author: phodal

cpp/src/main/kotlin/cc/unitmesh/cpp/context/CppMethodContextBuilder.kt

@@ -19,7 +19,7 @@ class CppMethodContextBuilder : MethodContextBuilder {
 
         val symbol: OCSymbolWithQualifiedName = psiElement.symbol ?: return null
         val locateDefinition = symbol.locateDefinition(psiElement.project) ?: return null
-        val function = (locateDefinition as? OCFunctionDefinition) ?: return null
+        val function = (psiElement as? OCFunctionDefinition) ?: return null
 
         val structDeclaration = if (includeClassContext) (psiElement as? OCMethod)?.containingClass else null
 
@@ -33,7 +33,7 @@ class CppMethodContextBuilder : MethodContextBuilder {
             function.name!!,
             symbol.getSignature(psiElement.project),
             structDeclaration,
-            "Cpp",
+            "c++",
             returnType,
             parameterList,
             includeClassContext,

cpp/src/main/kotlin/cc/unitmesh/cpp/context/CppVariableContextBuilder.kt

@@ -0,0 +1,42 @@
+package cc.unitmesh.cpp.context
+
+import cc.unitmesh.devti.context.VariableContext
+import cc.unitmesh.devti.context.builder.VariableContextBuilder
+import com.intellij.psi.PsiElement
+import com.jetbrains.cidr.lang.psi.*
+import com.intellij.psi.util.PsiTreeUtil
+
+class CppVariableContextBuilder : VariableContextBuilder {
+    override fun getVariableContext(
+        psiElement: PsiElement,
+        includeMethodContext: Boolean,
+        includeClassContext: Boolean,
+        gatherUsages: Boolean
+    ): VariableContext? {
+        if (psiElement !is OCDeclaration) return null
+
+        val declarators = psiElement.declarators
+        val symbols = declarators.filterIsInstance<OCDeclarator>()
+            .mapNotNull { it.symbol }
+            .filter { it.kind.isVariable }
+
+        val variableContexts = symbols.map { declarator ->
+            val text = psiElement.text
+            val enclosingMethod = PsiTreeUtil.getParentOfType(psiElement, OCFunctionDeclaration::class.java, true)
+            val enclosingClass = PsiTreeUtil.getParentOfType(psiElement, OCStructLike::class.java, true)
+
+            VariableContext(
+                psiElement,
+                text,
+                declarator.name,
+                enclosingMethod,
+                enclosingClass,
+                emptyList(),
+                includeMethodContext,
+                includeClassContext
+            )
+        }
+
+        return variableContexts.firstOrNull()
+    }
+}

cpp/src/main/resources/cc.unitmesh.cpp.xml

@@ -13,6 +13,8 @@
                              implementationClass="cc.unitmesh.cpp.context.CppStructContextBuilder"/>
         <methodContextBuilder language="ObjectiveC"
                               implementationClass="cc.unitmesh.cpp.context.CppMethodContextBuilder"/>
+        <variableContextBuilder language="ObjectiveC"
+                                implementationClass="cc.unitmesh.cpp.context.CppVariableContextBuilder"/>
 
         <chatContextProvider implementation="cc.unitmesh.cpp.provider.CLionWorkspaceContextProvider"/>
     </extensions>

cpp/src/test/kotlin/cc/unitmesh/cpp/context/CppMethodContextBuilderTest.kt

@@ -29,15 +29,17 @@ class CppMethodContextBuilderTest : BasePlatformTestCase() {
         val type = PsiTreeUtil.getChildrenOfTypeAsList(decl, OCTypeElement::class.java).first()
         val clz = PsiTreeUtil.getChildrenOfTypeAsList(type, OCStructLike::class.java).first()
         val function = PsiTreeUtil.getChildrenOfTypeAsList(clz, OCFunctionDeclaration::class.java).first()
-        println(function.text)
 
 //        // when
-//        val result = CppMethodContextBuilder()
-//            .getMethodContext(function, true, true)!!
+        val result = CppMethodContextBuilder()
+            .getMethodContext(function, false, true)!!
 //
 //        // then
-//        assertEquals("drive", result.name)
-//        assertEquals(1, result.paramNames)
-//        assertEquals(result.format(), "")
+        assertEquals("drive", result.name)
+        assertEquals("distance", result.paramNames.joinToString(", "))
+        assertEquals(result.format(), """path: /src/Car.h
+language: c++
+fun name: drive
+fun signature: void drive(int distance)""")
     }
 }

cpp/src/test/resources/META-INF/plugin.xml

@@ -0,0 +1,9 @@
+<idea-plugin package="cc.unitmesh" xmlns:xi="http://www.w3.org/2001/XInclude">
+    <id>cc.unitmesh.devti</id>
+    <xi:include href="/META-INF/autodev-core.xml" xpointer="xpointer(/idea-plugin/*)"/>
+
+    <!--suppress PluginXmlValidity -->
+    <content>
+        <module name="cc.unitmesh.cpp"/>
+    </content>
+</idea-plugin>