AssetMapMenuPanel.java

package net.squiz.matrix.assetmap;

import net.squiz.matrix.matrixtree.*;
import net.squiz.matrix.debug.*;
import net.squiz.matrix.core.*;
import net.squiz.matrix.ui.*;
import javax.swing.*;
import javax.swing.tree.*;
import java.awt.event.*;
import javax.swing.event.*;
import java.awt.*;
import java.io.IOException;
import java.util.*;


public class AssetMapMenuPanel extends JPanel {

    private JMenuBar menuBar;
    private MatrixTree tree;
    private JToolBar toolBar;

    public static final Color BG_COLOUR = new Color(0xF5F5F5);
    public static final int ICON_GAP = 1;

    public AssetMapMenuPanel(MatrixTree tree, boolean includeCreateButton) {
        this.tree = tree;

        setLayout(new BorderLayout());

        JPanel leftPanel  = new JPanel(new FlowLayout(FlowLayout.LEADING));
        JPanel rightPanel = new JPanel(new FlowLayout(FlowLayout.TRAILING));

        rightPanel.add(createRefreshAssetsButton());
        rightPanel.add(createRestoreRootButton());
        rightPanel.add(createCollapseAllButton());
        rightPanel.add(createPaintStatusesButton());

        if (includeCreateButton) {
            leftPanel.add(createAddMenuButton());
            add(leftPanel, BorderLayout.WEST);
        }

        add(rightPanel, BorderLayout.EAST);

        leftPanel.setBackground(BG_COLOUR);
        rightPanel.setBackground(BG_COLOUR);
        setBackground(BG_COLOUR);
    }

    public Dimension getPreferredSize() {
        return new Dimension(300, 25);
    }

    public Dimension getMinimumSize() {
        return new Dimension(300, 25);
    }

    public Dimension getMaximumSize() {
        return new Dimension(300, 25);
    }

    private JButton createButton(
            String iconName,
            ActionListener listener,
            String toolTipText) {

        Icon icon = GUIUtilities.getAssetMapIcon(iconName + "_off.png");
        Icon pressedIcon = GUIUtilities.getAssetMapIcon(iconName + "_on.png");

        JButton button = new JButton(icon);
        button.setBackground(BG_COLOUR);
        button.setBorderPainted(false);

        button.setPreferredSize(new Dimension(icon.getIconWidth(), icon.getIconHeight()));

        button.setPressedIcon(pressedIcon);
        button.addActionListener(listener);
        button.setToolTipText(toolTipText);

        return button;
    }

    private JButton createRefreshAssetsButton() {

        ActionListener refreshListener = new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                AssetRefreshWorker worker = new AssetRefreshWorker(true);
                worker.start();
            }
        };
        JButton refreshButton
            = createButton("refresh", refreshListener, Matrix.translate("asset_map_tooltip_refresh_all"));

        return refreshButton;
    }

    private JButton createRestoreRootButton() {

        ActionListener restoreListener = new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                String newRoot = Matrix.getProperty("parameter.rootlineage");
                // user can switch between the two root nodes
                if ((newRoot.length() > 0) && (tree.getModel().getRoot() == AssetManager.getRootFolderNode())) {
                    // root folder is not the actual root
                    String[] info = newRoot.split("~");
                    String[] assetIds = info[0].split("\\|");
                    String[] sort_orders = info[1].split("\\|");
                    tree.collapsePath(tree.getPathToRoot((MatrixTreeNode)tree.getModel().getRoot()));
                    tree.loadChildAssets(assetIds, sort_orders, false, true);
                } else {
                    tree.setRootVisible(false);
                    ((DefaultTreeModel) tree.getModel()).setRoot(AssetManager.getRootFolderNode());
                }
            }
        };
        JButton restoreButton
            = createButton("teleport", restoreListener, Matrix.translate("asset_map_tooltip_restore_root"));

        return restoreButton;
    }

    private JButton createCollapseAllButton() {

        ActionListener collapseListener = new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                ((DefaultTreeModel) tree.getModel()).reload();
                tree.repaint();
            }
        };
        JButton collapseButton
            = createButton("collapse", collapseListener, Matrix.translate("asset_map_tooltip_collapse_all"));

        return collapseButton;
    }

    private JButton createPaintStatusesButton() {

        ActionListener statusListener = new ActionListener() {
            public void actionPerformed(ActionEvent evt) {
                ((MatrixTreeCellRenderer) tree.getCellRenderer()).flipSelection();
                tree.repaint();
            }
        };
        JButton paintStatusButton
            = createButton("status", statusListener, Matrix.translate("asset_map_tooltip_toggle_status"));

        return paintStatusButton;
    }

    private ButtonMenu createAddMenuButton() {

        Icon icon = GUIUtilities.getAssetMapIcon("add_off.png");
        Icon pressedIcon = GUIUtilities.getAssetMapIcon("add_on.png");

        final ButtonMenu button = new ButtonMenu(icon, pressedIcon);

        // we need to do this because the asset map may not have made a request
        // to matrix yet, so the add menu elements might not yet be known
        ActionListener bListener = new ActionListener() {
            private JPopupMenu addMenu;

            public void actionPerformed(ActionEvent evt) {
                if (addMenu == null) {
                    ActionListener listener = MatrixMenus.getMatrixTreeAddMenuListener(tree);
                    addMenu = MatrixMenus.getPopupAddMenu(listener);
                    button.setPopupMenu(addMenu);
                }
            }
        };

        button.addActionListener(bListener);
        button.setPressedIcon(pressedIcon);
        button.setBackground(BG_COLOUR);
        button.setBorderPainted(false);

        button.setPreferredSize(new Dimension(icon.getIconWidth(), icon.getIconHeight()));
        button.setToolTipText(Matrix.translate("asset_map_tooltip_add_asset"));

        return button;
    }
}