Adding some logging capabilities to RefaktorDebug.

[ifi-stolz-refaktor.git] / software / no.uio.ifi.refaktor / src / no / uio / ifi / refaktor / utils / RefaktorHandleUtils.java

package no.uio.ifi.refaktor.utils;

import static no.uio.ifi.refaktor.RefaktorAssert.assertThat;
import no.uio.ifi.refaktor.exceptions.RefaktorException;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.IMethod;
import org.eclipse.jdt.core.IType;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.ui.IEditorPart;

public class RefaktorHandleUtils {

	public static ICompilationUnit getCompilationUnitFromEditor(IEditorPart editor) {
		return findCompilationUnitFromFile(DocumentUtils.getFileFromEditor(editor));
	}

	public static ICompilationUnit findCompilationUnitFromFile(IFile file) {
		return JavaCore.createCompilationUnitFrom(file);
	}

	public static ICompilationUnit findCompilationUnitFromProjectPackageAndTypeName(IProject project, String packageName, String simpleTypeName) {
		try {
			return findType(project, packageName, simpleTypeName).getCompilationUnit();
		} catch (JavaModelException e) {
			throw new RuntimeException(e);
		}
	}

	public static IType findType(IProject project, String packageName, String simpleTypeName) throws JavaModelException {
		return createJavaProjectFrom(project).findType(packageName + "." + simpleTypeName);
	}

	public static IMethod findMethodHandleChecked(IProject project, String packageName, String simpleTypeName, 
			String methodName, String[] methodSignature) {
		try {
			IMethod method = findMethodHandle(project, packageName, simpleTypeName, methodName, methodSignature);
			assertThat(method.exists());
			return method;
		} catch (JavaModelException e) {
			throw new RefaktorException("The requested method handle could not be found");
		}
	}

	public static IMethod findMethodHandle(IProject project, String packageName, String simpleTypeName, 
			String methodName, String[] methodSignature) throws JavaModelException {
		return findType(project, packageName, simpleTypeName).getMethod(methodName, methodSignature);
	}

	public static IMethod findMethodHandle(IProject project,
			String packageName, String simpleTypeName, String simpleSignature) throws JavaModelException {
		IMethod[] methods = findMethodsForType(project, packageName, simpleTypeName);
		return findMethodHandleThatMatchesSimpleSignature(methods, simpleSignature);
	}

	private static IMethod[] findMethodsForType(IProject project, String packageName, String simpleTypeName)
			throws JavaModelException {
		return findType(project, packageName, simpleTypeName).getMethods();
	}

	private static IMethod findMethodHandleThatMatchesSimpleSignature(IMethod[] methods, String simpleSignature)
			throws JavaModelException {
		for (IMethod method: methods) {
			String methodSource = method.getSource();
			if (methodSource == null)
				continue;
			if (makeSimpleSignature(methodSource).equals(simpleSignature)) 
				return method;
		}
		return null;
	}

	private static String makeSimpleSignature(String methodSource) {
		return methodSource.substring(0, methodSource.indexOf(')') + 1).trim();
	}

	public static IJavaProject createJavaProjectFrom(IProject project) {
		assertThat(project.exists());
		return JavaCore.create(project);
	}
}