fix: fix not controller has simliary chunk issues

unit-mesh · Jul 20, 2023 · 905306a · 905306a
1 parent 9b4677b
commit 905306a
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/idea/src/main/kotlin/cc/unitmesh/idea/provider/JavaContextPrompter.kt b/idea/src/main/kotlin/cc/unitmesh/idea/provider/JavaContextPrompter.kt
@@ -21,12 +21,14 @@ class JavaContextPrompter : ContextPrompter() {
     private var additionContext: String = ""
     private val autoDevSettingsState = AutoDevSettingsState.getInstance()
     private var promptConfig: PromptConfig? = null
-    lateinit var mvcContextService: MvcContextService
+    private lateinit var mvcContextService: MvcContextService
     private var fileName = ""
-    private val isController = fileName.endsWith("Controller.java")
-    private val isService = fileName.endsWith("Service.java") || fileName.endsWith("ServiceImpl.java")
     private lateinit var changeListManager: ChangeListManager
 
+    private fun isController() = fileName.endsWith("Controller.java")
+    private fun isService() = fileName.endsWith("Service.java") || fileName.endsWith("ServiceImpl.java")
+
+
     override fun initContext(actionType: ChatBotActionType, selectedText: String, file: PsiFile?, project: Project, offset: Int) {
         super.initContext(actionType, selectedText, file, project, offset)
         changeListManager = ChangeListManagerImpl.getInstance(project)
@@ -109,15 +111,15 @@ class JavaContextPrompter : ContextPrompter() {
                 }
 
                 when {
-                    isController -> {
+                    isController() -> {
                         val spec = PromptConfig.load().spec["controller"]
                         if (!spec.isNullOrEmpty()) {
                             additionContext = "requirements: \n$spec"
                         }
                         additionContext += mvcContextService.controllerPrompt(file)
                     }
 
-                    isService -> {
+                    isService() -> {
                         val spec = PromptConfig.load().spec["service"]
                         if (!spec.isNullOrEmpty()) {
                             additionContext = "requirements: \n$spec"
@@ -191,11 +193,11 @@ examples:
         val techStackProvider = TechStackProvider.stack(file?.language?.displayName ?: "")
         val techStacks = techStackProvider!!.prepareLibrary()
         when {
-            isController -> {
+            isController() -> {
                 additionContext = "// tech stacks: " + techStacks.controllerString()
             }
 
-            isService -> {
+            isService() -> {
                 additionContext = "// tech stacks: " + techStacks.serviceString()
             }
         }