CompileFilesHandler.java

/*  Copyright (C) 2009 Mobile Sorcery AB

    This program is free software; you can redistribute it and/or modify it
    under the terms of the Eclipse Public License v1.0.

    This program is distributed in the hope that it will be useful, but WITHOUT
    ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
    FITNESS FOR A PARTICULAR PURPOSE. See the Eclipse Public License v1.0 for
    more details.

    You should have received a copy of the Eclipse Public License v1.0 along
    with this program. It is also available at http://www.eclipse.org/legal/epl-v10.html
 */
package com.mobilesorcery.sdk.ui.internal.handlers;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;

import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.commands.HandlerEvent;
import org.eclipse.core.commands.IHandlerListener;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.IncrementalProjectBuilder;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.OperationCanceledException;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.SubProgressMonitor;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.actions.BuildAction;
import org.eclipse.ui.handlers.HandlerUtil;

import com.mobilesorcery.sdk.core.IBuildSession;
import com.mobilesorcery.sdk.core.IBuildVariant;
import com.mobilesorcery.sdk.core.MoSyncBuilder;
import com.mobilesorcery.sdk.core.MoSyncProject;
import com.mobilesorcery.sdk.core.ResourceSet;
import com.mobilesorcery.sdk.ui.MoSyncCommandHandler;
import com.mobilesorcery.sdk.ui.MosyncUIPlugin;

public class CompileFilesHandler extends MoSyncCommandHandler {

	class CompileFilesJob extends Job {

		private List<IResource> resources;
        private Shell shell;

		public CompileFilesJob() {
			super("Quick Compile");
			setUser(true);
		}

		public void setResources(List<IResource> resources) {
			this.resources = resources;
		}

		@Override
		protected IStatus run(IProgressMonitor monitor) {
			try {
				HashMap<MoSyncProject, List<IResource>> projectResourceMap = new HashMap<MoSyncProject, List<IResource>>();
				HashSet<MoSyncProject> projectsToCompileEntirely = new HashSet<MoSyncProject>();

				int resourceCount = 0;

				MoSyncBuilder builder = new MoSyncBuilder();
				for (IResource resource : resources) {
					MoSyncProject project = MoSyncProject.create(resource.getProject());
					if (project != null) {
						List<IResource> resourceList = projectResourceMap.get(project);
						if (resourceList == null) {
							resourceList = new ArrayList<IResource>();
							projectResourceMap.put(project, resourceList);
						}
						if (resource.getType() == IResource.PROJECT) {
							projectsToCompileEntirely.add(project);
						} else {
							resourceList.add(resource);
						}
						resourceCount++;
					}
				}

				monitor.beginTask("Quick Compile", resourceCount);

				boolean cancelled = !MoSyncBuilder.saveAllEditors(resources);

				if (!cancelled) {
  					for (MoSyncProject project : projectResourceMap.keySet()) {
  						boolean compileEntireProject = projectsToCompileEntirely.contains(project);
    					List<IResource> resources = projectResourceMap.get(project);
    					HashSet<IResource> resourceSet = new HashSet<IResource>(resources);
    					ResourceSet resourcesToCompile = compileEntireProject ? null : new ResourceSet(resourceSet);
    					IBuildVariant variant = MoSyncBuilder.getActiveVariant(project);
    					IBuildSession session = MoSyncBuilder.createCompileOnlySession(variant);
    					builder.build(project.getWrappedProject(), session, variant, resourcesToCompile, new SubProgressMonitor(monitor, resourceSet.size()));
    				}
    				return Status.OK_STATUS;
				} else {
				    return Status.CANCEL_STATUS;
				}
			} catch (OperationCanceledException e) {
			    return Status.CANCEL_STATUS;
			} catch (Exception e) {
				return new Status(IStatus.ERROR, MosyncUIPlugin.PLUGIN_ID, "Quick Compile failed.", e);
			}
		}

        public void setShell(Shell shell) {
            this.shell = shell;
        }

	}

	@Override
	public Object execute(ExecutionEvent event) throws ExecutionException {
		Shell shell = HandlerUtil.getActiveShell(event);
		// Hmm, very weird hack again.
		BuildAction dummy = new BuildAction(shell, IncrementalProjectBuilder.INCREMENTAL_BUILD);
		dummy.isEnabled();
		IStructuredSelection selection = dummy.getStructuredSelection();
		List<IResource> resources = extractResources(selection);
		CompileFilesJob job = new CompileFilesJob();
		job.setShell(shell);
		job.setResources(resources);
		job.schedule();
		return null;
	}

}