Skip to content

Commit

Permalink
Respect recurse flag
Browse files Browse the repository at this point in the history
  • Loading branch information
xxDark committed Jan 7, 2022
1 parent d6d0911 commit 6340a6d
Showing 1 changed file with 9 additions and 1 deletion.
10 changes: 9 additions & 1 deletion src/main/java/me/coley/recaf/compiler/JavacCompiler.java
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.*;
import java.util.function.Predicate;
import java.util.stream.Collectors;
import java.util.stream.Stream;

Expand Down Expand Up @@ -200,9 +201,16 @@ public Iterable<JavaFileObject> list(Location location, String packageName,
if ("CLASS_PATH".equals(location.getName()) && kinds.contains(Kind.CLASS)) {
String formatted = packageName.isEmpty() ? "" : packageName.replace('.', '/') + '/';
Set<JavaFileObject> result = Sets.newHashSet(list);
Predicate<String> check;
if (recurse) {
check = s -> s.startsWith(formatted);
} else {
check = s -> s.startsWith(formatted)
&& s.indexOf('/', formatted.length()) == -1;
}
classpath.stream()
.flatMap(x -> x.getClasses().entrySet().stream())
.filter(e -> e.getKey().startsWith(formatted))
.filter(e -> check.test(e.getKey()))
.forEach(x -> result.add(new ResourceVirtualJavaFileObject(x.getKey(),
x.getValue(), Kind.CLASS)));
return result;
Expand Down

0 comments on commit 6340a6d

Please # to comment.