blob: d7ae6d576a8618afae0fe2a855c60038521a6911 [file] [log] [blame]
/*******************************************************************************
* Copyright (c) 2000, 2021 IBM Corporation and others.
*
* This program and the accompanying materials
* are made available under the terms of the Eclipse Public License 2.0
* which accompanies this distribution, and is available at
* https://www.eclipse.org/legal/epl-2.0/
*
* SPDX-License-Identifier: EPL-2.0
*
* Contributors:
* IBM Corporation - initial API and implementation
* Benjamin Muskalla <bmuskalla@eclipsesource.com> - [extract method] Does not replace similar code in parent class of anonymous class - https://bugs.eclipse.org/bugs/show_bug.cgi?id=160853
* Benjamin Muskalla <bmuskalla@eclipsesource.com> - [extract method] Extract method and continue https://bugs.eclipse.org/bugs/show_bug.cgi?id=48056
* Benjamin Muskalla <bmuskalla@eclipsesource.com> - [extract method] should declare method static if extracted from anonymous in static method - https://bugs.eclipse.org/bugs/show_bug.cgi?id=152004
* Samrat Dhillon <samrat.dhillon@gmail.com> - [extract method] Extracted method should be declared static if extracted expression is also used in another static method https://bugs.eclipse.org/bugs/show_bug.cgi?id=393098
* Samrat Dhillon <samrat.dhillon@gmail.com> - [extract method] Extracting expression of parameterized type that is passed as argument to this constructor yields compilation error https://bugs.eclipse.org/bugs/show_bug.cgi?id=394030
*******************************************************************************/
package org.eclipse.jdt.internal.corext.refactoring.code;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.StringTokenizer;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.OperationCanceledException;
import org.eclipse.core.runtime.SubProgressMonitor;
import org.eclipse.core.resources.IFile;
import org.eclipse.text.edits.MultiTextEdit;
import org.eclipse.text.edits.TextEdit;
import org.eclipse.text.edits.TextEditGroup;
import org.eclipse.ltk.core.refactoring.Change;
import org.eclipse.ltk.core.refactoring.Refactoring;
import org.eclipse.ltk.core.refactoring.RefactoringChangeDescriptor;
import org.eclipse.ltk.core.refactoring.RefactoringDescriptor;
import org.eclipse.ltk.core.refactoring.RefactoringStatus;
import org.eclipse.ltk.core.refactoring.TextFileChange;
import org.eclipse.ltk.core.refactoring.participants.ResourceChangeChecker;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.IJavaElement;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.dom.AST;
import org.eclipse.jdt.core.dom.ASTNode;
import org.eclipse.jdt.core.dom.ASTVisitor;
import org.eclipse.jdt.core.dom.AbstractTypeDeclaration;
import org.eclipse.jdt.core.dom.AnnotationTypeDeclaration;
import org.eclipse.jdt.core.dom.AnonymousClassDeclaration;
import org.eclipse.jdt.core.dom.Assignment;
import org.eclipse.jdt.core.dom.Block;
import org.eclipse.jdt.core.dom.BodyDeclaration;
import org.eclipse.jdt.core.dom.ChildListPropertyDescriptor;
import org.eclipse.jdt.core.dom.CompilationUnit;
import org.eclipse.jdt.core.dom.ContinueStatement;
import org.eclipse.jdt.core.dom.DoStatement;
import org.eclipse.jdt.core.dom.EnhancedForStatement;
import org.eclipse.jdt.core.dom.EnumDeclaration;
import org.eclipse.jdt.core.dom.Expression;
import org.eclipse.jdt.core.dom.ExpressionStatement;
import org.eclipse.jdt.core.dom.FieldAccess;
import org.eclipse.jdt.core.dom.FieldDeclaration;
import org.eclipse.jdt.core.dom.ForStatement;
import org.eclipse.jdt.core.dom.IMethodBinding;
import org.eclipse.jdt.core.dom.ITypeBinding;
import org.eclipse.jdt.core.dom.IVariableBinding;
import org.eclipse.jdt.core.dom.Initializer;
import org.eclipse.jdt.core.dom.Javadoc;
import org.eclipse.jdt.core.dom.LabeledStatement;
import org.eclipse.jdt.core.dom.LambdaExpression;
import org.eclipse.jdt.core.dom.MethodDeclaration;
import org.eclipse.jdt.core.dom.MethodInvocation;
import org.eclipse.jdt.core.dom.Modifier;
import org.eclipse.jdt.core.dom.ParenthesizedExpression;
import org.eclipse.jdt.core.dom.QualifiedName;
import org.eclipse.jdt.core.dom.ReturnStatement;
import org.eclipse.jdt.core.dom.SimpleName;
import org.eclipse.jdt.core.dom.SingleVariableDeclaration;
import org.eclipse.jdt.core.dom.Statement;
import org.eclipse.jdt.core.dom.StructuralPropertyDescriptor;
import org.eclipse.jdt.core.dom.ThisExpression;
import org.eclipse.jdt.core.dom.Type;
import org.eclipse.jdt.core.dom.TypeDeclaration;
import org.eclipse.jdt.core.dom.TypeParameter;
import org.eclipse.jdt.core.dom.VariableDeclaration;
import org.eclipse.jdt.core.dom.VariableDeclarationFragment;
import org.eclipse.jdt.core.dom.VariableDeclarationStatement;
import org.eclipse.jdt.core.dom.WhileStatement;
import org.eclipse.jdt.core.dom.rewrite.ASTRewrite;
import org.eclipse.jdt.core.dom.rewrite.ImportRewrite;
import org.eclipse.jdt.core.dom.rewrite.ImportRewrite.ImportRewriteContext;
import org.eclipse.jdt.core.dom.rewrite.ImportRewrite.TypeLocation;
import org.eclipse.jdt.core.dom.rewrite.ListRewrite;
import org.eclipse.jdt.core.manipulation.CodeGeneration;
import org.eclipse.jdt.core.refactoring.CompilationUnitChange;
import org.eclipse.jdt.core.refactoring.IJavaRefactorings;
import org.eclipse.jdt.core.refactoring.descriptors.ExtractMethodDescriptor;
import org.eclipse.jdt.core.refactoring.descriptors.JavaRefactoringDescriptor;
import org.eclipse.jdt.internal.core.manipulation.StubUtility;
import org.eclipse.jdt.internal.core.manipulation.dom.ASTResolving;
import org.eclipse.jdt.internal.core.manipulation.util.BasicElementLabels;
import org.eclipse.jdt.internal.core.refactoring.descriptors.RefactoringSignatureDescriptorFactory;
import org.eclipse.jdt.internal.corext.codemanipulation.ContextSensitiveImportRewriteContext;
import org.eclipse.jdt.internal.corext.dom.ASTNodeFactory;
import org.eclipse.jdt.internal.corext.dom.ASTNodes;
import org.eclipse.jdt.internal.corext.dom.Bindings;
import org.eclipse.jdt.internal.corext.dom.BodyDeclarationRewrite;
import org.eclipse.jdt.internal.corext.dom.LinkedNodeFinder;
import org.eclipse.jdt.internal.corext.dom.Selection;
import org.eclipse.jdt.internal.corext.dom.StatementRewrite;
import org.eclipse.jdt.internal.corext.fix.LinkedProposalModel;
import org.eclipse.jdt.internal.corext.fix.LinkedProposalPositionGroup;
import org.eclipse.jdt.internal.corext.refactoring.Checks;
import org.eclipse.jdt.internal.corext.refactoring.JDTRefactoringDescriptorComment;
import org.eclipse.jdt.internal.corext.refactoring.JavaRefactoringArguments;
import org.eclipse.jdt.internal.corext.refactoring.JavaRefactoringDescriptorUtil;
import org.eclipse.jdt.internal.corext.refactoring.ParameterInfo;
import org.eclipse.jdt.internal.corext.refactoring.RefactoringCoreMessages;
import org.eclipse.jdt.internal.corext.refactoring.code.SnippetFinder.Match;
import org.eclipse.jdt.internal.corext.refactoring.util.RefactoringASTParser;
import org.eclipse.jdt.internal.corext.refactoring.util.ResourceUtil;
import org.eclipse.jdt.internal.corext.refactoring.util.SelectionAwareSourceRangeComputer;
import org.eclipse.jdt.internal.corext.util.JdtFlags;
import org.eclipse.jdt.internal.corext.util.Messages;
import org.eclipse.jdt.ui.JavaElementLabels;
import org.eclipse.jdt.internal.ui.text.correction.ModifierCorrectionSubProcessorCore;
import org.eclipse.jdt.internal.ui.viewsupport.BindingLabelProvider;
/**
* Extracts a method in a compilation unit based on a text selection range.
*/
public class ExtractMethodRefactoring extends Refactoring {
private static final String ATTRIBUTE_VISIBILITY= "visibility"; //$NON-NLS-1$
private static final String ATTRIBUTE_DESTINATION= "destination"; //$NON-NLS-1$
private static final String ATTRIBUTE_COMMENTS= "comments"; //$NON-NLS-1$
private static final String ATTRIBUTE_REPLACE= "replace"; //$NON-NLS-1$
private static final String ATTRIBUTE_EXCEPTIONS= "exceptions"; //$NON-NLS-1$
private ICompilationUnit fCUnit;
private CompilationUnit fRoot;
private ImportRewrite fImportRewriter;
private int fSelectionStart;
private int fSelectionLength;
private AST fAST;
private ASTRewrite fRewriter;
private ExtractMethodAnalyzer fAnalyzer;
private int fVisibility;
private String fMethodName;
private boolean fThrowRuntimeExceptions;
private List<ParameterInfo> fParameterInfos;
private Set<String> fUsedNames;
private boolean fGenerateJavadoc;
private boolean fReplaceDuplicates;
private List<SnippetFinder.Match> fDuplicates;
private int fDestinationIndex= 0;
// either of type TypeDeclaration or AnonymousClassDeclaration
private ASTNode fDestination;
// either of type TypeDeclaration or AnonymousClassDeclaration
private ASTNode[] fDestinations;
private LinkedProposalModel fLinkedProposalModel;
private static final String EMPTY= ""; //$NON-NLS-1$
private static final String KEY_TYPE= "type"; //$NON-NLS-1$
private static final String KEY_NAME= "name"; //$NON-NLS-1$
private static class UsedNamesCollector extends ASTVisitor {
private Set<String> result= new HashSet<>();
private Set<SimpleName> fIgnore= new HashSet<>();
public static Set<String> perform(ASTNode[] nodes) {
UsedNamesCollector collector= new UsedNamesCollector();
for (ASTNode node : nodes) {
node.accept(collector);
}
return collector.result;
}
@Override
public boolean visit(FieldAccess node) {
Expression exp= node.getExpression();
if (exp != null)
fIgnore.add(node.getName());
return true;
}
@Override
public void endVisit(FieldAccess node) {
fIgnore.remove(node.getName());
}
@Override
public boolean visit(MethodInvocation node) {
Expression exp= node.getExpression();
if (exp != null)
fIgnore.add(node.getName());
return true;
}
@Override
public void endVisit(MethodInvocation node) {
fIgnore.remove(node.getName());
}
@Override
public boolean visit(QualifiedName node) {
fIgnore.add(node.getName());
return true;
}
@Override
public void endVisit(QualifiedName node) {
fIgnore.remove(node.getName());
}
@Override
public boolean visit(SimpleName node) {
if (!fIgnore.contains(node))
result.add(node.getIdentifier());
return true;
}
@Override
public boolean visit(TypeDeclaration node) {
return visitType(node);
}
@Override
public boolean visit(AnnotationTypeDeclaration node) {
return visitType(node);
}
@Override
public boolean visit(EnumDeclaration node) {
return visitType(node);
}
private boolean visitType(AbstractTypeDeclaration node) {
result.add(node.getName().getIdentifier());
// don't dive into type declaration since they open a new
// context.
return false;
}
}
/**
* Creates a new extract method refactoring
* @param unit the compilation unit, or <code>null</code> if invoked by scripting
* @param selectionStart selection start
* @param selectionLength selection end
*/
public ExtractMethodRefactoring(ICompilationUnit unit, int selectionStart, int selectionLength) {
fCUnit= unit;
fRoot= null;
fMethodName= "extracted"; //$NON-NLS-1$
fSelectionStart= selectionStart;
fSelectionLength= selectionLength;
fVisibility= -1;
}
public ExtractMethodRefactoring(JavaRefactoringArguments arguments, RefactoringStatus status) {
this((ICompilationUnit) null, 0, 0);
RefactoringStatus initializeStatus= initialize(arguments);
status.merge(initializeStatus);
}
/**
* Creates a new extract method refactoring
* @param astRoot the AST root of an AST created from a compilation unit
* @param selectionStart start
* @param selectionLength length
*/
public ExtractMethodRefactoring(CompilationUnit astRoot, int selectionStart, int selectionLength) {
this((ICompilationUnit) astRoot.getTypeRoot(), selectionStart, selectionLength);
fRoot= astRoot;
}
public void setLinkedProposalModel(LinkedProposalModel linkedProposalModel) {
fLinkedProposalModel= linkedProposalModel;
}
@Override
public String getName() {
return RefactoringCoreMessages.ExtractMethodRefactoring_name;
}
/**
* Checks if the refactoring can be activated. Activation typically means, if a
* corresponding menu entry can be added to the UI.
*
* @param pm a progress monitor to report progress during activation checking.
* @return the refactoring status describing the result of the activation check.
* @throws CoreException if checking fails
*/
@Override
public RefactoringStatus checkInitialConditions(IProgressMonitor pm) throws CoreException {
RefactoringStatus result= new RefactoringStatus();
pm.beginTask("", 100); //$NON-NLS-1$
if (fSelectionStart < 0 || fSelectionLength == 0)
return mergeTextSelectionStatus(result);
IFile[] changedFiles= ResourceUtil.getFiles(new ICompilationUnit[]{fCUnit});
result.merge(Checks.validateModifiesFiles(changedFiles, getValidationContext(), pm));
if (result.hasFatalError())
return result;
result.merge(ResourceChangeChecker.checkFilesToBeChanged(changedFiles, new SubProgressMonitor(pm, 1)));
if (fRoot == null) {
fRoot= RefactoringASTParser.parseWithASTProvider(fCUnit, true, new SubProgressMonitor(pm, 99));
}
fImportRewriter= StubUtility.createImportRewrite(fRoot, true);
fAST= fRoot.getAST();
fRoot.accept(createVisitor());
fSelectionStart= fAnalyzer.getSelection().getOffset();
fSelectionLength= fAnalyzer.getSelection().getLength();
result.merge(fAnalyzer.checkInitialConditions(fImportRewriter));
if (result.hasFatalError())
return result;
if (fVisibility == -1) {
setVisibility(Modifier.PRIVATE);
}
initializeParameterInfos();
initializeUsedNames();
initializeDuplicates();
initializeDestinations();
return result;
}
private ASTVisitor createVisitor() throws CoreException {
fAnalyzer= new ExtractMethodAnalyzer(fCUnit, Selection.createFromStartLength(fSelectionStart, fSelectionLength));
return fAnalyzer;
}
/**
* Sets the method name to be used for the extracted method.
*
* @param name the new method name.
*/
public void setMethodName(String name) {
fMethodName= name;
}
/**
* Returns the method name to be used for the extracted method.
* @return the method name to be used for the extracted method.
*/
public String getMethodName() {
return fMethodName;
}
/**
* Sets the visibility of the new method.
*
* @param visibility the visibility of the new method. Valid values are
* "public", "protected", "", and "private"
*/
public void setVisibility(int visibility) {
fVisibility= visibility;
}
/**
* Returns the visibility of the new method.
*
* @return the visibility of the new method
*/
public int getVisibility() {
return fVisibility;
}
/**
* Returns the parameter infos.
* @return a list of parameter infos.
*/
public List<ParameterInfo> getParameterInfos() {
return fParameterInfos;
}
/**
* Sets whether the new method signature throws runtime exceptions.
*
* @param throwRuntimeExceptions flag indicating if the new method
* throws runtime exceptions
*/
public void setThrowRuntimeExceptions(boolean throwRuntimeExceptions) {
fThrowRuntimeExceptions= throwRuntimeExceptions;
}
/**
* Checks if the new method name is a valid method name. This method doesn't
* check if a method with the same name already exists in the hierarchy. This
* check is done in <code>checkInput</code> since it is expensive.
* @return validation status
*/
public RefactoringStatus checkMethodName() {
return Checks.checkMethodName(fMethodName, fCUnit);
}
public ASTNode[] getDestinations() {
return fDestinations;
}
public void setDestination(int index) {
fDestination= fDestinations[index];
fDestinationIndex= index;
}
/**
* Checks if the parameter names are valid.
* @return validation status
*/
public RefactoringStatus checkParameterNames() {
RefactoringStatus result= new RefactoringStatus();
for (ParameterInfo parameter : fParameterInfos) {
result.merge(Checks.checkIdentifier(parameter.getNewName(), fCUnit));
for (ParameterInfo other : fParameterInfos) {
if (parameter != other && other.getNewName().equals(parameter.getNewName())) {
result.addError(Messages.format(
RefactoringCoreMessages.ExtractMethodRefactoring_error_sameParameter,
BasicElementLabels.getJavaElementName(other.getNewName())));
return result;
}
}
if (parameter.isRenamed() && fUsedNames.contains(parameter.getNewName())) {
result.addError(Messages.format(
RefactoringCoreMessages.ExtractMethodRefactoring_error_nameInUse,
BasicElementLabels.getJavaElementName(parameter.getNewName())));
return result;
}
}
return result;
}
/**
* Checks if varargs are ordered correctly.
* @return validation status
*/
public RefactoringStatus checkVarargOrder() {
for (Iterator<ParameterInfo> iter= fParameterInfos.iterator(); iter.hasNext();) {
ParameterInfo info= iter.next();
if (info.isOldVarargs() && iter.hasNext()) {
return RefactoringStatus.createFatalErrorStatus(Messages.format(
RefactoringCoreMessages.ExtractMethodRefactoring_error_vararg_ordering,
BasicElementLabels.getJavaElementName(info.getOldName())));
}
}
return new RefactoringStatus();
}
/**
* Returns the names already in use in the selected statements/expressions.
*
* @return names already in use.
*/
public Set<String> getUsedNames() {
return fUsedNames;
}
@Override
public RefactoringStatus checkFinalConditions(IProgressMonitor pm) throws CoreException {
pm.beginTask(RefactoringCoreMessages.ExtractMethodRefactoring_checking_new_name, 2);
pm.subTask(EMPTY);
RefactoringStatus result= checkMethodName();
result.merge(checkParameterNames());
result.merge(checkVarargOrder());
pm.worked(1);
if (pm.isCanceled())
throw new OperationCanceledException();
BodyDeclaration node= fAnalyzer.getEnclosingBodyDeclaration();
if (node != null) {
fAnalyzer.checkInput(result, fMethodName, fDestination);
pm.worked(1);
}
pm.done();
return result;
}
@Override
public Change createChange(IProgressMonitor pm) throws CoreException {
if (fMethodName == null)
return null;
pm.beginTask("", 2); //$NON-NLS-1$
try {
fAnalyzer.aboutToCreateChange();
BodyDeclaration declaration= fAnalyzer.getEnclosingBodyDeclaration();
fRewriter= ASTRewrite.create(declaration.getAST());
final CompilationUnitChange result= new CompilationUnitChange(RefactoringCoreMessages.ExtractMethodRefactoring_change_name, fCUnit);
result.setSaveMode(TextFileChange.KEEP_SAVE_STATE);
result.setDescriptor(new RefactoringChangeDescriptor(getRefactoringDescriptor()));
MultiTextEdit root= new MultiTextEdit();
result.setEdit(root);
ASTNode[] selectedNodes= fAnalyzer.getSelectedNodes();
fRewriter.setTargetSourceRangeComputer(new SelectionAwareSourceRangeComputer(selectedNodes, fCUnit.getBuffer(), fSelectionStart, fSelectionLength));
TextEditGroup substituteDesc= new TextEditGroup(Messages.format(RefactoringCoreMessages.ExtractMethodRefactoring_substitute_with_call, BasicElementLabels.getJavaElementName(fMethodName)));
result.addTextEditGroup(substituteDesc);
MethodDeclaration mm= createNewMethod(selectedNodes, fCUnit.findRecommendedLineSeparator(), substituteDesc);
if (fLinkedProposalModel != null) {
LinkedProposalPositionGroup typeGroup= fLinkedProposalModel.getPositionGroup(KEY_TYPE, true);
typeGroup.addPosition(fRewriter.track(mm.getReturnType2()), false);
ITypeBinding typeBinding= fAnalyzer.getReturnTypeBinding();
if (typeBinding != null) {
ITypeBinding[] relaxingTypes= ASTResolving.getNarrowingTypes(fAST, typeBinding);
for (int i= 0; i < relaxingTypes.length; i++) {
typeGroup.addProposal(relaxingTypes[i], fCUnit, relaxingTypes.length - i);
}
}
LinkedProposalPositionGroup nameGroup= fLinkedProposalModel.getPositionGroup(KEY_NAME, true);
nameGroup.addPosition(fRewriter.track(mm.getName()), false);
ModifierCorrectionSubProcessorCore.installLinkedVisibilityProposals(fLinkedProposalModel, fRewriter, mm.modifiers(), false);
}
TextEditGroup insertDesc= new TextEditGroup(Messages.format(RefactoringCoreMessages.ExtractMethodRefactoring_add_method, BasicElementLabels.getJavaElementName(fMethodName)));
result.addTextEditGroup(insertDesc);
if (fDestination == ASTResolving.findParentType(declaration.getParent())) {
ChildListPropertyDescriptor desc= (ChildListPropertyDescriptor)declaration.getLocationInParent();
ListRewrite container= fRewriter.getListRewrite(declaration.getParent(), desc);
container.insertAfter(mm, declaration, insertDesc);
} else {
BodyDeclarationRewrite container= BodyDeclarationRewrite.create(fRewriter, fDestination);
container.insert(mm, insertDesc);
}
replaceDuplicates(result, mm.getModifiers());
replaceBranches(result);
if (fImportRewriter.hasRecordedChanges()) {
TextEdit edit= fImportRewriter.rewriteImports(null);
root.addChild(edit);
result.addTextEditGroup(new TextEditGroup(
RefactoringCoreMessages.ExtractMethodRefactoring_organize_imports,
new TextEdit[] {edit}
));
}
root.addChild(fRewriter.rewriteAST());
return result;
} finally {
pm.done();
}
}
private void replaceBranches(final CompilationUnitChange result) {
for (ASTNode astNode : fAnalyzer.getSelectedNodes()) {
astNode.accept(new ASTVisitor() {
private LinkedList<String> fOpenLoopLabels= new LinkedList<>();
private void registerLoopLabel(Statement node) {
String identifier;
if (node.getParent() instanceof LabeledStatement) {
LabeledStatement labeledStatement= (LabeledStatement)node.getParent();
identifier= labeledStatement.getLabel().getIdentifier();
} else {
identifier= null;
}
fOpenLoopLabels.add(identifier);
}
@Override
public boolean visit(ForStatement node) {
registerLoopLabel(node);
return super.visit(node);
}
@Override
public void endVisit(ForStatement node) {
fOpenLoopLabels.removeLast();
}
@Override
public boolean visit(WhileStatement node) {
registerLoopLabel(node);
return super.visit(node);
}
@Override
public void endVisit(WhileStatement node) {
fOpenLoopLabels.removeLast();
}
@Override
public boolean visit(EnhancedForStatement node) {
registerLoopLabel(node);
return super.visit(node);
}
@Override
public void endVisit(EnhancedForStatement node) {
fOpenLoopLabels.removeLast();
}
@Override
public boolean visit(DoStatement node) {
registerLoopLabel(node);
return super.visit(node);
}
@Override
public void endVisit(DoStatement node) {
fOpenLoopLabels.removeLast();
}
@Override
public void endVisit(ContinueStatement node) {
final SimpleName label= node.getLabel();
if (fOpenLoopLabels.isEmpty() || (label != null && !fOpenLoopLabels.contains(label.getIdentifier()))) {
TextEditGroup description= new TextEditGroup(RefactoringCoreMessages.ExtractMethodRefactoring_replace_continue);
result.addTextEditGroup(description);
ReturnStatement rs= fAST.newReturnStatement();
IVariableBinding returnValue= fAnalyzer.getReturnValue();
if (returnValue != null) {
rs.setExpression(fAST.newSimpleName(getName(returnValue)));
}
fRewriter.replace(node, rs, description);
}
}
});
}
}
private ExtractMethodDescriptor getRefactoringDescriptor() {
final Map<String, String> arguments= new HashMap<>();
String project= null;
IJavaProject javaProject= fCUnit.getJavaProject();
if (javaProject != null)
project= javaProject.getElementName();
ITypeBinding type= null;
if (fDestination instanceof AbstractTypeDeclaration) {
final AbstractTypeDeclaration decl= (AbstractTypeDeclaration) fDestination;
type= decl.resolveBinding();
} else if (fDestination instanceof AnonymousClassDeclaration) {
final AnonymousClassDeclaration decl= (AnonymousClassDeclaration) fDestination;
type= decl.resolveBinding();
}
IMethodBinding method= null;
final BodyDeclaration enclosing= fAnalyzer.getEnclosingBodyDeclaration();
if (enclosing instanceof MethodDeclaration) {
final MethodDeclaration node= (MethodDeclaration) enclosing;
method= node.resolveBinding();
}
final int flags= RefactoringDescriptor.STRUCTURAL_CHANGE | JavaRefactoringDescriptor.JAR_REFACTORING | JavaRefactoringDescriptor.JAR_SOURCE_ATTACHMENT;
final String description= Messages.format(RefactoringCoreMessages.ExtractMethodRefactoring_descriptor_description_short, BasicElementLabels.getJavaElementName(fMethodName));
final String label= method != null ? BindingLabelProvider.getBindingLabel(method, JavaElementLabels.ALL_FULLY_QUALIFIED) : '{' + JavaElementLabels.ELLIPSIS_STRING + '}';
final String header= Messages.format(RefactoringCoreMessages.ExtractMethodRefactoring_descriptor_description, new String[] { BasicElementLabels.getJavaElementName(getSignature()), label, BindingLabelProvider.getBindingLabel(type, JavaElementLabels.ALL_FULLY_QUALIFIED)});
final JDTRefactoringDescriptorComment comment= new JDTRefactoringDescriptorComment(project, this, header);
comment.addSetting(Messages.format(RefactoringCoreMessages.ExtractMethodRefactoring_name_pattern, BasicElementLabels.getJavaElementName(fMethodName)));
comment.addSetting(Messages.format(RefactoringCoreMessages.ExtractMethodRefactoring_destination_pattern, BindingLabelProvider.getBindingLabel(type, JavaElementLabels.ALL_FULLY_QUALIFIED)));
String visibility= JdtFlags.getVisibilityString(fVisibility);
if ("".equals(visibility)) //$NON-NLS-1$
visibility= RefactoringCoreMessages.ExtractMethodRefactoring_default_visibility;
comment.addSetting(Messages.format(RefactoringCoreMessages.ExtractMethodRefactoring_visibility_pattern, visibility));
if (fThrowRuntimeExceptions)
comment.addSetting(RefactoringCoreMessages.ExtractMethodRefactoring_declare_thrown_exceptions);
if (fReplaceDuplicates)
comment.addSetting(RefactoringCoreMessages.ExtractMethodRefactoring_replace_occurrences);
if (fGenerateJavadoc)
comment.addSetting(RefactoringCoreMessages.ExtractMethodRefactoring_generate_comment);
final ExtractMethodDescriptor descriptor= RefactoringSignatureDescriptorFactory.createExtractMethodDescriptor(project, description, comment.asString(), arguments, flags);
arguments.put(JavaRefactoringDescriptorUtil.ATTRIBUTE_INPUT, JavaRefactoringDescriptorUtil.elementToHandle(project, fCUnit));
arguments.put(JavaRefactoringDescriptorUtil.ATTRIBUTE_NAME, fMethodName);
arguments.put(JavaRefactoringDescriptorUtil.ATTRIBUTE_SELECTION, Integer.toString(fSelectionStart) + " " + Integer.toString(fSelectionLength)); //$NON-NLS-1$
arguments.put(ATTRIBUTE_VISIBILITY, Integer.toString(fVisibility));
arguments.put(ATTRIBUTE_DESTINATION, Integer.toString(fDestinationIndex));
arguments.put(ATTRIBUTE_EXCEPTIONS, Boolean.toString(fThrowRuntimeExceptions));
arguments.put(ATTRIBUTE_COMMENTS, Boolean.toString(fGenerateJavadoc));
arguments.put(ATTRIBUTE_REPLACE, Boolean.toString(fReplaceDuplicates));
return descriptor;
}
/**
* Returns the signature of the new method.
*
* @return the signature of the extracted method
*/
public String getSignature() {
return getSignature(fMethodName);
}
/**
* Returns the signature of the new method.
*
* @param methodName the method name used for the new method
* @return the signature of the extracted method
*/
public String getSignature(String methodName) {
MethodDeclaration methodDecl= createNewMethodDeclaration();
methodDecl.setBody(null);
String str= ASTNodes.asString(methodDecl);
return str.substring(0, str.indexOf(';'));
}
/**
* Returns the number of duplicate code snippets found.
*
* @return the number of duplicate code fragments
*/
public int getNumberOfDuplicates() {
if (fDuplicates == null)
return 0;
int result=0;
for (Match duplicate : fDuplicates) {
if (!duplicate.isInvalidNode())
result++;
}
return result;
}
public boolean getReplaceDuplicates() {
return fReplaceDuplicates;
}
public void setReplaceDuplicates(boolean replace) {
fReplaceDuplicates= replace;
}
public void setGenerateJavadoc(boolean generate) {
fGenerateJavadoc= generate;
}
public boolean getGenerateJavadoc() {
return fGenerateJavadoc;
}
public boolean isDestinationInterface() {
return fDestination instanceof TypeDeclaration && ((TypeDeclaration) fDestination).isInterface();
}
//---- Helper methods ------------------------------------------------------------------------
private void initializeParameterInfos() {
IVariableBinding[] arguments= fAnalyzer.getArguments();
fParameterInfos= new ArrayList<>(arguments.length);
ASTNode root= fAnalyzer.getEnclosingBodyDeclaration();
ParameterInfo vararg= null;
for (int i= 0; i < arguments.length; i++) {
IVariableBinding argument= arguments[i];
if (argument == null)
continue;
VariableDeclaration declaration= ASTNodes.findVariableDeclaration(argument, root);
boolean isVarargs= declaration instanceof SingleVariableDeclaration
? ((SingleVariableDeclaration)declaration).isVarargs()
: false;
ParameterInfo info= new ParameterInfo(argument, getType(declaration, isVarargs, false), argument.getName(), i);
if (isVarargs) {
vararg= info;
} else {
fParameterInfos.add(info);
}
}
if (vararg != null) {
fParameterInfos.add(vararg);
}
}
private void initializeUsedNames() {
fUsedNames= UsedNamesCollector.perform(fAnalyzer.getSelectedNodes());
for (ParameterInfo parameter : fParameterInfos) {
fUsedNames.remove(parameter.getOldName());
}
}
private void initializeDuplicates() {
ASTNode start= fAnalyzer.getEnclosingBodyDeclaration();
while (!(start instanceof AbstractTypeDeclaration)) {
start= start.getParent();
}
fDuplicates= findValidDuplicates(start);
fReplaceDuplicates= fDuplicates.size() > 0 && !fAnalyzer.isLiteralNodeSelected();
}
private List<SnippetFinder.Match> findValidDuplicates(ASTNode startNode) {
List<Match> duplicates= SnippetFinder.perform(startNode, fAnalyzer.getSelectedNodes());
List<SnippetFinder.Match> validDuplicates= new ArrayList<>();
for (Match duplicate : duplicates) {
if (duplicate != null && !duplicate.isInvalidNode()) {
try {
ASTNode[] nodes= duplicate.getNodes();
int duplicateStart= nodes[0].getStartPosition();
ASTNode lastNode= nodes[nodes.length - 1];
int duplicateEnd= lastNode.getStartPosition() + lastNode.getLength();
int duplicateLength= duplicateEnd - duplicateStart;
ExtractMethodAnalyzer analyzer= new ExtractMethodAnalyzer(fCUnit, Selection.createFromStartLength(duplicateStart, duplicateLength));
fRoot.accept(analyzer);
RefactoringStatus result= new RefactoringStatus();
result.merge(analyzer.checkInitialConditions(fImportRewriter));
if (!result.hasFatalError()) {
ITypeBinding originalReturnTypeBinding= fAnalyzer.getReturnTypeBinding();
ITypeBinding duplicateReturnTypeBinding= analyzer.getReturnTypeBinding();
if (originalReturnTypeBinding == null && duplicateReturnTypeBinding == null) {
validDuplicates.add(duplicate);
} else if (originalReturnTypeBinding != null && duplicateReturnTypeBinding != null) {
if (!originalReturnTypeBinding.equals(duplicateReturnTypeBinding)) {
if (duplicateReturnTypeBinding.equals(startNode.getAST().resolveWellKnownType("void"))) { //$NON-NLS-1$
// extracted snippet returns non-void and duplicate snippet returns void => OK
validDuplicates.add(duplicate);
}
} else {
IVariableBinding originalReturnValBinding= fAnalyzer.getReturnValue();
IVariableBinding duplicateReturnValBinding= analyzer.getReturnValue();
if (originalReturnValBinding == null && duplicateReturnValBinding == null) {
validDuplicates.add(duplicate);
} else if (originalReturnValBinding != null && duplicateReturnValBinding != null) {
BodyDeclaration originalEnclosingBodyDeclaration= fAnalyzer.getEnclosingBodyDeclaration();
BodyDeclaration duplicateEnclosingBodyDeclaration= analyzer.getEnclosingBodyDeclaration();
VariableDeclaration originalReturnNode= ASTNodes.findVariableDeclaration(originalReturnValBinding, originalEnclosingBodyDeclaration);
VariableDeclaration duplicateReturnNode= ASTNodes.findVariableDeclaration(duplicateReturnValBinding, duplicateEnclosingBodyDeclaration);
if (originalReturnNode != null && duplicateReturnNode != null) {
boolean matches;
if (!fAnalyzer.getSelection().covers(originalReturnNode) && !analyzer.getSelection().covers(duplicateReturnNode)) {
// returned variables are defined outside of the selection => always OK
matches= true;
} else {
matches= matchesLocationInEnclosingBodyDecl(originalEnclosingBodyDeclaration, duplicateEnclosingBodyDeclaration, originalReturnNode, duplicateReturnNode);
}
if (matches) {
validDuplicates.add(duplicate);
}
}
}
}
}
}
} catch (CoreException e) {
// consider as invalid duplicate
}
}
}
return validDuplicates;
}
private boolean matchesLocationInEnclosingBodyDecl(BodyDeclaration originalEnclosingBodyDeclaration, BodyDeclaration duplicateEnclosingBodyDeclaration,
VariableDeclaration originalReturnNode, VariableDeclaration duplicateReturnNode) {
boolean matches= true;
ASTNode original= originalReturnNode;
ASTNode dupliacte= duplicateReturnNode;
// walk up the parent chains to check if the location of the return nodes in their respective parent chains is same
do {
ASTNode originalParent= original.getParent();
ASTNode duplicateParent= dupliacte.getParent();
StructuralPropertyDescriptor originalLoc= original.getLocationInParent();
StructuralPropertyDescriptor duplicateLoc= dupliacte.getLocationInParent();
if (originalParent != null && duplicateParent != null
&& originalLoc.getNodeClass().equals(duplicateLoc.getNodeClass())
&& originalLoc.getId().equals(duplicateLoc.getId())) {
if (originalLoc.isChildListProperty() && duplicateLoc.isChildListProperty()) {
int indexOfOriginal= ((List<?>) originalParent.getStructuralProperty(originalLoc)).indexOf(original);
int indexOfDuplicate= ((List<?>) duplicateParent.getStructuralProperty(duplicateLoc)).indexOf(dupliacte);
if (indexOfOriginal != indexOfDuplicate) {
matches= false;
break;
}
}
} else {
matches= false;
break;
}
original= originalParent;
dupliacte= duplicateParent;
if ((originalEnclosingBodyDeclaration.equals(original) && !duplicateEnclosingBodyDeclaration.equals(dupliacte))
|| (!originalEnclosingBodyDeclaration.equals(original) && duplicateEnclosingBodyDeclaration.equals(dupliacte))) {
matches= false;
break;
}
} while (!originalEnclosingBodyDeclaration.equals(original) && !duplicateEnclosingBodyDeclaration.equals(dupliacte));
return matches;
}
private void initializeDestinations() {
List<ASTNode> result= new ArrayList<>();
BodyDeclaration decl= fAnalyzer.getEnclosingBodyDeclaration();
ASTNode current= ASTResolving.findParentType(decl.getParent());
if (fAnalyzer.isValidDestination(current)) {
result.add(current);
}
if (current != null && (decl instanceof MethodDeclaration || decl instanceof Initializer || decl instanceof FieldDeclaration)) {
ITypeBinding binding= ASTNodes.getEnclosingType(current);
ASTNode next= ASTResolving.findParentType(current.getParent());
while (next != null && binding != null && binding.isNested()) {
if (fAnalyzer.isValidDestination(next)) {
result.add(next);
}
current= next;
binding= ASTNodes.getEnclosingType(current);
next= ASTResolving.findParentType(next.getParent());
}
}
fDestinations= result.toArray(new ASTNode[result.size()]);
fDestination= fDestinations[fDestinationIndex];
}
private RefactoringStatus mergeTextSelectionStatus(RefactoringStatus status) {
status.addFatalError(RefactoringCoreMessages.ExtractMethodRefactoring_no_set_of_statements);
return status;
}
private String getType(VariableDeclaration declaration, boolean isVarargs) {
String type= ASTNodes.asString(ASTNodeFactory.newType(declaration.getAST(), declaration, fImportRewriter, new ContextSensitiveImportRewriteContext(declaration, fImportRewriter)));
if (isVarargs)
return type + ParameterInfo.ELLIPSIS;
else
return type;
}
private String getType(VariableDeclaration declaration, boolean isVarargs, boolean isVarTypeAllowed) {
if (isVarTypeAllowed) {
return getType(declaration, isVarargs);
} else {
String type= ASTNodes.asString(ASTNodeFactory.newNonVarType(declaration.getAST(), declaration, fImportRewriter, new ContextSensitiveImportRewriteContext(declaration, fImportRewriter)));
if (isVarargs)
return type + ParameterInfo.ELLIPSIS;
else
return type;
}
}
//---- Code generation -----------------------------------------------------------------------
private ASTNode[] createCallNodes(SnippetFinder.Match duplicate, int modifiers) {
List<ASTNode> result= new ArrayList<>(2);
for (IVariableBinding local : fAnalyzer.getCallerLocals()) {
result.add(createDeclaration(local, null));
}
MethodInvocation invocation= fAST.newMethodInvocation();
invocation.setName(fAST.newSimpleName(fMethodName));
ASTNode typeNode= ASTResolving.findParentType(fAnalyzer.getEnclosingBodyDeclaration());
RefactoringStatus status= new RefactoringStatus();
while (fDestination != typeNode) {
fAnalyzer.checkInput(status, fMethodName, typeNode);
if (!status.isOK()) {
SimpleName destinationTypeName= fAST.newSimpleName(ASTNodes.getEnclosingType(fDestination).getName());
if ((modifiers & Modifier.STATIC) == 0) {
ThisExpression thisExpression= fAST.newThisExpression();
thisExpression.setQualifier(destinationTypeName);
invocation.setExpression(thisExpression);
} else {
invocation.setExpression(destinationTypeName);
}
break;
}
typeNode= typeNode.getParent();
}
List<Expression> arguments= invocation.arguments();
for (ParameterInfo parameter : fParameterInfos) {
arguments.add(ASTNodeFactory.newName(fAST, getMappedName(duplicate, parameter)));
}
if (fLinkedProposalModel != null) {
LinkedProposalPositionGroup nameGroup= fLinkedProposalModel.getPositionGroup(KEY_NAME, true);
nameGroup.addPosition(fRewriter.track(invocation.getName()), false);
}
ASTNode call;
int returnKind= fAnalyzer.getReturnKind();
switch (returnKind) {
case ExtractMethodAnalyzer.ACCESS_TO_LOCAL:
IVariableBinding binding= fAnalyzer.getReturnLocal();
if (binding != null) {
VariableDeclarationStatement decl= createDeclaration(getMappedBinding(duplicate, binding), invocation);
call= decl;
} else {
Assignment assignment= fAST.newAssignment();
assignment.setLeftHandSide(ASTNodeFactory.newName(fAST,
getMappedBinding(duplicate, fAnalyzer.getReturnValue()).getName()));
assignment.setRightHandSide(invocation);
call= assignment;
}
break;
case ExtractMethodAnalyzer.RETURN_STATEMENT_VALUE:
ReturnStatement rs= fAST.newReturnStatement();
rs.setExpression(invocation);
call= rs;
break;
default:
call= invocation;
}
if (call instanceof Expression && !fAnalyzer.isExpressionSelected()) {
call= fAST.newExpressionStatement((Expression)call);
}
result.add(call);
return result.toArray(new ASTNode[result.size()]);
}
private IVariableBinding getMappedBinding(SnippetFinder.Match duplicate, IVariableBinding org) {
if (duplicate == null)
return org;
return duplicate.getMappedBinding(org);
}
private String getMappedName(SnippetFinder.Match duplicate, ParameterInfo paramter) {
if (duplicate == null)
return paramter.getOldName();
return duplicate.getMappedName(paramter.getOldBinding()).getIdentifier();
}
private void replaceDuplicates(CompilationUnitChange result, int modifiers) {
int numberOf= getNumberOfDuplicates();
if (numberOf == 0 || !fReplaceDuplicates)
return;
String label= null;
if (numberOf == 1)
label= Messages.format(RefactoringCoreMessages.ExtractMethodRefactoring_duplicates_single, BasicElementLabels.getJavaElementName(fMethodName));
else
label= Messages.format(RefactoringCoreMessages.ExtractMethodRefactoring_duplicates_multi, BasicElementLabels.getJavaElementName(fMethodName));
TextEditGroup description= new TextEditGroup(label);
result.addTextEditGroup(description);
for (Match duplicate : fDuplicates) {
if (!duplicate.isInvalidNode()) {
if (isDestinationReachable(duplicate.getEnclosingMethod())) {
ASTNode[] callNodes= createCallNodes(duplicate, modifiers);
ASTNode[] duplicateNodes= duplicate.getNodes();
for (int i= 0; i < duplicateNodes.length; i++) {
while (duplicateNodes[i].getParent() instanceof ParenthesizedExpression) {
duplicateNodes[i]= duplicateNodes[i].getParent();
}
}
new StatementRewrite(fRewriter, duplicateNodes).replace(callNodes, description);
}
}
}
}
private boolean forceStatic(){
if(!fReplaceDuplicates){
return false;
}
for (Match duplicate : fDuplicates) {
if(!duplicate.isInvalidNode() && duplicate.isNodeInStaticContext()) {
return true;
}
}
return false;
}
private boolean isDestinationReachable(MethodDeclaration methodDeclaration) {
ASTNode start= methodDeclaration;
while (start != null && start != fDestination) {
start= start.getParent();
}
return start == fDestination;
}
private MethodDeclaration createNewMethod(ASTNode[] selectedNodes, String lineDelimiter, TextEditGroup substitute) throws CoreException {
MethodDeclaration result= createNewMethodDeclaration();
result.setBody(createMethodBody(selectedNodes, substitute, result.getModifiers()));
if (fGenerateJavadoc) {
AbstractTypeDeclaration enclosingType=
ASTNodes.getParent(fAnalyzer.getEnclosingBodyDeclaration(), AbstractTypeDeclaration.class);
String string= CodeGeneration.getMethodComment(fCUnit, enclosingType.getName().getIdentifier(), result, null, lineDelimiter);
if (string != null) {
Javadoc javadoc= (Javadoc)fRewriter.createStringPlaceholder(string, ASTNode.JAVADOC);
result.setJavadoc(javadoc);
}
}
return result;
}
private MethodDeclaration createNewMethodDeclaration() {
MethodDeclaration result= fAST.newMethodDeclaration();
int modifiers= fVisibility;
BodyDeclaration enclosingBodyDeclaration= fAnalyzer.getEnclosingBodyDeclaration();
boolean isDestinationInterface= isDestinationInterface();
if (isDestinationInterface
&& ((!(enclosingBodyDeclaration instanceof MethodDeclaration)
|| (enclosingBodyDeclaration.getParent() != fDestination))
|| !Modifier.isPublic(enclosingBodyDeclaration.getModifiers()))) {
modifiers= Modifier.NONE;
}
boolean shouldBeStatic= false;
ASTNode currentParent= enclosingBodyDeclaration;
do {
if (currentParent instanceof BodyDeclaration) {
shouldBeStatic= shouldBeStatic || JdtFlags.isStatic((BodyDeclaration) currentParent);
}
currentParent= currentParent.getParent();
} while (!shouldBeStatic && currentParent != null && currentParent != fDestination);
if (shouldBeStatic || fAnalyzer.getForceStatic() || forceStatic()) {
modifiers|= Modifier.STATIC;
} else if (isDestinationInterface) {
modifiers|= Modifier.DEFAULT;
}
List<TypeParameter> typeParameters= result.typeParameters();
for (ITypeBinding typeVariable : computeLocalTypeVariables(modifiers)) {
TypeParameter parameter= fAST.newTypeParameter();
parameter.setName(fAST.newSimpleName(typeVariable.getName()));
for (ITypeBinding bound : typeVariable.getTypeBounds()) {
if (!"java.lang.Object".equals(bound.getQualifiedName())) { //$NON-NLS-1$
parameter.typeBounds().add(fImportRewriter.addImport(bound, fAST));
}
}
typeParameters.add(parameter);
}
result.modifiers().addAll(ASTNodeFactory.newModifiers(fAST, modifiers));
result.setReturnType2((Type)ASTNode.copySubtree(fAST, fAnalyzer.getReturnType()));
result.setName(fAST.newSimpleName(fMethodName));
ImportRewriteContext context= new ContextSensitiveImportRewriteContext(enclosingBodyDeclaration, fImportRewriter);
List<SingleVariableDeclaration> parameters= result.parameters();
for (ParameterInfo info : fParameterInfos) {
VariableDeclaration infoDecl= getVariableDeclaration(info);
SingleVariableDeclaration parameter= fAST.newSingleVariableDeclaration();
parameter.modifiers().addAll(ASTNodeFactory.newModifiers(fAST, ASTNodes.getModifiers(infoDecl)));
parameter.setType(ASTNodeFactory.newNonVarType(fAST, infoDecl, fImportRewriter, context));
parameter.setName(fAST.newSimpleName(info.getNewName()));
parameter.setVarargs(info.isNewVarargs());
parameters.add(parameter);
}
List<Type> exceptions= result.thrownExceptionTypes();
for (ITypeBinding exceptionType : fAnalyzer.getExceptions(fThrowRuntimeExceptions)) {
exceptions.add(fImportRewriter.addImport(exceptionType, fAST, context, TypeLocation.EXCEPTION));
}
return result;
}
private ITypeBinding[] computeLocalTypeVariables(int modifier) {
List<ITypeBinding> result= new ArrayList<>(Arrays.asList(fAnalyzer.getTypeVariables()));
for (ParameterInfo info : fParameterInfos) {
processVariable(result, info.getOldBinding(), modifier);
}
for (IVariableBinding methodLocal : fAnalyzer.getMethodLocals()) {
processVariable(result, methodLocal, modifier);
}
return result.toArray(new ITypeBinding[result.size()]);
}
private void processVariable(List<ITypeBinding> result, IVariableBinding variable, int modifier) {
if (variable == null)
return;
ITypeBinding binding= variable.getType();
if (binding != null && binding.isParameterizedType()) {
for (ITypeBinding arg : binding.getTypeArguments()) {
if (arg.isTypeVariable() && !result.contains(arg)) {
ASTNode decl= fRoot.findDeclaringNode(arg);
if (decl != null) {
ASTNode parent= decl.getParent();
if (parent instanceof MethodDeclaration || (parent instanceof TypeDeclaration && Modifier.isStatic(modifier))) {
result.add(arg);
}
}
} else {
ITypeBinding bound= arg.getBound();
if (arg.isWildcardType() && bound != null && !result.contains(bound)) {
ASTNode decl= fRoot.findDeclaringNode(bound);
if (decl != null) {
ASTNode parent= decl.getParent();
if (parent instanceof MethodDeclaration || (parent instanceof TypeDeclaration && Modifier.isStatic(modifier))) {
result.add(bound);
}
}
}
}
}
}
}
private Block createMethodBody(ASTNode[] selectedNodes, TextEditGroup substitute, int modifiers) {
Block result= fAST.newBlock();
ListRewrite statements= fRewriter.getListRewrite(result, Block.STATEMENTS_PROPERTY);
// Locals that are not passed as an arguments since the extracted method only
// writes to them
for (IVariableBinding methodLocal : fAnalyzer.getMethodLocals()) {
if (methodLocal != null) {
result.statements().add(createDeclaration(methodLocal, null));
}
}
for (ParameterInfo parameter : fParameterInfos) {
if (parameter.isRenamed()) {
for (ASTNode selectedNode : selectedNodes) {
for (SimpleName oldName : LinkedNodeFinder.findByBinding(selectedNode, parameter.getOldBinding())) {
fRewriter.replace(oldName, fAST.newSimpleName(parameter.getNewName()), null);
}
}
}
}
boolean extractsExpression= fAnalyzer.isExpressionSelected();
ASTNode[] callNodes= createCallNodes(null, modifiers);
ASTNode replacementNode;
if (callNodes.length == 1) {
replacementNode= callNodes[0];
} else {
replacementNode= fRewriter.createGroupNode(callNodes);
}
if (extractsExpression) {
// if we have an expression then only one node is selected.
ITypeBinding binding= fAnalyzer.getExpressionBinding();
if (binding != null && (!binding.isPrimitive() || !"void".equals(binding.getName()))) { //$NON-NLS-1$
ReturnStatement rs= fAST.newReturnStatement();
rs.setExpression((Expression)fRewriter.createMoveTarget(ASTNodes.getUnparenthesedExpression(selectedNodes[0])));
statements.insertLast(rs, null);
} else {
ExpressionStatement st= fAST.newExpressionStatement((Expression)fRewriter.createMoveTarget(selectedNodes[0]));
statements.insertLast(st, null);
}
ASTNode parenthesizedNode= selectedNodes[0];
while (parenthesizedNode.getParent() instanceof ParenthesizedExpression) {
parenthesizedNode= parenthesizedNode.getParent();
}
fRewriter.replace(parenthesizedNode, replacementNode, substitute);
} else {
boolean isReturnVoid= selectedNodes[selectedNodes.length - 1] instanceof ReturnStatement &&
fAnalyzer.getReturnTypeBinding().equals(fAST.resolveWellKnownType("void")); //$NON-NLS-1$
if (selectedNodes.length == 1) {
if (!isReturnVoid) {
if (selectedNodes[0] instanceof Block) {
Block block= (Block)selectedNodes[0];
List<Statement> blockStatements= block.statements();
for (Statement blockStatement : blockStatements) {
statements.insertLast(fRewriter.createMoveTarget(blockStatement), substitute);
}
} else {
statements.insertLast(fRewriter.createMoveTarget(selectedNodes[0]), substitute);
}
}
if (selectedNodes[0].getLocationInParent() == LambdaExpression.BODY_PROPERTY) {
if (replacementNode instanceof ExpressionStatement) {
fRewriter.replace(selectedNodes[0], ((ExpressionStatement)replacementNode).getExpression(), substitute);
} else if (replacementNode instanceof ReturnStatement) {
fRewriter.replace(selectedNodes[0], ((ReturnStatement)replacementNode).getExpression(), substitute);
}
} else {
fRewriter.replace(selectedNodes[0], replacementNode, substitute);
}
} else if (selectedNodes.length > 1) {
if (isReturnVoid) {
fRewriter.remove(selectedNodes[selectedNodes.length - 1], substitute);
}
ListRewrite source= fRewriter.getListRewrite(
selectedNodes[0].getParent(),
(ChildListPropertyDescriptor) selectedNodes[0].getLocationInParent());
// if last statement is a void return statement then we skip it
int index= isReturnVoid ? selectedNodes.length - 2 : selectedNodes.length - 1;
ASTNode toMove= source.createMoveTarget(
selectedNodes[0], selectedNodes[index],
replacementNode, substitute);
statements.insertLast(toMove, substitute);
}
IVariableBinding returnValue= fAnalyzer.getReturnValue();
if (returnValue != null) {
ReturnStatement rs= fAST.newReturnStatement();
rs.setExpression(fAST.newSimpleName(getName(returnValue)));
statements.insertLast(rs, null);
}
}
return result;
}
private String getName(IVariableBinding binding) {
for (ParameterInfo info : fParameterInfos) {
if (Bindings.equals(binding, info.getOldBinding())) {
return info.getNewName();
}
}
return binding.getName();
}
private VariableDeclaration getVariableDeclaration(ParameterInfo parameter) {
return ASTNodes.findVariableDeclaration(parameter.getOldBinding(), fAnalyzer.getEnclosingBodyDeclaration());
}
private VariableDeclarationStatement createDeclaration(IVariableBinding binding, Expression intilizer) {
VariableDeclaration original= ASTNodes.findVariableDeclaration(binding, fAnalyzer.getEnclosingBodyDeclaration());
VariableDeclarationFragment fragment= fAST.newVariableDeclarationFragment();
fragment.setName((SimpleName)ASTNode.copySubtree(fAST, original.getName()));
fragment.setInitializer(intilizer);
VariableDeclarationStatement result= fAST.newVariableDeclarationStatement(fragment);
result.modifiers().addAll(ASTNode.copySubtrees(fAST, ASTNodes.getModifiers(original)));
result.setType(ASTNodeFactory.newType(fAST, original, fImportRewriter, new ContextSensitiveImportRewriteContext(original, fImportRewriter)));
return result;
}
public ICompilationUnit getCompilationUnit() {
return fCUnit;
}
private RefactoringStatus initialize(JavaRefactoringArguments arguments) {
final String selection= arguments.getAttribute(JavaRefactoringDescriptorUtil.ATTRIBUTE_SELECTION);
if (selection == null)
return RefactoringStatus.createFatalErrorStatus(Messages.format(RefactoringCoreMessages.InitializableRefactoring_argument_not_exist, JavaRefactoringDescriptorUtil.ATTRIBUTE_SELECTION));
int offset= -1;
int length= -1;
final StringTokenizer tokenizer= new StringTokenizer(selection);
if (tokenizer.hasMoreTokens())
offset= Integer.parseInt(tokenizer.nextToken());
if (tokenizer.hasMoreTokens())
length= Integer.parseInt(tokenizer.nextToken());
if (offset < 0 || length < 0)
return RefactoringStatus.createFatalErrorStatus(Messages.format(RefactoringCoreMessages.InitializableRefactoring_illegal_argument, new Object[] { selection, JavaRefactoringDescriptorUtil.ATTRIBUTE_SELECTION}));
fSelectionStart= offset;
fSelectionLength= length;
final String handle= arguments.getAttribute(JavaRefactoringDescriptorUtil.ATTRIBUTE_INPUT);
if (handle == null)
return RefactoringStatus.createFatalErrorStatus(Messages.format(RefactoringCoreMessages.InitializableRefactoring_argument_not_exist, JavaRefactoringDescriptorUtil.ATTRIBUTE_INPUT));
IJavaElement element= JavaRefactoringDescriptorUtil.handleToElement(arguments.getProject(), handle, false);
if (element == null || !element.exists() || element.getElementType() != IJavaElement.COMPILATION_UNIT)
return JavaRefactoringDescriptorUtil.createInputFatalStatus(element, getName(), IJavaRefactorings.EXTRACT_METHOD);
fCUnit= (ICompilationUnit) element;
final String visibility= arguments.getAttribute(ATTRIBUTE_VISIBILITY);
if (visibility != null && visibility.length() != 0) {
int flag= 0;
try {
flag= Integer.parseInt(visibility);
} catch (NumberFormatException exception) {
return RefactoringStatus.createFatalErrorStatus(Messages.format(RefactoringCoreMessages.InitializableRefactoring_argument_not_exist, ATTRIBUTE_VISIBILITY));
}
fVisibility= flag;
}
final String name= arguments.getAttribute(JavaRefactoringDescriptorUtil.ATTRIBUTE_NAME);
if (name == null || name.length() == 0)
return RefactoringStatus.createFatalErrorStatus(Messages.format(RefactoringCoreMessages.InitializableRefactoring_argument_not_exist, JavaRefactoringDescriptorUtil.ATTRIBUTE_NAME));
fMethodName= name;
final String destination= arguments.getAttribute(ATTRIBUTE_DESTINATION);
if (destination != null && destination.length() == 0) {
int index= 0;
try {
index= Integer.parseInt(destination);
} catch (NumberFormatException exception) {
return RefactoringStatus.createFatalErrorStatus(Messages.format(RefactoringCoreMessages.InitializableRefactoring_argument_not_exist, ATTRIBUTE_DESTINATION));
}
fDestinationIndex= index;
}
final String replace= arguments.getAttribute(ATTRIBUTE_REPLACE);
if (replace == null)
return RefactoringStatus.createFatalErrorStatus(Messages.format(RefactoringCoreMessages.InitializableRefactoring_argument_not_exist, ATTRIBUTE_REPLACE));
fReplaceDuplicates= Boolean.parseBoolean(replace);
final String comments= arguments.getAttribute(ATTRIBUTE_COMMENTS);
if (comments == null)
return RefactoringStatus.createFatalErrorStatus(Messages.format(RefactoringCoreMessages.InitializableRefactoring_argument_not_exist, ATTRIBUTE_COMMENTS));
fGenerateJavadoc= Boolean.parseBoolean(comments);
final String exceptions= arguments.getAttribute(ATTRIBUTE_EXCEPTIONS);
if (exceptions == null)
return RefactoringStatus.createFatalErrorStatus(Messages.format(RefactoringCoreMessages.InitializableRefactoring_argument_not_exist, ATTRIBUTE_EXCEPTIONS));
fThrowRuntimeExceptions= Boolean.parseBoolean(exceptions);
return new RefactoringStatus();
}
}