feat(rust): add support for EnumContext

Author: phodal

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

@@ -12,6 +12,8 @@ import org.rust.lang.core.psi.RsStructItem
 import org.rust.lang.core.psi.ext.RsStructOrEnumItemElement
 import org.rust.lang.core.psi.ext.expandedFields
 import org.rust.lang.core.psi.ext.implementingType
+import org.rust.lang.core.types.asTy
+import org.rust.lang.core.types.implLookup
 
 class RustClassContextBuilder : ClassContextBuilder {
     override fun getClassContext(psiElement: PsiElement, gatherUsages: Boolean): ClassContext? {
@@ -61,7 +63,22 @@ class RustClassContextBuilder : ClassContextBuilder {
             }
 
             is RsEnumItem -> {
-                // TODO: Implement
+                val fields = psiElement.enumBody?.enumVariantList?.map { it } ?: emptyList()
+//                val impls = psiElement.implLookup.findImplsAndTraits(psiElement.asTy()).toList()
+//                val methods = impls.flatMap {
+//                    it.implementedTrait?.element?.children?.filterIsInstance<RsFunction>() ?: emptyList()
+//                }
+
+                return ClassContext(
+                    psiElement,
+                    psiElement.text,
+                    psiElement.name,
+                    emptyList(),
+                    fields,
+                    emptyList(),
+                    emptyList(),
+                    psiElement.name
+                )
             }
         }
 

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

@@ -4,6 +4,7 @@ import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.extensions.ExtensionPoint
 import com.intellij.psi.util.PsiTreeUtil
 import com.intellij.testFramework.fixtures.BasePlatformTestCase
+import org.rust.lang.core.psi.RsEnumItem
 import org.rust.lang.core.psi.RsStructItem
 
 class RustClassContextBuilderTest: BasePlatformTestCase() {
@@ -55,5 +56,42 @@ class RustClassContextBuilderTest: BasePlatformTestCase() {
             """.trimIndent()
         )
     }
+
+    fun testShouldHandleForEnum() {
+        val code = myFixture.configureByText(
+            "test.rs", """
+            
+            enum List {
+                Cons(u32, Box<List>),
+                Nil,
+            }
+            
+            impl List {
+                fn new() -> List {
+                    Nil
+                }
+            
+                fn prepend(self, elem: u32) -> List {
+                    Cons(elem, Box::new(self))
+                }
+            }
+            """.trimIndent()
+        )
+
+        val decl = PsiTreeUtil.getChildrenOfTypeAsList(code, RsEnumItem::class.java).first()
+        val result = RustClassContextBuilder().getClassContext(decl, false)!!
+
+        assertEquals("List", result.name)
+        assertEquals(
+            result.format(), """
+            'package: List
+            class List {
+              Cons(u32, Box<List>)
+              Nil
+              
+            }
+            """.trimIndent()
+        )
+    }
 }
 

src/main/kotlin/cc/unitmesh/devti/actions/chat/base/ChatBaseAction.kt

@@ -33,6 +33,7 @@ abstract class ChatBaseAction : AnAction() {
         val file = event.getData(CommonDataKeys.PSI_FILE)
 
         val lineEndOffset = document?.getLineEndOffset(document.getLineNumber(caretModel?.offset ?: 0)) ?: 0
+
         // if selectedText is empty, then we use the cursor position to get the text
         if (prefixText.isEmpty()) {
             prefixText = document?.text?.substring(0, lineEndOffset) ?: ""

src/main/kotlin/cc/unitmesh/devti/actions/quick/QuickAssistantAction.kt

@@ -98,15 +98,14 @@ class QuickAssistantAction : AnAction() {
         val promptInlay: InlayPanel<QuickPromptField>? =
             InlayPanel.add(editor as EditorEx, offset, QuickPromptField())
 
-        promptInlay?.let { doExecute(it, project, editor, element, sourceFile) }
+        promptInlay?.let { doExecute(it, project, editor, element) }
     }
 
     private fun doExecute(
         inlay: InlayPanel<QuickPromptField>,
         project: Project,
         editor: EditorEx,
         element: PsiElement?,
-        sourceFile: PsiFile,
     ) {
         val component = inlay.component
 

src/main/kotlin/cc/unitmesh/devti/actions/template/NewPromptTemplateAction.kt

@@ -11,9 +11,7 @@ import com.intellij.psi.PsiDirectory
 class NewPromptTemplateAction : CreateFileFromTemplateAction(
     "AutoDev Customize", "Creates new AutoDev customize", AutoDevIcons.AI_COPILOT
 ), DumbAware {
-    override fun getDefaultTemplateProperty(): String {
-        return "DefaultAutoDevTemplate"
-    }
+    override fun getDefaultTemplateProperty(): String = "DefaultAutoDevTemplate"
 
     override fun buildDialog(project: Project, psiDir: PsiDirectory, builder: CreateFileFromTemplateDialog.Builder) {
         builder

src/main/kotlin/cc/unitmesh/devti/llms/openai/OpenAIProvider.kt

@@ -74,6 +74,7 @@ class OpenAIProvider(val project: Project) : LLMProvider {
             if (project.coderSetting.state.recordingInLocal) {
                 return project.service<JsonlRecording>()
             }
+
             return EmptyRecording()
         }