plugins/org.polarsys.chess.wizards/src/org/polarsys/chess/wizards/providers/WorkspaceNewModelStorageProvider.java

/*******************************************************************************
 * Copyright (C) 2020 
 * 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
 * http://www.eclipse.org/legal/epl-v20.html
 ******************************************************************************/
package org.polarsys.chess.wizards.providers;

import static org.eclipse.papyrus.uml.diagram.wizards.utils.WizardsHelper.adapt;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.ResourcesPlugin;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Path;
import org.eclipse.core.runtime.Status;
import org.eclipse.emf.common.ui.URIEditorInput;
import org.eclipse.emf.common.util.URI;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.wizard.IWizardPage;
import org.eclipse.papyrus.uml.diagram.wizards.pages.NewModelFilePage;
import org.eclipse.papyrus.uml.diagram.wizards.utils.WizardsHelper;
import org.eclipse.ui.IEditorInput;
import org.eclipse.ui.part.FileEditorInput;
import org.polarsys.chess.wizards.wizards.CreateCHESSModelWizard;
import org.polarsys.chess.wizards.wizards.InitModelWizard.NewDiagramForExistingModelPage;

/**
 * The Class WorkspaceNewModelStorageProvider.
 */
public class WorkspaceNewModelStorageProvider {

	/** The wizard. */
	private CreateCHESSModelWizard wizard;

	/** The new model file page. */
	private NewModelFilePage newModelFilePage;

	/**
	 * Instantiates a new workspace new model storage provider.
	 */
	public WorkspaceNewModelStorageProvider() {
		super();
	}

	/**
	 * Checks if the WorkspaceNewModelStorageProvider can handle the structured selection in input
	 *
	 * @param initialSelection the initial structured selection
	 * @return true, if successful
	 */
	public boolean canHandle(IStructuredSelection initialSelection) {
		boolean result = false;

		for(Object next : initialSelection.toList()) {
			if(adapt(next, IResource.class) != null) {
				result = true;
				break;
			}
		}

		return result;
	}

	/**
	 * Initializes the wizard.
	 *
	 * @param wizard the wizard
	 * @param selection the selection
	 */
	public void init(CreateCHESSModelWizard wizard, IStructuredSelection selection) {
		this.wizard = wizard;
		newModelFilePage = createNewModelFilePage(selection);
	}

	/**
	 * Creates the wizard pages.
	 *
	 * @return the list of wizards pages
	 */
	public List<? extends IWizardPage> createPages() {
		if(newModelFilePage == null) {
			return Collections.emptyList();
		}

		return Arrays.asList(newModelFilePage);
	}

	/**
	 * Validates the diagram categories.
	 *
	 * @param newCategories the new categories
	 * @return the status of the validation
	 */
	public IStatus validateDiagramCategories(String... newCategories) {
		if(newModelFilePage != null) {
			String firstCategory = newCategories.length > 0 ? newCategories[0] : null;
			if(newCategories.length > 0) {
				return newModelFilePage.diagramExtensionChanged(wizard.getDiagramFileExtension(firstCategory));
			}
		}

		return Status.OK_STATUS;
	}


	/**
	 * Creates the new model file page.
	 *
	 * @param selection the current structured selection
	 * @return the new model file page
	 */
	protected NewModelFilePage createNewModelFilePage(IStructuredSelection selection) {

		if(wizard.isCreateProjectWizard() || wizard.isCreateMultipleModelsWizard()) {

			return null;
		}

		//		IFile selectedFile = getSelectedFile(selection);
		URI selectedResourceURI = WizardsHelper.getSelectedResourceURI(selection);

		if(isCreateFromExistingDomainModel() && selectedResourceURI != null) {
			return new NewDiagramForExistingModelPage(selection, wizard.getModelKindName(), getDiagramFileName(selectedResourceURI) + "." + wizard.getDiagramFileExtension(null), wizard.getDiagramFileExtension(null)); //$NON-NLS-1$
		}

		return new NewModelFilePage(selection, wizard.getModelKindName());
	}

	/**
	 * Gets the diagram file name.
	 *
	 * @param domainModelURI the domain model URI
	 * @return the diagram file name
	 */
	protected String getDiagramFileName(URI domainModelURI) {
		return domainModelURI.trimFileExtension().lastSegment();
	}

	/**
	 * Checks if the wizards creates the from existing domain model.
	 *
	 * @return false
	 */
	protected boolean isCreateFromExistingDomainModel() {
		return false;
	}

	/**
	 * Creates the new model URI
	 *
	 * @param categoryId the category id
	 * @return the URI
	 */
	public URI createNewModelURI(String categoryId) {
		IFile newFile = newModelFilePage.createNewFile();
		return (newFile == null) ? null : URI.createPlatformResourceURI(newFile.getFullPath().toString(), true);
	}
	
	/**
	 * Creates the editor input.
	 *
	 * @param uri the URI
	 * @return the editor input
	 */
	public IEditorInput createEditorInput(URI uri) {
		if (uri.isPlatformResource()) {
			return new FileEditorInput(ResourcesPlugin.getWorkspace().getRoot()
				.getFile(new Path(uri.toPlatformString(true))));
		} else {
			return new URIEditorInput(uri);
		}
	}

}