CompilationUnitTextSelection: can now find local declaring types

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

package no.uio.ifi.refaktor.utils;

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

import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.resources.IFile;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jface.text.Document;
import org.eclipse.jface.text.IDocument;
import org.eclipse.jface.text.ITextSelection;
import org.eclipse.jface.text.TextSelection;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.handlers.HandlerUtil;
import org.eclipse.ui.ide.ResourceUtil;
import org.eclipse.ui.texteditor.ITextEditor;

public class DocumentUtils {

	private static TextSelection getTextSelectionFrom(ExecutionEvent event) throws ExecutionException {
		ISelection selection = HandlerUtil.getCurrentSelectionChecked(event);
		assertThat(selection, instanceOf(TextSelection.class));
		return (TextSelection) selection;
	}

	public static ITextEditor getEditorFrom(ExecutionEvent event) throws ExecutionException {
		IEditorPart part = HandlerUtil.getActiveEditorChecked(event);
		assertThat(part, instanceOf(ITextEditor.class));
		return (ITextEditor) part;
	}

	public static IDocument getDocumentFrom(ITextEditor editor) {
		return editor.getDocumentProvider().getDocument(editor.getEditorInput());
	}

	public static CompilationUnitTextSelection getStrippedCompilationUnitTextSelectionFromExecutionEvent(ExecutionEvent event) 
			throws ExecutionException {
		ICompilationUnit compilationUnit = getCompilationUnitFromExecutionEvent(event);
		IDocument document = DocumentUtils.getDocumentFromCompilationUnit(compilationUnit);
		ITextSelection strippedTextSelection = ParseUtils.stripTextSelection(document, getTextSelectionFrom(event));
		return new CompilationUnitTextSelection(compilationUnit, strippedTextSelection);
	}

	public static CompilationUnitTextSelection getCompilationUnitTextSelectionFromExecutionEvent(ExecutionEvent event) 
			throws ExecutionException {
		return new CompilationUnitTextSelection(getCompilationUnitFromExecutionEvent(event), getTextSelectionFrom(event));
	}

	private static ICompilationUnit getCompilationUnitFromExecutionEvent(ExecutionEvent event)
			throws ExecutionException {
		return RefaktorHandleUtils.getCompilationUnitFromEditor(getEditorFrom(event));
	}

	public static void setSelectionForEditor(ISelection selection, ITextEditor editor) {
		editor.getSite().getSelectionProvider().setSelection(selection);
	}

	public static IDocument getDocumentFromCompilationUnit(ICompilationUnit compilationUnit) {
		assertThat(compilationUnit, notNullValue());
		try {
			return new Document(compilationUnit.getSource());
		} catch (JavaModelException e) {
			throw new RuntimeException(e);
		}
	}

	public static IFile getFileFromEditor(IEditorPart editor) {
		return ResourceUtil.getFile(editor.getEditorInput());
	}
}