From 905306ae4953284447c17ceb2b57a3750a658ab6 Mon Sep 17 00:00:00 2001
From: Phodal Huang <h@phodal.com>
Date: Thu, 20 Jul 2023 15:45:28 +0800
Subject: [PATCH] fix: fix not controller has simliary chunk issues

---
 .../idea/provider/JavaContextPrompter.kt         | 16 +++++++++-------
 1 file changed, 9 insertions(+), 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
index 1a97209041..4fd75ba118 100644
--- 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,7 +111,7 @@ class JavaContextPrompter : ContextPrompter() {
                 }
 
                 when {
-                    isController -> {
+                    isController() -> {
                         val spec = PromptConfig.load().spec["controller"]
                         if (!spec.isNullOrEmpty()) {
                             additionContext = "requirements: \n$spec"
@@ -117,7 +119,7 @@ class JavaContextPrompter : ContextPrompter() {
                         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()
             }
         }