jsf/plugins/org.eclipse.jst.jsf.facesconfig.ui/src/org/eclipse/jst/jsf/facesconfig/ui/pageflow/command/ConnectionCommand.java

/*******************************************************************************
 * Copyright (c) 2004, 2005 Sybase, Inc. 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:
 *     Sybase, Inc. - initial API and implementation
 *******************************************************************************/

package org.eclipse.jst.jsf.facesconfig.ui.pageflow.command;

import org.eclipse.gef.commands.Command;
import org.eclipse.jst.jsf.facesconfig.ui.EditorResources;
import org.eclipse.jst.jsf.facesconfig.ui.pageflow.layout.PageflowLayoutManager;
import org.eclipse.jst.jsf.facesconfig.ui.pageflow.model.PFLink;
import org.eclipse.jst.jsf.facesconfig.ui.pageflow.model.Pageflow;
import org.eclipse.jst.jsf.facesconfig.ui.pageflow.model.PageflowNode;
import org.eclipse.jst.jsf.facesconfig.ui.pageflow.util.PageflowValidation;

/**
 * This is the connection command for pageflow editpart
 * 
 * @author Xiao-guang Zhang
 */
public class ConnectionCommand extends Command {
	/** Old source pageflow node */
	protected PageflowNode oldSource;

	/** Old target pageflow node */
	protected PageflowNode oldTarget;

	/** New source pageflow node */
	protected PageflowNode pageflowNode;

	/** New target pageflow node */
	protected PageflowNode target;

	/** pageflow link */
	protected PFLink link;

	public ConnectionCommand() {
		// Pageflow.Commands.ConnectionCommand.Label = Connection
		super(EditorResources.getInstance().getString(
				"Pageflow.Commands.ConnectionCommand.Label"));
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see Command#canExecute()
	 */
	public boolean canExecute() {
		// if user don't set PFLink object before, this command can't be
		// executed.
		if (link == null) {
			return false;
		}

		// It is a connection create command
		if (oldSource == null && oldTarget == null) {
			// It is a connection create command
			// Source and target must be pointing to some
			// real connection point
			if (pageflowNode == null || target == null) {
				return false;
			}

			if (!PageflowValidation.getInstance().isValidLinkForCreation(
					pageflowNode, target)) {
				return false;
			}
		}
		// It is a reconnect both of source and target command
		if (oldSource != null && pageflowNode != null && oldTarget != null
				&& target != null) {
			if (!PageflowValidation.getInstance().isValidLinkForCreation(
					pageflowNode, target)) {
				return false;
			}
		}

		// It is a reconnect only source command
		if (oldSource != null && pageflowNode != null) {
			if (!PageflowValidation.getInstance().isValidLinkForCreation(
					pageflowNode, oldTarget)) {
				return false;
			}
		}
		// It is a reconnect only target command
		if (oldTarget != null && target != null) {
			if (!PageflowValidation.getInstance().isValidLinkForCreation(
					oldSource, target)) {
				return false;
			}
		}

		return true;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see Command#execute()
	 */
	public void execute() {
		// It is a delete connection command
		if (pageflowNode == null && target == null) {
			oldSource.getOutlinks().remove(link);
			oldTarget.getInlinks().remove(link);
			oldSource.getPageflow().getLinks().remove(link);
			// deleteNavigationRule(oldSource, oldTarget);
		}
		// It is a reconnect source command
		if (oldSource != null && pageflowNode != null) {
			// The link is still linked to the oldSource
			if (link.getSource() != null) {
				link.getSource().getOutlinks().remove(link);
			}
			// No containment link between link and input and output port
			// Two add method need to be called
			link.setSource(pageflowNode);
			pageflowNode.getOutlinks().add(link);
		}
		// It is a reconnect target command
		if (oldTarget != null && target != null) {
			// The target is still linked to the oldTarget
			if (link.getTarget() != null) {
				link.getTarget().getInlinks().remove(link);
			}
			// No containment link between link and input and output port
			// Two add method need to be called
			link.setTarget(target);
			target.getInlinks().add(link);
		}

		// It is a connection create command
		if (oldSource == null && oldTarget == null) {
			Pageflow pageflow = null;
			// Get a reference to the pageflow
			if (pageflowNode != null) {
				pageflow = pageflowNode.getPageflow();
			} else if (target != null) {
				pageflow = target.getPageflow();
			}
			// Need to modify the SSE model directly.
			pageflow.connect(pageflowNode, target, link);

			// self loop link
			if (pageflowNode == target) {
				PageflowLayoutManager.updateSelfLoopLink(link);
			}
		}
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see Command#redo()
	 */
	public void redo() {
		execute();
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see ommand#undo()
	 */
	public void undo() {
		if (link == null) {
			return;
		}
		// It is a delete connection command
		if (pageflowNode == null && target == null) {
			oldSource.getOutlinks().add(link);
			oldTarget.getInlinks().add(link);
			oldSource.getPageflow().getLinks().add(link);
		}

		// It was a reconnect source command
		if (oldSource != null && pageflowNode != null) {
			// The link source must be replaced by the oldSource
			if (link.getSource() != null) {
				link.getSource().getOutlinks().remove(link);
			}
			// Source should not know link anymore
			pageflowNode.getOutlinks().remove(link);
			// Re-link with oldSource
			// No containment link between link and input and output
			// Two add method need to be called
			link.setSource(oldSource);
			oldSource.getOutlinks().add(link);
		}
		// It was a reconnect target command
		if (oldTarget != null && target != null) {
			// The link target must be replaced by the oldTarget
			if (link.getTarget() != null) {
				link.getTarget().getInlinks().remove(link);
			}
			// Target should not know link anymore
			target.getInlinks().remove(link);
			// Re-link with oldTarget
			// No containment link between link and input and output port
			// Two add method need to be called
			link.setTarget(oldTarget);
			oldTarget.getInlinks().add(link);
		}

		// It was a connection create command
		if (oldSource == null && oldTarget == null) {

			Pageflow pageflow = null;
			// Get a reference to the pageflow
			if (pageflowNode != null) {
				pageflow = pageflowNode.getPageflow();
			} else if (target != null) {
				pageflow = target.getPageflow();
			}

			// Remove all reference to the link
			pageflowNode.getOutlinks().remove(link);
			target.getInlinks().remove(link);

			// Ensure that link knows nothing about nodes anymore.
			link.setSource(null);
			link.setTarget(null);
			// Remove link from pageflow
			pageflow.getLinks().remove(link);
		}
	}

	/**
	 * Get the source pageflow node
	 * 
	 * @return - the source pageflow node
	 */
	public PageflowNode getSource() {
		return pageflowNode;
	}

	/**
	 * Get the target pageflow node
	 * 
	 * @return - the target pageflow node.
	 */
	public PageflowNode getTarget() {
		return target;
	}

	/**
	 * Get the pageflow link
	 * 
	 * @return - the pageflow link.
	 */
	public PFLink getPFLink() {
		return link;
	}

	/**
	 * Set the source pageflow node
	 * 
	 * @param newSource -
	 *            new source pageflow node
	 */
	public void setSource(PageflowNode newSource) {
		pageflowNode = newSource;
	}

	/**
	 * Set the target pageflow node
	 * 
	 * @param newTarget -
	 *            new target pageflow node
	 */
	public void setTarget(PageflowNode newTarget) {
		target = newTarget;
	}

	/**
	 * Set the pageflow link
	 * 
	 * @param link -
	 *            new pageflow link
	 */
	public void setPFLink(PFLink link) {
		this.link = link;
		oldSource = link.getSource();
		oldTarget = link.getTarget();
	}
}