Bpmn2ProjectWizard.java

/**
 * Copyright (c) 2011-2013 Obeo.
 * 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:
 *     Obeo - initial API and implementation
 * 
 */
package org.obeonetwork.bpmn2.design.ui.wizards.newmodel;

import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.util.HashMap;

import org.obeonetwork.dsl.bpmn2.Bpmn2Factory;
import org.obeonetwork.dsl.bpmn2.Bpmn2Package;
import org.obeonetwork.dsl.bpmn2.Definitions;
import org.obeonetwork.dsl.bpmn2.Process;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.core.runtime.Path;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EClassifier;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.emf.transaction.RecordingCommand;
import org.eclipse.jface.dialogs.ProgressMonitorDialog;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.actions.WorkspaceModifyOperation;
import org.eclipse.ui.dialogs.WizardNewProjectCreationPage;
import org.eclipse.ui.wizards.newresource.BasicNewResourceWizard;
import org.obeonetwork.bpmn2.design.Activator;
import org.eclipse.sirius.ext.base.Option;
import org.eclipse.sirius.ext.base.Options;
import org.eclipse.sirius.business.api.componentization.ViewpointRegistry;
import org.eclipse.sirius.business.api.modelingproject.ModelingProject;
import org.eclipse.sirius.business.api.session.Session;
import org.eclipse.sirius.viewpoint.description.Viewpoint;
import org.eclipse.sirius.ui.business.api.viewpoint.ViewpointSelectionCallback;
import org.eclipse.sirius.ui.tools.api.project.ModelingProjectManager;

/**
 * 
 * @author ymortier
 */
public class Bpmn2ProjectWizard extends BasicNewResourceWizard {
	/**
	 * Dot constant.
	 */
	public static final String DOT = ".";

	/**
	 * The UML file extension.
	 */
	public static final String MODEL_FILE_EXTENSION = "bpmn"; //$NON-NLS-1$

	private Bpmn2ModelWizardInitModelPage modelPage;

	private WizardNewProjectCreationPage newProjectPage;

	@Override
	public void addPages() {
		super.addPages();

		newProjectPage = new WizardNewProjectCreationPage("Project"); //$NON-NLS-1$
		newProjectPage.setInitialProjectName("");
		newProjectPage.setTitle("Create a BPMN Modeling project");
		newProjectPage.setDescription("Enter a project name"); //$NON-NLS-1$
		addPage(newProjectPage);

		modelPage = new Bpmn2ModelWizardInitModelPage(Messages.Bpmn2ModelWizard_UI_InitModelPageId);
		modelPage.setTitle(Messages.Bpmn2ModelWizard_UI_InitModelPageTitle);
		modelPage.setDescription(Messages.Bpmn2ModelWizard_UI_InitModelPageDescription);
		addPage(modelPage);
	}

	@Override
	public boolean performFinish() {

		try {
			IProject project = ModelingProjectManager.INSTANCE.createNewModelingProject(
					newProjectPage.getProjectName(), newProjectPage.getLocationPath(), true, new NullProgressMonitor());
			Option<IFile> optionalNewfile = createEcoreResource(project);
			if (optionalNewfile.some() && optionalNewfile.get().exists()) {
				selectAndReveal(optionalNewfile.get());
			} else {
				selectAndReveal(project);
			}
			return true;

		} catch (CoreException e) {
			Activator.log(IStatus.ERROR, Messages.Bpmn2ModelWizard_UI_Error_CreatingBpmn2Model, e);
		}
		return false;

	}

	private Option<IFile> createEcoreResource(IProject project) {
		final Session session;
		Option<ModelingProject> modelingProject = ModelingProject.asModelingProject(project);
		if (modelingProject.some()) {
			session = modelingProject.get().getSession();
		} else {
			session = null;
		}

		if (session == null) {
			return Options.newNone();
		}

		/*
		 * Create a resource for this file. Don't specify acontent type, as it could be Ecore or EMOF.Create
		 * in a other resourceset and let the workspace monitor for modeling project add it as semantic
		 * resource.
		 */
		final ResourceSet rs = new ResourceSetImpl();
		String platformPath = '/' + project.getName() + '/' + modelPage.getInitialObjectName().toLowerCase()
				+ DOT + MODEL_FILE_EXTENSION;
		final URI semanticModelURI = URI.createPlatformResourceURI(platformPath, true);
		final Resource resource = rs.createResource(semanticModelURI);

		/* Add the initial model object to the contents. */
		final EObject rootObject = createInitialModel();
		if (rootObject != null) {
			resource.getContents().add(rootObject);
		}

		WorkspaceModifyOperation operation = new WorkspaceModifyOperation() {

			@Override
			protected void execute(IProgressMonitor monitor) throws CoreException, InvocationTargetException,
					InterruptedException {
				try {
					resource.save(new HashMap<Object, Object>());
				} catch (IOException e) {
					/* do nothing it should always work */
				}
			}
		};
		try {
			new ProgressMonitorDialog(PlatformUI.getWorkbench().getDisplay().getActiveShell()).run(true,
					false, operation);
		} catch (InvocationTargetException e) {
			/* do nothing it should always work */
		} catch (InterruptedException e) {
			/* do nothing it should always work */
		}

		session.getTransactionalEditingDomain().getCommandStack()
				.execute(new RecordingCommand(session.getTransactionalEditingDomain()) {
					@Override
					protected void doExecute() {
						ViewpointSelectionCallback callback = new ViewpointSelectionCallback();

						for (Viewpoint vp : ViewpointRegistry.getInstance().getViewpoints()) {
							if ("Process".equals(vp.getName())) {
								callback.selectViewpoint(vp, session, new NullProgressMonitor());
							} 
						}
					}
				});
		return Options.newSome(ResourcesPlugin.getWorkspace().getRoot().getFile(new Path(platformPath)));
	}

	/**
	 * Creates the semantic root element from the given operation arguments.
	 * 
	 * @return the semantic root {@link EObject}
	 */
	private EObject createInitialModel() {
		Process process=Bpmn2Factory.eINSTANCE.createProcess();
		EClassifier found = Bpmn2Package.eINSTANCE.getEClassifier(modelPage.getInitialObjectName());
		if (found instanceof EClass) {	
			Definitions definitions = (Definitions) Bpmn2Factory.eINSTANCE.create((EClass)found);
			definitions.getRootElements().add(process);			 	 
			return definitions;
		} else {	
			Definitions definitions = Bpmn2Factory.eINSTANCE.createDefinitions();
			definitions.getRootElements().add(process);	
			return definitions;
		}
	}

}