summaryrefslogtreecommitdiff
path: root/enigma-swing/src/main/java/cuchaz/enigma/gui/EnigmaQuickFindDialog.java
blob: c912be3a0c145c061aafa2c649cdb8483765b756 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
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();

		try {
			super.showFor(target);
		} catch (Exception e) {
			e.printStackTrace();
			return;
		}

		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);
	}
}