Case Study: adding data and statistics

[ifi-stolz-refaktor.git] / case-study / refaktor-before / src / no / uio / ifi / refaktor / refactorings / visitors / InsertAssertsVisitor.java

package no.uio.ifi.refaktor.refactorings.visitors;

import java.util.List;

import org.eclipse.jdt.core.dom.ASTVisitor;
import org.eclipse.jdt.core.dom.AssertStatement;
import org.eclipse.jdt.core.dom.Block;
import org.eclipse.jdt.core.dom.InfixExpression;
import org.eclipse.jdt.core.dom.MethodDeclaration;
import org.eclipse.jdt.core.dom.Name;
import org.eclipse.jdt.core.dom.SingleVariableDeclaration;
import org.eclipse.jdt.core.dom.Statement;
import org.eclipse.jdt.core.dom.StringLiteral;
import org.eclipse.jdt.core.dom.rewrite.ASTRewrite;
import org.eclipse.jdt.core.dom.rewrite.ListRewrite;

public class InsertAssertsVisitor extends ASTVisitor {
	private ASTRewrite rewrite;

	@Override
	public boolean visit(MethodDeclaration node) {
		makeTheAssertRewrite(node);
		return false;
	}

	protected void makeTheAssertRewrite(MethodDeclaration node) {
		Block body = node.getBody();
		this.rewrite = ASTRewrite.create(body.getAST());
		ListRewrite statementsListRewrite = this.rewrite.getListRewrite(body, Block.STATEMENTS_PROPERTY);
		@SuppressWarnings("unchecked")
		List<Statement> originalList = (List<Statement>) statementsListRewrite.getOriginalList();
		
		for (Statement stmt: originalList) {
			if (stmt == originalList.get(originalList.size()-1))
				break;
			// TODO: Do we have to call getAST() every time again?
			AssertStatement assertStatement = body.getAST().newAssertStatement();
			InfixExpression infixExpression = body.getAST().newInfixExpression();
			Name qualifier = ((SingleVariableDeclaration) node.parameters().get(0)).getName();
			// TODO: how can we make sure that "x" is free?
			infixExpression.setLeftOperand(body.getAST().newQualifiedName(body.getAST().newName(qualifier.getFullyQualifiedName()) , body.getAST().newSimpleName("x")));
			infixExpression.setOperator(InfixExpression.Operator.EQUALS);
			infixExpression.setRightOperand(body.getAST().newThisExpression());
			assertStatement.setExpression(infixExpression);
			StringLiteral stringLiteral = body.getAST().newStringLiteral();
			stringLiteral.setLiteralValue("The Move Method refactoring has broken the program due to aliasing issues!");
			assertStatement.setMessage(stringLiteral);
			statementsListRewrite.insertAfter(assertStatement, stmt, null);
		}
	}
	
	public ASTRewrite getRewrite() {
		return rewrite;
	}
}