//@category Experimental


import java.awt.AWTEvent;
import java.awt.Toolkit;
import java.awt.event.AWTEventListener;
import java.awt.event.MouseEvent;
import java.util.Set;

import docking.DefaultActionContext;
import ghidra.app.script.GhidraScript;
import docking.action.DockingActionIf;
import ghidra.framework.plugintool.PluginTool;
import ghidra.util.Swing;

public class MouseNavigationScript extends GhidraScript {

    private static final String ACTION_OWNER = "NextPrevAddressPlugin";
    private static AWTEventListener LISTENER;

    @Override
    protected void run() throws Exception {
        if (isRunningHeadless()) {
            return;
        }

        createListener();

        Toolkit tk = Toolkit.getDefaultToolkit();

        // don't repeatedly add the listener
        tk.removeAWTEventListener(LISTENER);
        tk.addAWTEventListener(LISTENER, AWTEvent.MOUSE_EVENT_MASK);

    }

    private void createListener() {

        if (LISTENER != null) {
            return;
        }

        LISTENER = e -> {
            MouseEvent mousey = (MouseEvent) e;

            int id = mousey.getID();
            int button = mousey.getButton();

            if (id != MouseEvent.MOUSE_PRESSED) {
                return;
            }
            // 可以在这里打印看看自己鼠标对应的按键
            if (button == 4) {
                mousey.consume();
                DockingActionIf back = getNavigationAction("Previous Function in History");
                if (back != null) {
                    Swing.runLater(() -> {
                        back.actionPerformed(new DefaultActionContext());
                    });
                }
            } else if (button == 5) {
                mousey.consume();
                DockingActionIf forward = getNavigationAction("Next Function in History");
                if (forward != null) {
                    Swing.runLater(() -> {
                        forward.actionPerformed(new DefaultActionContext());
                    });
                }
            }
        };
    }

    private DockingActionIf getNavigationAction(String name) {
        PluginTool tool = state.getTool();
        Set<DockingActionIf> actions = tool.getDockingActionsByOwnerName(ACTION_OWNER);
        for (DockingActionIf action : actions) {
            if (action.getName().equals(name)) {
                return action;
            }
        }
        return null;
    }
}