search

Author: Enaium

src/main/java/cn/enaium/joe/dialog/SearchDialog.java

@@ -29,7 +29,7 @@ public class SearchDialog extends Dialog {
     public SearchDialog() {
         super("Search");
         setLayout(new BorderLayout());
-        setSize(400, 300);
+        setSize(700, 400);
         add(resultPanel, BorderLayout.CENTER);
     }
 }

src/main/java/cn/enaium/joe/dialog/search/SearchFieldDialog.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.FieldInsnNode;
+import org.objectweb.asm.tree.AbstractInsnNode;
+import org.objectweb.asm.tree.ClassNode;
+import org.objectweb.asm.tree.MethodNode;
+
+import javax.swing.*;
+import java.awt.*;
+import java.util.Map;
+
+/**
+ * @author Enaium
+ * @since 0.5.0
+ */
+public class SearchFieldDialog extends SearchDialog {
+    public SearchFieldDialog() {
+        setTitle("Search Field");
+        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);
+            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 FieldInsnNode) {
+                                        FieldInsnNode fieldInsnNode = (FieldInsnNode) instruction;
+                                        if ((fieldInsnNode.owner.contains(owner.getText()) || StringUtil.isBlank(owner.getText())) &&
+                                                (fieldInsnNode.name.contains(name.getText()) || StringUtil.isBlank(name.getText())) &&
+                                                (fieldInsnNode.desc.contains(description.getText()) || StringUtil.isBlank(description.getText()))
+                                        ) {
+                                            ((DefaultListModel<ResultNode>) resultPanel.getList().getModel()).addElement(new ResultNode(stringClassNodeEntry.getValue(), fieldInsnNode.name + ":" + fieldInsnNode.desc));
+                                        }
+                                    }
+                                    JavaOctetEditor.getInstance().bottomPanel.setProcess((int) ((loaded++ / total) * 100f));
+                                }
+                            }
+                        }
+                        JavaOctetEditor.getInstance().bottomPanel.setProcess(0);
+                    });
+                });
+            }});
+        }}, BorderLayout.SOUTH);
+    }
+}

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

@@ -54,7 +54,6 @@ public SearchLdcDialog() {
 
                             for (Map.Entry<String, ClassNode> stringClassNodeEntry : jar.classes.entrySet()) {
                                 for (MethodNode method : stringClassNodeEntry.getValue().methods) {
-
                                     for (AbstractInsnNode instruction : method.instructions) {
                                         if (instruction instanceof LdcInsnNode) {
                                             String ldc = ((LdcInsnNode) instruction).cst.toString();

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

@@ -16,6 +16,7 @@
 
 package cn.enaium.joe.gui.panel.menu;
 
+import cn.enaium.joe.gui.panel.menu.search.FieldMenuItem;
 import cn.enaium.joe.gui.panel.menu.search.LdcMenuItem;
 
 import javax.swing.*;
@@ -27,5 +28,6 @@ public class SearchMenu extends JMenu {
     public SearchMenu() {
         super("Search");
         add(new LdcMenuItem());
+        add(new FieldMenuItem());
     }
 }

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

@@ -0,0 +1,40 @@
+/*
+ * 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.jar.Jar;
+
+import javax.swing.*;
+
+/**
+ * @author Enaium
+ * @since 0.5.0
+ */
+public class FieldMenuItem extends JMenuItem {
+    public FieldMenuItem() {
+        super("Field");
+        addActionListener(e -> {
+            Jar jar = JavaOctetEditor.getInstance().jar;
+            if (jar == null) {
+                return;
+            }
+            new SearchFieldDialog().setVisible(true);
+        });
+    }
+}

src/main/java/cn/enaium/joe/gui/panel/search/ResultNode.java

@@ -16,8 +16,11 @@
 
 package cn.enaium.joe.gui.panel.search;
 
+import cn.enaium.joe.util.HtmlUtil;
 import org.objectweb.asm.tree.ClassNode;
 
+import java.awt.*;
+
 /**
  * @author Enaium
  */
@@ -40,6 +43,6 @@ public String getResult() {
 
     @Override
     public String toString() {
-        return result;
+        return HtmlUtil.toHtml(HtmlUtil.setColor(classNode.name, new Color(255, 255, 255)) + "#" + HtmlUtil.setColor(result, new Color(151, 194, 120)));
     }
 }