Improve SQL block comment formatting and handling for parser level comments

Author: xterao

src/main/kotlin/org/domaframework/doma/intellij/formatter/block/SqlFileBlock.kt

@@ -24,8 +24,11 @@ import com.intellij.formatting.Spacing
 import com.intellij.formatting.SpacingBuilder
 import com.intellij.formatting.Wrap
 import com.intellij.lang.ASTNode
+import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiWhiteSpace
 import com.intellij.psi.formatter.common.AbstractBlock
+import com.intellij.psi.util.PsiTreeUtil
+import com.intellij.psi.util.elementType
 import org.domaframework.doma.intellij.common.util.TypeUtil
 import org.domaframework.doma.intellij.formatter.block.comma.SqlCommaBlock
 import org.domaframework.doma.intellij.formatter.block.comment.SqlCommentBlock
@@ -262,15 +265,23 @@ open class SqlFileBlock(
                 child,
                 createBlockDirectiveCommentSpacingBuilder(),
             )
-        if (tempBlock !is SqlElConditionLoopCommentBlock) {
-            if (lastGroup is SqlWithQueryGroupBlock || lastGroupFilteredDirective is SqlWithQueryGroupBlock) {
-                return SqlWithCommonTableGroupBlock(child, defaultFormatCtx)
+        val commentType = PsiTreeUtil.getChildrenOfType(child.psi, PsiElement::class.java)?.get(1)?.elementType
+        val defaultBlockComment =
+            commentType == SqlTypes.EL_PARSER_LEVEL_COMMENT || commentType == SqlTypes.BLOCK_COMMENT_CONTENT ||
+                child.elementType == SqlTypes.LINE_COMMENT
+        if (!defaultBlockComment) {
+            if (tempBlock !is SqlElConditionLoopCommentBlock) {
+                if (lastGroup is SqlWithQueryGroupBlock || lastGroupFilteredDirective is SqlWithQueryGroupBlock) {
+                    return SqlWithCommonTableGroupBlock(child, defaultFormatCtx)
+                }
+            }
+            return if (lastGroup is SqlWithCommonTableGroupBlock) {
+                SqlWithCommonTableGroupBlock(child, defaultFormatCtx)
+            } else {
+                tempBlock
             }
-        }
-        return if (lastGroup is SqlWithCommonTableGroupBlock) {
-            SqlWithCommonTableGroupBlock(child, defaultFormatCtx)
         } else {
-            tempBlock
+            return tempBlock
         }
     }
 

src/test/testData/sql/formatter/BlockCommentParse.sql

@@ -1,8 +1,9 @@
-/*%! This is Parser Level Comment*/
+
+ /*%! This is Parser Level Comment*/
 select * from table1
 /*%!This is Parser Level Comment*/
 where column1 = /*# value1*/
 /*+
    * This is Block Comment
    */
-and column2 = /* value2 */ 'value2'     or column3 = /*^ value3 */ 'value3'
+and column2 = /* value2 */ 'value2' /** line break */    or column3 = /*^ value3 */ 'value3'

src/test/testData/sql/formatter/BlockCommentParse_format.sql

@@ -7,4 +7,5 @@ SELECT *
    * This is Block Comment
    */
    AND column2 = /* value2 */'value2'
+    /** line break */
     OR column3 = /*^ value3 */'value3'