feat: add for prompt by text

Author: phodal

src/main/kotlin/cc/unitmesh/devti/connector/custom/CustomConnector.kt

@@ -2,10 +2,36 @@ package cc.unitmesh.devti.connector.custom
 
 import cc.unitmesh.devti.connector.CodeCopilot
 import cc.unitmesh.devti.settings.DevtiSettingsState
+import kotlinx.serialization.decodeFromString
+import kotlinx.serialization.json.Json
+
 
 class CustomConnector : CodeCopilot {
-    val url = DevtiSettingsState.getInstance()?.customEngineServer ?: ""
-    val key = DevtiSettingsState.getInstance()?.customEngineToken ?: ""
+    val devtiSettingsState = DevtiSettingsState.getInstance()
+
+    val url = devtiSettingsState?.customEngineServer ?: ""
+    val key = devtiSettingsState?.customEngineToken ?: ""
+    var promptConfig: PromptConfig? = null
+
+    init {
+        val prompts = devtiSettingsState?.customEnginePrompts
+        try {
+            if (prompts != null) {
+                promptConfig = Json.decodeFromString(prompts)
+            }
+        } catch (e: Exception) {
+            println("Error parsing prompts: $e")
+        }
+
+        if (promptConfig == null) {
+            promptConfig = PromptConfig(
+                PromptItem("Auto complete", "{code}"),
+                PromptItem("Auto comment", "{code}"),
+                PromptItem("Code review", "{code}"),
+                PromptItem("Find bug", "{code}")
+            )
+        }
+    }
 
     private fun prompt(instruction: String, input: String): String {
 //        val retrofit = Retrofit.Builder()
@@ -17,19 +43,23 @@ class CustomConnector : CodeCopilot {
     }
 
     override fun codeCompleteFor(text: String, className: String): String {
-        return ""
+        val complete = promptConfig!!.autoComplete
+        return prompt(complete.instruction, complete.input.replace("{code}", text))
     }
 
     override fun autoComment(text: String): String {
-        return ""
+        val comment = promptConfig!!.autoComment
+        return prompt(comment.instruction, comment.input.replace("{code}", text))
     }
 
     override fun codeReviewFor(text: String): String {
-        return ""
+        val review = promptConfig!!.codeReview
+        return prompt(review.instruction, review.input.replace("{code}", text))
     }
 
     override fun findBug(text: String): String {
-        return ""
+        val bug = promptConfig!!.findBug
+        return prompt(bug.instruction, bug.input.replace("{code}", text))
     }
 
 }

src/main/kotlin/cc/unitmesh/devti/settings/DevtiSettingsConfigurable.kt

@@ -32,7 +32,7 @@ class DevtiSettingsConfigurable : Configurable {
         modified = modified or (!component!!.aiEngine.equals(settings.aiEngine))
         modified = modified or (!component!!.customEngineServer.equals(settings.customEngineServer))
         modified = modified or (!component!!.customEngineToken.equals(settings.customEngineToken))
-        modified = modified or (!component!!.customEnginePrompt.equals(settings.customEnginePrompt))
+        modified = modified or (!component!!.customEnginePrompt.equals(settings.customEnginePrompts))
         return modified
     }
 
@@ -45,7 +45,7 @@ class DevtiSettingsConfigurable : Configurable {
         settings.aiEngine = component!!.getAiEngine()
         settings.customEngineServer = component!!.getCustomEngineServer()
         settings.customEngineToken = component!!.getCustomEngineToken()
-        settings.customEnginePrompt = component!!.getCustomEnginePrompt()
+        settings.customEnginePrompts = component!!.getCustomEnginePrompt()
     }
 
     override fun reset() {
@@ -57,7 +57,7 @@ class DevtiSettingsConfigurable : Configurable {
         component!!.setAiEngine(settings.aiEngine)
         component!!.setCustomEngineServer(settings.customEngineServer)
         component!!.setCustomEngineToken(settings.customEngineToken)
-        component!!.setCustomEnginePrompt(settings.customEnginePrompt)
+        component!!.setCustomEnginePrompt(settings.customEnginePrompts)
     }
 
     override fun disposeUIResources() {

src/main/kotlin/cc/unitmesh/devti/settings/DevtiSettingsState.kt

@@ -17,15 +17,15 @@ class DevtiSettingsState : PersistentStateComponent<DevtiSettingsState> {
     var customOpenAiHost = ""
     var customEngineServer = ""
     var customEngineToken = ""
-    var customEnginePrompt = ""
+    var customEnginePrompts = ""
 
     override fun getState(): DevtiSettingsState {
         return this
     }
 
     override fun loadState(state: DevtiSettingsState) {
-        if (customEnginePrompt == "") {
-            customEnginePrompt = DEFAULT_PROMPTS.trimIndent()
+        if (customEnginePrompts == "") {
+            customEnginePrompts = DEFAULT_PROMPTS.trimIndent()
         }
 
         XmlSerializerUtil.copyBean(state, this)