Fixed overflow when referencing itself in for directive search

xterao · xterao · commit 931ab4852639 · 2025-04-25T20:22:05.000+09:00
diff --git a/src/main/kotlin/org/domaframework/doma/intellij/inspection/ForDirectiveInspection.kt b/src/main/kotlin/org/domaframework/doma/intellij/inspection/ForDirectiveInspection.kt
@@ -218,6 +218,13 @@ class ForDirectiveInspection(
                                 }
                             }
                     val stack = mutableListOf<BlockToken>()
+                    val filterPosition = directiveBlocks.filter { it.position < targetElement.textOffset }
+                    filterPosition.forEach { block ->
+                        when (block.type) {
+                            BlockType.FOR, BlockType.IF -> stack.add(block)
+                            BlockType.END -> if (stack.isNotEmpty()) stack.removeAt(stack.lastIndex)
+                        }
+                    }
                     directiveBlocks.forEach { block ->
                         when (block.type) {
                             BlockType.FOR, BlockType.IF -> stack.add(block)

Original file line number	Diff line number	Diff line change
`@@ -218,6 +218,13 @@ class ForDirectiveInspection(`
`218`	`218`	`}`
`219`	`219`	`}`
`220`	`220`	`val stack = mutableListOf<BlockToken>()`
	`221`	`+ val filterPosition = directiveBlocks.filter { it.position < targetElement.textOffset }`
	`222`	`+ filterPosition.forEach { block ->`
	`223`	`+ when (block.type) {`
	`224`	`+ BlockType.FOR, BlockType.IF -> stack.add(block)`
	`225`	`+ BlockType.END -> if (stack.isNotEmpty()) stack.removeAt(stack.lastIndex)`
	`226`	`+ }`
	`227`	`+ }`
`221`	`228`	`directiveBlocks.forEach { block ->`
`222`	`229`	`when (block.type) {`
`223`	`230`	`BlockType.FOR, BlockType.IF -> stack.add(block)`