Horribly hacked together xref feature

Author: rcx

src/main/java/club/bytecode/the/jda/FileContainer.java

@@ -24,8 +24,8 @@ public FileContainer(File f) {
 
     public HashMap<String, byte[]> files = new HashMap<>(); // this is assigned outside the class?!
 
-    public ClassNode loadClass(String name) {
-        byte[] bytes = files.get(name);
+    public ClassNode loadClassFile(String filename) {
+        byte[] bytes = files.get(filename);
         if (bytes == null)
             return null;
         ClassReader reader = new ClassReader(bytes);

src/main/java/club/bytecode/the/jda/decompilers/FernflowerDecompiler.java

@@ -73,7 +73,7 @@ public String decompileClassNode(FileContainer container, final ClassNode cn) {
                 if (classCache.containsKey(className)) {
                     requestedCn = classCache.get(className);
                 } else {
-                    requestedCn = container.loadClass(container.findClassfile(className));
+                    requestedCn = container.loadClassFile(container.findClassfile(className));
                     if (requestedCn == null) {
                         System.err.println("Couldn't load " + externalPath);
                         throw new IOException(container + "$" + cn + " is missing");
@@ -139,7 +139,7 @@ public void closeArchive(String s, String s1) {
                     if (classCache.containsKey(innerClass.name)) {
                         innerCn = classCache.get(innerClass.name);
                     } else {
-                        innerCn = container.loadClass(container.findClassfile(innerClass.name));
+                        innerCn = container.loadClassFile(container.findClassfile(innerClass.name));
                         if (innerCn != null) {
                             applyFilters(innerCn);
                             classCache.put(innerCn.name, innerCn);

src/main/java/club/bytecode/the/jda/decompilers/bytecode/BytecodeDecompiler.java

@@ -88,7 +88,7 @@ protected PrefixedStringBuilder decompile(PrefixedStringBuilder sb, ArrayList<St
                 String innerClassName = innerClassNode.name;
                 if ((innerClassName != null) && !decompiledClasses.contains(innerClassName)) {
                     decompiledClasses.add(innerClassName);
-                    ClassNode cn1 = container.loadClass(container.findClassfile(innerClassName));
+                    ClassNode cn1 = container.loadClassFile(container.findClassfile(innerClassName));
                     applyFilters(cn1);
                     if (cn1 != null) {
                         sb.appendPrefix("     ");

src/main/java/club/bytecode/the/jda/gui/fileviewer/DecompileThread.java

@@ -38,7 +38,7 @@ public void run() {
         try {
             String decompileResult;
 
-            ClassNode cn = viewer.getFile().container.loadClass(viewer.getFile().name);
+            ClassNode cn = viewer.getFile().container.loadClassFile(viewer.getFile().name);
             if (cn == null) {
                 decompileResult = "// The file was removed during the reload.";
             } else {

src/main/java/club/bytecode/the/jda/gui/fileviewer/JDATextArea.java

@@ -1,9 +1,13 @@
 package club.bytecode.the.jda.gui.fileviewer;
 
+import club.bytecode.the.jda.FileContainer;
+import club.bytecode.the.jda.JDA;
 import club.bytecode.the.jda.settings.Settings;
+import net.miginfocom.swing.MigLayout;
 import org.fife.ui.rsyntaxtextarea.RSyntaxTextArea;
 import org.fife.ui.rsyntaxtextarea.Token;
 import org.fife.ui.rsyntaxtextarea.TokenTypes;
+import org.mapleir.stdlib.util.Pair;
 
 import javax.swing.*;
 import javax.swing.event.CaretEvent;
@@ -123,6 +127,46 @@ private void setComment(int line, String comment) {
         resetLine(line);
     }
 
+    private void search(String needle) {
+        List<ViewerFile> matches = new ArrayList<>();
+        for (FileContainer fc : JDA.getOpenFiles()) {
+            for (Map.Entry<String, byte[]> e : fc.getFiles().entrySet()) {
+                if (e.getKey().endsWith(".class")) {
+                    try {
+                        // ClassNode cn = fc.loadClassFile(e.getKey());
+                        String fileBytes = new String(e.getValue());
+                        if (fileBytes.contains(needle)) {
+                            matches.add(new ViewerFile(fc, e.getKey()));
+                        }
+                    } catch (Exception ex) {
+                        ex.printStackTrace();
+                    }
+                }
+            }
+        }
+        final JDialog frame = new JDialog(new JFrame(), "Search Results", true);
+        Container pane = frame.getContentPane();
+        pane.setLayout(new MigLayout());
+        pane.add(new JLabel(needle + "found in:"), "spanx, grow, wrap, align center");
+        JList<Pair<FileContainer, String>> list = new JList(matches.toArray());
+        list.setSelectionMode(ListSelectionModel.SINGLE_INTERVAL_SELECTION);
+        list.setLayoutOrientation(JList.VERTICAL);
+        list.addMouseListener(new MouseAdapter() {
+            public void mouseClicked(MouseEvent evt) {
+                JList list = (JList)evt.getSource();
+                if (evt.getClickCount() == 2) {
+                    int index = list.locationToIndex(evt.getPoint());
+                    ViewerFile vf = matches.get(index);
+                    JDA.viewer.navigator.openClassFileToWorkSpace(vf);
+                }
+            }
+        });
+        JScrollPane listScroller = new JScrollPane(list);
+        pane.add(listScroller);
+        frame.pack();
+        frame.setVisible(true);
+    }
+
     private void resetLine(int line) {
         if (line > lines.size())
             return;
@@ -144,7 +188,9 @@ private boolean isIdentifierSelected() {
     private void doXrefDialog() {
         if (!isIdentifierSelected())
             return;
-        JOptionPane.showMessageDialog(this, "Not implemented");
+        String oldName = currentlySelectedToken.getLexeme();
+        search(oldName);
+        // JOptionPane.showMessageDialog(this, "Not implemented");
     }
 
     private void doRenameDialog() {