gitlight/plugins/org.eclipse.papyrus.gitlight.compare.ui/src/org/eclipse/papyrus/gitlight/compare/ui/SaveParameterHook.java

/*****************************************************************************
 * Copyright (c) 2019 CEA LIST, and others.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License 2.0
 * which accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *   Nicolas FAUVERGUE (CEA LIST) nicolas.fauvergue@cea.fr - Initial API and implementation
 *
 *****************************************************************************/
package org.eclipse.papyrus.gitlight.compare.ui;

import java.util.Collection;
import java.util.Map;

import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.xmi.XMLResource;
import org.eclipse.papyrus.gitlight.compare.util.ModelExtensionUtil;
import org.eclipse.papyrus.infra.core.resource.sasheditor.DiModel;
import org.eclipse.papyrus.infra.gmfdiag.common.model.NotationModel;

/**
 * Hook into the EMF Compare {@link org.eclipse.emf.ecore.resource.ResourceSet} in order to adjust the default
 * save parameters.
 */
public class SaveParameterHook extends AbstractPapyrusResourceSetHook {

	/**
	 * Adjust the default save parameters of Papyrus resources. The {@link ResourceSet} is checked if it
	 * "really" contains Papyrus resources by looking for either a {@code .di} or {@code .notation} file since
	 * they are pretty unique to Papyrus. If either one is found all default save parameters of files with
	 * file extensions registered in Papyrus are checked for additional save parameters. {@inheritDoc}
	 */
	@Override
	public void postLoadingHook(final ResourceSet resourceSet, final Collection<? extends URI> uris) {
		if (containsPapyrusResources(resourceSet)) {
			for (final Resource resource : resourceSet.getResources()) {
				if (resource instanceof XMLResource && resource.getURI() != null) {
					final String resourceExtension = resource.getURI().fileExtension();
					if (fileExtensions.contains(resourceExtension)) {
						final Map<?, ?> saveParameters = ModelExtensionUtil.getSaveParameters(resourceExtension);
						final XMLResource xmlResource = (XMLResource) resource;
						xmlResource.getDefaultSaveOptions().putAll(saveParameters);
					}
				}
			}
		}
	}

	/**
	 * Walks through the {@link Resource}s of the {@link ResourceSet} to determine if it contains Papyrus
	 * resources.
	 * 
	 * @param resourceSet
	 *            The checked {@link ResourceSet}.
	 * @return {@code true} if the given {@code resourceSet} contains either a {@code .di} or
	 *         {@code .notation} file, {@code false} otherwise.
	 */
	private boolean containsPapyrusResources(final ResourceSet resourceSet) {
		for (final Resource resource : resourceSet.getResources()) {
			if (resource.getURI() != null) {
				final String resourceExtension = resource.getURI().fileExtension();
				if (DiModel.DI_FILE_EXTENSION.equals(resourceExtension) || NotationModel.NOTATION_FILE_EXTENSION.equals(resourceExtension)) {
					return true;
				}
			}
		}
		return false;
	}
}