Enhance SQL formatting for DELETE conditions with subqueries

Author: xterao

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

@@ -357,6 +357,25 @@ open class SqlBlock(
         return builder
     }
 
+    protected fun calculatePrevBlocksLength(
+        children: List<SqlBlock>,
+        parent: SqlBlock,
+    ): Int =
+        children
+            .sumOf { prev ->
+                prev
+                    .getChildrenTextLen()
+                    .plus(
+                        if (prev.node.elementType == SqlTypes.DOT ||
+                            prev.node.elementType == SqlTypes.RIGHT_PAREN
+                        ) {
+                            0
+                        } else {
+                            prev.getNodeText().length.plus(1)
+                        },
+                    )
+            }.plus(parent.indent.groupIndentLen)
+
     /**
      * Returns the child indentation for the block.
      *

src/main/kotlin/org/domaframework/doma/intellij/formatter/block/group/keyword/condition/SqlConditionKeywordGroupBlock.kt

@@ -19,6 +19,7 @@ import com.intellij.lang.ASTNode
 import org.domaframework.doma.intellij.formatter.block.SqlBlock
 import org.domaframework.doma.intellij.formatter.block.comment.SqlElConditionLoopCommentBlock
 import org.domaframework.doma.intellij.formatter.block.group.keyword.option.SqlSecondOptionKeywordGroupBlock
+import org.domaframework.doma.intellij.formatter.block.group.keyword.second.SqlWhereGroupBlock
 import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlSubGroupBlock
 import org.domaframework.doma.intellij.formatter.util.SqlBlockFormattingContext
 
@@ -63,4 +64,14 @@ class SqlConditionKeywordGroupBlock(
             }
         } ?: return 1
     }
+
+    override fun createGroupIndentLen(): Int {
+        parentBlock?.let { parent ->
+            if (parent is SqlWhereGroupBlock) {
+                return indent.indentLen.plus(getNodeText().length)
+            }
+            return super.createGroupIndentLen()
+        }
+        return 0
+    }
 }

src/main/kotlin/org/domaframework/doma/intellij/formatter/block/group/keyword/condition/SqlConditionalExpressionGroupBlock.kt

@@ -62,7 +62,7 @@ class SqlConditionalExpressionGroupBlock(
                     }
                 groupIndentLen + directiveParentTextLen
             } else {
-                parent.indent.groupIndentLen.plus(1)
+                calculatePrevBlocksLength(prevBlocks, parent).plus(1)
             }
         }
             ?: offset

src/main/kotlin/org/domaframework/doma/intellij/formatter/block/group/keyword/option/SqlInGroupBlock.kt

@@ -56,17 +56,9 @@ class SqlInGroupBlock(
                 }
 
             val dotCount = sumChildren.count { it.node.elementType == SqlTypes.DOT }
-            val parentText = (parent as? SqlElConditionLoopCommentBlock)?.parentBlock?.getNodeText()?.length ?: 0
+            val parentText = prevChildren.dropLast(1).filter { it !is SqlDefaultCommentBlock }
 
-            return sumChildren
-                .sumOf { prev ->
-                    prev
-                        .getChildrenTextLen()
-                        .plus(prev.getNodeText().length.plus(1))
-                }.minus(dotCount * 2)
-                .plus(parent.indent.groupIndentLen)
-                .plus(parentText)
-                .plus(1)
+            return calculatePrevBlocksLength(parentText, parent)
         }
         return 0
     }

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

@@ -50,25 +50,6 @@ class SqlFunctionGroupBlock(
         return baseIndent.plus(getNodeText().length)
     }
 
-    private fun calculatePrevBlocksLength(
-        children: List<SqlBlock>,
-        parent: SqlBlock,
-    ): Int =
-        children
-            .sumOf { prev ->
-                prev
-                    .getChildrenTextLen()
-                    .plus(
-                        if (prev.node.elementType == SqlTypes.DOT ||
-                            prev.node.elementType == SqlTypes.RIGHT_PAREN
-                        ) {
-                            0
-                        } else {
-                            prev.getNodeText().length.plus(1)
-                        },
-                    )
-            }.plus(parent.indent.groupIndentLen)
-
     private fun calculateBaseIndent(
         parent: SqlBlock,
         prevBlocksLength: Int,

src/test/kotlin/org/domaframework/doma/intellij/formatter/SqlFormatterTest.kt

@@ -166,6 +166,10 @@ class SqlFormatterTest : BasePlatformTestCase() {
         formatSqlFile("WithDelete.sql", "WithDelete$formatDataPrefix.sql")
     }
 
+    fun testDeleteWithSubQuery() {
+        formatSqlFile("DeleteWithSubQuery.sql", "DeleteWithSubQuery$formatDataPrefix.sql")
+    }
+
     fun testNestedDirectivesFormatter() {
         formatSqlFile("NestedDirectives.sql", "NestedDirectives$formatDataPrefix.sql")
     }

src/test/testData/sql/formatter/DeleteWithSubQuery.sql

@@ -0,0 +1,11 @@
+DELETE FROM user_session s
+ WHERE s.count = ( SELECT COUNT(*)
+                                FROM user u
+                                  WHERE u.id = /* id */1
+                                     AND u.session_id = u.id
+                                          AND u.time_stamp < /* current */'2099-12-31 00:00:00' )
+    OR EXISTS ( SELECT u.id
+       FROM user u
+     WHERE u.id = /* id */1
+AND u.session_id = u.id
+AND u.time_stamp < /* current */'2099-12-31 00:00:00')

src/test/testData/sql/formatter/DeleteWithSubQuery_format.sql

@@ -0,0 +1,11 @@
+DELETE FROM user_session s
+ WHERE s.count = ( SELECT COUNT(*)
+                     FROM user u
+                    WHERE u.id = /* id */1
+                      AND u.session_id = u.id
+                      AND u.time_stamp < /* current */'2099-12-31 00:00:00' )
+    OR EXISTS ( SELECT u.id
+                  FROM user u
+                 WHERE u.id = /* id */1
+                   AND u.session_id = u.id
+                   AND u.time_stamp < /* current */'2099-12-31 00:00:00' )