NO_BRACES_IN_CONDITIONALS_AND_LOOPS fail within variable initialization block

diktat-rules/src/main/kotlin/com/saveourtool/diktat/ruleset/rules/chapter3/BracesInConditionalsAndLoopsRule.kt

@@ -13,6 +13,7 @@
 import org.jetbrains.kotlin.KtNodeTypes.BLOCK
 import org.jetbrains.kotlin.KtNodeTypes.CALL_EXPRESSION
 import org.jetbrains.kotlin.KtNodeTypes.IF
+import org.jetbrains.kotlin.KtNodeTypes.LAMBDA_EXPRESSION
 import org.jetbrains.kotlin.KtNodeTypes.REFERENCE_EXPRESSION
 import org.jetbrains.kotlin.KtNodeTypes.SAFE_ACCESS_EXPRESSION
 import org.jetbrains.kotlin.KtNodeTypes.WHEN
@@ -66,12 +67,7 @@
         val thenNode = ifPsi.then?.node
         val elseKeyword = ifPsi.elseKeyword
         val elseNode = ifPsi.`else`?.node
-        val indent = node
-            .prevSibling { it.elementType == WHITE_SPACE }
-            ?.text
-            ?.lines()
-            ?.last()
-            ?.count { it == ' ' } ?: 0
+        val indent = node.findIndentBeforeNode()
 
         if (node.isSingleLineIfElse()) {
             return
@@ -132,11 +128,8 @@
         if (loopBodyNode == null || loopBodyNode.elementType != BLOCK) {
             NO_BRACES_IN_CONDITIONALS_AND_LOOPS.warnAndFix(configRules, emitWarn, isFixMode, node.elementType.toString(), node.startOffset, node) {
                 // fixme proper way to calculate indent? or get step size (instead of hardcoded 4)
-                val indent = node.prevSibling { it.elementType == WHITE_SPACE }!!
-                    .text
-                    .lines()
-                    .last()
-                    .count { it == ' ' }
+                val indent = node.findIndentBeforeNode()
+
                 loopBody?.run {
                     replaceWithBlock(indent)
                 }
@@ -152,6 +145,20 @@
         }
     }
 
+    private fun ASTNode.findIndentBeforeNode(): Int {
+        val indentNode = if (treeParent?.treeParent?.treeParent?.elementType == LAMBDA_EXPRESSION) {
+            treeParent.prevSibling { it.elementType == WHITE_SPACE }
+        } else {
+            prevSibling { it.elementType == WHITE_SPACE }
+        }
+
+        return indentNode!!
+            .text
+            .lines()
+            .last()
+            .count { it == ' ' }
+    }
+
     @Suppress("UnsafeCallOnNullableType")
     private fun checkWhenBranches(node: ASTNode) {
         (node.psi as KtWhenExpression)

diktat-rules/src/test/kotlin/com/saveourtool/diktat/ruleset/chapter3/BracesRuleFixTest.kt

@@ -22,6 +22,18 @@ class BracesRuleFixTest : FixTestBase("test/paragraph3/braces", ::BracesInCondit
         fixAndCompare("LoopsBracesExpected.kt", "LoopsBracesTest.kt")
     }
 
+    @Test
+    @Tag(WarningNames.NO_BRACES_IN_CONDITIONALS_AND_LOOPS)
+    fun `should add braces to if-else statements inside scope functions`() {
+        fixAndCompare("IfElseBracesInsideScopeFunctionsExpected.kt", "IfElseBracesInsideScopeFunctionsTest.kt")
+    }
+
+    @Test
+    @Tag(WarningNames.NO_BRACES_IN_CONDITIONALS_AND_LOOPS)
+    fun `should add braces to loops inside scope functions`() {
+        fixAndCompare("LoopsBracesInsideScopeFunctionsExpected.kt", "LoopsBracesInsideScopeFunctionsTest.kt")
+    }
+
     @Test
     @Tag(WarningNames.NO_BRACES_IN_CONDITIONALS_AND_LOOPS)
     @Disabled("https://github.com/saveourtool/diktat/issues/1737")

diktat-rules/src/test/resources/test/paragraph3/braces/IfElseBracesInsideScopeFunctionsExpected.kt

@@ -0,0 +1,51 @@
+package test.paragraph3.braces
+
+fun foo1() {
+    str.apply {
+        if (x > 0) {
+            foo()
+        } else {
+            bar()
+        }
+    }
+}
+
+fun foo2() {
+    str.let { if (x > 0) { foo() }
+    else {
+     bar()
+ }
+    }
+}
+
+fun foo3() {
+    str.run {
+        while (x > 0) {
+            if (x > 0) {
+                foo()
+            } else {
+                bar()
+            }
+        }
+    }
+}
+
+fun foo4() {
+    str.with { while (x > 0) {
+        if (x > 0) {
+            foo()
+        } else { bar() }
+    }
+    }
+}
+
+fun foo5() {
+    str.also {
+        while (x > 0) { if (x > 0) {
+     foo()
+ } else {
+     bar()
+ }
+        }
+    }
+}

diktat-rules/src/test/resources/test/paragraph3/braces/IfElseBracesInsideScopeFunctionsTest.kt

@@ -0,0 +1,39 @@
+package test.paragraph3.braces
+
+fun foo1() {
+    str.apply {
+        if (x > 0) foo()
+        else bar()
+    }
+}
+
+fun foo2() {
+    str.let { if (x > 0) { foo() }
+    else bar()
+    }
+}
+
+fun foo3() {
+    str.run {
+        while (x > 0) {
+            if (x > 0) foo()
+            else bar()
+        }
+    }
+}
+
+fun foo4() {
+    str.with { while (x > 0) {
+        if (x > 0) foo()
+        else { bar() }
+    }
+    }
+}
+
+fun foo5() {
+    str.also {
+        while (x > 0) { if (x > 0) foo()
+            else bar()
+        }
+    }
+}

diktat-rules/src/test/resources/test/paragraph3/braces/LoopsBracesInsideScopeFunctionsExpected.kt

@@ -0,0 +1,44 @@
+package test.paragraph3.braces
+
+fun foo1() {
+    str.apply {
+        for (i in 1..100) {
+            println(i)
+        }
+    }
+}
+
+fun foo2() {
+    str.let { while (x > 0) {
+     println(i)
+ }
+    }
+}
+
+fun foo3() {
+    str.run {
+        do {
+            println(i)
+        }
+        while (x > 0)
+    }
+}
+
+fun foo4() {
+    str.with { do {
+     println(i)
+ }
+    while (x > 0)
+    }
+}
+
+fun foo5() {
+    str.also {
+        for (i in 1..100) {
+            while (x > 0)
+                {
+                println(i)
+            }
+        }
+    }
+}

diktat-rules/src/test/resources/test/paragraph3/braces/LoopsBracesInsideScopeFunctionsTest.kt

@@ -0,0 +1,34 @@
+package test.paragraph3.braces
+
+fun foo1() {
+    str.apply {
+        for (i in 1..100) println(i)
+    }
+}
+
+fun foo2() {
+    str.let { while (x > 0) println(i)
+    }
+}
+
+fun foo3() {
+    str.run {
+        do println(i)
+        while (x > 0)
+    }
+}
+
+fun foo4() {
+    str.with { do println(i)
+    while (x > 0)
+    }
+}
+
+fun foo5() {
+    str.also {
+        for (i in 1..100) {
+            while (x > 0)
+                println(i)
+        }
+    }
+}