/** This plugin implements the Plugins/Utilities/Find Commands 
    command. It provides an easy user interface to finding commands 
    you might know the name of without having to go through
    all the menus.  If you type a part of a command name, the box
    below will only show commands that match that substring (case
    insensitively).  If only a single command matches then that
    command can be run by hitting Enter.  If multiple commands match,
    they can be selected by selecting with the mouse and clicking
    "Run"; alternatively hitting the up or down arrows will move the
    keyboard focus to the list and the selected command can be run
    with Enter.  Double-clicking on a command in the list should
    also run the appropriate command.

    @author Mark Longair <mark-imagej@longair.net>
 */

package ij.plugin;

import ij.*;

import java.awt.*;
import java.awt.event.*;
import java.util.Hashtable;
import java.util.Arrays;
import java.util.Iterator;
import java.util.ArrayList;
import java.util.Set;

public class CommandFinder implements PlugIn, TextListener, ActionListener, WindowListener, KeyListener, ItemListener, MouseListener {

    public CommandFinder() {
        Toolkit toolkit=Toolkit.getDefaultToolkit();
        Integer interval=(Integer)toolkit.getDesktopProperty("awt.multiClickInterval");
        if (interval==null)
            // Hopefully 300ms is a sensible default when the property
            // is not available.
            multiClickInterval=300;
        else
            multiClickInterval=interval.intValue();
    }

    class CommandAction {
        CommandAction(String classCommand, MenuItem menuItem, String menuLocation) {
            this.classCommand = classCommand;
            this.menuItem = menuItem;
            this.menuLocation = menuLocation;
        }
        String classCommand;
        MenuItem menuItem;
        String menuLocation;
        public String toString() {
            return "classCommand: " + classCommand + ", menuItem: "+menuItem+", menuLocation: "+menuLocation;
        }
    }

    int multiClickInterval;
    long lastClickTime=Long.MIN_VALUE;
    String lastClickedItem;
    Dialog d;
    TextField prompt;
    List completions;
    Button runButton;
    Button closeButton;
    Checkbox fullInfoCheckbox, closeCheckbox;
    Hashtable commandsHash;
    String [] commands;
    Hashtable listLabelToCommand;

    protected String makeListLabel(String command, CommandAction ca, boolean fullInfo) {
        if (fullInfo) {
            String result = command;
            if( ca.menuLocation != null)
                result += " (in " + ca.menuLocation + ")";
            if( ca.classCommand != null )
                result += " [" + ca.classCommand + "]";
            return result;
        } else {
            return command;
        }
    }

    protected void populateList(String matchingSubstring) {
        boolean fullInfo=fullInfoCheckbox.getState();
        String substring = matchingSubstring.toLowerCase();
        completions.removeAll();
        for(int i=0; i<commands.length; ++i) {
            String commandName = commands[i];
            if (commandName.length()==0)
                continue;
            String lowerCommandName = commandName.toLowerCase();
            if( lowerCommandName.indexOf(substring) >= 0 ) {
                CommandAction ca = (CommandAction)commandsHash.get(commandName);
                String listLabel = makeListLabel(commandName, ca, fullInfo);
                completions.add(listLabel);
            }
        }
    }

    public void actionPerformed(ActionEvent ae) {
        Object source = ae.getSource();
        if (source==runButton) {
            String selected = completions.getSelectedItem();
            if(selected==null) {
                IJ.error("You must select a plugin to run");
                return;
            }
            runFromLabel(selected);
        } else if (source == closeButton) {
            d.dispose();
        }
    }

    public void itemStateChanged(ItemEvent ie) {
        populateList(prompt.getText());
    }

    public void mouseClicked(MouseEvent e) {
        long now=System.currentTimeMillis();
        String justClickedItem=completions.getSelectedItem();
        // Is this fast enough to be a double-click?
        long thisClickInterval=now-lastClickTime;
        if (thisClickInterval<multiClickInterval) {
            if (justClickedItem!=null&&
                lastClickedItem!=null&&
                justClickedItem.equals(lastClickedItem)) {
                runFromLabel(justClickedItem);
            }
        }
        lastClickTime=now;
        lastClickedItem=justClickedItem;
    }

    public void mousePressed(MouseEvent e) {}
    public void mouseReleased(MouseEvent e) {}
    public void mouseEntered(MouseEvent e) {}
    public void mouseExited(MouseEvent e) {}

    protected void runFromLabel(String listLabel) {
        String command = (String)listLabelToCommand.get(listLabel);
        CommandAction ca = (CommandAction)commandsHash.get(command);
        if (ca.classCommand != null ) {
            IJ.showStatus("Running command "+ca.classCommand);
            IJ.doCommand(command);
        } else if (ca.menuItem != null) {
            IJ.showStatus("Clicking menu item "+ca.menuLocation+" > "+command);
            ActionEvent ae = new ActionEvent(ca.menuItem, ActionEvent.ACTION_PERFORMED, command);
            ActionListener [] als = ca.menuItem.getActionListeners();
            for (int i=0; i<als.length; ++i)
                als[i].actionPerformed(ae);
        } else {
            IJ.error("BUG: nothing to run found for '"+listLabel+"'");
            return;
        }
        if (closeCheckbox.getState())
            d.dispose();
    }

    public void keyPressed(KeyEvent ke) {
        int key = ke.getKeyCode();
        int items = completions.getItemCount();
        Object source = ke.getSource();
        if (key==KeyEvent.VK_ESCAPE) {
            d.dispose();
        } else if (source==prompt) {
            /* If you hit enter in the text field, and
               there's only one command that matches, run
               that: */
            if (key==KeyEvent.VK_ENTER) {
                if (1==items) {
                    String selected = completions.getItem(0);
                    runFromLabel(selected);
                }
            /* If you hit the up or down arrows in the
               text field, move the focus to the
               completions list and select the item at the
               bottom or top of that list. */
            } else if (key==KeyEvent.VK_UP) {
                completions.requestFocus();
                if(items>0)
                    completions.select(completions.getItemCount()-1);
            } else if (key==KeyEvent.VK_DOWN)  {
                completions.requestFocus();
                if (items>0)
                    completions.select(0);
            }
        } else if (key==KeyEvent.VK_BACK_SPACE) {
            /* If someone presses backspace they probably want to
               remove the last letter from the search string, so
               switch the focus back to the prompt: */
            prompt.requestFocus();
        } else if (source==completions) {
            /* If you hit enter with the focus in the
               completions list, run the selected
               command */
            if (key==KeyEvent.VK_ENTER) {
                String selected = completions.getSelectedItem();
                if (selected!=null)
                    runFromLabel(selected);
            }
        } else if (source==runButton) {
            if (key==KeyEvent.VK_ENTER) {
                String selected = completions.getSelectedItem();
                if (selected!=null)
                    runFromLabel(selected);
            }
        } else if (source==closeButton) {
            if (key==KeyEvent.VK_ENTER)
                d.dispose();
        }
    }

    public void keyReleased(KeyEvent ke) { }

    public void keyTyped(KeyEvent ke) { }

    public void textValueChanged(TextEvent te) {
        populateList(prompt.getText());
    }

    /* This function recurses down through a menu, adding to
       commandsHash the location and MenuItem of any items it
       finds that aren't submenus. */

    public void parseMenu(String path, Menu menu) {
        int n=menu.getItemCount();
        for (int i=0; i<n; ++i) {
            MenuItem m=menu.getItem(i);
            String label=m.getLabel();
            if (m instanceof Menu) {
                Menu subMenu=(Menu)m;
                parseMenu(path+" > "+label,subMenu);
            } else {
                String trimmedLabel = label.trim();
                if (trimmedLabel.length()==0 || trimmedLabel.equals("-"))
                    continue;
                CommandAction ca=(CommandAction)commandsHash.get(label);
                if( ca == null )
                    commandsHash.put(label, new CommandAction(null,m,path));
                else {
                    ca.menuItem=m;
                    ca.menuLocation=path;
                }
                CommandAction caAfter=(CommandAction)commandsHash.get(label);
            }
        }
    }

    /* Finds all the top level menus from the menu bar and
       recurses down through each. */

    public void findAllMenuItems() {
        MenuBar menuBar = Menus.getMenuBar();
        int topLevelMenus = menuBar.getMenuCount();
        for (int i=0; i<topLevelMenus; ++i) {
            Menu topLevelMenu=menuBar.getMenu(i);
            parseMenu(topLevelMenu.getLabel(), topLevelMenu);
        }
    }

    public void run(String ignored) {

        commandsHash = new Hashtable();

        /* Find the "normal" commands; those which are
           registered plugins: */

        Hashtable realCommandsHash = ij.Menus.getCommands();

        Set realCommandSet = realCommandsHash.keySet();

        for (Iterator i = realCommandSet.iterator();
             i.hasNext();) {
            String command = (String)i.next();
            // Some of these are whitespace only or separators - ignore them:
            String trimmedCommand = command.trim();
            if (trimmedCommand.length()>0 && !trimmedCommand.equals("-")) {
                commandsHash.put(command,
                         new CommandAction((String)realCommandsHash.get(command),
                                   null,
                                   null));
            }
        }

        /* There are some menu items that don't correspond to
           plugins, such as those added by RefreshScripts, so
           look through all the menus as well: */

        findAllMenuItems();

        /* Sort the commands, generate list labels for each
           and put them into a hash: */

        commands = (String[])commandsHash.keySet().toArray(new String[0]);
        Arrays.sort(commands);

        listLabelToCommand = new Hashtable();

        for (int i=0; i<commands.length; ++i) {
            CommandAction ca = (CommandAction)commandsHash.get(commands[i]);
            listLabelToCommand.put(makeListLabel(commands[i], ca, true), commands[i]);
            listLabelToCommand.put(makeListLabel(commands[i], ca, false), commands[i]);
        }

        /* The code below just constructs the dialog: */

        ImageJ imageJ = IJ.getInstance();

        d = new Dialog(imageJ, "Command Finder");
        d.setLayout(new BorderLayout());
        d.addWindowListener(this);

        fullInfoCheckbox = new Checkbox("Show full information", false);
        fullInfoCheckbox.addItemListener(this);
        closeCheckbox = new Checkbox("Close when running", true);
        closeCheckbox.addItemListener(this);

        Panel northPanel = new Panel();

        northPanel.add(new Label("Type part of a command:"));

        prompt = new TextField("", 30);
        prompt.addTextListener(this);
        prompt.addKeyListener(this);

        northPanel.add(prompt);

        d.add(northPanel, BorderLayout.NORTH);

        completions = new List(20);
        completions.addKeyListener(this);
        populateList("");

        d.add(completions, BorderLayout.CENTER);
        // Add a mouse listener so we can detect double-clicks
        completions.addMouseListener(this);

        runButton = new Button("Run");
        closeButton = new Button("Close");

        runButton.addActionListener(this);
        closeButton.addActionListener(this);
        runButton.addKeyListener(this);
        closeButton.addKeyListener(this);

        Panel southPanel = new Panel();
        southPanel.setLayout(new BorderLayout());

        Panel optionsPanel = new Panel();
        optionsPanel.add(fullInfoCheckbox);
        optionsPanel.add(closeCheckbox);

        Panel buttonsPanel = new Panel();
        buttonsPanel.add(runButton);
        buttonsPanel.add(closeButton);

        southPanel.add(optionsPanel, BorderLayout.CENTER);
        southPanel.add(buttonsPanel, BorderLayout.SOUTH);

        d.add(southPanel, BorderLayout.SOUTH);

        Dimension screenSize = Toolkit.getDefaultToolkit().getScreenSize();

        d.pack();

        int dialogWidth = d.getWidth();
        int dialogHeight = d.getHeight();

        int screenWidth = (int)screenSize.getWidth();
        int screenHeight = (int)screenSize.getHeight();

        Point pos=imageJ.getLocationOnScreen();

        /* Generally try to position the dialog slightly
           offset from the main ImageJ window, but if that
           would push the dialog off to the screen to any
           side, adjust it so that it's on the screen.
        */
        int initialX = (int)pos.getX() + 38;
        int initialY = (int)pos.getY() + 84;

        if (initialX+dialogWidth>screenWidth)
            initialX = screenWidth-dialogWidth;
        if (initialX<0)
            initialX = 0;
        if (initialY+dialogHeight>screenHeight)
            initialY = screenHeight-dialogHeight;
        if (initialY<0)
            initialY = 0;

        d.setLocation(initialX,initialY);

        d.setVisible(true);
        d.toFront();
    }

    /* Make sure that clicks on the close icon close the window: */

    public void windowClosing(WindowEvent e) {
        d.dispose();
    }

    public void windowActivated(WindowEvent e) { }
    public void windowDeactivated(WindowEvent e) { }
    public void windowClosed(WindowEvent e) { }
    public void windowOpened(WindowEvent e) { }
    public void windowIconified(WindowEvent e) { }
    public void windowDeiconified(WindowEvent e) { }
}