Quick find fixes (#133)

* Correctly offset quick find dialog, select all text on Ctrl+F, and set text to highlighted on open

* Support quick find navigation with enter & shift+enter

1 files changed, 85 insertions, 0 deletions

diff --git a/src/main/java/cuchaz/enigma/gui/EnigmaQuickFindDialog.java b/src/main/java/cuchaz/enigma/gui/EnigmaQuickFindDialog.java
new file mode 100644
index 0000000..0c1d4cd
--- /dev/null
+++ b/src/main/java/cuchaz/enigma/gui/EnigmaQuickFindDialog.java
@@ -0,0 +1,85 @@
+package cuchaz.enigma.gui;
+
+import de.sciss.syntaxpane.actions.DocumentSearchData;
+import de.sciss.syntaxpane.actions.gui.QuickFindDialog;
+
+import javax.swing.*;
+import javax.swing.text.JTextComponent;
+import java.awt.*;
+import java.awt.event.KeyAdapter;
+import java.awt.event.KeyEvent;
+import java.util.stream.IntStream;
+import java.util.stream.Stream;
+
+public class EnigmaQuickFindDialog extends QuickFindDialog {
+        public EnigmaQuickFindDialog(JTextComponent target) {
+                super(target, DocumentSearchData.getFromEditor(target));
+
+                JToolBar toolBar = getToolBar();
+                JTextField textField = getTextField(toolBar);
+
+                textField.addKeyListener(new KeyAdapter() {
+                        @Override
+                        public void keyPressed(KeyEvent e) {
+                                super.keyPressed(e);
+                                if (e.getKeyCode() == KeyEvent.VK_ENTER) {
+                                        JToolBar toolBar = getToolBar();
+                                        boolean next = !e.isShiftDown();
+                                        JButton button = next ? getNextButton(toolBar) : getPrevButton(toolBar);
+                                        button.doClick();
+                                }
+                        }
+                });
+        }
+
+        @Override
+        public void showFor(JTextComponent target) {
+                String selectedText = target.getSelectedText();
+
+                super.showFor(target);
+
+                Container view = target.getParent();
+                Point loc = new Point(0, view.getHeight() - getSize().height);
+                setLocationRelativeTo(view);
+                SwingUtilities.convertPointToScreen(loc, view);
+                setLocation(loc);
+
+                JToolBar toolBar = getToolBar();
+                JTextField textField = getTextField(toolBar);
+
+                if (selectedText != null) {
+                        textField.setText(selectedText);
+                }
+
+                textField.selectAll();
+        }
+
+        private JToolBar getToolBar() {
+                return components(getContentPane(), JToolBar.class).findFirst().orElse(null);
+        }
+
+        private JTextField getTextField(JToolBar toolBar) {
+                return components(toolBar, JTextField.class).findFirst().orElse(null);
+        }
+
+        private JButton getNextButton(JToolBar toolBar) {
+                Stream<JButton> buttons = components(toolBar, JButton.class);
+                return buttons.skip(1).findFirst().orElse(null);
+        }
+
+        private JButton getPrevButton(JToolBar toolBar) {
+                Stream<JButton> buttons = components(toolBar, JButton.class);
+                return buttons.findFirst().orElse(null);
+        }
+
+        private static Stream<Component> components(Container container) {
+                return IntStream.range(0, container.getComponentCount())
+                                .mapToObj(container::getComponent);
+        }
+
+        private static <T extends Component> Stream<T> components(Container container, Class<T> type) {
+                return components(container)
+                                .filter(type::isInstance)
+                                .map(type::cast);
+        }
+}