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
|
package org.zwobble.mammoth.internal.xml.parsing;
import org.xml.sax.Attributes;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import org.xml.sax.XMLReader;
import org.xml.sax.helpers.DefaultHandler;
import org.zwobble.mammoth.internal.util.PassThroughException;
import javax.xml.XMLConstants;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.util.Map;
import java.util.stream.IntStream;
import static java.util.stream.Collectors.toMap;
class SimpleSax {
static void parseStream(InputStream input, SimpleSaxHandler handler) {
parseInputSource(new InputSource(input), handler);
}
static void parseString(String value, SimpleSaxHandler handler) {
parseInputSource(new InputSource(new StringReader(value)), handler);
}
private static void parseInputSource(InputSource inputSource, SimpleSaxHandler handler) {
SAXParserFactory parserFactory = SAXParserFactory.newInstance();
parserFactory.setNamespaceAware(true);
try {
parserFactory.setFeature("http://xml.org/sax/features/external-general-entities", false);
parserFactory.setFeature("http://apache.org/xml/features/disallow-doctype-decl", true);
parserFactory.setFeature(XMLConstants.FEATURE_SECURE_PROCESSING, true);
SAXParser saxParser = parserFactory.newSAXParser();
XMLReader xmlReader = saxParser.getXMLReader();
xmlReader.setContentHandler(new DefaultHandler() {
@Override
public void startElement(String uri, String localName, String qName, Attributes attributes) throws SAXException {
ElementName name = new ElementName(uri, localName);
Map<ElementName, String> attributesMap = IntStream.range(0, attributes.getLength())
.boxed()
.collect(toMap(
index -> new ElementName(attributes.getURI(index), attributes.getLocalName(index)),
attributes::getValue
));
handler.startElement(name, attributesMap);
}
@Override
public void endElement(String uri, String localName, String qName) throws SAXException {
handler.endElement();
}
@Override
public void characters(char[] ch, int start, int length) throws SAXException {
handler.characters(new String(ch, start, length));
}
});
xmlReader.parse(inputSource);
} catch (IOException exception) {
throw new PassThroughException(exception);
} catch (ParserConfigurationException | SAXException exception) {
throw new RuntimeException(exception);
}
}
}
|
