1 /*
2 * Copyright (c) 2005, 2020, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation. Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
25
26 package com.sun.org.apache.xalan.internal.xsltc.trax;
27
28 import java.util.ArrayList;
29 import java.util.List;
30 import javax.xml.stream.Location;
31 import javax.xml.stream.XMLReporter;
32 import javax.xml.stream.XMLStreamException;
33 import org.xml.sax.Attributes;
34 import org.xml.sax.Locator;
35 import org.xml.sax.SAXException;
36 import org.xml.sax.SAXParseException;
37 import org.xml.sax.ext.LexicalHandler;
38 import org.xml.sax.ext.Locator2;
39 import org.xml.sax.helpers.DefaultHandler;
40
41 public abstract class SAX2StAXBaseWriter extends DefaultHandler
42 implements
43 LexicalHandler {
44
45 protected boolean isCDATA;
46
47 protected StringBuffer CDATABuffer;
48
49 protected List<String> namespaces;
50
51 protected Locator docLocator;
52
53 protected XMLReporter reporter;
54
55 String xmlVersion = null, encoding = null;
56
57 public SAX2StAXBaseWriter() {
58 }
59
60 public SAX2StAXBaseWriter(XMLReporter reporter) {
61 this.reporter = reporter;
62 }
63
64 public void setXMLReporter(XMLReporter reporter) {
65 this.reporter = reporter;
66 }
67
68 public void setDocumentLocator(Locator locator) {
69 this.docLocator = locator;
70 }
71
72 private void updateVersionAndEncoding() {
73 if (docLocator instanceof Locator2) {
74 Locator2 l2 = (Locator2) docLocator;
75 xmlVersion = l2.getXMLVersion();
76 encoding = l2.getEncoding();
77 }
78 }
79
80 public void setXmlVersion(String version) {
81 this.xmlVersion = version;
82 }
83
84 public void setEncoding(String encoding) {
85 this.encoding = encoding;
86 }
87
88 void writeStartDocument() throws SAXException {
89 updateVersionAndEncoding();
90 }
91
92 public Location getCurrentLocation() {
93 if (docLocator != null) {
94 return new SAXLocation(docLocator);
95 } else {
96 return null;
97 }
98 }
99
100 public void error(SAXParseException e) throws SAXException {
101 reportException("ERROR", e);
102 }
103
104 public void fatalError(SAXParseException e) throws SAXException {
105 reportException("FATAL", e);
106 }
107
108 public void warning(SAXParseException e) throws SAXException {
109 reportException("WARNING", e);
110 }
111
112 public void startDocument() throws SAXException {
113 namespaces = new ArrayList<>(2);
114 }
115
116 public void endDocument() throws SAXException {
117 namespaces = null;
118 }
119
120 public void startElement(String uri, String localName, String qName,
121 Attributes attributes) throws SAXException {
122 namespaces = null;
123 }
124
125 public void endElement(String uri, String localName, String qName)
126 throws SAXException {
127 namespaces = null;
128 }
129
130 public void startPrefixMapping(String prefix, String uri)
131 throws SAXException {
132
133 if (prefix == null) {
134 prefix = "";
135 } else if (prefix.equals("xml")) {
136 return;
137 }
138
139 if (namespaces == null) {
140 namespaces = new ArrayList<>(2);
141 }
142 namespaces.add(prefix);
143 namespaces.add(uri);
144 }
145
146 public void endPrefixMapping(String prefix) throws SAXException {
147 }
148
149 public void startCDATA() throws SAXException {
150 isCDATA = true;
151 if (CDATABuffer == null) {
152 CDATABuffer = new StringBuffer();
153 } else {
154 CDATABuffer.setLength(0);
155 }
156 }
157
158 public void characters(char[] ch, int start, int length)
159 throws SAXException {
160 if (isCDATA) {
161 CDATABuffer.append(ch, start, length);
162 }
163 }
164
165 public void endCDATA() throws SAXException {
166 isCDATA = false;
167 CDATABuffer.setLength(0);
168 }
169
170 public void comment(char[] ch, int start, int length) throws SAXException {
171 }
172
173 public void endDTD() throws SAXException {
174 }
175
176 public void endEntity(String name) throws SAXException {
177 }
178
179 public void startDTD(String name, String publicId, String systemId)
180 throws SAXException {
181 }
182
183 public void startEntity(String name) throws SAXException {
184 }
185
186 /**
187 * Used to report a {@link SAXException}to the {@link XMLReporter}
188 * registered with this handler.
189 */
190 protected void reportException(String type, SAXException e)
191 throws SAXException {
192
193 if (reporter != null) {
194 try {
195 reporter.report(e.getMessage(), type, e, getCurrentLocation());
196 } catch (XMLStreamException e1) {
197 throw new SAXException(e1);
198 }
199 }
200 }
201
202 /**
203 * Parses an XML qualified name, and places the resulting prefix and local
204 * name in the provided String array.
205 *
206 * @param qName The qualified name to parse.
207 * @param results An array where parse results will be placed. The prefix
208 * will be placed at <code>results[0]</code>, and the local
209 * part at <code>results[1]</code>
210 */
211 public static final void parseQName(String qName, String[] results) {
212
213 String prefix, local;
214 int idx = qName.indexOf(':');
215 if (idx >= 0) {
216 prefix = qName.substring(0, idx);
217 local = qName.substring(idx + 1);
218 } else {
219 prefix = "";
220 local = qName;
221 }
222 results[0] = prefix;
223 results[1] = local;
224 }
225
226 /**
227 * {@Link Location}implementation used to expose details from a SAX
228 * {@link Locator}.
229 *
230 * @author christian
231 */
232 private static final class SAXLocation implements Location {
233
234 private int lineNumber;
235 private int columnNumber;
236 private String publicId;
237 private String systemId;
238 private SAXLocation(Locator locator) {
239 lineNumber = locator.getLineNumber();
240 columnNumber = locator.getColumnNumber();
241 publicId = locator.getPublicId();
242 systemId = locator.getSystemId();
243 }
244
245 public int getLineNumber() {
246 return lineNumber;
247 }
248
249 public int getColumnNumber() {
250 return columnNumber;
251 }
252
253 public int getCharacterOffset() {
254 return -1;
255 }
256
257 public String getPublicId() {
258 return publicId;
259 }
260
261 public String getSystemId() {
262 return systemId;
263 }
264 }
265 }