search

Author: Enaium

src/main/java/cn/enaium/joe/JavaOctetEditor.java

@@ -58,8 +58,8 @@ public void run() {
 
         window.setJMenuBar(new JMenuBar() {{
             add(new FileMenu());
-            add(new HelpMenu());
             add(new SearchMenu());
+            add(new HelpMenu());
         }});
 
         window.setContentPane(new JPanel(new BorderLayout()) {

src/main/java/cn/enaium/joe/dialog/search/SearchMethodDialog.java

@@ -0,0 +1,85 @@
+/*
+ * Copyright 2022 Enaium
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package cn.enaium.joe.dialog.search;
+
+import cn.enaium.joe.JavaOctetEditor;
+import cn.enaium.joe.dialog.SearchDialog;
+import cn.enaium.joe.gui.panel.search.ResultNode;
+import cn.enaium.joe.jar.Jar;
+import cn.enaium.joe.util.ASyncUtil;
+import cn.enaium.joe.util.StringUtil;
+import org.objectweb.asm.tree.*;
+
+import javax.swing.*;
+import java.awt.*;
+import java.util.Map;
+
+/**
+ * @author Enaium
+ * @since 0.5.0
+ */
+public class SearchMethodDialog extends SearchDialog {
+    public SearchMethodDialog() {
+        setTitle("Search Method");
+        add(new JPanel(new FlowLayout()) {{
+            add(new JLabel("Owner:"));
+            JTextField owner = new JTextField();
+            add(owner);
+            add(new JLabel("Name:"));
+            JTextField name = new JTextField();
+            add(name);
+            add(new JLabel("Description:"));
+            JTextField description = new JTextField();
+            add(description);
+            JCheckBox anInterface = new JCheckBox("Interface");
+            add(anInterface);
+            add(new JButton("Search") {{
+                addActionListener(e -> {
+                    ASyncUtil.execute(() -> {
+                        float loaded = 0;
+                        float total = 0;
+                        Jar jar = JavaOctetEditor.getInstance().jar;
+                        for (Map.Entry<String, ClassNode> stringClassNodeEntry : jar.classes.entrySet()) {
+                            for (MethodNode method : stringClassNodeEntry.getValue().methods) {
+                                total += method.instructions.size();
+                            }
+                        }
+
+                        for (Map.Entry<String, ClassNode> stringClassNodeEntry : jar.classes.entrySet()) {
+                            for (MethodNode method : stringClassNodeEntry.getValue().methods) {
+                                for (AbstractInsnNode instruction : method.instructions) {
+                                    if (instruction instanceof MethodInsnNode) {
+                                        MethodInsnNode methodInsnNode = (MethodInsnNode) instruction;
+                                        if ((methodInsnNode.owner.contains(owner.getText()) || StringUtil.isBlank(owner.getText())) &&
+                                                (methodInsnNode.name.contains(name.getText()) || StringUtil.isBlank(name.getText())) &&
+                                                (methodInsnNode.desc.contains(description.getText()) || StringUtil.isBlank(description.getText())) &&
+                                                (methodInsnNode.itf && anInterface.isSelected() || !anInterface.isSelected())
+                                        ) {
+                                            ((DefaultListModel<ResultNode>) resultPanel.getList().getModel()).addElement(new ResultNode(stringClassNodeEntry.getValue(), methodInsnNode.name + "#" + methodInsnNode.desc));
+                                        }
+                                    }
+                                    JavaOctetEditor.getInstance().bottomPanel.setProcess((int) ((loaded++ / total) * 100f));
+                                }
+                            }
+                        }
+                        JavaOctetEditor.getInstance().bottomPanel.setProcess(0);
+                    });
+                });
+            }});
+        }}, BorderLayout.SOUTH);
+    }
+}

src/main/java/cn/enaium/joe/gui/panel/menu/SearchMenu.java

@@ -18,6 +18,7 @@
 
 import cn.enaium.joe.gui.panel.menu.search.FieldMenuItem;
 import cn.enaium.joe.gui.panel.menu.search.LdcMenuItem;
+import cn.enaium.joe.gui.panel.menu.search.MethodMenuItem;
 
 import javax.swing.*;
 
@@ -29,5 +30,6 @@ public SearchMenu() {
         super("Search");
         add(new LdcMenuItem());
         add(new FieldMenuItem());
+        add(new MethodMenuItem());
     }
 }

src/main/java/cn/enaium/joe/gui/panel/menu/search/MethodMenuItem.java

@@ -0,0 +1,41 @@
+/*
+ * Copyright 2022 Enaium
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package cn.enaium.joe.gui.panel.menu.search;
+
+import cn.enaium.joe.JavaOctetEditor;
+import cn.enaium.joe.dialog.search.SearchFieldDialog;
+import cn.enaium.joe.dialog.search.SearchMethodDialog;
+import cn.enaium.joe.jar.Jar;
+
+import javax.swing.*;
+
+/**
+ * @author Enaium
+ * @since 0.5.0
+ */
+public class MethodMenuItem extends JMenuItem {
+    public MethodMenuItem() {
+        super("Method");
+        addActionListener(e -> {
+            Jar jar = JavaOctetEditor.getInstance().jar;
+            if (jar == null) {
+                return;
+            }
+            new SearchMethodDialog().setVisible(true);
+        });
+    }
+}