ITextFileBuffer textFileBuffer = bufferManager.getTextFileBuffer(path,
LocationKind.IFILE);
IDocument document = textFileBuffer.getDocument();
DocumentChange anotherChange = new DocumentChange(RefactoringMessages.ExtractMethodPreviewPage_TextChangeName, document);
MultiTextEdit rootEdit = new MultiTextEdit();
anotherChange.setEdit(rootEdit);
TextEditGroup newMethodEdit = new TextEditGroup(Messages.format(RefactoringMessages.ExtractMethodPreviewPage_TextChangeNewMethod, fMethodName));
TextEditGroup inlineReplacementEdit = new TextEditGroup(Messages.format(RefactoringMessages.ExtractMethodPreviewPage_TextChangeSubstituteStatements, fMethodName));
TextEditGroup additionalInlineReplacementEdit = new TextEditGroup(Messages.format(RefactoringMessages.ExtractMethodPreviewPage_TextChangeSubsituteDuplicateStatements, fMethodName));
anotherChange.addTextEditGroup(newMethodEdit);
anotherChange.addTextEditGroup(inlineReplacementEdit);
anotherChange.addTextEditGroup(additionalInlineReplacementEdit);
AST ast = fProgram.getAST();
MethodDeclaration method = ast.newMethodDeclaration();
Block extractedMethodBody = ast.newBlock();
FunctionDeclaration functionDec = ast.newFunctionDeclaration(ast.newIdentifier(fMethodName), computeArguments(ast), extractedMethodBody, false);
method.setModifier(fModifierAccessFlag);
method.setFunction(functionDec);
ASTRewrite rewriter = ASTRewrite.create(ast);
ListRewrite classListRewrite = rewriter.getListRewrite( fCoveringDeclarationFinder.getCoveringClassDeclaration().getBody(), Block.STATEMENTS_PROPERTY);
VariableBase dispatcher = ast.newVariable(THIS_VARIABLE_NAME);
FunctionInvocation calledExtractedMethod = ast.newFunctionInvocation(ast.newFunctionName(ast.newIdentifier(fMethodName)), computeParameters(ast));
MethodInvocation inlineMethodCall = ast.newMethodInvocation(dispatcher, calledExtractedMethod);
List<List<ASTNode>> Occurences = new ArrayList<List<ASTNode>>();
if(fReplaceDuplicates) {
for(Match replace : fDuplicates) {
Occurences.add(Arrays.asList(replace.getNodes()));
}
} else {
Occurences.add(fSelectedNodesFinder.getNodes());
}
boolean createdMethodBody = false;
TextEditGroup inlineReplacementEditGroup = inlineReplacementEdit;
for(List<ASTNode> selectedNodeOccurence : Occurences) {
// this is also an indicator, whether this loop was already gone through
if(createdMethodBody) {
inlineReplacementEditGroup = additionalInlineReplacementEdit;
}
ASTNode parent = selectedNodeOccurence.get(0).getParent();
inlineMethodCall = ASTNode.copySubtree(ast, inlineMethodCall);
ListRewrite lrw;
if(parent instanceof Block) {
if(!createdMethodBody) {
extractedMethodBody.statements().addAll(ASTNode.copySubtrees(ast, selectedNodeOccurence));
addReturnStatement(ast, extractedMethodBody, fReturnStatement);
createdMethodBody = true;
}
lrw = rewriter.getListRewrite(parent, Block.STATEMENTS_PROPERTY);
ExpressionStatement inlineReplacement;
if (fReturnStatement != null) {
inlineReplacement = ast.newExpressionStatement(ast.newAssignment(
ast.newVariable(fReturnStatement.getParameterName()),
Assignment.OP_EQUAL, inlineMethodCall));
} else {
inlineReplacement = ast.newExpressionStatement(inlineMethodCall);
}
lrw.replace(selectedNodeOccurence.get(0),inlineReplacement, inlineReplacementEditGroup);
for (int i = 1; i < selectedNodeOccurence.size(); ++i) {
lrw.remove(selectedNodeOccurence.get(i), inlineReplacementEditGroup);
}
} else {
if(!createdMethodBody) {
addReturnStatement(ast, extractedMethodBody, ASTNode.copySubtree(ast, selectedNodeOccurence.get(0)));
createdMethodBody = true;
}
rewriter.replace( selectedNodeOccurence.get(0), inlineMethodCall, inlineReplacementEditGroup);
}
}
classListRewrite.insertAfter(method, fCoveringDeclarationFinder.getCoveringMethodDeclaration(), newMethodEdit);
TextEdit fullDocumentEdit = rewriter.rewriteAST(document, null);
anotherChange.addEdit(fullDocumentEdit);
return anotherChange;
}