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
|
/*******************************************************************************
* Copyright (c) 2015 Jeff Martin.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the GNU Lesser General Public
* License v3.0 which accompanies this distribution, and is available at
* http://www.gnu.org/licenses/lgpl.html
* <p>
* Contributors:
* Jeff Martin - initial API and implementation
******************************************************************************/
package cuchaz.enigma;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableListMultimap;
import cuchaz.enigma.analysis.ClassCache;
import cuchaz.enigma.analysis.index.JarIndex;
import cuchaz.enigma.api.EnigmaPlugin;
import cuchaz.enigma.api.EnigmaPluginContext;
import cuchaz.enigma.api.service.EnigmaService;
import cuchaz.enigma.api.service.EnigmaServiceFactory;
import cuchaz.enigma.api.service.EnigmaServiceType;
import cuchaz.enigma.api.service.JarIndexerService;
import cuchaz.enigma.utils.Utils;
import java.io.IOException;
import java.nio.file.Path;
import java.util.List;
import java.util.ServiceLoader;
public class Enigma {
private final EnigmaProfile profile;
private final EnigmaServices services;
private Enigma(EnigmaProfile profile, EnigmaServices services) {
this.profile = profile;
this.services = services;
}
public static Enigma create() {
return new Builder().build();
}
public static Builder builder() {
return new Builder();
}
public EnigmaProject openJar(Path path, ProgressListener progress) throws IOException {
ClassCache classCache = ClassCache.of(path);
JarIndex jarIndex = classCache.index(progress);
services.get(JarIndexerService.TYPE).forEach(indexer -> indexer.acceptJar(classCache, jarIndex));
return new EnigmaProject(this, classCache, jarIndex, Utils.zipSha1(path));
}
public EnigmaProfile getProfile() {
return profile;
}
public EnigmaServices getServices() {
return services;
}
public static class Builder {
private EnigmaProfile profile = EnigmaProfile.EMPTY;
private Iterable<EnigmaPlugin> plugins = ServiceLoader.load(EnigmaPlugin.class);
private Builder() {
}
public Builder setProfile(EnigmaProfile profile) {
Preconditions.checkNotNull(profile, "profile cannot be null");
this.profile = profile;
return this;
}
public Builder setPlugins(Iterable<EnigmaPlugin> plugins) {
Preconditions.checkNotNull(plugins, "plugins cannot be null");
this.plugins = plugins;
return this;
}
public Enigma build() {
PluginContext pluginContext = new PluginContext(profile);
for (EnigmaPlugin plugin : plugins) {
plugin.init(pluginContext);
}
EnigmaServices services = pluginContext.buildServices();
return new Enigma(profile, services);
}
}
private static class PluginContext implements EnigmaPluginContext {
private final EnigmaProfile profile;
private final ImmutableListMultimap.Builder<EnigmaServiceType<?>, EnigmaService> services = ImmutableListMultimap.builder();
PluginContext(EnigmaProfile profile) {
this.profile = profile;
}
@Override
public <T extends EnigmaService> void registerService(String id, EnigmaServiceType<T> serviceType, EnigmaServiceFactory<T> factory) {
List<EnigmaProfile.Service> serviceProfiles = profile.getServiceProfiles(serviceType);
for (EnigmaProfile.Service serviceProfile : serviceProfiles) {
if (serviceProfile.matches(id)) {
T service = factory.create(serviceProfile::getArgument);
services.put(serviceType, service);
break;
}
}
}
EnigmaServices buildServices() {
return new EnigmaServices(services.build());
}
}
}
|
