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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.config.application;
import com.yahoo.config.application.FileSystemWrapper.FileWrapper;
import com.yahoo.text.XML;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import javax.xml.transform.TransformerException;
import java.io.File;
import java.io.IOException;
import java.io.StringReader;
import java.io.StringWriter;
import java.util.List;
import java.util.NoSuchElementException;
import static java.nio.charset.StandardCharsets.UTF_8;
/**
* Handles preprocess:include statements and returns a Document which has all the include statements resolved
*
* @author hmusum
* @since 5.22
*/
class IncludeProcessor implements PreProcessor {
private final FileWrapper application;
public IncludeProcessor(File application) {
this(FileSystemWrapper.getDefault(application.toPath()).wrap(application.toPath()));
}
public IncludeProcessor(FileWrapper application) {
this.application = application;
}
public Document process(Document input) throws IOException, TransformerException {
Document doc = Xml.copyDocument(input);
includeFile(application, doc.getDocumentElement());
return doc;
}
private static void includeFile(FileWrapper currentFolder, Element currentElement) throws IOException {
NodeList list = currentElement.getElementsByTagNameNS(XmlPreProcessor.preprocessNamespaceUri, "include");
while (list.getLength() > 0) {
Element elem = (Element) list.item(0);
Element parent = (Element) elem.getParentNode();
String filename = elem.getAttribute("file");
boolean required = ! elem.hasAttribute("required") || Boolean.parseBoolean(elem.getAttribute("required"));
FileWrapper file = currentFolder.child(filename);
Document subFile = IncludeProcessor.parseIncludeFile(file, parent.getTagName(), required);
includeFile(file.parent().orElseThrow(() -> new NoSuchElementException(file + " has no parent")),
subFile.getDocumentElement());
//System.out.println("document before merging: " + documentAsString(doc));
IncludeProcessor.mergeInto(parent, XML.getChildren(subFile.getDocumentElement()));
//System.out.println("document after merging: " + documentAsString(doc));
parent.removeChild(elem);
//System.out.println("document after removing child: " + documentAsString(doc));
list = currentElement.getElementsByTagNameNS(XmlPreProcessor.preprocessNamespaceUri, "include");
}
}
private static void mergeInto(Element destination, List<Element> subElements) {
// System.out.println("merging " + subElements.size() + " elements into " + destination.getTagName());
for (Element subElement : subElements) {
Node copiedNode = destination.getOwnerDocument().importNode(subElement, true);
destination.appendChild(copiedNode);
}
}
private static Document parseIncludeFile(FileWrapper file, String parentTagName, boolean required) throws IOException {
StringWriter w = new StringWriter();
final String startTag = "<" + parentTagName + " " + XmlPreProcessor.deployNamespace + "='" + XmlPreProcessor.deployNamespaceUri + "' " + XmlPreProcessor.preprocessNamespace + "='" + XmlPreProcessor.preprocessNamespaceUri + "'>";
w.append(startTag);
if (file.exists() || required) {
w.append(new String(file.content(), UTF_8));
}
final String endTag = "</" + parentTagName + ">";
w.append(endTag);
return XML.getDocument(new StringReader(w.toString()));
}
}
|
