core/plugins/org.eclipse.dltk.ui/core refactoring/org/eclipse/dltk/internal/corext/refactoring/reorg/CopyModifications.java

/*******************************************************************************
 * Copyright (c) 2000, 2017 IBM Corporation and others.
 * 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
 *
 *******************************************************************************/
package org.eclipse.dltk.internal.corext.refactoring.reorg;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import org.eclipse.core.resources.IContainer;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.resources.mapping.IResourceChangeDescriptionFactory;
import org.eclipse.core.resources.mapping.ResourceMapping;
import org.eclipse.core.runtime.Assert;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.dltk.core.IModelElement;
import org.eclipse.dltk.core.IProjectFragment;
import org.eclipse.dltk.core.IScriptFolder;
import org.eclipse.dltk.core.ISourceModule;
import org.eclipse.dltk.internal.corext.refactoring.participants.ResourceModifications;
import org.eclipse.dltk.internal.ui.model.DLTKElementResourceMapping;
import org.eclipse.ltk.core.refactoring.RefactoringStatus;
import org.eclipse.ltk.core.refactoring.participants.CopyArguments;
import org.eclipse.ltk.core.refactoring.participants.IParticipantDescriptorFilter;
import org.eclipse.ltk.core.refactoring.participants.ParticipantManager;
import org.eclipse.ltk.core.refactoring.participants.RefactoringArguments;
import org.eclipse.ltk.core.refactoring.participants.RefactoringParticipant;
import org.eclipse.ltk.core.refactoring.participants.RefactoringProcessor;
import org.eclipse.ltk.core.refactoring.participants.SharableParticipants;

public class CopyModifications extends RefactoringModifications {

	private List fCopies;
	private List fCopyArguments;
	private List fParticipantDescriptorFilter;

	public CopyModifications() {
		fCopies = new ArrayList();
		fCopyArguments = new ArrayList();
		fParticipantDescriptorFilter = new ArrayList();
	}

	public void copy(IResource resource, CopyArguments args) {
		add(resource, args, null);
	}

	public void copy(IModelElement element, CopyArguments javaArgs,
			CopyArguments resourceArgs) throws CoreException {
		switch (element.getElementType()) {
		case IModelElement.PROJECT_FRAGMENT:
			copy((IProjectFragment) element, javaArgs, resourceArgs);
			break;
		case IModelElement.SCRIPT_FOLDER:
			copy((IScriptFolder) element, javaArgs, resourceArgs);
			break;
		case IModelElement.SOURCE_MODULE:
			copy((ISourceModule) element, javaArgs, resourceArgs);
			break;
		default:
			add(element, javaArgs, null);
		}
	}

	public void copy(IProjectFragment sourceFolder, CopyArguments javaArgs,
			CopyArguments resourceArgs) {
		add(sourceFolder, javaArgs, null);
		ResourceMapping mapping = DLTKElementResourceMapping
				.create(sourceFolder);
		if (mapping != null) {
			add(mapping, resourceArgs, null);
		}
		IResource sourceResource = sourceFolder.getResource();
		if (sourceResource != null) {
			getResourceModifications().addCopyDelta(sourceResource,
					resourceArgs);
			IFile buildpath = getBuildpathFile(
					(IResource) resourceArgs.getDestination());
			if (buildpath != null) {
				getResourceModifications().addChanged(buildpath);
			}
		}
	}

	public void copy(IScriptFolder pack, CopyArguments javaArgs,
			CopyArguments resourceArgs) throws CoreException {
		add(pack, javaArgs, null);
		ResourceMapping mapping = DLTKElementResourceMapping.create(pack);
		if (mapping != null) {
			add(mapping, resourceArgs, null);
		}
		IProjectFragment scriptDestination = (IProjectFragment) javaArgs
				.getDestination();
		if (scriptDestination.getResource() == null)
			return;
		IScriptFolder newPack = scriptDestination
				.getScriptFolder(pack.getElementName());
		// Here we have a special case. When we copy a package into the same
		// source
		// folder than the user will choose an "unused" name at the end which
		// will
		// lead to the fact that we can copy the pack. Unfortunately we don't
		// know
		// the new name yet, so we use the current package name.
		if (!pack.hasSubfolders()
				&& (!newPack.exists() || pack.equals(newPack))) {
			// we can do a simple move
			IContainer resourceDestination = newPack.getResource().getParent();
			createIncludingParents(resourceDestination);
			getResourceModifications().addCopyDelta(pack.getResource(),
					resourceArgs);
		} else {
			IContainer resourceDestination = (IContainer) newPack.getResource();
			createIncludingParents(resourceDestination);
			CopyArguments arguments = new CopyArguments(resourceDestination,
					resourceArgs.getExecutionLog());
			IResource[] resourcesToCopy = collectResourcesOfInterest(pack);
			for (int i = 0; i < resourcesToCopy.length; i++) {
				IResource toCopy = resourcesToCopy[i];
				getResourceModifications().addCopyDelta(toCopy, arguments);
			}
		}
	}

	public void copy(ISourceModule unit, CopyArguments javaArgs,
			CopyArguments resourceArgs) throws CoreException {
		add(unit, javaArgs, null);
		ResourceMapping mapping = DLTKElementResourceMapping.create(unit);
		if (mapping != null) {
			add(mapping, resourceArgs, null);
		}
		if (unit.getResource() != null) {
			getResourceModifications().addCopyDelta(unit.getResource(),
					resourceArgs);
		}
	}

	@Override
	public void buildDelta(IResourceChangeDescriptionFactory builder) {
		for (int i = 0; i < fCopies.size(); i++) {
			Object element = fCopies.get(i);
			if (element instanceof IResource) {
				ResourceModifications.buildCopyDelta(builder,
						(IResource) element,
						(CopyArguments) fCopyArguments.get(i));
			}
		}
		getResourceModifications().buildDelta(builder);
	}

	@Override
	public RefactoringParticipant[] loadParticipants(RefactoringStatus status,
			RefactoringProcessor owner, String[] natures,
			SharableParticipants shared) {
		List<RefactoringParticipant> result = new ArrayList<>();
		for (int i = 0; i < fCopies.size(); i++) {
			result.addAll(Arrays.asList(ParticipantManager.loadCopyParticipants(
					status, owner, fCopies.get(i),
					(CopyArguments) fCopyArguments.get(i),
					(IParticipantDescriptorFilter) fParticipantDescriptorFilter
							.get(i),
					natures, shared)));
		}
		result.addAll(Arrays.asList(getResourceModifications()
				.getParticipants(status, owner, natures, shared)));
		return result.toArray(new RefactoringParticipant[result.size()]);
	}

	private void add(Object element, RefactoringArguments args,
			IParticipantDescriptorFilter filter) {
		Assert.isNotNull(element);
		Assert.isNotNull(args);
		fCopies.add(element);
		fCopyArguments.add(args);
		fParticipantDescriptorFilter.add(filter);
	}
}