Fixed bug in join lines

compiler/frontend/src/org/jetbrains/jet/lang/psi/JetProperty.java

@@ -33,7 +33,6 @@ import org.jetbrains.jet.lexer.JetTokens;
 
 import java.util.List;
 
-import static org.jetbrains.jet.JetNodeTypes.PROPERTY_ACCESSOR;
 import static org.jetbrains.jet.JetNodeTypes.PROPERTY_DELEGATE;
 import static org.jetbrains.jet.lexer.JetTokens.*;
 
@@ -233,6 +232,34 @@ public class JetProperty extends JetTypeParameterListOwnerStub<PsiJetPropertyStu
         return hasDelegateExpression() || hasInitializer();
     }
 
+    @Nullable
+    public JetExpression setInitializer(@Nullable JetExpression initializer) {
+        JetExpression oldInitializer = getInitializer();
+
+        if (oldInitializer != null) {
+            if (initializer != null) {
+                return (JetExpression) oldInitializer.replace(initializer);
+            }
+            else {
+                deleteChildRange(findChildByType(EQ), oldInitializer);
+                return null;
+            }
+        }
+        else {
+            if (initializer != null) {
+                PsiElement addAfter = getTypeRef();
+                if (addAfter == null) {
+                    addAfter = getNameIdentifier();
+                }
+                PsiElement eq = addAfter(new JetPsiFactory(getProject()).createEQ(), addAfter);
+                return (JetExpression) addAfter(initializer, eq);
+            }
+            else {
+                return null;
+            }
+        }
+    }
+
     @Nullable
     public JetExpression getDelegateExpressionOrInitializer() {
         JetExpression expression = getDelegateExpression();

idea/src/org/jetbrains/jet/plugin/intentions/declarations/JetDeclarationJoinLinesHandler.kt

@@ -38,7 +38,8 @@ public class JetDeclarationJoinLinesHandler : JoinRawLinesHandlerDelegate {
                            .firstOrNull() ?: return -1
         val (property, assignment) = pair
 
-        return doJoin(property, assignment).getTextRange()!!.getStartOffset()
+        doJoin(property, assignment)
+        return property.getTextRange()!!.getStartOffset()
     }
 
     override fun tryJoinLines(document: Document, file: PsiFile, start: Int, end: Int)
@@ -58,10 +59,9 @@ public class JetDeclarationJoinLinesHandler : JoinRawLinesHandlerDelegate {
         return property to assignment
     }
 
-    private fun doJoin(property: JetProperty, assignment: JetBinaryExpression): JetProperty {
-        val newProperty = DeclarationUtils.changePropertyInitializer(property, assignment.getRight())
-        property.getParent()!!.deleteChildRange(property.getNextSibling(), assignment)
-        return property.replace(newProperty) as JetProperty
+    private fun doJoin(property: JetProperty, assignment: JetBinaryExpression) {
+        property.setInitializer(assignment.getRight())
+        property.getParent()!!.deleteChildRange(property.getNextSibling(), assignment) //TODO: should we remove range?
     }
 
 }

idea/testData/joinLines/propertyWithAnnotation.kt

@@ -0,0 +1,4 @@
+fun foo() {
+    <caret>[volatile] var v: Int
+    v = 1
+}

idea/testData/joinLines/propertyWithAnnotation.kt.after

@@ -0,0 +1,3 @@
+fun foo() {
+    <caret>[volatile] var v: Int = 1
+}

idea/testData/joinLines/simpleInitWithComments.kt.after

@@ -1,5 +1,5 @@
 fun foo(n: Int) {
     // foo3
-    <caret>val x: String = "" // bar3
+    <caret>val /* foo */ x: String = "" // bar3
     /* baz */
 }

idea/testData/joinLines/simpleInitWithComments2.kt.after

@@ -1,5 +1,5 @@
 fun foo(n: Int) {
     // foo3
-    <caret>val x: String = "" // bar3
+    <caret>val /* foo */ x: String = "" // bar3
     /* baz */
 }

idea/tests/org/jetbrains/jet/plugin/intentions/declarations/JoinLinesTestGenerated.java

@@ -17,13 +17,9 @@
 package org.jetbrains.jet.plugin.intentions.declarations;
 
 import com.intellij.testFramework.TestDataPath;
-import junit.framework.Test;
-import junit.framework.TestSuite;
-import org.junit.runner.RunWith;
 import org.jetbrains.jet.JetTestUtils;
-import org.jetbrains.jet.test.InnerTestClasses;
 import org.jetbrains.jet.test.TestMetadata;
-import org.jetbrains.jet.JUnit3RunnerWithInners;
+import org.junit.runner.RunWith;
 
 import java.io.File;
 import java.util.regex.Pattern;
@@ -50,6 +46,12 @@ public class JoinLinesTestGenerated extends AbstractJoinLinesTest {
         doTest(fileName);
     }
     
+    @TestMetadata("propertyWithAnnotation.kt")
+    public void testPropertyWithAnnotation() throws Exception {
+        String fileName = JetTestUtils.navigationMetadata("idea/testData/joinLines/propertyWithAnnotation.kt");
+        doTest(fileName);
+    }
+    
     @TestMetadata("simpleInit.kt")
     public void testSimpleInit() throws Exception {
         String fileName = JetTestUtils.navigationMetadata("idea/testData/joinLines/simpleInit.kt");