Skip to content

Commit 3e117bd

Browse files
xteraoxterao
committed
Deleted line breaks in list test data and organized the code
1 parent 111e528 commit 3e117bd

File tree

11 files changed

+474
-290
lines changed

11 files changed

+474
-290
lines changed

‎src/main/kotlin/org/domaframework/doma/intellij/extension/expr/SqlElExtensions.kt‎

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,15 @@ package org.domaframework.doma.intellij.extension.expr
1818
import com.intellij.psi.PsiElement
1919
import com.intellij.psi.util.PsiTreeUtil
2020
import com.intellij.psi.util.elementType
21+
import org.domaframework.doma.intellij.psi.SqlCustomElCommentExpr
2122
import org.domaframework.doma.intellij.psi.SqlElClass
23+
import org.domaframework.doma.intellij.psi.SqlElElseifDirective
24+
import org.domaframework.doma.intellij.psi.SqlElForDirective
25+
import org.domaframework.doma.intellij.psi.SqlElIfDirective
2226
import org.domaframework.doma.intellij.psi.SqlElPrimaryExpr
2327
import org.domaframework.doma.intellij.psi.SqlElStaticFieldAccessExpr
2428
import org.domaframework.doma.intellij.psi.SqlTypes
29+
import kotlin.invoke
2530

2631
val SqlElStaticFieldAccessExpr.accessElements: List<PsiElement>
2732
get() {
@@ -43,3 +48,13 @@ val SqlElStaticFieldAccessExpr.fqdn: String
4348
val fqdn = PsiTreeUtil.getChildrenOfTypeAsList(elClazz, PsiElement::class.java)
4449
return fqdn.toList().joinToString("") { it.text }
4550
}
51+
52+
fun SqlCustomElCommentExpr.isConditionOrLoopDirective(): Boolean =
53+
PsiTreeUtil.getChildOfType(this, SqlElIfDirective::class.java) != null ||
54+
PsiTreeUtil.getChildOfType(this, SqlElForDirective::class.java) != null ||
55+
PsiTreeUtil.getChildOfType(
56+
this,
57+
SqlElElseifDirective::class.java,
58+
) != null ||
59+
this.findElementAt(2)?.elementType == SqlTypes.EL_END ||
60+
this.findElementAt(2)?.elementType == SqlTypes.EL_ELSE

‎src/main/kotlin/org/domaframework/doma/intellij/formatter/SqlBlockUtil.kt‎

Lines changed: 286 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,286 @@
1+
/*
2+
* Copyright Doma Tools Authors
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License");
5+
* you may not use this file except in compliance with the License.
6+
* You may obtain a copy of the License at
7+
*
8+
* https://www.apache.org/licenses/LICENSE-2.0
9+
*
10+
* Unless required by applicable law or agreed to in writing, software
11+
* distributed under the License is distributed on an "AS IS" BASIS,
12+
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
* See the License for the specific language governing permissions and
14+
* limitations under the License.
15+
*/
16+
package org.domaframework.doma.intellij.formatter
17+
18+
import com.intellij.lang.ASTNode
19+
import com.intellij.psi.PsiComment
20+
import com.intellij.psi.util.PsiTreeUtil
21+
import org.domaframework.doma.intellij.extension.expr.isConditionOrLoopDirective
22+
import org.domaframework.doma.intellij.formatter.block.SqlBlock
23+
import org.domaframework.doma.intellij.formatter.block.SqlBlockCommentBlock
24+
import org.domaframework.doma.intellij.formatter.block.SqlColumnBlock
25+
import org.domaframework.doma.intellij.formatter.block.SqlCommaBlock
26+
import org.domaframework.doma.intellij.formatter.block.SqlCommentBlock
27+
import org.domaframework.doma.intellij.formatter.block.SqlKeywordBlock
28+
import org.domaframework.doma.intellij.formatter.block.SqlTableBlock
29+
import org.domaframework.doma.intellij.formatter.block.SqlWordBlock
30+
import org.domaframework.doma.intellij.formatter.block.expr.SqlElBlockCommentBlock
31+
import org.domaframework.doma.intellij.formatter.block.expr.SqlElConditionLoopCommentBlock
32+
import org.domaframework.doma.intellij.formatter.block.group.SqlColumnDefinitionRawGroupBlock
33+
import org.domaframework.doma.intellij.formatter.block.group.keyword.SqlCreateKeywordGroupBlock
34+
import org.domaframework.doma.intellij.formatter.block.group.keyword.SqlInlineGroupBlock
35+
import org.domaframework.doma.intellij.formatter.block.group.keyword.SqlInlineSecondGroupBlock
36+
import org.domaframework.doma.intellij.formatter.block.group.keyword.SqlInsertKeywordGroupBlock
37+
import org.domaframework.doma.intellij.formatter.block.group.keyword.SqlJoinGroupBlock
38+
import org.domaframework.doma.intellij.formatter.block.group.keyword.SqlKeywordGroupBlock
39+
import org.domaframework.doma.intellij.formatter.block.group.keyword.SqlUpdateKeywordGroupBlock
40+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlColumnDefinitionGroupBlock
41+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlColumnGroupBlock
42+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlDataTypeParamBlock
43+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlFunctionParamBlock
44+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlInsertColumnGroupBlock
45+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlParallelListBlock
46+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlSubQueryGroupBlock
47+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlUpdateColumnGroupBlock
48+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlUpdateValueGroupBlock
49+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlViewGroupBlock
50+
import org.domaframework.doma.intellij.psi.SqlCustomElCommentExpr
51+
import org.domaframework.doma.intellij.psi.SqlTypes
52+
53+
class SqlBlockUtil(
54+
private val sqlBlock: SqlBlock,
55+
) {
56+
val wrap = sqlBlock.wrap
57+
val alignment = sqlBlock.alignment
58+
val spacingBuilder = sqlBlock.spacingBuilder
59+
60+
fun getKeywordBlock(
61+
child: ASTNode,
62+
lastGroupBlock: SqlBlock?,
63+
): SqlBlock {
64+
// Because we haven't yet set the parent-child relationship of the block,
65+
// the parent group references groupTopNodeIndexHistory.
66+
val indentLevel = SqlKeywordUtil.getIndentType(child.text)
67+
val keywordText = child.text.lowercase()
68+
if (indentLevel.isNewLineGroup()) {
69+
when (indentLevel) {
70+
IndentType.JOIN -> {
71+
return if (SqlKeywordUtil.isJoinKeyword(child.text)) {
72+
SqlJoinGroupBlock(child, wrap, alignment, spacingBuilder)
73+
} else if (lastGroupBlock is SqlJoinGroupBlock) {
74+
SqlKeywordBlock(child, IndentType.ATTACHED, wrap, alignment, spacingBuilder)
75+
} else {
76+
SqlJoinGroupBlock(child, wrap, alignment, spacingBuilder)
77+
}
78+
}
79+
80+
IndentType.INLINE_SECOND -> {
81+
return SqlInlineSecondGroupBlock(child, wrap, alignment, spacingBuilder)
82+
}
83+
84+
IndentType.TOP -> {
85+
if (keywordText == "create") {
86+
return SqlCreateKeywordGroupBlock(child, wrap, alignment, spacingBuilder)
87+
}
88+
if (keywordText == "insert") {
89+
return SqlInsertKeywordGroupBlock(child, wrap, alignment, spacingBuilder)
90+
}
91+
92+
return SqlKeywordGroupBlock(child, indentLevel, wrap, alignment, spacingBuilder)
93+
}
94+
95+
IndentType.SECOND -> {
96+
return if (keywordText == "set") {
97+
SqlUpdateKeywordGroupBlock(child, wrap, alignment, spacingBuilder)
98+
} else {
99+
SqlKeywordGroupBlock(child, indentLevel, wrap, alignment, spacingBuilder)
100+
}
101+
}
102+
103+
else -> {
104+
return SqlKeywordGroupBlock(child, indentLevel, wrap, alignment, spacingBuilder)
105+
}
106+
}
107+
}
108+
109+
when (indentLevel) {
110+
IndentType.INLINE -> {
111+
if (!SqlKeywordUtil.isSetLineKeyword(
112+
child.text,
113+
lastGroupBlock?.node?.text ?: "",
114+
)
115+
) {
116+
return SqlInlineGroupBlock(child, wrap, alignment, spacingBuilder)
117+
}
118+
}
119+
120+
IndentType.ATTACHED -> {
121+
if (lastGroupBlock is SqlCreateKeywordGroupBlock) {
122+
lastGroupBlock.setCreateQueryType(child.text)
123+
return SqlKeywordBlock(child, indentLevel, wrap, alignment, spacingBuilder)
124+
}
125+
}
126+
127+
IndentType.OPTIONS -> {
128+
if (child.text.lowercase() == "as") {
129+
val parentCreateBlock =
130+
lastGroupBlock as? SqlCreateKeywordGroupBlock
131+
?: lastGroupBlock?.parentBlock as? SqlCreateKeywordGroupBlock
132+
if (parentCreateBlock != null && parentCreateBlock.createType == CreateQueryType.VIEW) {
133+
return SqlViewGroupBlock(child, wrap, alignment, spacingBuilder)
134+
}
135+
}
136+
}
137+
138+
else -> return SqlKeywordBlock(child, indentLevel, wrap, alignment, spacingBuilder)
139+
}
140+
return SqlKeywordBlock(child, indentLevel, wrap, alignment, spacingBuilder)
141+
}
142+
143+
fun getSubGroupBlock(
144+
lastGroup: SqlBlock?,
145+
child: ASTNode,
146+
): SqlBlock {
147+
if (child.treePrev.elementType == SqlTypes.WORD) {
148+
return SqlFunctionParamBlock(child, wrap, alignment, spacingBuilder)
149+
}
150+
151+
when (lastGroup) {
152+
is SqlKeywordGroupBlock -> {
153+
val lastKeyword =
154+
lastGroup.childBlocks
155+
.lastOrNull { SqlKeywordUtil.isOptionSqlKeyword(it.node.text) }
156+
if (lastKeyword != null && lastKeyword.node.text.lowercase() == "in") {
157+
return SqlParallelListBlock(child, wrap, alignment, spacingBuilder)
158+
}
159+
if (lastGroup is SqlCreateKeywordGroupBlock) {
160+
return SqlColumnDefinitionGroupBlock(child, wrap, alignment, spacingBuilder)
161+
}
162+
if (lastGroup is SqlInsertKeywordGroupBlock) {
163+
return SqlInsertColumnGroupBlock(child, wrap, alignment, spacingBuilder)
164+
}
165+
if (lastGroup is SqlUpdateKeywordGroupBlock) {
166+
return if (lastGroup.childBlocks.firstOrNull { it is SqlUpdateColumnGroupBlock } == null) {
167+
SqlUpdateColumnGroupBlock(child, wrap, alignment, spacingBuilder)
168+
} else if (lastGroup.childBlocks.lastOrNull { it is SqlUpdateColumnGroupBlock } != null) {
169+
SqlUpdateValueGroupBlock(child, wrap, alignment, spacingBuilder)
170+
} else {
171+
SqlSubQueryGroupBlock(child, wrap, alignment, spacingBuilder)
172+
}
173+
}
174+
return SqlSubQueryGroupBlock(child, wrap, alignment, spacingBuilder)
175+
}
176+
177+
is SqlColumnDefinitionRawGroupBlock ->
178+
return SqlDataTypeParamBlock(child, wrap, alignment, spacingBuilder)
179+
180+
else ->
181+
return SqlSubQueryGroupBlock(child, wrap, alignment, spacingBuilder)
182+
}
183+
}
184+
185+
fun getCommaGroupBlock(
186+
lastGroup: SqlBlock?,
187+
child: ASTNode,
188+
): SqlBlock =
189+
when (lastGroup) {
190+
is SqlColumnDefinitionGroupBlock, is SqlColumnDefinitionRawGroupBlock ->
191+
SqlColumnDefinitionRawGroupBlock(
192+
child,
193+
wrap,
194+
alignment,
195+
spacingBuilder,
196+
)
197+
198+
is SqlColumnGroupBlock, is SqlKeywordGroupBlock -> {
199+
if (lastGroup.indent.indentLevel == IndentType.SECOND) {
200+
SqlCommaBlock(child, wrap, alignment, spacingBuilder)
201+
} else {
202+
SqlColumnGroupBlock(child, wrap, alignment, spacingBuilder)
203+
}
204+
}
205+
206+
else -> SqlCommaBlock(child, wrap, alignment, spacingBuilder)
207+
}
208+
209+
fun getWordBlock(
210+
lastGroup: SqlBlock?,
211+
child: ASTNode,
212+
): SqlBlock =
213+
when (lastGroup) {
214+
is SqlKeywordGroupBlock -> {
215+
when {
216+
SqlKeywordUtil.isBeforeTableKeyword(lastGroup.node.text) ->
217+
SqlTableBlock(
218+
child,
219+
wrap,
220+
alignment,
221+
spacingBuilder,
222+
)
223+
224+
else -> SqlWordBlock(child, wrap, alignment, spacingBuilder)
225+
}
226+
}
227+
228+
is SqlColumnDefinitionGroupBlock -> {
229+
lastGroup.alignmentColumnName = child.text
230+
SqlColumnDefinitionRawGroupBlock(
231+
child,
232+
wrap,
233+
alignment,
234+
spacingBuilder,
235+
)
236+
}
237+
238+
is SqlColumnDefinitionRawGroupBlock -> {
239+
if (lastGroup.childBlocks.isEmpty()) {
240+
lastGroup.columnName = child.text
241+
SqlColumnBlock(
242+
child,
243+
wrap,
244+
alignment,
245+
spacingBuilder,
246+
)
247+
} else {
248+
SqlWordBlock(child, wrap, alignment, spacingBuilder)
249+
}
250+
}
251+
252+
else -> SqlWordBlock(child, wrap, alignment, spacingBuilder)
253+
}
254+
255+
fun getBlockCommentBlock(
256+
child: ASTNode,
257+
blockCommentSpacingBuilder: SqlCustomSpacingBuilder?,
258+
): SqlCommentBlock {
259+
if (PsiTreeUtil.getChildOfType(child.psi, PsiComment::class.java) != null) {
260+
return SqlBlockCommentBlock(
261+
child,
262+
wrap,
263+
alignment,
264+
spacingBuilder,
265+
)
266+
}
267+
if (child.psi is SqlCustomElCommentExpr &&
268+
(child.psi as SqlCustomElCommentExpr).isConditionOrLoopDirective()
269+
) {
270+
return SqlElConditionLoopCommentBlock(
271+
child,
272+
wrap,
273+
alignment,
274+
blockCommentSpacingBuilder,
275+
spacingBuilder,
276+
)
277+
}
278+
return SqlElBlockCommentBlock(
279+
child,
280+
wrap,
281+
alignment,
282+
blockCommentSpacingBuilder,
283+
spacingBuilder,
284+
)
285+
}
286+
}

‎src/main/kotlin/org/domaframework/doma/intellij/formatter/SqlCustomSpacingBuilder.kt‎

Lines changed: 42 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,15 @@ import com.intellij.formatting.Spacing
2121
import com.intellij.psi.tree.IElementType
2222
import org.domaframework.doma.intellij.formatter.block.SqlBlock
2323
import org.domaframework.doma.intellij.formatter.block.SqlColumnBlock
24+
import org.domaframework.doma.intellij.formatter.block.SqlRightPatternBlock
2425
import org.domaframework.doma.intellij.formatter.block.SqlWhitespaceBlock
2526
import org.domaframework.doma.intellij.formatter.block.group.SqlColumnDefinitionRawGroupBlock
26-
import org.domaframework.doma.intellij.formatter.block.group.SqlNewGroupBlock
27+
import org.domaframework.doma.intellij.formatter.block.group.keyword.SqlKeywordGroupBlock
28+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlColumnDefinitionGroupBlock
29+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlDataTypeParamBlock
30+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlParallelListBlock
31+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlUpdateColumnGroupBlock
32+
import org.domaframework.doma.intellij.formatter.block.group.subgroup.SqlUpdateValueGroupBlock
2733

2834
class SqlCustomSpacingBuilder {
2935
companion object {
@@ -68,10 +74,14 @@ class SqlCustomSpacingBuilder {
6874
child1: SqlBlock?,
6975
child2: SqlBlock,
7076
): Spacing? {
71-
val indentLen: Int = child2.indent.indentLen
77+
if (child2.parentBlock is SqlParallelListBlock) {
78+
return nonSpacing
79+
}
80+
7281
when (child1) {
73-
null -> return Spacing.createSpacing(0, 0, 0, false, 0, 0)
82+
null -> return nonSpacing
7483
is SqlWhitespaceBlock -> {
84+
val indentLen: Int = child2.indent.indentLen
7585
val afterNewLine = child1.node.text.substringAfterLast("\n", "")
7686
if (child1.node.text.contains("\n")) {
7787
val currentIndent = afterNewLine.length
@@ -84,14 +94,15 @@ class SqlCustomSpacingBuilder {
8494
return Spacing.createSpacing(newIndent, newIndent, 0, false, 0, 0)
8595
}
8696
}
97+
8798
else -> {
88-
return Spacing.createSpacing(indentLen, indentLen, 1, false, 0, 1)
99+
return getSpacing(child2)
89100
}
90101
}
91102
return null
92103
}
93104

94-
fun getSpacing(child2: SqlNewGroupBlock): Spacing? =
105+
fun getSpacing(child2: SqlBlock): Spacing =
95106
Spacing.createSpacing(
96107
child2.indent.indentLen,
97108
child2.indent.indentLen,
@@ -110,4 +121,30 @@ class SqlCustomSpacingBuilder {
110121
val indentLen = child.createBlockIndentLen()
111122
return Spacing.createSpacing(indentLen, indentLen, 0, false, 0, 0)
112123
}
124+
125+
fun getSpacingRightPattern(block: SqlRightPatternBlock): Spacing? {
126+
return when {
127+
block.parentBlock is SqlColumnDefinitionGroupBlock ||
128+
block.parentBlock is SqlUpdateColumnGroupBlock ||
129+
block.parentBlock is SqlUpdateValueGroupBlock -> {
130+
return getSpacing(block)
131+
}
132+
133+
block.parentBlock is SqlParallelListBlock -> {
134+
if (block.parentBlock
135+
?.childBlocks
136+
?.dropLast(1)
137+
?.lastOrNull() is SqlKeywordGroupBlock
138+
) {
139+
return normalSpacing
140+
}
141+
return nonSpacing
142+
}
143+
144+
block.parentBlock is SqlDataTypeParamBlock -> nonSpacing
145+
146+
block.preSpaceRight -> normalSpacing
147+
else -> nonSpacing
148+
}
149+
}
113150
}

0 commit comments

Comments
 (0)