Adding block to control structure when needed

idea/testData/refactoring/inline/function/returnAtEnd/AddBlockToControlStatement.kt

@@ -0,0 +1,23 @@
+fun <caret>f(p1: Int, p2: Int): Int {
+    println(p1)
+    println(p2)
+    return p1 + p2
+}
+
+fun main(args: Array<String>) {
+    if (args.size > 0)
+        f(1, 2)
+    else
+        f(3, 4)
+
+    for (i in 1..10)
+        f(0, 1)
+
+    when (args.size) {
+        0, 1 -> println(f(1, 1))
+        else -> println(f(2, 2))
+    }
+
+    while (true)
+        println(f(5, 6))
+}

idea/testData/refactoring/inline/function/returnAtEnd/AddBlockToControlStatement.kt.after

@@ -0,0 +1,37 @@
+fun main(args: Array<String>) {
+    if (args.size > 0) {
+        println(1)
+        println(2)
+        1 + 2
+    }
+    else {
+        println(3)
+        println(4)
+        3 + 4
+    }
+
+    for (i in 1..10) {
+        println(0)
+        println(1)
+        0 + 1
+    }
+
+    when (args.size) {
+        0, 1 -> {
+            println(1)
+            println(1)
+            println(1 + 1)
+        }
+        else -> {
+            println(2)
+            println(2)
+            println(2 + 2)
+        }
+    }
+
+    while (true) {
+        println(5)
+        println(6)
+        println(5 + 6)
+    }
+}