feat(rust): add RustClassContextBuilderTest and modify RustClassContextBuilder

- Add RustClassContextBuilderTest to test the functionality of the RustClassContextBuilder class.
- Modify RustClassContextBuilder to include functions in the ClassContext and remove unnecessary code.
- The test case in RustClassContextBuilderTest checks if the struct is formatted correctly and returns the expected result.

Author: phodal

prompts/templates/Test.kt.vm

@@ -1,5 +1,6 @@
-// Here is the Test template code.
-// for test intellij plugin
+// Here is the Test template code, please copy and paste to the test file
+// and replace the class name and the test method name
+//
 //import com.intellij.testFramework.LightPlatformTestCase
 //class /*TestClassName*/Test : LightPlatformTestCase() {
 //    // the Intellij test should start with test

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

@@ -3,7 +3,10 @@ package cc.unitmesh.rust.context
 import cc.unitmesh.devti.context.ClassContext
 import cc.unitmesh.devti.context.builder.ClassContextBuilder
 import com.intellij.psi.PsiElement
+import com.intellij.psi.util.PsiTreeUtil
 import org.rust.lang.core.psi.RsEnumItem
+import org.rust.lang.core.psi.RsFunction
+import org.rust.lang.core.psi.RsImplItem
 import org.rust.lang.core.psi.RsStructItem
 import org.rust.lang.core.psi.ext.RsStructOrEnumItemElement
 import org.rust.lang.core.psi.ext.fields
@@ -15,13 +18,17 @@ class RustClassContextBuilder : ClassContextBuilder {
         when (psiElement) {
             is RsStructItem -> {
                 val fields: List<PsiElement> = psiElement.fields.map {
-                    it.typeReference?.reference?.resolve() ?: it
+                    it
                 }
+                val impls = PsiTreeUtil.getChildrenOfTypeAsList(psiElement.containingFile, RsImplItem::class.java)
+                val functions = impls.filter { it.name == psiElement.name }
+                    .flatMap { PsiTreeUtil.getChildrenOfTypeAsList(it, RsFunction::class.java) }
+
                 return ClassContext(
                     psiElement,
                     psiElement.text,
                     psiElement.name,
-                    emptyList(),
+                    functions,
                     fields,
                     emptyList(),
                     emptyList(),

rust/src/test/kotlin/cc/unitmesh/rust/context/RustClassContextBuilderTest.kt

@@ -0,0 +1,54 @@
+package cc.unitmesh.rust.context;
+
+import cc.unitmesh.devti.context.ClassContext
+import cc.unitmesh.devti.context.builder.ClassContextBuilder
+import com.intellij.psi.PsiElement
+import com.intellij.psi.PsiFileFactory
+import com.intellij.psi.util.PsiTreeUtil
+import com.intellij.testFramework.fixtures.BasePlatformTestCase
+import com.jetbrains.cidr.lang.psi.OCDeclaration
+import org.junit.Assert.assertEquals
+import org.junit.Assert.assertNull
+import org.junit.Test
+import org.rust.lang.core.psi.RsEnumItem
+import org.rust.lang.core.psi.RsStructItem
+import org.rust.lang.core.psi.ext.RsStructOrEnumItemElement
+import org.rust.lang.core.psi.ext.fields
+
+class RustClassContextBuilderTest: BasePlatformTestCase() {
+
+    fun testShouldFormatStruct() {
+        // given
+        val code = myFixture.configureByText("test.rs", """
+        use crate::embedding::Embedding;
+        use crate::similarity::{CosineSimilarity, RelevanceScore};
+        
+        #[derive(Debug, Clone)]
+        pub struct Entry {
+            id: String,
+            embedding: Embedding,
+            embedded: Document,
+        }
+        
+        impl Entry {
+            fn new(id: String, embedding: Embedding, embedded: Document) -> Self {
+                Entry { id, embedding, embedded }
+            }
+        }
+        """.trimIndent())
+
+        // when
+        val decl = PsiTreeUtil.getChildrenOfTypeAsList(code, RsStructItem::class.java).first()
+
+        // then
+        val result = RustClassContextBuilder().getClassContext(decl, false)!!
+        assertEquals("Entry", result.name)
+        assertEquals(result.format(), """
+            'package: Entry
+            class Entry {
+              
+              
+            }
+            """.trimIndent())
+    }
+}