org.eclipse.jdt.apt.core/src/org/eclipse/jdt/apt/core/internal/APTDispatch.java

/*******************************************************************************
 * Copyright (c) 2005 BEA Systems, Inc.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    tyeung@bea.com - initial API and implementation
 *    mkaufman@bea.com
 *******************************************************************************/

package org.eclipse.jdt.apt.core.internal;

import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.IWorkspace;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.IJavaProject;
import org.eclipse.jdt.core.compiler.IProblem;

import com.sun.mirror.apt.AnnotationProcessorFactory;

/**
 * Dispatch APT. 
 * @author tyeung
 *
 */
public class APTDispatch 
{	
	public static APTResult runAPTDuringBuild(
			final List<AnnotationProcessorFactory> factories, final IFile file,
			final IJavaProject javaProj) 
	{
		return runAPT( factories, javaProj, file, null );
	}

	/**
	 * Run annnotation processing.
	 * @param factories the list of annotation processor factories to be run.
	 * @return the set of files that need to be compiled.
	 */
	public static APTResult runAPTDuringReconcile(
			final List<AnnotationProcessorFactory> factories,
			ICompilationUnit compilationUnit, IJavaProject javaProj) 
	{
		return runAPT( factories, javaProj, null, compilationUnit );
	}
		
	private static APTResult runAPT(final List<AnnotationProcessorFactory> factories,
			IJavaProject javaProj, IFile file, 
			ICompilationUnit compilationUnit )
	{
		
		APTDispatchRunnable runnable;
		if ( file != null )
			 runnable = new APTDispatchRunnable( file, javaProj, factories );
		else
			runnable = new APTDispatchRunnable( compilationUnit, javaProj, factories );
		
		IWorkspace w = ResourcesPlugin.getWorkspace();
		try
		{	
			// need to do something here to avoid the "Invalid Begin Rule..." errors.  Not sure what is correct.
			IResource r;
			r = w.getRoot();
			
			w.run( runnable, r, IWorkspace.AVOID_UPDATE, null );
		}
		catch( CoreException ce )
		{
			// TODO:  deal with this exception
			ce.printStackTrace();
		}
			
		return runnable.getResult();
	}


	public static class APTResult
	{
		/**
		 * For creating an empty result. i.e. no file changes, no new dependencies and not
		 * new problems.
		 */
		APTResult()
		{
			_newFiles = Collections.emptySet();
			_deletedFiles = Collections.emptySet();
			_newDependencies = Collections.emptySet();
			_newProblems = Collections.emptyMap();
		}
		APTResult( Set<IFile> newFiles, Set<IFile> deletedFiles, Set<String> deps, Map<IFile, List<IProblem>> problems )
		{
			_newFiles = newFiles;
			_newDependencies = deps;
			_deletedFiles = deletedFiles;
			_newProblems = problems;
		}
		
		private final Set<IFile> _newFiles;
		private final Set<IFile> _deletedFiles;
		private final Set<String> _newDependencies;
		private final Map<IFile, List<IProblem>> _newProblems;
		
		Set<IFile> getNewFiles() { return _newFiles; }
		Set<IFile> getDeletedFiles() { return _deletedFiles; }
		Set<String> getNewDependencies() { return _newDependencies; }
		Map<IFile, List<IProblem>> getProblems(){return _newProblems;}
	}

	
}