NodeDragTargetListener.java

/**
 * Licensed under the GNU LESSER GENERAL PUBLIC LICENSE, Version 3, 29 June 2007;
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.gnu.org/licenses/lgpl-3.0.txt
 *
 * 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.
 */
/**
 * Licensed under the GNU LESSER GENERAL PUBLIC LICENSE, Version 3, 29 June 2007;
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * 	http://www.gnu.org/licenses/lgpl-3.0.txt
 *
 * 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.
 */
/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package org.kevoree.tools.ui.editor.listener;

import org.kevoree.ComponentInstance;
import org.kevoree.ContainerNode;
import org.kevoree.tools.ui.editor.KevoreeUIKernel;
import org.kevoree.tools.ui.editor.command.AddComponentCommand;
import org.kevoree.tools.ui.editor.command.AddGroupBindingCommand;
import org.kevoree.tools.ui.editor.command.AddNodeCommand;
import org.kevoree.tools.ui.editor.command.MoveComponentCommand;
import org.kevoree.tools.ui.framework.elements.*;

import javax.swing.*;
import java.awt.*;
import java.awt.datatransfer.DataFlavor;
import java.awt.dnd.DropTarget;
import java.awt.dnd.DropTargetDragEvent;
import java.awt.dnd.DropTargetDropEvent;
import java.awt.dnd.DropTargetEvent;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * implementation of the target listener
 *
 * @author francoisfouquet
 */
public class NodeDragTargetListener extends DropTarget {

    KevoreeUIKernel kernel;
    NodePanel target;


    /**
     * constructor
     *
     * @param _kernel the table view panel
     * @param _target the view of the component
     */
    public NodeDragTargetListener(NodePanel _target, KevoreeUIKernel _kernel) {
        kernel = _kernel;
        target = _target;
    }

    private Boolean isDropAccept(Object o) {
        if (o instanceof ComponentTypePanel) {
            return true;
        }
        if (o instanceof ComponentPanel) {
            //CHECK IF THIS COMPONENT IS NOT IN NODE
            ComponentInstance component = (ComponentInstance) kernel.getUifactory().getMapping().get(o);
            ContainerNode node = (ContainerNode) kernel.getUifactory().getMapping().get(target);
            if (node.getComponents().contains(component)) {
                return false;
            } else {
                return true;
            }
        }
        if (o instanceof GroupAnchorPanel) {
            return true;
        }
        if ( (o instanceof NodeTypePanel)) {
            return true;
        }
        if (o instanceof NodePanel) {
            return true;
        }
        return false;
    }

    /**
     * callback when DnD is finished
     *
     * @param arg0
     */
    @Override
    public void drop(DropTargetDropEvent arg0) {
        target.setSelected(false);
        target.repaint();
        target.revalidate();

        try {
            Object o = arg0.getTransferable().getTransferData(new DataFlavor(DataFlavor.javaJVMLocalObjectMimeType));
            if (isDropAccept(o)) {
                if (o instanceof ComponentTypePanel) {
                    AddComponentCommand command = new AddComponentCommand();
                    command.setKernel(kernel);
                    command.setNodepanel(target);
                    command.execute(o);
                }
                if (o instanceof ComponentPanel) {
                    MoveComponentCommand command = new MoveComponentCommand();
                    command.setKernel(kernel);
                    command.setNodepanel(target);
                    command.execute(o);
                }
                if (o instanceof GroupAnchorPanel) {
                    AddGroupBindingCommand command = new AddGroupBindingCommand();
                    command.setKernel(kernel);
                    command.setTarget(target);
                    command.execute(o);
                }
                if (o instanceof NodeTypePanel) {
                    AddNodeCommand command = new AddNodeCommand();
                    command.setParentPanel(target);
                    command.setPoint(arg0.getLocation());
                    command.setKernel(kernel);
                    command.execute(o);
                } else {
                    if (o instanceof NodePanel) {
                        ContainerNode newParent = (ContainerNode) kernel.getUifactory().getMapping().get(target);
                        if (((NodePanel) o).getParent() != null) {
                            ((NodePanel) o).getParent().remove((NodePanel) o);
                        }
                        target.add((NodePanel) o);
                        NodeDragSourceListener lis = new NodeDragSourceListener((NodePanel) o,kernel);

                        for(ContainerNode node : kernel.getModelHandler().getActualModel().getNodes()){
                            if(node.getHosts().contains((ContainerNode) kernel.getUifactory().getMapping().get(o))){
                                node.removeHosts((ContainerNode) kernel.getUifactory().getMapping().get(o));
                            }
                        }
                        newParent.addHosts((ContainerNode) kernel.getUifactory().getMapping().get(o));
                        kernel.getModelHandler().notifyChanged();
                    }
                }
                kernel.getModelPanel().repaint();
                kernel.getModelPanel().revalidate();
                arg0.dropComplete(true);
            } else {
                arg0.rejectDrop();
            }
        } catch (Exception ex) {
            Logger.getLogger(NodeDragTargetListener.class.getName()).log(Level.SEVERE, null, ex);
            arg0.rejectDrop();
        }

    }

    /**
     * not implemented
     *
     * @param dtde
     */
    @Override
    public void dragEnter(DropTargetDragEvent dtde) {
        target.setSelected(true);
        target.repaint();
        target.revalidate();
    }

    /**
     * not implemented
     *
     * @param arg0
     */
    @Override
    public void dragExit(DropTargetEvent arg0) {
       target.setSelected(false);
        target.repaint();
        target.revalidate();
    }

    /**
     * not implemented
     *
     * @param arg0
     */
    @Override
    public void dragOver(DropTargetDragEvent arg0) {
        if (kernel.getModelPanel().getFlightObject() != null) {

            Point p2 = arg0.getLocation();
            SwingUtilities.convertPointToScreen(p2, target);
            SwingUtilities.convertPointFromScreen(p2, kernel.getModelPanel().getFlightObject().getParent());

            kernel.getModelPanel().getFlightObject().setBounds(
                    (int) p2.getX() - (kernel.getModelPanel().getFlightObject().getWidth() / 2),
                    (int) p2.getY() - (kernel.getModelPanel().getFlightObject().getHeight() / 2),
                    kernel.getModelPanel().getFlightObject().getWidth(),
                    kernel.getModelPanel().getFlightObject().getHeight());
            kernel.getModelPanel().repaint();
        }


    }

    /**
     * not implemented
     *
     * @param arg0
     */
    @Override
    public void dropActionChanged(DropTargetDragEvent arg0) {
    }
}