NewPuppetModuleProjectWizard.java

/**
 * Copyright (c) 2011 Cloudsmith Inc. and other contributors, as listed below.
 * 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:
 *   Cloudsmith
 * 
 */
package org.cloudsmith.geppetto.ui.wizard;

import static org.cloudsmith.geppetto.forge.Forge.MODULEFILE_NAME;

import java.io.File;
import java.lang.reflect.InvocationTargetException;
import java.util.Collections;

import org.cloudsmith.geppetto.common.os.FileUtils;
import org.cloudsmith.geppetto.forge.Forge;
import org.cloudsmith.geppetto.forge.util.ModuleUtils;
import org.cloudsmith.geppetto.forge.v2.model.Metadata;
import org.cloudsmith.geppetto.forge.v2.model.ModuleName;
import org.cloudsmith.geppetto.pp.dsl.ui.builder.PPBuildJob;
import org.cloudsmith.geppetto.pp.dsl.ui.pptp.PptpTargetProjectHandler;
import org.cloudsmith.geppetto.pp.dsl.ui.preferences.PPPreferencesHelper;
import org.cloudsmith.geppetto.semver.Version;
import org.cloudsmith.geppetto.ui.UIPlugin;
import org.cloudsmith.geppetto.ui.util.ResourceUtil;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IPath;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.Platform;
import org.eclipse.core.runtime.SubMonitor;
import org.eclipse.emf.common.util.URI;
import org.eclipse.jface.dialogs.ErrorDialog;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.wizard.Wizard;
import org.eclipse.ui.INewWizard;
import org.eclipse.ui.IWorkbench;
import org.eclipse.ui.PartInitException;
import org.eclipse.ui.actions.WorkspaceModifyOperation;
import org.eclipse.ui.dialogs.WizardNewProjectCreationPage;
import org.eclipse.ui.internal.ide.DialogUtil;

import com.google.inject.Inject;

public class NewPuppetModuleProjectWizard extends Wizard implements INewWizard {

	protected class PuppetProjectCreationPage extends WizardNewProjectCreationPage {

		protected PuppetProjectCreationPage(String pageName) {
			super(pageName);
			setInitialProjectName("unnamed");
		}

		@Override
		protected boolean validatePage() {

			if(super.validatePage()) {
				IPath locationPath = getLocationPath();
				projectLocation = Platform.getLocation().equals(locationPath)
						? null
						: locationPath;
				projectContainer = getProjectHandle().getFullPath();
				try {
					ModuleName.checkName(getProjectName(), true);
				}
				catch(IllegalArgumentException e) {
					setErrorMessage("Project name must be a valid module name: " + e.getMessage());
					return false;
				}
				return true;
			}

			return false;
		}

	}

	@Inject
	private Forge forge;

	@Inject
	private PptpTargetProjectHandler pptpHandler;

	@Inject
	protected PPPreferencesHelper preferenceHelper;

	protected IPath projectLocation;

	protected IPath projectContainer;

	protected IProject project;

	@Override
	public void addPages() {
		WizardNewProjectCreationPage newProjectCreationPage = newProjectCreationPage("NewProjectCreationPage"); //$NON-NLS-1$

		newProjectCreationPage.setTitle(getProjectCreationPageTitle());
		newProjectCreationPage.setDescription(getProjectCreationPageDescription());

		addPage(newProjectCreationPage);
	}

	protected Forge getForge() {
		return forge;
	}

	private String getModuleOwner() {
		String moduleOwner = preferenceHelper.getForgeLogin();
		if(moduleOwner == null)
			moduleOwner = ModuleName.safeOwner(System.getProperty("user.name"));
		return moduleOwner;
	}

	protected String getProjectCreationPageDescription() {
		return UIPlugin.getLocalString("_UI_PuppetModuleProject_description"); //$NON-NLS-1$
	}

	protected String getProjectCreationPageTitle() {
		return UIPlugin.getLocalString(getProjectCreationPageTitleKey()); //$NON-NLS-1$
	}

	protected String getProjectCreationPageTitleKey() {
		return "_UI_PuppetModuleProject_title";
	}

	@Override
	public void init(IWorkbench workbench, IStructuredSelection selection) {
		setDefaultPageImageDescriptor(UIPlugin.getImageDesc("full/wizban/NewPuppetProject.png")); //$NON-NLS-1$
		setWindowTitle(UIPlugin.getLocalString("_UI_NewPuppetModuleProject_title")); //$NON-NLS-1$
	}

	protected void initializeProjectContents(IProgressMonitor monitor) throws Exception {
		SubMonitor submon = SubMonitor.convert(monitor, 100);
		Metadata metadata = new Metadata();
		metadata.setName(new ModuleName(getModuleOwner(), project.getName().toLowerCase(), true));
		metadata.setVersion(Version.create("0.1.0"));

		if(ResourceUtil.getFile(project.getFullPath().append("manifests/init.pp")).exists()) { //$NON-NLS-1$
			File modulefile = project.getLocation().append(MODULEFILE_NAME).toFile(); //$NON-NLS-1$
			submon.worked(20);

			if(!modulefile.exists()) {
				ModuleUtils.saveAsModulefile(metadata, modulefile);
			}
			submon.worked(80);
		}
		else {
			forge.generate(project.getLocation().toFile(), metadata);
			submon.worked(70);
			// This will cause a build. The build will recreate the metadata.json file
			project.refreshLocal(IResource.DEPTH_INFINITE, submon.newChild(30));
		}
		monitor.done();
	}

	protected WizardNewProjectCreationPage newProjectCreationPage(String pageName) {
		return new PuppetProjectCreationPage(pageName);
	}

	@Override
	public boolean performFinish() {
		try {
			project = null;
			getContainer().run(false, false, new WorkspaceModifyOperation() {

				@Override
				protected void execute(IProgressMonitor progressMonitor) throws InvocationTargetException {
					SubMonitor monitor = SubMonitor.convert(progressMonitor, 100);
					try {
						String projectName = projectContainer.segment(0);
						if(projectLocation == null)
							projectLocation = ResourcesPlugin.getWorkspace().getRoot().getLocation().append(projectName);
						File projectDir = projectLocation.toFile();
						if(projectDir.exists()) {
							if(!MessageDialog.openConfirm(
								getShell(), UIPlugin.getLocalString("_UI_Confirm_Overwrite"),
								UIPlugin.getLocalString("_UI_Directory_not_empty", projectDir.getAbsolutePath())))
								// User don't want us to overwrite
								return;

							FileUtils.rmR(projectDir);
						}

						project = ResourceUtil.createProject(
							projectContainer, URI.createFileURI(projectDir.getAbsolutePath()),
							Collections.<IProject> emptyList(), monitor.newChild(1));

						initializeProjectContents(monitor.newChild(80));
						pptpHandler.ensureStateOfPuppetProjects(monitor.newChild(10));

						IFile modulefile = ResourceUtil.getFile(project.getFullPath().append(MODULEFILE_NAME)); //$NON-NLS-1$
						if(modulefile.exists()) {
							NewModulefileWizard.ensureMetadataJSONExists(modulefile, monitor.newChild(1));
							ResourceUtil.selectFile(modulefile);

							try {
								ResourceUtil.openEditor(modulefile);
							}
							catch(PartInitException partInitException) {
								MessageDialog.openError(
									getShell(),
									UIPlugin.getLocalString("_UI_OpenEditor_title"), partInitException.getMessage()); //$NON-NLS-1$
							}
						}
					}
					catch(Exception exception) {
						throw new InvocationTargetException(exception);
					}
					finally {
						progressMonitor.done();
					}
				}
			});
			if(project == null)
				return false;

			new PPBuildJob(project.getWorkspace(), true).schedule(1000);
			return true;
		}
		catch(InvocationTargetException e) {
			Throwable t = e.getTargetException();
			String title = UIPlugin.getLocalString("_UI_CreateProject_title");
			if(t instanceof PartInitException)
				DialogUtil.openError(getShell(), title, t.getMessage(), (PartInitException) t);
			else if(t instanceof CoreException)
				ErrorDialog.openError(getShell(), title, t.getMessage(), ((CoreException) t).getStatus());
			else
				MessageDialog.openError(getShell(), title, t.getMessage());
		}
		catch(InterruptedException e) {
		}
		return false;
	}
}