Moving analyzers to no.uio.ifi.refaktor.analyze.analyzers

[ifi-stolz-refaktor.git] / software / no.uio.ifi.refaktor / src / no / uio / ifi / refaktor / analyze / analyzers / StatementsCreator.java

package no.uio.ifi.refaktor.analyze.analyzers;

import static no.uio.ifi.refaktor.RefaktorAssert.assertThat;
import static org.hamcrest.CoreMatchers.notNullValue;

import java.util.LinkedList;
import java.util.List;

import no.uio.ifi.refaktor.analyze.GenericAnalyzerException;
import no.uio.ifi.refaktor.utils.ParseUtils;

import org.eclipse.jdt.core.IMethod;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jdt.core.dom.CompilationUnit;
import org.eclipse.jdt.core.dom.MethodDeclaration;
import org.eclipse.jdt.core.dom.Statement;
import org.eclipse.jdt.internal.corext.refactoring.structure.ASTNodeSearchUtil;

@SuppressWarnings("restriction")
class StatementsCreator {

	private IMethod method;
	private List<Statement> statements;

	public StatementsCreator(IMethod method) {
		this.method = method;
	}

	List<Statement> getStatements() {
		if (statements == null) 
			createStatements();

		return statements;
	}

	private void createStatements() {
		statements = new LinkedList<Statement>();
		MethodDeclaration methodDeclaration = createMethodDeclaration();
		if (hasBody(methodDeclaration))
			addStatementsFrom(methodDeclaration);
	}

	private MethodDeclaration createMethodDeclaration() {
		try {
			CompilationUnit compilationUnit = ParseUtils.parse(method.getCompilationUnit());
			MethodDeclaration methodDeclaration = ASTNodeSearchUtil.getMethodDeclarationNode(method, compilationUnit);
			assertThat(methodDeclaration, notNullValue());
			return methodDeclaration;
		} catch (JavaModelException e) {
			throw new GenericAnalyzerException(e);
		}
	}

	private boolean hasBody(MethodDeclaration methodDecl) {
		return methodDecl.getBody() != null;
	}

	private void addStatementsFrom(MethodDeclaration methodDecl) {
		for (Object element: methodDecl.getBody().statements())
			statements.add((Statement) element);
	}

	public int getNumberOfStatements() {
		if (statements == null)
			return -1;

		return statements.size();
	}
}