// ============================================================================
//
// Copyright (C) 2006-2012 Talend Inc. - www.talend.com
//
// This source code is available under agreement available at
// %InstallDIR%\features\org.talend.rcp.branding.%PRODUCTNAME%\%PRODUCTNAME%license.txt
//
// You should have received a copy of the agreement
// along with this program; if not, write to Talend SA
// 9 rue Pages 92150 Suresnes, France
//
// ============================================================================
package org.talend.designer.core.ui.editor.connections;

import java.util.List;

import org.eclipse.draw2d.Figure;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.gef.commands.Command;
import org.eclipse.gef.editpolicies.NonResizableEditPolicy;
import org.eclipse.gef.requests.ChangeBoundsRequest;
import org.talend.core.model.process.Element;
import org.talend.designer.core.ui.editor.cmd.MoveConnTextCommand;
import org.talend.designer.core.ui.editor.cmd.MoveConnTraceCommand;

/**
 * Policy that will allow to move the label of the connection. <br/>
 * 
 * $Id: ConnTextMovePolicy.java 77219 2012-01-24 01:14:15Z mhirt $
 * 
 */
public class ConnTextMovePolicy extends NonResizableEditPolicy {

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.eclipse.gef.editpolicies.NonResizableEditPolicy#getMoveCommand(org.eclipse.gef.requests.ChangeBoundsRequest)
     */
    public Command getMoveCommand(ChangeBoundsRequest request) {
        if (((Connection) getHost().getParent().getModel()).isReadOnly()) {
            return null;
        }
        if (getHost().getModel() instanceof ConnectionTrace) {
            ConnectionTrace model = (ConnectionTrace) getHost().getModel();
            Point delta = request.getMoveDelta();
            ConnectionPart edge = (ConnectionPart) getHost().getParent();
            MoveConnTraceCommand command = new MoveConnTraceCommand(model, (Figure) edge.getFigure(), delta);
            return command;
        } else if (getHost().getModel() instanceof ConnectionLabel) {
            ConnectionLabel model = (ConnectionLabel) getHost().getModel();
            Point delta = request.getMoveDelta();
            ConnectionPart edge = (ConnectionPart) getHost().getParent();
            List<Element> elements = edge.getModelChildren();
            for (Element e : elements) {
                if (e instanceof ConnectionResuming) {
                    MoveConnTextCommand command = new MoveConnTextCommand(model, (ConnectionResuming) e, (Figure) edge
                            .getFigure(), delta);
                    return command;
                }
            }
            MoveConnTextCommand command = new MoveConnTextCommand(model, null, (Figure) edge.getFigure(), delta);
            return command;
        }
        return null;
    }
}