blob: db84fdc05df1edce281c33dd76ee09f4cd306ae0 (
plain)
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
|
package org.jetbrains
import com.github.jengelman.gradle.plugins.shadow.relocation.Relocator
import com.github.jengelman.gradle.plugins.shadow.transformers.Transformer
import groovy.xml.XmlUtil
import org.apache.tools.zip.ZipEntry
import org.apache.tools.zip.ZipOutputStream
import org.gradle.api.file.FileTreeElement
public class PluginXmlTransformer implements Transformer {
private Map<String, Node> transformedPluginXmlFiles = new HashMap<>();
@Override
boolean canTransformResource(FileTreeElement fileTreeElement) {
return fileTreeElement.relativePath.segments.contains("META-INF") && fileTreeElement.name.endsWith(".xml")
}
@Override
void transform(String path, InputStream inputStream, List<Relocator> relocators) {
System.out.println(path)
Node node = new XmlParser().parse(inputStream)
relocateXml(node, relocators)
transformedPluginXmlFiles.put(path, node)
}
@Override
boolean hasTransformedResource() {
return !transformedPluginXmlFiles.isEmpty()
}
@Override
void modifyOutputStream(ZipOutputStream zipOutputStream) {
for (Map.Entry<String, Node> entry : transformedPluginXmlFiles.entrySet()) {
zipOutputStream.putNextEntry(new ZipEntry(entry.key))
XmlUtil.serialize(entry.value, zipOutputStream)
}
}
private static void relocateXml(Node node, List<Relocator> relocators) {
Map attributes = node.attributes()
for (Map.Entry entry : attributes.entrySet()) {
entry.setValue(relocateClassName((String) entry.getValue(), relocators))
}
List<String> localText = node.localText()
if (localText.size() == 1) {
node.setValue(relocateClassName(localText[0], relocators))
}
node.children().each {
if (it instanceof Node) {
relocateXml((Node) it, relocators)
}
}
}
private static String relocateClassName(String className, List<Relocator> relocators) {
for (Relocator relocator : relocators) {
if (relocator.canRelocateClass(className)) {
return relocator.relocateClass(className)
}
}
return className
}
}
|
