MainMethodSearchHelper.java

package org.lejos.ev3.ldt.launch;

import java.lang.reflect.InvocationTargetException;
import java.util.Collection;

import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.SubProgressMonitor;
import org.eclipse.jdt.core.IMethod;
import org.eclipse.jdt.core.IType;
import org.eclipse.jdt.core.JavaModelException;
import org.eclipse.jdt.core.search.IJavaSearchConstants;
import org.eclipse.jdt.core.search.IJavaSearchScope;
import org.eclipse.jdt.core.search.SearchEngine;
import org.eclipse.jdt.core.search.SearchMatch;
import org.eclipse.jdt.core.search.SearchParticipant;
import org.eclipse.jdt.core.search.SearchPattern;
import org.eclipse.jdt.core.search.SearchRequestor;
import org.eclipse.jface.operation.IRunnableContext;
import org.eclipse.jface.operation.IRunnableWithProgress;
import org.lejos.ev3.ldt.LeJOSEV3Plugin;
import org.lejos.ev3.ldt.util.LeJOSEV3Util;

public class MainMethodSearchHelper
{
	private static class ResultAggregator extends SearchRequestor
	{
		private Collection<IType> result;

		public ResultAggregator(Collection<IType> dst)
		{
			this.result = dst;
		}

		@Override
		public void acceptSearchMatch(SearchMatch match) throws CoreException
		{
			Object element = match.getElement();
			if (element instanceof IMethod)
			{
				IMethod method = (IMethod) element;
				try
				{
					if (method.isMainMethod())
					{
						IType type = method.getDeclaringType();
						result.add(type);
					}
				}
				catch (JavaModelException e)
				{
					LeJOSEV3Util.log(e);
				}
			}
		}
	}

	public void searchMainMethods(IProgressMonitor pm, IJavaSearchScope scope, Collection<IType> dst) throws CoreException
	{
		pm.beginTask("Searching for main methods...", 100);
		try
		{
			SearchPattern pattern = SearchPattern.createPattern("main(String[]) void",
					IJavaSearchConstants.METHOD, IJavaSearchConstants.DECLARATIONS,
					SearchPattern.R_EXACT_MATCH | SearchPattern.R_CASE_SENSITIVE); //$NON-NLS-1$
			SearchParticipant[] participants = new SearchParticipant[] { SearchEngine.getDefaultSearchParticipant() };
			IProgressMonitor searchMonitor = new SubProgressMonitor(pm, 100);
			ResultAggregator collector = new ResultAggregator(dst);
			new SearchEngine().search(pattern, participants, scope,	collector, searchMonitor);
		}
		finally
		{
			pm.done();
		}
	}

	public void searchMainMethods(IRunnableContext context, final IJavaSearchScope scope, final Collection<IType> dst)
		throws InterruptedException, CoreException
	{
		try
		{
			context.run(true, true, new IRunnableWithProgress()
				{
					public void run(IProgressMonitor pm) throws InvocationTargetException
					{
						try
						{
							searchMainMethods(pm, scope, dst);
						}
						catch (CoreException e)
						{
							throw new InvocationTargetException(e);
						}
					}
				});
		}
		catch (InvocationTargetException e)
		{
			Throwable e2 = e.getTargetException();
			if (e2 instanceof Error)
				throw (Error)e2;
			if (e2 instanceof RuntimeException)
				throw (RuntimeException)e2;
			if (e2 instanceof CoreException)
				throw (CoreException)e2;
			
			throw new CoreException(new Status(IStatus.ERROR, LeJOSEV3Plugin.ID, "unexpected exception", e2));
		}
	}

}