design/designer-form/src/main/java/com/fr/design/mainframe/FormToolBarPane.java

package com.fr.design.mainframe;

import com.fr.design.beans.BasicBeanPane;
import com.fr.design.dialog.BasicDialog;
import com.fr.design.dialog.DialogActionAdapter;
import com.fr.design.gui.core.WidgetOption;
import com.fr.design.layout.FRGUIPaneFactory;
import com.fr.form.ui.ToolBar;
import com.fr.form.ui.Widget;

import javax.swing.*;
import java.awt.*;
import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.UnsupportedFlavorException;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
import java.util.List;

/**
 * Created by harry on 2017-3-2.
 */
public class FormToolBarPane extends BasicBeanPane<ToolBar> {

    private FormToolBar ftoolbar = new FormToolBar();

    public FormToolBarPane() {
        super();
        this.initComponent();
    }


    /**
     * 添加鼠标监听
     *
     * @param mouselistener 鼠标监听
     */
    public void addAuthorityListener(MouseListener mouselistener) {
        List<FormToolBarButton> list = ftoolbar.getButtonlist();
        for (int i = 0; i < list.size(); i++) {
            list.get(i).addMouseListener(mouselistener);
        }
    }

    public FormToolBarPane(FormToolBarButton button) {
        super();
        this.initComponent();
        this.add(button);
    }

    /**
     * 初始化组件
     */
    public void initComponent() {
        this.addMouseListener(listener);
        this.setLayout(FRGUIPaneFactory.createBoxFlowLayout());
        this.setTransferHandler(new ToolBarHandler(TransferHandler.COPY));
        this.setBorder(BorderFactory.createTitledBorder(""));
    }

    /**
     * 删除鼠标事件
     */
    public void removeDefaultMouseListener() {
        this.removeMouseListener(listener);
    }

    @Override
    protected String title4PopupWindow() {
        return "Toolbar";
    }

    public void setSelectedButton(FormToolBarButton button) {
        this.ftoolbar.addButton(button);
    }

    /**
     * 添加组件
     *
     * @param comp 组件
     *
     * @return 被添加的组件
     */
    public Component add(Component comp) {
        if (comp instanceof FormToolBarButton) {
            this.ftoolbar.addButton((FormToolBarButton) comp);
        }
        return super.add(comp);
    }

    private Component addComp(Component comp) {
        return super.add(comp);
    }


    public void removeButtonList() {
        this.ftoolbar.clearButton();
    }

    protected void setFToolBar(FormToolBar ftoolbar) {
        if (ftoolbar == null) {
            ftoolbar = new FormToolBar();
        }
        this.ftoolbar = ftoolbar;
        this.setToolBar(this.ftoolbar.getButtonlist());
    }

    public List<FormToolBarButton> getToolBarButtons() {
        return ftoolbar.getButtonlist();
    }

    protected FormToolBar getFToolBar() {
        return this.ftoolbar;
    }

    public boolean isEmpty() {
        return this.ftoolbar.getButtonlist().size() <= 0;
    }

    private void setToolBar(List<FormToolBarButton> list) {
        if (list == null) {
            return;
        }
        this.removeAll();
        for (int i = 0; i < list.size(); i++) {
            this.addComp(list.get(i));
        }
        this.validate();
        this.repaint();

    }

    @Override
    public void populateBean(ToolBar toolbar) {
        this.removeAll();
        this.getFToolBar().clearButton();
        for (int j = 0; j < toolbar.getWidgetSize(); j++) {
            Widget widget = toolbar.getWidget(j);
            WidgetOption no = WidgetOption.getToolBarButton(widget.getClass());
            if (no == null){
                //如果装了什么插件, 放到了工具栏上, 后来删除了插件, 模板里还存着之前的控件
                continue;
            }

            FormToolBarButton button = new FormToolBarButton(no.optionIcon(), widget);
            button.setNameOption(no);
            this.add(button);
            this.validate();
            this.repaint();
        }
        this.getFToolBar().setBackground(toolbar.getBackground());
        this.getFToolBar().setDefault(toolbar.isDefault());
    }

    @Override
    public ToolBar updateBean() {
        return this.ftoolbar.getToolBar();
    }

    MouseListener listener = new MouseAdapter() {
        @Override
        public void mouseClicked(MouseEvent e) {
            if (e.getClickCount() >= 2 && !SwingUtilities.isRightMouseButton(e)) {
                final FormEditToolBar tb = new FormEditToolBar();
                tb.populate(getFToolBar());
                BasicDialog dialog = tb.showWindow(SwingUtilities.getWindowAncestor(FormToolBarPane.this));
                dialog.addDialogActionListener(new DialogActionAdapter() {
                    @Override
                    public void doOk() {
                        FormToolBarPane.this.setFToolBar(tb.update());
                    }
                });
                dialog.setVisible(true);
            }
        }
    };


    /*
     * 拖拽属性设置
     */
    private class ToolBarHandler extends TransferHandler {
        private int action;

        public ToolBarHandler(int action) {
            this.action = action;
        }


        @Override
        public boolean canImport(TransferHandler.TransferSupport support) {
            if (!support.isDrop()) {
                return false;
            }

            if (!support.isDataFlavorSupported(DataFlavor.stringFlavor)) {
                return false;
            }

            boolean actionSupported = (action & support.getSourceDropActions()) == action;
            if (actionSupported) {
                support.setDropAction(action);
                return true;
            }

            return false;
        }

        @Override
        public boolean importData(TransferHandler.TransferSupport support) {
            if (!canImport(support)) {
                return false;
            }
            WidgetOption data;
            try {
                data = (WidgetOption) support.getTransferable().getTransferData(DataFlavor.stringFlavor);
            } catch (UnsupportedFlavorException e) {
                return false;
            } catch (java.io.IOException e) {
                return false;
            }

            Widget widget = data.createWidget();
            FormToolBarButton btn = new FormToolBarButton(data.optionIcon(), widget);
            btn.setNameOption(data);
            FormToolBarPane.this.add(btn);
            FormToolBarPane.this.validate();
            FormToolBarPane.this.repaint();
            return true;
        }

    }
}