Do not re-execute PreProcessor for elements that were already text blocks.

Author: xterao

src/main/kotlin/org/domaframework/doma/intellij/formatter/processor/SqlInjectionPostProcessor.kt

@@ -19,14 +19,27 @@ import com.intellij.lang.injection.InjectedLanguageManager
 import com.intellij.openapi.util.TextRange
 import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiFile
-import com.intellij.psi.PsiLiteralExpression
 import com.intellij.psi.codeStyle.CodeStyleSettings
 import org.domaframework.doma.intellij.common.dao.getDaoClass
 import org.domaframework.doma.intellij.common.isJavaOrKotlinFileType
 import org.domaframework.doma.intellij.common.isSupportFileType
+import org.domaframework.doma.intellij.common.util.InjectionSqlUtil
 import org.domaframework.doma.intellij.formatter.visitor.DaoInjectionSqlVisitor
 import org.domaframework.doma.intellij.formatter.visitor.FormattingTask
 
+/**
+ * Post-processor for SQL injection formatting.
+ *
+ * This processor handles SQL formatting in two contexts:
+ * 1. **File formatting**: When formatting entire DAO files (Java/Kotlin) containing SQL annotations
+ * 2. **Code formatting**: When formatting injected SQL fragments within string literals
+ *
+ * The context is determined by checking:
+ * - If the source is an injected fragment (`InjectedLanguageManager.isInjectedFragment()` returns true),
+ *   it's being called from code formatting for a specific SQL string literal
+ * - If the source is a regular DAO file (Java/Kotlin with @Dao annotation),
+ *   it's being called from file formatting to process all SQL strings in the file
+ */
 class SqlInjectionPostProcessor : SqlPostProcessor() {
     override fun processElement(
         element: PsiElement,
@@ -38,49 +51,53 @@ class SqlInjectionPostProcessor : SqlPostProcessor() {
         rangeToReformat: TextRange,
         settings: CodeStyleSettings,
     ): TextRange {
-        if (!shouldProcessFile(source)) {
+        if (!isProcessFile(source)) {
             return rangeToReformat
         }
 
         val manager = InjectedLanguageManager.getInstance(source.project)
         if (manager.isInjectedFragment(source)) {
-            processInjectedFragment(source, manager)
+            processInjectedFragment(source)
         } else {
             processRegularFile(source)
         }
 
         return rangeToReformat
     }
 
-    private fun shouldProcessFile(source: PsiFile): Boolean {
+    private fun isProcessFile(source: PsiFile): Boolean {
         val manager = InjectedLanguageManager.getInstance(source.project)
         val isInjectedSql = if (isSupportFileType(source)) manager.isInjectedFragment(source) else false
         val isDaoFile = isJavaOrKotlinFileType(source) && getDaoClass(source) != null
 
         return isInjectedSql || isDaoFile
     }
 
-    private fun processInjectedFragment(
-        source: PsiFile,
-        manager: InjectedLanguageManager,
-    ) {
-        val host = manager.getInjectionHost(source) as? PsiLiteralExpression ?: return
+    /**
+     * Processes all SQL injections in a DAO file during file formatting.
+     * This is called when formatting an entire DAO file containing SQL annotations.
+     */
+    private fun processInjectedFragment(source: PsiFile) {
+        val host = InjectionSqlUtil.getLiteralExpressionHost(source) ?: return
         val originalText = host.value?.toString() ?: return
 
         val visitor = DaoInjectionSqlVisitor(source.project)
-        val formattingTask = FormattingTask(host, originalText)
+        val formattingTask = FormattingTask(host, originalText, host.isTextBlock)
 
         visitor.convertExpressionToTextBlock(formattingTask.expression)
         visitor.processFormattingTask(formattingTask) { text ->
             processDocumentText(text)
         }
     }
 
+    /**
+     * Processes injected SQL fragments during code formatting.
+     * This is called when formatting a specific SQL string literal within a DAO file.
+     */
     private fun processRegularFile(source: PsiFile) {
         val visitor = DaoInjectionSqlVisitor(source.project)
         source.accept(visitor)
         visitor.processAllTextBlock()
-        visitor.initFormattingTasks()
         source.accept(visitor)
         visitor.processAllReFormat { text ->
             processDocumentText(text)