plugins/org.eclipse.graphiti.ui/src/org/eclipse/graphiti/ui/internal/policy/GFConnectionEndpointTracker.java

/*********************************************************************
* Copyright (c) 2005, 2019 SAP SE
*
* This program and the accompanying materials are made
* available under the terms of the Eclipse Public License 2.0
* which is available at https://www.eclipse.org/legal/epl-2.0/
*
* Contributors:
*    mgorning - Bug 343983 - Notification for Cancelled Reconnection Events
*
* SPDX-License-Identifier: EPL-2.0
**********************************************************************/
package org.eclipse.graphiti.ui.internal.policy;

import org.eclipse.gef.ConnectionEditPart;
import org.eclipse.gef.commands.Command;
import org.eclipse.gef.tools.ConnectionEndpointTracker;
import org.eclipse.graphiti.ui.internal.command.ReconnectCommand;

public class GFConnectionEndpointTracker extends ConnectionEndpointTracker {

	public GFConnectionEndpointTracker(ConnectionEditPart cep) {
		super(cep);
	}

	@Override
	public void deactivate() {
		if (getCurrentCommand() instanceof ReconnectCommand) {
			ReconnectCommand cmd = (ReconnectCommand) getCurrentCommand();
			if (cmd != null) {
				cmd.deactivate();
			}
		}
		super.deactivate();
	}

	@Override
	protected boolean handleButtonUp(int button) {
		// Store current command for later usage in deactivate(). Call to
		// super.handleButtonUp() sets current command after execution always to
		// null.
		Command cmd = getCurrentCommand();
		boolean ret = super.handleButtonUp(button);
		setCurrentCommand(cmd);
		return ret;
	}
}