add option to view calls to a method in general, not just the specific instance of the method

1 files changed, 15 insertions, 1 deletions

diff --git a/src/main/java/cuchaz/enigma/analysis/JarIndex.java b/src/main/java/cuchaz/enigma/analysis/JarIndex.java
index 820ecbb..361c8e7 100644
--- a/src/main/java/cuchaz/enigma/analysis/JarIndex.java
+++ b/src/main/java/cuchaz/enigma/analysis/JarIndex.java
@@ -447,8 +447,22 @@ public class JarIndex {
                 return this.methodsReferencingClasses.get(classEntry);
         }
 
+        @Deprecated
         public Collection<EntryReference<MethodEntry, MethodDefEntry>> getMethodsReferencing(MethodEntry methodEntry) {
-                return this.methodsReferencing.get(methodEntry);
+                return getMethodsReferencing(methodEntry, false);
+        }
+
+        public Collection<EntryReference<MethodEntry, MethodDefEntry>> getMethodsReferencing(MethodEntry methodEntry, boolean recurse) {
+                if (!recurse) {
+                        return this.methodsReferencing.get(methodEntry);
+                }
+
+                List<EntryReference<MethodEntry, MethodDefEntry>> references = new ArrayList<>();
+                Set<MethodEntry> methodEntries = getRelatedMethodImplementations(methodEntry);
+                for (MethodEntry entry : methodEntries) {
+                        references.addAll(getMethodsReferencing(entry, false));
+                }
+                return references;
         }
 
         public Collection<MethodEntry> getReferencedMethods(MethodDefEntry methodEntry) {