/
XmlAppendingTransformer.groovy
113 lines (97 loc) · 3.59 KB
/
XmlAppendingTransformer.groovy
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
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License") you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package com.github.jengelman.gradle.plugins.shadow.transformers
import org.apache.tools.zip.ZipEntry
import org.apache.tools.zip.ZipOutputStream
import org.gradle.api.file.FileTreeElement
import org.jdom2.Attribute
import org.jdom2.Content
import org.jdom2.Document
import org.jdom2.Element
import org.jdom2.JDOMException
import org.jdom2.input.SAXBuilder
import org.jdom2.output.Format
import org.jdom2.output.XMLOutputter
import org.xml.sax.EntityResolver
import org.xml.sax.InputSource
import org.xml.sax.SAXException
/**
* Appends multiple occurrences of some XML file.
* <p>
* Modified from org.apache.maven.plugins.shade.resource.XmlAppendingTransformer.java
*
* @author John Engelman
*/
class XmlAppendingTransformer implements Transformer {
static final String XSI_NS = "http://www.w3.org/2001/XMLSchema-instance"
boolean ignoreDtd = true
String resource
Document doc
boolean canTransformResource(FileTreeElement element) {
def path = element.relativePath.pathString
if (resource != null && resource.equalsIgnoreCase(path)) {
return true
}
return false
}
void transform(TransformerContext context) {
Document r
try {
SAXBuilder builder = new SAXBuilder(false)
builder.setExpandEntities(false)
if (ignoreDtd) {
builder.setEntityResolver(new EntityResolver() {
InputSource resolveEntity(String publicId, String systemId)
throws SAXException, IOException {
return new InputSource(new StringReader(""))
}
})
}
r = builder.build(context.is)
}
catch (JDOMException e) {
throw new RuntimeException("Error processing resource " + resource + ": " + e.getMessage(), e)
}
if (doc == null) {
doc = r
} else {
Element root = r.getRootElement()
root.attributes.each { Attribute a ->
Element mergedEl = doc.getRootElement()
Attribute mergedAtt = mergedEl.getAttribute(a.getName(), a.getNamespace())
if (mergedAtt == null) {
mergedEl.setAttribute(a)
}
}
root.children.each { Content n ->
doc.getRootElement().addContent(n.clone())
}
}
}
boolean hasTransformedResource() {
return doc != null
}
void modifyOutputStream(ZipOutputStream os, boolean preserveFileTimestamps) {
ZipEntry entry = new ZipEntry(resource)
entry.time = TransformerContext.getEntryTimestamp(preserveFileTimestamps, entry.time)
os.putNextEntry(entry)
new XMLOutputter(Format.getPrettyFormat()).output(doc, os)
doc = null
}
}