Add braces to 'if' statement: save/restore comments correctly

So #KT-16332 Fixed

idea/testData/intentions/addBraces/doWhileWithComment.kt

@@ -0,0 +1,6 @@
+fun foo() {}
+
+fun test() {
+    <caret>do /* aaa */ foo() // comment
+    while(true)
+}

idea/testData/intentions/addBraces/doWhileWithComment.kt.after

@@ -0,0 +1,7 @@
+fun foo() {}
+
+fun test() {
+    do {
+        /* aaa */ foo() // comment
+    } while(true)
+}

idea/testData/intentions/addBraces/elseWithComment.kt

@@ -0,0 +1,6 @@
+fun foo() {}
+
+fun test(b: Boolean) {
+    if (b) foo() <caret>else
+        /* aaa */ foo() // bbb
+}

idea/testData/intentions/addBraces/elseWithComment.kt.after

@@ -0,0 +1,7 @@
+fun foo() {}
+
+fun test(b: Boolean) {
+    if (b) foo() else {
+        /* aaa */ foo() // bbb
+    }
+}

idea/testData/intentions/addBraces/forWithComment.kt

@@ -0,0 +1,5 @@
+fun foo() {}
+
+fun test() {
+    <caret>for (i in 1..4) /* aaa */ foo() // bbb
+}

idea/testData/intentions/addBraces/forWithComment.kt.after

@@ -0,0 +1,7 @@
+fun foo() {}
+
+fun test() {
+    for (i in 1..4) {
+        /* aaa */ foo() // bbb
+    }
+}

idea/testData/intentions/addBraces/ifElseWithComment.kt

@@ -0,0 +1,5 @@
+fun foo() {}
+
+fun test(b: Boolean) {
+    <caret>if (b) foo() /* if comment */ else foo() // else comment
+}

idea/testData/intentions/addBraces/ifElseWithComment.kt.after

@@ -0,0 +1,7 @@
+fun foo() {}
+
+fun test(b: Boolean) {
+    if (b) {
+        foo() /* if comment */
+    } else foo() // else comment
+}

idea/testData/intentions/addBraces/ifWithComment.kt

@@ -0,0 +1,5 @@
+fun foo() {}
+
+fun test(b: Boolean) {
+    <caret>if (b) /* aaa */ /* bbb */ foo() // ccc
+}

idea/testData/intentions/addBraces/ifWithComment.kt.after

@@ -0,0 +1,7 @@
+fun foo() {}
+
+fun test(b: Boolean) {
+    if (b) {
+        /* aaa */ /* bbb */ foo() // ccc
+    }
+}

idea/testData/intentions/addBraces/whenWithComment.kt

@@ -0,0 +1,7 @@
+fun foo() {}
+
+fun test(a: Int) {
+    when (a) {
+        <caret>1 -> /* aaa */ foo() // bbb
+    }
+}

idea/testData/intentions/addBraces/whenWithComment.kt.after

@@ -0,0 +1,9 @@
+fun foo() {}
+
+fun test(a: Int) {
+    when (a) {
+        1 -> {
+            /* aaa */ foo() // bbb
+        }
+    }
+}

idea/testData/intentions/addBraces/whileWithComment.kt

@@ -0,0 +1,5 @@
+fun foo() {}
+
+fun test(b: Boolean) {
+    <caret>while (b) /* aaa */ foo() // bbb
+}

idea/testData/intentions/addBraces/whileWithComment.kt.after

@@ -0,0 +1,7 @@
+fun foo() {}
+
+fun test(b: Boolean) {
+    while (b) {
+        /* aaa */ foo() // bbb
+    }
+}