feat(java): add code prompter methods and documentation

This commit adds new methods and documentation to the JavaContextPrompter and JvmPromptStrategy classes. The JavaContextPrompter class now has a private additionContext property, and the JvmPromptStrategy class has a new getByFields method that retrieves lines matching a specified field regex. Additionally, the JvmPromptStrategy class now has a new advice method that generates code prompt text for implementing missing methods in a service file. These changes improve the functionality and usability of the code prompter feature in the Java context.

Author: phodal

java/src/main/kotlin/cc/unitmesh/idea/context/JavaCodeModifier.kt

@@ -74,10 +74,14 @@ open class JavaCodeModifier : CodeModifier {
         }
 
         WriteCommandAction.runWriteCommandAction(project) {
-            val classEndOffset = rootElement.textRange.endOffset
-            val document = PsiDocumentManager.getInstance(project).getDocument(rootElement.containingFile)
-            document?.insertString(classEndOffset - 1, "\n    ")
-            document?.insertString(classEndOffset - 1 + "\n    ".length, newTestMethod.text)
+            try {
+                rootElement.add(newTestMethod)
+            } catch (e: Throwable) {
+                val classEndOffset = rootElement.textRange.endOffset
+                val document = PsiDocumentManager.getInstance(project).getDocument(rootElement.containingFile)
+                document?.insertString(classEndOffset - 1, "\n    ")
+                document?.insertString(classEndOffset - 1 + "\n    ".length, newTestMethod.text)
+            }
         }
 
         project.guessProjectDir()?.refresh(true, true)

java/src/main/kotlin/cc/unitmesh/idea/prompting/JavaContextPrompter.kt

@@ -21,7 +21,7 @@ import com.intellij.temporary.similar.chunks.SimilarChunksWithPaths
 import kotlinx.coroutines.runBlocking
 
 open class JavaContextPrompter : ContextPrompter() {
-    protected var additionContext: String = ""
+    private var additionContext: String = ""
     protected open val testDataBuilder: TestDataBuilder = JavaTestDataBuilder()
 
     private val autoDevSettingsState = AutoDevSettingsState.getInstance()

java/src/main/kotlin/cc/unitmesh/idea/prompting/JvmPromptStrategy.kt

@@ -120,7 +120,13 @@ class JvmPromptStrategy : PromptStrategy() {
         return methodCodes
     }
 
-    // get all line when match by field usage (by Regex)
+    /**
+     * Retrieves all lines from the given code string that match the specified field regex.
+     *
+     * @param codeString The code string to search for field usage.
+     * @param firstFieldRegex The regular expression pattern to match the first field usage in each line.
+     * @return A string containing all the lines that match the field regex, with additional code added before each line.
+     */
     private fun getByFields(
         codeString: @NlsSafe String,
         firstFieldRegex: Regex
@@ -144,6 +150,14 @@ class JvmPromptStrategy : PromptStrategy() {
         }
     }
 
+    /**
+     * Generates a code prompt text for implementing missing methods in a given service file.
+     *
+     * @param serviceFile the PsiJavaFile representing the service file
+     * @param usedMethod a list of used methods in the service file
+     * @param noExistMethods a list of methods that do not exist in the service file
+     * @return a CodePromptText object containing the generated code prompt text
+     */
     fun advice(serviceFile: PsiJavaFile, usedMethod: List<String>, noExistMethods: List<String>): CodePromptText {
         val code = serviceFile.text
         val filterNeedImplementMethods = usedMethod.filter {