UnfixesCollector: checking explicitly for interface types

[ifi-stolz-refaktor.git] / software / no.uio.ifi.refaktor / src / no / uio / ifi / refaktor / analyze / collectors / EveryConditionalBranchEndsInReturnCheckerCollector.java

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

import no.uio.ifi.refaktor.analyze.CollectorManager;
import no.uio.ifi.refaktor.analyze.ReturnStatementFinder;
import no.uio.ifi.refaktor.analyze.analyzers.Checker;
import no.uio.ifi.refaktor.analyze.exceptions.IllegalExpressionFoundException;
import no.uio.ifi.refaktor.analyze.exceptions.IllegalStatementFoundException;
import no.uio.ifi.refaktor.utils.CompilationUnitTextSelection;

import org.eclipse.jdt.core.dom.IfStatement;
import org.eclipse.jdt.core.dom.ReturnStatement;
import org.eclipse.jdt.core.dom.Statement;
import org.eclipse.jdt.core.dom.SwitchStatement;

public class EveryConditionalBranchEndsInReturnCheckerCollector extends PropertyCollector implements Checker {

	public EveryConditionalBranchEndsInReturnCheckerCollector(CompilationUnitTextSelection selection) {
		super(selection);
	}

	@Override
	public void clearData() {
	}

	@Override
	public boolean visit(IfStatement node) {
		if (!nodeInSelection(node))
			return false;
		
		ReturnStatementFinder returnStatementFinder = new ReturnStatementFinder();
		node.getThenStatement().accept(returnStatementFinder);

		if (returnStatementFinder.hasFoundReturnStatement()) {
			Statement elseStatement = node.getElseStatement();

			if (elseStatement == null) {
				throw new IllegalStatementFoundException(ReturnStatement.class);
			}

			returnStatementFinder.reset();
			elseStatement.accept(returnStatementFinder);

			if (!returnStatementFinder.hasFoundReturnStatement()) {
				throw new IllegalStatementFoundException(ReturnStatement.class);
			}
		}

		return true;
	}
	
	@Override
	public boolean visit(SwitchStatement node) {
		// TODO: Handle switch statements in another way?
		if (!nodeInSelection(node))
			return false;
		
		node.accept(new IllegalReturnStatementChecker());
		
		return false;
	}

	@Override
	public void check() throws IllegalStatementFoundException, IllegalExpressionFoundException {
		CollectorManager.collectProperties(selection, this);
	}
}