2 * Copyright 2000-2009 JetBrains s.r.o.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
16 package com
.intellij
.refactoring
.util
.duplicates
;
18 import com
.intellij
.codeInsight
.PsiEquivalenceUtil
;
19 import com
.intellij
.psi
.*;
20 import com
.intellij
.psi
.codeStyle
.CodeStyleManager
;
21 import com
.intellij
.util
.IncorrectOperationException
;
26 public class ExpressionReturnValue
implements ReturnValue
{
27 private final PsiExpression myExpression
;
29 public ExpressionReturnValue(PsiExpression expression
) {
30 myExpression
= expression
;
33 public PsiExpression
getExpression() {
37 public boolean isEquivalent(ReturnValue other
) {
38 if (!(other
instanceof ExpressionReturnValue
)) return false;
39 return PsiEquivalenceUtil
.areElementsEquivalent(myExpression
, ((ExpressionReturnValue
)other
).myExpression
);
42 public PsiStatement
createReplacement(final PsiMethod extractedMethod
, final PsiMethodCallExpression methodCallExpression
)
43 throws IncorrectOperationException
{
44 final PsiElementFactory elementFactory
= JavaPsiFacade
.getInstance(methodCallExpression
.getProject()).getElementFactory();
45 final CodeStyleManager styleManager
= CodeStyleManager
.getInstance(methodCallExpression
.getProject());
46 PsiExpressionStatement expressionStatement
;
47 expressionStatement
= (PsiExpressionStatement
)elementFactory
.createStatementFromText("x = y();", null);
48 expressionStatement
= (PsiExpressionStatement
)styleManager
.reformat(expressionStatement
);
49 final PsiAssignmentExpression assignmentExpression
= (PsiAssignmentExpression
)expressionStatement
.getExpression();
50 assignmentExpression
.getLExpression().replace(getExpression());
51 assignmentExpression
.getRExpression().replace(methodCallExpression
);
52 return expressionStatement
;