enigma-swing/src/main/java/cuchaz/enigma/gui/elements/DeobfPanelPopupMenu.java


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

package cuchaz.enigma.gui.elements;

import javax.swing.JMenuItem;
import javax.swing.JPopupMenu;
import javax.swing.tree.TreePath;

import cuchaz.enigma.gui.ClassSelector;
import cuchaz.enigma.gui.panels.DeobfPanel;
import cuchaz.enigma.utils.I18n;

public class DeobfPanelPopupMenu {

    private final JPopupMenu ui;
    private final JMenuItem renamePackage = new JMenuItem();
    private final JMenuItem renameClass = new JMenuItem();
    private final JMenuItem expandAll = new JMenuItem();
    private final JMenuItem collapseAll = new JMenuItem();

    public DeobfPanelPopupMenu(DeobfPanel panel) {
        this.ui = new JPopupMenu();

        this.ui.add(this.renamePackage);
        this.ui.add(this.renameClass);
        this.ui.addSeparator();
        this.ui.add(this.expandAll);
        this.ui.add(this.collapseAll);

        ClassSelector deobfClasses = panel.deobfClasses;

        this.renamePackage.addActionListener(a -> {
            TreePath path;

            if (deobfClasses.getSelectedClass() != null) {
                // Rename parent package if selected path is a class
                path = deobfClasses.getSelectionPath().getParentPath();
            } else {
                // Rename selected path if it's already a package
                path = deobfClasses.getSelectionPath();
            }

            deobfClasses.getUI().startEditingAtPath(deobfClasses, path);
        });
        this.renameClass.addActionListener(a -> deobfClasses.getUI().startEditingAtPath(deobfClasses, deobfClasses.getSelectionPath()));
        this.expandAll.addActionListener(a -> deobfClasses.expandAll());
        this.collapseAll.addActionListener(a -> deobfClasses.collapseAll());

        this.retranslateUi();
    }

    public void show(ClassSelector deobfClasses, int x, int y) {
        // Only enable rename class if selected path is a class
        this.renameClass.setEnabled(deobfClasses.getSelectedClass() != null);

        this.ui.show(deobfClasses, x, y);
    }

    public void retranslateUi() {
        this.renamePackage.setText(I18n.translate("popup_menu.deobf_panel.rename_package"));
        this.renameClass.setText(I18n.translate("popup_menu.deobf_panel.rename_class"));
        this.expandAll.setText(I18n.translate("popup_menu.deobf_panel.expand_all"));
        this.collapseAll.setText(I18n.translate("popup_menu.deobf_panel.collapse_all"));
    }
}