blob: c5a6443b68ba927cbddd6ec9947dfc9d43db61db (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
package com.yahoo.abicheck.classtree;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayDeque;
import java.util.Arrays;
import java.util.Collection;
import java.util.Deque;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
public abstract class ClassFileTree implements AutoCloseable {
public static ClassFileTree fromJar(JarFile jarFile) throws IOException {
Map<String, Package> rootPackages = new HashMap<>();
Enumeration<JarEntry> jarEntries = jarFile.entries();
while (jarEntries.hasMoreElements()) {
JarEntry entry = jarEntries.nextElement();
if (!entry.isDirectory() && entry.getName().endsWith(".class")) {
Deque<String> parts = new ArrayDeque<>(Arrays.asList(entry.getName().split("/")));
String className = parts.removeLast();
Package pkg = rootPackages
.computeIfAbsent(parts.removeFirst(), name -> new Package(null, name));
for (String part : parts) {
pkg = pkg.getOrCreateSubPackage(part);
}
pkg.addClass(new ClassFile(pkg, className) {
@Override
public InputStream getInputStream() throws IOException {
return jarFile.getInputStream(entry);
}
});
}
}
return new ClassFileTree() {
@Override
public Collection<Package> getRootPackages() {
return rootPackages.values();
}
@Override
public void close() throws IOException {
jarFile.close();
}
};
}
public abstract Collection<Package> getRootPackages();
public static abstract class ClassFile {
private final Package parent;
private final String name;
private ClassFile(Package parent, String name) {
this.parent = parent;
this.name = name;
}
public abstract InputStream getInputStream() throws IOException;
public String getName() {
return name;
}
@Override
public String toString() {
return "ClassFile(" + parent.getFullyQualifiedName() + "." + name + ")";
}
}
public static class Package {
private final Package parent;
private final String name;
private final Map<String, Package> subPackages = new HashMap<>();
private final Set<ClassFile> classFiles = new HashSet<>();
private Package(Package parent, String name) {
this.parent = parent;
this.name = name;
}
private Package getOrCreateSubPackage(String name) {
return subPackages.computeIfAbsent(name, n -> new Package(this, n));
}
private void addClass(ClassFile klazz) {
classFiles.add(klazz);
}
public String getFullyQualifiedName() {
if (parent == null) {
return name;
} else {
return parent.getFullyQualifiedName() + "." + name;
}
}
public Collection<Package> getSubPackages() {
return subPackages.values();
}
public Collection<ClassFile> getClassFiles() {
return classFiles;
}
@Override
public String toString() {
return "Package(" + getFullyQualifiedName() + ")";
}
}
}
|