/*******************************************************************************
 * Copyright (c) 2005, 2012 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
 *
 * Contributors:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.bpel.ui.commands;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.eclipse.bpel.common.ui.editmodel.AbstractEditModelCommand;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.EditPartViewer;


/**
 * A Command that knows how to capture and restore a selection in the EditPartViewer.
 * 
 * On execute(), this Command remembers the set of model objects whose EditParts are
 * selected in the EditPartViewer.  On undo(), the EditParts representing those model
 * objects are re-selected in the same order.
 * 
 * This is used to restore the user context when a DeleteSelectedAction is undone.
 */
public class RestoreViewerSelectionCommand extends AbstractEditModelCommand {

	EditPartViewer viewer;
	List selectedModelObjects;
	boolean restoreOnUndo, restoreOnRedo;

	public RestoreViewerSelectionCommand(EditPartViewer viewer, boolean restoreOnUndo,
		boolean restoreOnRedo)
	{
		this.viewer = viewer;
		this.restoreOnUndo = restoreOnUndo;
		this.restoreOnRedo = restoreOnRedo;
	}

	// This method may be called directly before execute() is called.  Otherwise,
	// it will be called by execute().
	public void captureSelection() {
		// Remember the model objects whose EditParts are selected in the viewer.
		// (note: the EditParts themselves may be destroyed and re-created)
		selectedModelObjects = new ArrayList();
		for (Iterator it = viewer.getSelectedEditParts().iterator(); it.hasNext(); ) {
			Object modelObject = ((EditPart)it.next()).getModel();
			if (modelObject != null)  selectedModelObjects.add(modelObject);
		}
	}

	@Override
	public void execute() {
		if (selectedModelObjects == null)  captureSelection();
	}
	
	protected void restoreSelection() {
		// Re-select the current EditParts for the formerly selected model objects.
		viewer.deselectAll();
		Map modelObject2EditPart = viewer.getEditPartRegistry();
		for (Iterator it = selectedModelObjects.iterator(); it.hasNext(); ) {
			EditPart editPart = (EditPart)modelObject2EditPart.get(it.next());
			if (editPart != null)  viewer.appendSelection(editPart);
		}
	}

	@Override
	public void undo() { if (restoreOnUndo)  restoreSelection(); }
	@Override
	public void redo() { if (restoreOnRedo)  restoreSelection(); }

	@Override
	public Resource[] getResources() { return EMPTY_RESOURCE_ARRAY; }
	@Override
	public Resource[] getModifiedResources() { return EMPTY_RESOURCE_ARRAY; }
}