/**
* Copyright 2010 JBoss Inc
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.drools.eclipse.flow.common.editor.core.command;
/*
* Copyright 2005 JBoss Inc
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
import org.drools.eclipse.flow.common.editor.core.ElementConnection;
import org.drools.eclipse.flow.common.editor.core.ElementWrapper;
import org.drools.eclipse.flow.common.editor.core.ProcessWrapper;
import org.eclipse.gef.commands.Command;
/**
* A command for splitting a connection.
*
* @author <a href="mailto:kris_verlaenen@hotmail.com">Kris Verlaenen</a>
*/
public class SplitConnectionCommand extends Command {
private ProcessWrapper parent;
private ElementWrapper oldSource;
private ElementWrapper oldTarget;
private ElementConnection oldConnection;
private ElementConnection secondConnection;
private ElementWrapper newElement;
public void setNewSecondConnection(ElementConnection connection) {
if (connection == null) {
throw new IllegalArgumentException("new second connection cannot be null");
}
this.secondConnection = connection;
}
public void execute() {
if (secondConnection == null) {
throw new IllegalStateException("new second connection is still null");
}
oldConnection.disconnect();
parent.addElement(newElement);
newElement.setParent(parent);
oldConnection.connect(oldSource, newElement);
secondConnection.connect(newElement, oldTarget);
}
public void setParent(ProcessWrapper process) {
if (process == null) {
throw new IllegalArgumentException("process is null");
}
parent = process;
}
public void setElementConnection(ElementConnection connection) {
if (connection == null) {
throw new IllegalArgumentException("Element connection is null");
}
this.oldConnection = connection;
oldSource = connection.getSource();
oldTarget = connection.getTarget();
}
public void setNewElement(ElementWrapper newElement) {
if (newElement == null) {
throw new IllegalArgumentException("NewElement is null");
}
this.newElement = newElement;
}
public void undo() {
oldConnection.disconnect();
secondConnection.disconnect();
parent.removeElement(newElement);
newElement.setParent(null);
oldConnection.connect(oldSource, oldTarget);
}
}