feat: update for context

unit-mesh · Jul 26, 2023 · d79a6f6 · d79a6f6
1 parent 0182c29
commit d79a6f6
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 11 deletions.
diff --git a/.../provider/JavaTechStackContextProvider.kt → .../provider/JavaTechStackContextProvider.kt b/.../provider/JavaTechStackContextProvider.kt → .../provider/JavaTechStackContextProvider.kt
@@ -1,4 +1,4 @@
-package cc.unitmesh.idea.provider
+package cc.unitmesh.idea.java.provider
 
 import cc.unitmesh.devti.prompting.code.TestStack
 import cc.unitmesh.devti.provider.context.ChatContextItem
@@ -12,7 +12,9 @@ import com.intellij.openapi.project.ProjectManager
 import com.intellij.psi.PsiJavaFile
 import org.jetbrains.plugins.gradle.util.GradleConstants
 
-class JavaTechStackContextProvider : ChatContextProvider {
+open class JavaTechStackContextProvider : ChatContextProvider {
+    open val fileExt = "java"
+
     override fun isApplicable(project: Project, creationContext: ChatCreationContext): Boolean {
         logger<JavaTechStackContextProvider>().warn("psiFile: $creationContext")
         val psiFile = creationContext.sourceFile ?: return false
@@ -28,9 +30,9 @@ class JavaTechStackContextProvider : ChatContextProvider {
 
         val fileName = creationContext.sourceFile?.name ?: ""
 
-        fun isController() = fileName.endsWith("Controller.java")
+        fun isController() = fileName.endsWith("Controller.$fileExt")
         fun isService() =
-            fileName.endsWith("Service.java") || fileName.endsWith("ServiceImpl.java")
+            fileName.endsWith("Service.$fileExt") || fileName.endsWith("ServiceImpl.$fileExt")
 
         when {
             isController() -> {

diff --git a/...mesh/idea/provider/JavaVersionProvider.kt → ...idea/java/provider/JavaVersionProvider.kt b/...mesh/idea/provider/JavaVersionProvider.kt → ...idea/java/provider/JavaVersionProvider.kt
@@ -1,4 +1,4 @@
-package cc.unitmesh.idea.provider
+package cc.unitmesh.idea.java.provider
 
 import cc.unitmesh.devti.provider.context.ChatContextItem
 import cc.unitmesh.devti.provider.context.ChatContextProvider

diff --git a/...sh/idea/provider/KotlinVersionProvider.kt → .../kotlin/provider/KotlinVersionProvider.kt b/...sh/idea/provider/KotlinVersionProvider.kt → .../kotlin/provider/KotlinVersionProvider.kt
@@ -1,4 +1,4 @@
-package cc.unitmesh.idea.provider
+package cc.unitmesh.idea.kotlin.provider
 
 import cc.unitmesh.devti.provider.context.ChatContextItem
 import cc.unitmesh.devti.provider.context.ChatContextProvider
@@ -16,7 +16,4 @@ class KotlinVersionProvider : ChatContextProvider {
     override fun collect(project: Project, creationContext: ChatCreationContext): List<ChatContextItem> {
         return emptyList()
     }
-
-    private fun isKotlinFile(psiFile: PsiFile?) =
-        psiFile?.containingFile?.virtualFile?.extension?.equals("kt", true) ?: false
 }
diff --git a/idea/src/main/resources/cc.unitmesh.idea.xml b/idea/src/main/resources/cc.unitmesh.idea.xml
@@ -18,8 +18,8 @@
         <variableContextBuilder language="JAVA"
                                 implementationClass="cc.unitmesh.idea.java.context.JavaVariableContextBuilder"/>
 
-        <chatContextProvider implementation="cc.unitmesh.idea.provider.JavaVersionProvider"/>
-        <chatContextProvider implementation="cc.unitmesh.idea.provider.JavaTechStackContextProvider"/>
+        <chatContextProvider implementation="cc.unitmesh.idea.java.provider.JavaVersionProvider"/>
+        <chatContextProvider implementation="cc.unitmesh.idea.java.provider.JavaTechStackContextProvider"/>
 
         <contextPrompter
                 language="JAVA"

diff --git a/idea/src/main/resources/cc.unitmesh.kotlin.xml b/idea/src/main/resources/cc.unitmesh.kotlin.xml
@@ -18,6 +18,8 @@
         <variableContextBuilder language="kotlin"
                                 implementationClass="cc.unitmesh.idea.kotlin.context.KotlinVariableContextBuilder"/>
 
+        <chatContextProvider implementation="cc.unitmesh.idea.kotlin.provider.KotlinVersionProvider"/>
+
 <!--        <contextPrompter-->
 <!--                language="Kotlin"-->
 <!--                implementation="cc.unitmesh.idea.provider.JvmIdeaContextPrompter"/>-->