org.eclipse.opencert.sam.arg.editor.dawn/src/org/eclipse/opencert/sam/arg/presentation/DawnArgModelWizard.java

/*******************************************************************************
 * Copyright (c) 2016 Fundación Tecnalia Research & Innovation.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v2.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/org/documents/epl-2.0/EPL-2.0.html
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *   Huascar Espinoza - initial API and implementation
 *   Alejandra Ruíz - initial API and implementation
 *   Idoya Del Río - initial API and implementation
 *   Mari Carmen Palacios - initial API and implementation
 *   Angel López - initial API and implementation
 *******************************************************************************/
package org.eclipse.opencert.sam.arg.presentation;

import org.eclipse.emf.cdo.dawn.preferences.PreferenceConstants;
import org.eclipse.emf.cdo.dawn.ui.DawnEditorInput;
import org.eclipse.emf.cdo.dawn.ui.wizards.DawnCreateNewResourceWizardPage;
import org.eclipse.emf.cdo.dawn.util.connection.CDOConnectionUtil;
import org.eclipse.emf.cdo.eresource.CDOResource;
import org.eclipse.emf.cdo.session.CDOSession;
import org.eclipse.emf.cdo.transaction.CDOTransaction;
import org.eclipse.emf.cdo.view.CDOView;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.ui.INewWizard;
import org.eclipse.ui.IWorkbenchPage;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.PartInitException;
import org.eclipse.ui.actions.WorkspaceModifyOperation;
import org.eclipse.opencert.sam.arg.arg.presentation.ArgEditorPlugin;
import org.eclipse.opencert.sam.arg.arg.presentation.ArgModelWizard;

import java.util.Collections;

public class DawnArgModelWizard extends ArgModelWizard implements INewWizard {
	private DawnCreateNewResourceWizardPage newResourceCreationPage;

	private CDOView view;

	private CDOResource resource;

	public DawnArgModelWizard() {
		super();
		CDOConnectionUtil.instance.init(
				PreferenceConstants.getRepositoryName(),
				PreferenceConstants.getProtocol(),
				PreferenceConstants.getServerName());
		CDOSession session = CDOConnectionUtil.instance.getCurrentSession();
		view = CDOConnectionUtil.instance.openView(session);
	}

	@Override
	public void addPages() {
		newResourceCreationPage = new DawnCreateNewResourceWizardPage("arg",
				false, view);
		addPage(newResourceCreationPage);

		initialObjectCreationPage = new ArgModelWizardInitialObjectCreationPage(
				"Whatever2");
		initialObjectCreationPage.setTitle(ArgEditorPlugin.INSTANCE
				.getString("_UI_ArgModelWizard_label"));
		initialObjectCreationPage.setDescription(ArgEditorPlugin.INSTANCE
				.getString("_UI_Wizard_initial_object_description"));
		//addPage(initialObjectCreationPage);
	}

	@Override
	public boolean performFinish() {
		try {
			// Do the work within an operation.
			//
			WorkspaceModifyOperation operation = new WorkspaceModifyOperation() {

				@Override
				protected void execute(IProgressMonitor progressMonitor) {
					try {
						ResourceSet resourceSet = new ResourceSetImpl();

						URI resourceURI = newResourceCreationPage.getURI();

						CDOTransaction transaction = CDOConnectionUtil.instance
								.openCurrentTransaction(resourceSet,
										resourceURI.toString());

						resource = transaction.getOrCreateResource(resourceURI
								.path());

						EObject rootObject = createInitialModel();
						if (rootObject != null) {
							resource.getContents().add(rootObject);
						}

						resource.save(Collections.EMPTY_MAP);
						transaction.close();
					} catch (Exception exception) {
						ArgEditorPlugin.INSTANCE.log(exception);
						throw new RuntimeException(exception);
					} finally {
						progressMonitor.done();
					}
				}
			};

			getContainer().run(false, false, operation);

			openEditor(newResourceCreationPage.getURI());

			return true;
		} catch (Exception exception) {
			ArgEditorPlugin.INSTANCE.log(exception);
			return false;
		}
	}

	private void openEditor(URI uri) {
		IWorkbenchWindow workbenchWindow = workbench.getActiveWorkbenchWindow();
		IWorkbenchPage page = workbenchWindow.getActivePage();
		DawnEditorInput dawnEditorInput = new DawnEditorInput(uri);
		try {
			page.openEditor(dawnEditorInput, DawnArgEditor.ID);
		} catch (PartInitException exception) {
			MessageDialog.openError(workbenchWindow.getShell(),
					ArgEditorPlugin.INSTANCE
							.getString("_UI_OpenEditorError_label"), exception
							.getMessage());
			throw new RuntimeException(exception);
		}
	}
}