openide/source/com/intellij/codeInsight/PopupActionChooser.java

package com.intellij.codeInsight;

import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.command.CommandProcessor;
import com.intellij.openapi.editor.Editor;
import com.intellij.openapi.editor.LogicalPosition;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.roots.ui.util.CellAppearanceUtils;
import com.intellij.ui.ColoredListCellRenderer;
import com.intellij.ui.ListPopup;
import com.intellij.ui.SimpleTextAttributes;

import javax.swing.*;
import java.awt.*;
import java.awt.event.KeyAdapter;
import java.awt.event.KeyEvent;
import java.util.ArrayList;

public class PopupActionChooser {
  private final String myTitle;
  private final String[] myActionNames;
  private final Object[] myActionObjects;
  private boolean myShowNumbers;

  public static interface Callback {
    void execute(Object actionObject);
    boolean isApplicable(Object actionObject);
  }

  public PopupActionChooser(String title, String[] actionNames, Object[] actionObjects) {
    myTitle = title;
    myActionNames = actionNames;
    myActionObjects = actionObjects;
  }

  public void setShowNumbers(boolean showNumbers) {
    myShowNumbers = showNumbers;
  }

  public void invoke(final Project project, Editor editor, final Callback callback){
    final ArrayList actionObjects = new ArrayList();
    final ArrayList actionNames = new ArrayList();
    for(int i = 0; i < myActionObjects.length; i++){
      Object actionObject = myActionObjects[i];
      if (callback.isApplicable(actionObject)){
        actionObjects.add(actionObject);
        actionNames.add(myActionNames[i]);
      }
    }
    if (actionObjects.size() == 0) return;

    final JList list = new JList(actionNames.toArray());

    list.setCellRenderer(new MyListCellRenderer());

    final Runnable listener = new Runnable(){
      public void run() {
        int index = list.getSelectedIndex();
        if (index >= 0){
          final Object actionObject = actionObjects.get(index);
          CommandProcessor.getInstance().executeCommand(
              project, new Runnable(){
              public void run(){
                final Runnable action = new Runnable(){
                  public void run(){
                    callback.execute(actionObject);
                  }
                };
                ApplicationManager.getApplication().runWriteAction(action);
              }
            },
            null,
            null
          );
        }
      }
    };

    final ListPopup listPopup = new ListPopup(myTitle, list, listener, project);

    if (myShowNumbers){
      list.addKeyListener(
        new KeyAdapter() {
          public void keyPressed(KeyEvent e) {
            char c = e.getKeyChar();
            int index = -1;
            if ('0' <= c && c <= '9'){
              index = c == '0' ? 9 : c - '1';
            }
            else if ('A' <= c && c <= 'Z'){
              index = c - 'A' + 10;
            }
            else if ('a' <= c && c <= 'z'){
              index = c - 'a' + 10;
            }
            if (index >= 0 && index < list.getModel().getSize()) {
              list.setSelectedIndex(index);
              listener.run();
              listPopup.closePopup(false);
            }
          }
        }
      );
    }

    LogicalPosition caretLogicalPosition = editor.getCaretModel().getLogicalPosition();
    Point caretLocation = editor.logicalPositionToXY(new LogicalPosition(caretLogicalPosition.line + 1, caretLogicalPosition.column));
    Point location = editor.getContentComponent().getLocationOnScreen();
    listPopup.show(caretLocation.x + location.x, caretLocation.y + location.y); //TODO : check screen bounds!
  }

  private final class MyListCellRenderer extends ColoredListCellRenderer{
    private final SimpleTextAttributes myAttributes;

    public MyListCellRenderer(){
      myAttributes=SimpleTextAttributes.SIMPLE_CELL_ATTRIBUTES;
    }

    protected void customizeCellRenderer(
      JList list,
      Object value,
      int index,
      boolean selected,
      boolean hasFocus
    ){
      String name = (String)value;
      if (myShowNumbers){
        if (index < 9){
          name = (index + 1) + ". " + name;
        }
        else if (index == 9){
          name = 0 + ". " + name;
        }
        else {
          name = (char)('A' + index - 10) + ". " + name;
        }
      }
      append(name,myAttributes);
    }
  }
}