StateContainerEditPolicy.java

/*******************************************************************************
 * Copyright (c) 2007 Spring IDE Developers
 * 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:
 *     Spring IDE Developers - initial API and implementation
 *******************************************************************************/
package org.springframework.ide.eclipse.webflow.ui.graph.policies;

import java.util.List;

import org.eclipse.gef.EditPart;
import org.eclipse.gef.commands.Command;
import org.eclipse.gef.commands.CompoundCommand;
import org.eclipse.gef.editpolicies.ContainerEditPolicy;
import org.eclipse.gef.requests.ChangeBoundsRequest;
import org.eclipse.gef.requests.CreateRequest;
import org.eclipse.gef.requests.GroupRequest;
import org.springframework.ide.eclipse.webflow.core.internal.model.SubflowState;
import org.springframework.ide.eclipse.webflow.core.model.IActionElement;
import org.springframework.ide.eclipse.webflow.core.model.IActionState;
import org.springframework.ide.eclipse.webflow.core.model.IAttributeMapper;
import org.springframework.ide.eclipse.webflow.core.model.IDecisionState;
import org.springframework.ide.eclipse.webflow.core.model.IExceptionHandler;
import org.springframework.ide.eclipse.webflow.core.model.IIf;
import org.springframework.ide.eclipse.webflow.core.model.IInlineFlowState;
import org.springframework.ide.eclipse.webflow.core.model.IState;
import org.springframework.ide.eclipse.webflow.core.model.ISubflowState;
import org.springframework.ide.eclipse.webflow.core.model.ITransitionableFrom;
import org.springframework.ide.eclipse.webflow.core.model.IViewState;
import org.springframework.ide.eclipse.webflow.core.model.IWebflowModelElement;
import org.springframework.ide.eclipse.webflow.core.model.IWebflowState;
import org.springframework.ide.eclipse.webflow.ui.graph.commands.ActionCloneCommand;
import org.springframework.ide.eclipse.webflow.ui.graph.commands.ActionOrphanChildCommand;
import org.springframework.ide.eclipse.webflow.ui.graph.commands.AttributeMapperCloneCommand;
import org.springframework.ide.eclipse.webflow.ui.graph.commands.AttributeMapperOrphanChildCommand;
import org.springframework.ide.eclipse.webflow.ui.graph.commands.ExceptionHandlerCloneCommand;
import org.springframework.ide.eclipse.webflow.ui.graph.commands.ExceptionHandlerOrphanChildCommand;
import org.springframework.ide.eclipse.webflow.ui.graph.commands.IfOrphanChildCommand;

/**
 * @author Christian Dupuis
 */
public class StateContainerEditPolicy extends ContainerEditPolicy {

	/*
	 * (non-Javadoc)
	 * @see org.eclipse.gef.editpolicies.ContainerEditPolicy#getCreateCommand(org.eclipse.gef.requests.CreateRequest)
	 */
	protected Command getCreateCommand(CreateRequest request) {
		return null;
	}

	/*
	 * (non-Javadoc)
	 * @see org.eclipse.gef.editpolicies.ContainerEditPolicy#getOrphanChildrenCommand(org.eclipse.gef.requests.GroupRequest)
	 */
	protected Command getOrphanChildrenCommand(GroupRequest request) {
		List parts = request.getEditParts();
		CompoundCommand result = new CompoundCommand();
		for (int i = 0; i < parts.size(); i++) {
			EditPart part = (EditPart) parts.get(i);
			if (part.getModel() instanceof IActionElement) {
				ActionOrphanChildCommand orphan = new ActionOrphanChildCommand();
				orphan.setChild((IActionElement) ((EditPart) parts.get(i))
						.getModel());
				result.add(orphan);
			}
			else if (part.getModel() instanceof IExceptionHandler) {
				ExceptionHandlerOrphanChildCommand orphan = new ExceptionHandlerOrphanChildCommand();
				orphan.setChild((IExceptionHandler) ((EditPart) parts.get(i))
						.getModel());
				result.add(orphan);
			}
			else if (part.getModel() instanceof IAttributeMapper) {
				AttributeMapperOrphanChildCommand orphan = new AttributeMapperOrphanChildCommand();
				orphan.setChild((IAttributeMapper) ((EditPart) parts.get(i))
						.getModel());
				orphan.setParent((ISubflowState) getHost().getModel());
				result.add(orphan);
			}
			else if (part.getModel() instanceof IIf) {
				IfOrphanChildCommand orphan = new IfOrphanChildCommand();
				orphan.setChild((IIf) ((EditPart) parts.get(i)).getModel());
				orphan.setParent((IDecisionState) getHost().getModel());
				result.add(orphan);
			}
		}
		return result.unwrap();
	}

	/**
	 * Override to contribute to clone requests.
	 * @param request the clone request
	 * @return the command contribution to the clone
	 */
	protected Command getCloneCommand(ChangeBoundsRequest request) {
		List parts = request.getEditParts();
		CompoundCommand result = new CompoundCommand();
		for (int i = 0; i < parts.size(); i++) {
			EditPart part = (EditPart) parts.get(i);
			if (part.getModel() instanceof IActionElement) {
				IActionElement ae = (IActionElement) part.getModel();
				IState state = (IState) getHost().getModel();
				if ((ae.getType() == IActionElement.ACTION_TYPE.ACTION && state instanceof IActionState)
						|| (ae.getType() == IActionElement.ACTION_TYPE.RENDER_ACTION && state instanceof IViewState)
						|| (ae.getType() == IActionElement.ACTION_TYPE.ENTRY_ACTION && state instanceof IState)
						|| (ae.getType() == IActionElement.ACTION_TYPE.EXIT_ACTION && state instanceof ITransitionableFrom)) {
					ActionCloneCommand orphan = new ActionCloneCommand();
					orphan.setChild((IActionElement) ((EditPart) parts.get(i))
							.getModel());
					orphan.setNewState((IWebflowModelElement) getHost()
							.getModel());
					result.add(orphan);
				}
			}
			else if (part.getModel() instanceof IExceptionHandler) {
				IState state = (IState) getHost().getModel();
				if (!(state instanceof IWebflowState)
						&& !(state instanceof IInlineFlowState)) {
					ExceptionHandlerCloneCommand orphan = new ExceptionHandlerCloneCommand();
					orphan.setChild((IExceptionHandler) ((EditPart) parts.get(i))
							.getModel());
					orphan.setNewState((IWebflowModelElement) getHost()
							.getModel());
					result.add(orphan);
				}
			}
			else if (part.getModel() instanceof IAttributeMapper) {
				IState state = (IState) getHost().getModel();
				if (state instanceof SubflowState
						&& ((SubflowState) state).getAttributeMapper() == null) {
					AttributeMapperCloneCommand orphan = new AttributeMapperCloneCommand();
					orphan.setChild((IAttributeMapper) ((EditPart) parts.get(i))
							.getModel());
					orphan.setNewState((IWebflowModelElement) getHost()
							.getModel());
					result.add(orphan);
				}
			}
		}
		return result.unwrap();
	}
}