feat(rust): init context

unit-mesh · Aug 23, 2023 · 4e7cd19 · 4e7cd19
1 parent 4489358
commit 4e7cd19
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 2 deletions.
diff --git a/rust/src/main/kotlin/cc/unitmesh/rust/context/RustClassContextBuilder.kt b/rust/src/main/kotlin/cc/unitmesh/rust/context/RustClassContextBuilder.kt
@@ -12,7 +12,15 @@ class RustClassContextBuilder : ClassContextBuilder {
         if (psiElement !is RsStructOrEnumItemElement) return null
         when (psiElement) {
             is RsStructItem -> {
-                // TODO: Implement
+                return ClassContext(
+                    psiElement,
+                    psiElement.text,
+                    psiElement.name,
+                    emptyList(),
+                    emptyList(),
+                    emptyList(),
+                    emptyList()
+                )
             }
 
             is RsEnumItem -> {

diff --git a/rust/src/main/kotlin/cc/unitmesh/rust/provider/RustContextProvider.kt b/rust/src/main/kotlin/cc/unitmesh/rust/provider/RustContextProvider.kt
@@ -0,0 +1,16 @@
+package cc.unitmesh.rust.provider
+
+import cc.unitmesh.devti.provider.context.ChatContextItem
+import cc.unitmesh.devti.provider.context.ChatContextProvider
+import cc.unitmesh.devti.provider.context.ChatCreationContext
+import com.intellij.openapi.project.Project
+
+class RustContextProvider : ChatContextProvider {
+    override fun isApplicable(project: Project, creationContext: ChatCreationContext): Boolean {
+        return true
+    }
+
+    override suspend fun collect(project: Project, creationContext: ChatCreationContext): List<ChatContextItem> {
+        return listOf()
+    }
+}
diff --git a/rust/src/main/resources/cc.unitmesh.rust.xml b/rust/src/main/resources/cc.unitmesh.rust.xml
@@ -15,6 +15,6 @@
 
         <contextPrompter language="Rust" implementation="cc.unitmesh.rust.provider.RustContextPrompter"/>
 
-<!--        <chatContextProvider implementation="cc.unitmesh.rust.provider.RustContextProvider"/>-->
+        <chatContextProvider implementation="cc.unitmesh.rust.provider.RustContextProvider"/>
     </extensions>
 </idea-plugin>