Introduce JetExpressionImplStub

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

@@ -39,6 +39,8 @@ public abstract class JetExpressionImpl extends JetElementImpl implements JetExp
         return containerNode.findChildByClass(JetExpression.class);
     }
 
+    //NOTE: duplicate with JetExpressionImplStub
+    @NotNull
     @Override
     public PsiElement replace(@NotNull PsiElement newElement) throws IncorrectOperationException {
         PsiElement parent = getParent();

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

@@ -0,0 +1,51 @@
+/*
+ * Copyright 2010-2014 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.jet.lang.psi;
+
+import com.intellij.lang.ASTNode;
+import com.intellij.psi.PsiElement;
+import com.intellij.psi.stubs.IStubElementType;
+import com.intellij.psi.stubs.StubElement;
+import com.intellij.util.IncorrectOperationException;
+import org.jetbrains.annotations.NotNull;
+
+public abstract class JetExpressionImplStub<T extends StubElement> extends JetElementImplStub<T> implements JetExpression {
+    public JetExpressionImplStub(@NotNull T stub, @NotNull IStubElementType nodeType) {
+        super(stub, nodeType);
+    }
+
+    public JetExpressionImplStub(@NotNull ASTNode node) {
+        super(node);
+    }
+
+    @Override
+    public <R, D> R accept(@NotNull JetVisitor<R, D> visitor, D data) {
+        return visitor.visitExpression(this, data);
+    }
+
+    //NOTE: duplicate with JetExpressionImpl
+    @NotNull
+    @Override
+    public PsiElement replace(@NotNull PsiElement newElement) throws IncorrectOperationException {
+        PsiElement parent = getParent();
+        if (parent instanceof JetExpression && newElement instanceof JetExpression &&
+            JetPsiUtil.areParenthesesNecessary((JetExpression) newElement, this, (JetExpression) parent)) {
+            return super.replace(JetPsiFactory.createExpression(getProject(), "(" + newElement.getText() + ")"));
+        }
+        return super.replace(newElement);
+    }
+}

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

@@ -29,7 +29,7 @@ import org.jetbrains.jet.lexer.JetTokens;
 
 import static org.jetbrains.jet.lexer.JetTokens.*;
 
-public class JetNameReferenceExpression extends JetElementImplStub<PsiJetNameReferenceExpressionStub> implements JetSimpleNameExpression {
+public class JetNameReferenceExpression extends JetExpressionImplStub<PsiJetNameReferenceExpressionStub> implements JetSimpleNameExpression {
 
     private static final TokenSet NAME_REFERENCE_EXPRESSIONS = TokenSet.create(IDENTIFIER, FIELD_IDENTIFIER, THIS_KEYWORD, SUPER_KEYWORD);