feat(rust): add RustVariableContextBuilder

This commit adds a new file `RustVariableContextBuilder.kt` to the `rust/src/main/kotlin/cc/unitmesh/rust/context` directory. The `RustVariableContextBuilder` class implements the `VariableContextBuilder` interface and provides the implementation for the `getVariableContext` method. This method is responsible for generating the variable context for a given `PsiElement` in Rust code. The implementation checks if the `PsiElement` is an instance of `RsLetDecl` and extracts relevant information such as the text, name, parent function, and containing class. The generated `VariableContext` object is then returned. Additionally, the commit modifies the `rust/src/main/resources/cc.unitmesh

Author: phodal

rust/src/main/kotlin/cc/unitmesh/rust/context/RustClassContextBuilder.kt

@@ -11,6 +11,7 @@ import org.rust.lang.core.psi.ext.fields
 class RustClassContextBuilder : ClassContextBuilder {
     override fun getClassContext(psiElement: PsiElement, gatherUsages: Boolean): ClassContext? {
         if (psiElement !is RsStructOrEnumItemElement) return null
+
         when (psiElement) {
             is RsStructItem -> {
                 val fields: List<PsiElement> = psiElement.fields.map {
@@ -29,7 +30,7 @@ class RustClassContextBuilder : ClassContextBuilder {
             }
 
             is RsEnumItem -> {
-
+                // TODO: Implement
             }
         }
 

rust/src/main/kotlin/cc/unitmesh/rust/context/RustVariableContextBuilder.kt

@@ -0,0 +1,35 @@
+package cc.unitmesh.rust.context
+
+import cc.unitmesh.devti.context.VariableContext
+import cc.unitmesh.devti.context.builder.VariableContextBuilder
+import com.intellij.psi.PsiElement
+import com.intellij.psi.util.PsiTreeUtil
+import org.rust.lang.core.psi.RsFunction
+import org.rust.lang.core.psi.RsImplItem
+import org.rust.lang.core.psi.RsLetDecl
+
+class RustVariableContextBuilder : VariableContextBuilder {
+    override fun getVariableContext(
+        psiElement: PsiElement,
+        includeMethodContext: Boolean,
+        includeClassContext: Boolean,
+        gatherUsages: Boolean
+    ): VariableContext? {
+        if (psiElement !is RsLetDecl) return null
+
+        val text = psiElement.text
+        val parentOfType = PsiTreeUtil.getParentOfType(psiElement, RsFunction::class.java, true)
+        val containingClass = PsiTreeUtil.getParentOfType(psiElement, RsImplItem::class.java, true)
+
+        return VariableContext(
+            psiElement,
+            text,
+            psiElement.name,
+            parentOfType,
+            containingClass,
+            emptyList(),
+            includeClassContext
+        )
+    }
+
+}

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

@@ -12,6 +12,8 @@
                              implementationClass="cc.unitmesh.rust.context.RustClassContextBuilder"/>
         <methodContextBuilder language="Rust"
                               implementationClass="cc.unitmesh.rust.context.RustMethodContextBuilder"/>
+        <variableContextBuilder language="Rust"
+                                implementationClass="cc.unitmesh.rust.context.RustVariableContextBuilder"/>
 
         <contextPrompter language="Rust" implementation="cc.unitmesh.rust.provider.RustContextPrompter"/>