Added indent handling for the case when condition or loop inside scop…

…e function

### What's done:
* added findIndentBeforeNode() function
* reworked test for loops
* added test for conditions

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

@@ -13,6 +13,7 @@ import org.jetbrains.kotlin.KtNodeTypes
 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 @@ class BracesInConditionalsAndLoopsRule(configRules: List<RulesConfig>) : DiktatR
  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 @@ class BracesInConditionalsAndLoopsRule(configRules: List<RulesConfig>) : DiktatR
  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 == ' ' } ?: 0
+ val indent = node.findIndentBeforeNode()
+
  loopBody?.run {
  replaceWithBlock(indent)
  }
@@ -152,6 +145,20 @@ class BracesInConditionalsAndLoopsRule(configRules: List<RulesConfig>) : DiktatR
  }
  }
 
+ 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,12 @@ 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`() {

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

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

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

@@ -7,25 +7,28 @@ fun foo1() {
 }
 
 fun foo2() {
- str.let {
- for (i in 1..100) println(i)
+ str.let { while (x > 0) println(i)
  }
 }
 
 fun foo3() {
  str.run {
- for (i in 1..100) println(i)
+ do println(i)
+ while (x > 0)
  }
 }
 
 fun foo4() {
- str.with {
-  for (i in 1..100) println(i)
+ str.with { do println(i)
+ while (x > 0)
  }
 }
 
 fun foo5() {
  str.also {
- for (i in 1..100) println(i)
+ for (i in 1..100) {
+ while (x > 0)
+ println(i)
+ }
  }
 }