rtm/org.eclipse.statet.rtm.base.ui/src/org/eclipse/statet/rtm/base/ui/actions/AbstractNewRTaskFileWizard.java - statet/org.eclipse.statet-r - Git at Google

 /*=============================================================================#
  # Copyright (c) 2012, 2021 Stephan Wahlbrink and others.
  #
  # This program and the accompanying materials are made available under the
  # terms of the Eclipse Public License 2.0 which is available at
  # https://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
  # which is available at https://www.apache.org/licenses/LICENSE-2.0.
  #
  # SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
  #
  # Contributors:
  #     Stephan Wahlbrink <sw@wahlbrink.eu> - initial API and implementation
  #=============================================================================*/

 package org.eclipse.statet.rtm.base.ui.actions;

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

 import org.eclipse.core.resources.IFile;
 import org.eclipse.core.resources.IFolder;
 import org.eclipse.core.resources.IProject;
 import org.eclipse.core.resources.IResource;
 import org.eclipse.core.resources.ResourcesPlugin;
 import org.eclipse.core.runtime.IProgressMonitor;
 import org.eclipse.core.runtime.IStatus;
 import org.eclipse.core.runtime.Platform;
 import org.eclipse.core.runtime.Status;
 import org.eclipse.emf.common.util.URI;
 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.ecore.xmi.XMLResource;
 import org.eclipse.jface.viewers.ISelection;
 import org.eclipse.jface.viewers.IStructuredSelection;
 import org.eclipse.jface.viewers.StructuredSelection;
 import org.eclipse.jface.wizard.Wizard;
 import org.eclipse.osgi.util.NLS;
 import org.eclipse.ui.INewWizard;
 import org.eclipse.ui.IWorkbench;
 import org.eclipse.ui.IWorkbenchPage;
 import org.eclipse.ui.IWorkbenchPart;
 import org.eclipse.ui.IWorkbenchWindow;
 import org.eclipse.ui.PartInitException;
 import org.eclipse.ui.actions.WorkspaceModifyOperation;
 import org.eclipse.ui.dialogs.WizardNewFileCreationPage;
 import org.eclipse.ui.part.FileEditorInput;
 import org.eclipse.ui.part.ISetSelectionTarget;
 import org.eclipse.ui.statushandlers.StatusManager;

 import org.eclipse.statet.internal.rtm.base.ui.actions.Messages;
 import org.eclipse.statet.rtm.base.ui.IRtDescriptor;
 import org.eclipse.statet.rtm.base.ui.RtModelUIPlugin;


 /**
  * This is a simple wizard for creating a new model file.
  * <!-- begin-user-doc -->
  * <!-- end-user-doc -->
  */
 public abstract class AbstractNewRTaskFileWizard extends Wizard implements INewWizard {


 	protected final IRtDescriptor rtDescriptor;

 	/**
 	 * This is the file creation page.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	protected ModelWizardNewFileCreationPage newFileCreationPage;

 	/**
 	 * Remember the selection during initialization for populating the default container.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	protected IStructuredSelection selection;

 	/**
 	 * Remember the workbench during initialization.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	protected IWorkbench workbench;


 	protected AbstractNewRTaskFileWizard(final IRtDescriptor descriptor) {
 		this.rtDescriptor= descriptor;
 	}

 	/**
 	 * This just records the information.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	@Override
 	public void init(final IWorkbench workbench, final IStructuredSelection selection) {
 		this.workbench= workbench;
 		this.selection= selection;
 		setWindowTitle(NLS.bind(Messages.NewRTaskWizard_title, this.rtDescriptor.getName()));
 //		setDefaultPageImageDescriptor(ExtendedImageRegistry.INSTANCE.getImageDescriptor(RtGGPlotEditorPlugin.INSTANCE.getImage("full/wizban/NewGGPlot"))); //$NON-NLS-1$
 	}

 	/**
 	 * Create a new model.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 */
 	protected EObject createInitialModel() {
 		return this.rtDescriptor.createInitialModelObject();
 	}

 	/**
 	 * Do the work after everything is specified.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	@Override
 	public boolean performFinish() {
 		try {
 			// Remember the file
 			final IFile modelFile= getModelFile();

 			// Do the work within an operation
 			final WorkspaceModifyOperation operation= new WorkspaceModifyOperation() {
 				@Override
 				protected void execute(final IProgressMonitor progressMonitor) throws InvocationTargetException {
 					try {
 						// Create a resource set
 						final ResourceSet resourceSet= new ResourceSetImpl();

 						// Get the URI of the model file
 						final URI fileURI= URI.createPlatformResourceURI(modelFile.getFullPath().toString(), true);

 						// Create a resource for this file
 						final Resource resource= resourceSet.createResource(fileURI, AbstractNewRTaskFileWizard.this.rtDescriptor.getDefaultContentTypeID());

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

 						// Save the contents of the resource to the file system
 						final Map<Object, Object> options= new HashMap<>();
 						options.put(XMLResource.OPTION_ENCODING, "UTF-8"); //$NON-NLS-1$
 						resource.save(options);
 					}
 					catch (final IOException e) {
 						throw new InvocationTargetException(e);
 					}
 					finally {
 						progressMonitor.done();
 					}
 				}
 			};

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

 			final PerspectiveUtil perspectiveUtil= new PerspectiveUtil();
 			perspectiveUtil.updatePerspective(this.rtDescriptor.getAssociatedPerspectiveId());

 			// Select the new file resource in the current view
 			final IWorkbenchWindow workbenchWindow= this.workbench.getActiveWorkbenchWindow();
 			final IWorkbenchPage page= workbenchWindow.getActivePage();
 			final IWorkbenchPart activePart= page.getActivePart();
 			if (activePart instanceof ISetSelectionTarget) {
 				final ISelection targetSelection= new StructuredSelection(modelFile);
 				getShell().getDisplay().asyncExec(new Runnable() {
 					@Override
 					public void run() {
 						((ISetSelectionTarget)activePart).selectReveal(targetSelection);
 					}
 				});
 			}

 			// Open an editor on the new file
 			try {
 				page.openEditor(new FileEditorInput(modelFile),
 						this.workbench.getEditorRegistry().getDefaultEditor(
 								modelFile.getFullPath().toString(),
 								Platform.getContentTypeManager().getContentType(this.rtDescriptor.getDefaultContentTypeID())).getId());
 			}
 			catch (final PartInitException e) {
 				StatusManager.getManager().handle(new Status(IStatus.ERROR, RtModelUIPlugin.BUNDLE_ID, 0,
 						NLS.bind(Messages.NewRTaskWizard_error_OpenEditor_message, this.rtDescriptor.getName()),
 						e ), StatusManager.LOG | StatusManager.SHOW );
 				return false;
 			}

 			return true;
 		}
 		catch (final Exception e) {
 			StatusManager.getManager().handle(new Status(IStatus.ERROR, RtModelUIPlugin.BUNDLE_ID, 0,
 					NLS.bind(Messages.NewRTaskWizard_error_CreateFile_message, this.rtDescriptor.getName()),
 					e ), StatusManager.LOG | StatusManager.SHOW );
 			return false;
 		}
 	}

 	/**
 	 * This is the one page of the wizard.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	public class ModelWizardNewFileCreationPage extends WizardNewFileCreationPage {
 		/**
 		 * Pass in the selection.
 		 * <!-- begin-user-doc -->
 		 * <!-- end-user-doc -->
 		 * @generated
 		 */
 		public ModelWizardNewFileCreationPage(final String pageId, final IStructuredSelection selection) {
 			super(pageId, selection);
 			setFileExtension(AbstractNewRTaskFileWizard.this.rtDescriptor.getDefaultFileExtension());
 		}

 		/**
 		 * <!-- begin-user-doc -->
 		 * <!-- end-user-doc -->
 		 * @generated
 		 */
 		public IFile getModelFile() {
 			return ResourcesPlugin.getWorkspace().getRoot().getFile(getContainerFullPath().append(getFileName()));
 		}
 	}

 	/**
 	 * The framework calls this to create the contents of the wizard.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	@Override
 	public void addPages() {
 		// Create a page, set the title, and the initial model file name

 		this.newFileCreationPage= new ModelWizardNewFileCreationPage("NewRTask.NewFile", this.selection); //$NON-NLS-1$
 		this.newFileCreationPage.setTitle(NLS.bind(Messages.NewRTaskWizard_NewFile_title, this.rtDescriptor.getName()));
 		this.newFileCreationPage.setDescription(NLS.bind(Messages.NewRTaskWizard_NewFile_description, this.rtDescriptor.getName()));
 		addPage(this.newFileCreationPage);

 		// Try and get the resource selection to determine a current directory for the file dialog
 		if (this.selection != null && !this.selection.isEmpty()) {
 			// Get the resource...
 			final Object selectedElement= this.selection.iterator().next();
 			if (selectedElement instanceof IResource) {
 				// Get the resource parent, if its a file
 				IResource selectedResource= (IResource)selectedElement;
 				if (selectedResource.getType() == IResource.FILE) {
 					selectedResource= selectedResource.getParent();
 				}

 				// This gives us a directory...
 				if (selectedResource instanceof IFolder || selectedResource instanceof IProject) {
 					// Set this for the container
 					this.newFileCreationPage.setContainerFullPath(selectedResource.getFullPath());

 //					// Make up a unique new name here
 //					String defaultModelBaseFilename= RtGGPlotEditorPlugin.INSTANCE.getString("_UI_GGPlotEditorFilenameDefaultBase"); //$NON-NLS-1$
 //					String defaultModelFilenameExtension= this.rtDescriptor.getDefaultContentTypeFileExtension();
 //					String modelFilename= defaultModelBaseFilename + "." + defaultModelFilenameExtension; //$NON-NLS-1$
 //					for (int i= 1; ((IContainer)selectedResource).findMember(modelFilename) != null; ++i) {
 //						modelFilename= defaultModelBaseFilename + i + "." + defaultModelFilenameExtension; //$NON-NLS-1$
 //					}
 //					this.newFileCreationPage.setFileName(modelFilename);
 				}
 			}
 		}
 	}

 	/**
 	 * Get the file from the page.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 */
 	public IFile getModelFile() {
 		return this.newFileCreationPage.getModelFile();
 	}

 }
	/*=============================================================================#
	# Copyright (c) 2012, 2021 Stephan Wahlbrink and others.
	#
	# This program and the accompanying materials are made available under the
	# terms of the Eclipse Public License 2.0 which is available at
	# https://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
	# which is available at https://www.apache.org/licenses/LICENSE-2.0.
	#
	# SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
	#
	# Contributors:
	# Stephan Wahlbrink <sw@wahlbrink.eu> - initial API and implementation
	#=============================================================================*/

	package org.eclipse.statet.rtm.base.ui.actions;

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

	import org.eclipse.core.resources.IFile;
	import org.eclipse.core.resources.IFolder;
	import org.eclipse.core.resources.IProject;
	import org.eclipse.core.resources.IResource;
	import org.eclipse.core.resources.ResourcesPlugin;
	import org.eclipse.core.runtime.IProgressMonitor;
	import org.eclipse.core.runtime.IStatus;
	import org.eclipse.core.runtime.Platform;
	import org.eclipse.core.runtime.Status;
	import org.eclipse.emf.common.util.URI;
	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.ecore.xmi.XMLResource;
	import org.eclipse.jface.viewers.ISelection;
	import org.eclipse.jface.viewers.IStructuredSelection;
	import org.eclipse.jface.viewers.StructuredSelection;
	import org.eclipse.jface.wizard.Wizard;
	import org.eclipse.osgi.util.NLS;
	import org.eclipse.ui.INewWizard;
	import org.eclipse.ui.IWorkbench;
	import org.eclipse.ui.IWorkbenchPage;
	import org.eclipse.ui.IWorkbenchPart;
	import org.eclipse.ui.IWorkbenchWindow;
	import org.eclipse.ui.PartInitException;
	import org.eclipse.ui.actions.WorkspaceModifyOperation;
	import org.eclipse.ui.dialogs.WizardNewFileCreationPage;
	import org.eclipse.ui.part.FileEditorInput;
	import org.eclipse.ui.part.ISetSelectionTarget;
	import org.eclipse.ui.statushandlers.StatusManager;

	import org.eclipse.statet.internal.rtm.base.ui.actions.Messages;
	import org.eclipse.statet.rtm.base.ui.IRtDescriptor;
	import org.eclipse.statet.rtm.base.ui.RtModelUIPlugin;


	/**
	* This is a simple wizard for creating a new model file.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	*/
	public abstract class AbstractNewRTaskFileWizard extends Wizard implements INewWizard {


	protected final IRtDescriptor rtDescriptor;

	/**
	* This is the file creation page.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	protected ModelWizardNewFileCreationPage newFileCreationPage;

	/**
	* Remember the selection during initialization for populating the default container.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	protected IStructuredSelection selection;

	/**
	* Remember the workbench during initialization.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	protected IWorkbench workbench;


	protected AbstractNewRTaskFileWizard(final IRtDescriptor descriptor) {
	this.rtDescriptor= descriptor;
	}

	/**
	* This just records the information.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	@Override
	public void init(final IWorkbench workbench, final IStructuredSelection selection) {
	this.workbench= workbench;
	this.selection= selection;
	setWindowTitle(NLS.bind(Messages.NewRTaskWizard_title, this.rtDescriptor.getName()));
	// setDefaultPageImageDescriptor(ExtendedImageRegistry.INSTANCE.getImageDescriptor(RtGGPlotEditorPlugin.INSTANCE.getImage("full/wizban/NewGGPlot"))); //$NON-NLS-1$
	}

	/**
	* Create a new model.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	*/
	protected EObject createInitialModel() {
	return this.rtDescriptor.createInitialModelObject();
	}

	/**
	* Do the work after everything is specified.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	@Override
	public boolean performFinish() {
	try {
	// Remember the file
	final IFile modelFile= getModelFile();

	// Do the work within an operation
	final WorkspaceModifyOperation operation= new WorkspaceModifyOperation() {
	@Override
	protected void execute(final IProgressMonitor progressMonitor) throws InvocationTargetException {
	try {
	// Create a resource set
	final ResourceSet resourceSet= new ResourceSetImpl();

	// Get the URI of the model file
	final URI fileURI= URI.createPlatformResourceURI(modelFile.getFullPath().toString(), true);

	// Create a resource for this file
	final Resource resource= resourceSet.createResource(fileURI, AbstractNewRTaskFileWizard.this.rtDescriptor.getDefaultContentTypeID());

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

	// Save the contents of the resource to the file system
	final Map<Object, Object> options= new HashMap<>();
	options.put(XMLResource.OPTION_ENCODING, "UTF-8"); //$NON-NLS-1$
	resource.save(options);
	}
	catch (final IOException e) {
	throw new InvocationTargetException(e);
	}
	finally {
	progressMonitor.done();
	}
	}
	};

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

	final PerspectiveUtil perspectiveUtil= new PerspectiveUtil();
	perspectiveUtil.updatePerspective(this.rtDescriptor.getAssociatedPerspectiveId());

	// Select the new file resource in the current view
	final IWorkbenchWindow workbenchWindow= this.workbench.getActiveWorkbenchWindow();
	final IWorkbenchPage page= workbenchWindow.getActivePage();
	final IWorkbenchPart activePart= page.getActivePart();
	if (activePart instanceof ISetSelectionTarget) {
	final ISelection targetSelection= new StructuredSelection(modelFile);
	getShell().getDisplay().asyncExec(new Runnable() {
	@Override
	public void run() {
	((ISetSelectionTarget)activePart).selectReveal(targetSelection);
	}
	});
	}

	// Open an editor on the new file
	try {
	page.openEditor(new FileEditorInput(modelFile),
	this.workbench.getEditorRegistry().getDefaultEditor(
	modelFile.getFullPath().toString(),
	Platform.getContentTypeManager().getContentType(this.rtDescriptor.getDefaultContentTypeID())).getId());
	}
	catch (final PartInitException e) {
	StatusManager.getManager().handle(new Status(IStatus.ERROR, RtModelUIPlugin.BUNDLE_ID, 0,
	NLS.bind(Messages.NewRTaskWizard_error_OpenEditor_message, this.rtDescriptor.getName()),
	e ), StatusManager.LOG \| StatusManager.SHOW );
	return false;
	}

	return true;
	}
	catch (final Exception e) {
	StatusManager.getManager().handle(new Status(IStatus.ERROR, RtModelUIPlugin.BUNDLE_ID, 0,
	NLS.bind(Messages.NewRTaskWizard_error_CreateFile_message, this.rtDescriptor.getName()),
	e ), StatusManager.LOG \| StatusManager.SHOW );
	return false;
	}
	}

	/**
	* This is the one page of the wizard.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	public class ModelWizardNewFileCreationPage extends WizardNewFileCreationPage {
	/**
	* Pass in the selection.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	public ModelWizardNewFileCreationPage(final String pageId, final IStructuredSelection selection) {
	super(pageId, selection);
	setFileExtension(AbstractNewRTaskFileWizard.this.rtDescriptor.getDefaultFileExtension());
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	public IFile getModelFile() {
	return ResourcesPlugin.getWorkspace().getRoot().getFile(getContainerFullPath().append(getFileName()));
	}
	}

	/**
	* The framework calls this to create the contents of the wizard.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	@Override
	public void addPages() {
	// Create a page, set the title, and the initial model file name

	this.newFileCreationPage= new ModelWizardNewFileCreationPage("NewRTask.NewFile", this.selection); //$NON-NLS-1$
	this.newFileCreationPage.setTitle(NLS.bind(Messages.NewRTaskWizard_NewFile_title, this.rtDescriptor.getName()));
	this.newFileCreationPage.setDescription(NLS.bind(Messages.NewRTaskWizard_NewFile_description, this.rtDescriptor.getName()));
	addPage(this.newFileCreationPage);

	// Try and get the resource selection to determine a current directory for the file dialog
	if (this.selection != null && !this.selection.isEmpty()) {
	// Get the resource...
	final Object selectedElement= this.selection.iterator().next();
	if (selectedElement instanceof IResource) {
	// Get the resource parent, if its a file
	IResource selectedResource= (IResource)selectedElement;
	if (selectedResource.getType() == IResource.FILE) {
	selectedResource= selectedResource.getParent();
	}

	// This gives us a directory...
	if (selectedResource instanceof IFolder \|\| selectedResource instanceof IProject) {
	// Set this for the container
	this.newFileCreationPage.setContainerFullPath(selectedResource.getFullPath());

	// // Make up a unique new name here
	// String defaultModelBaseFilename= RtGGPlotEditorPlugin.INSTANCE.getString("_UI_GGPlotEditorFilenameDefaultBase"); //$NON-NLS-1$
	// String defaultModelFilenameExtension= this.rtDescriptor.getDefaultContentTypeFileExtension();
	// String modelFilename= defaultModelBaseFilename + "." + defaultModelFilenameExtension; //$NON-NLS-1$
	// for (int i= 1; ((IContainer)selectedResource).findMember(modelFilename) != null; ++i) {
	// modelFilename= defaultModelBaseFilename + i + "." + defaultModelFilenameExtension; //$NON-NLS-1$
	// }
	// this.newFileCreationPage.setFileName(modelFilename);
	}
	}
	}
	}

	/**
	* Get the file from the page.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	*/
	public IFile getModelFile() {
	return this.newFileCreationPage.getModelFile();
	}

	}