/*
 *    Geotoolkit - An Open Source Java GIS Toolkit
 *    http://www.geotoolkit.org
 *
 *    (C) 2007 - 2008, Open Source Geospatial Foundation (OSGeo)
 *    (C) 2008 - 2011, Johann Sorel
 *
 *    This library is free software; you can redistribute it and/or
 *    modify it under the terms of the GNU Lesser General Public
 *    License as published by the Free Software Foundation;
 *    version 2.1 of the License.
 *
 *    This library is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *    Lesser General Public License for more details.
 */
package org.geotoolkit.gui.swing.contexttree;

import java.awt.Component;
import javax.swing.JPopupMenu;
import javax.swing.tree.TreePath;

import org.geotoolkit.gui.swing.contexttree.menu.SeparatorItem;
import org.geotoolkit.gui.swing.contexttree.menu.TitledSeparatorItem;

/**
 * TreePopup item.
 *
 * @author Johann Sorel (Puzzle-GIS)
 * @module
 */
final class TreePopup extends JPopupMenu {

    private final JContextTree tree;

    TreePopup(final JContextTree tree) {
        this.tree = tree;
    }

    /**
     * will not be set visible if nothing is in the popup
     * 
     * @param view 
     */
    @Override
    public void setVisible(final boolean view) {

        if (view && tree != null) {
            removeAll();

            TreePath[] selection = tree.getRealTree().getSelectionPaths();            
            if(selection == null){
                selection = new TreePath[0];
            }

            for (final TreePopupItem control : tree.controls()) {
                control.setTree(tree);
                if (control.isValid(selection)) {
                    add(control.getComponent(selection));
                }
            }
            removeLastSeparators();

            if (getComponentCount() > 0) {
                super.setVisible(getComponentCount() > 0);
            }
        } else {
            super.setVisible(view);
        }

    }

    private void removeLastSeparators() {
        while (getComponentCount() > 0 &&
            (   getComponent(getComponentCount() - 1) instanceof TitledSeparatorItem
            || getComponent(getComponentCount() - 1) instanceof SeparatorItem)) {
            remove(getComponentCount() - 1);
        }
    }

    /**
     * {@inheritDoc }
     */
    @Override
    public Component add(final Component menuItem) {

        if (getComponentCount() > 0) {
            if( getComponent(getComponentCount() - 1) instanceof SeparatorItem && menuItem instanceof SeparatorItem){
                return menuItem;
            } else if (!(getComponent(getComponentCount() - 1) instanceof TitledSeparatorItem && menuItem instanceof TitledSeparatorItem)) {
                return super.add(menuItem);
            } else {
                remove(getComponentCount() - 1);
                return super.add(menuItem);
            }
        }

        return super.add(menuItem);
    }
}