| /* |
| * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. Oracle designates this |
| * particular file as subject to the "Classpath" exception as provided |
| * by Oracle in the LICENSE file that accompanied this code. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| package com.sun.xml.internal.txw2.output; |
| |
| import com.sun.xml.internal.txw2.TxwException; |
| |
| import javax.xml.transform.stream.StreamResult; |
| import java.io.BufferedWriter; |
| import java.io.FileOutputStream; |
| import java.io.IOException; |
| import java.io.OutputStream; |
| import java.io.OutputStreamWriter; |
| import java.io.Writer; |
| import java.io.UnsupportedEncodingException; |
| |
| /** |
| * {@link XmlSerializer} for {@link javax.xml.transform.stream.StreamResult}. |
| * |
| * @author Ryan.Shoemaker@Sun.COM |
| */ |
| public class StreamSerializer implements XmlSerializer { |
| |
| // delegate to SaxSerializer |
| private final SaxSerializer serializer; |
| |
| private final XMLWriter writer; |
| |
| public StreamSerializer(OutputStream out) { |
| this(createWriter(out)); |
| } |
| |
| public StreamSerializer(OutputStream out,String encoding) throws UnsupportedEncodingException { |
| this(createWriter(out,encoding)); |
| } |
| |
| public StreamSerializer(Writer out) { |
| this(new StreamResult(out)); |
| } |
| |
| public StreamSerializer(StreamResult streamResult) { |
| // if this method opened a stream, let it close it |
| final OutputStream[] autoClose = new OutputStream[1]; |
| |
| if (streamResult.getWriter() != null) |
| writer = createWriter(streamResult.getWriter()); |
| else if (streamResult.getOutputStream() != null) |
| writer = createWriter(streamResult.getOutputStream()); |
| else if (streamResult.getSystemId() != null) { |
| String fileURL = streamResult.getSystemId(); |
| |
| fileURL = convertURL(fileURL); |
| |
| try { |
| FileOutputStream fos = new FileOutputStream(fileURL); |
| autoClose[0] = fos; |
| writer = createWriter(fos); |
| } catch (IOException e) { |
| throw new TxwException(e); |
| } |
| } else |
| throw new IllegalArgumentException(); |
| |
| // now delegate to the SaxSerializer |
| serializer = new SaxSerializer(writer,writer,false) { |
| public void endDocument() { |
| super.endDocument(); |
| if(autoClose[0]!=null) { |
| try { |
| autoClose[0].close(); |
| } catch (IOException e) { |
| throw new TxwException(e); |
| } |
| autoClose[0] = null; |
| } |
| } |
| }; |
| } |
| |
| private StreamSerializer(XMLWriter writer) { |
| this.writer = writer; |
| // now delegate to the SaxSerializer |
| serializer = new SaxSerializer(writer,writer,false); |
| } |
| |
| private String convertURL(String url) { |
| url = url.replace('\\', '/'); |
| url = url.replaceAll("//","/"); |
| url = url.replaceAll("//","/"); |
| if (url.startsWith("file:/")) { |
| if (url.substring(6).indexOf(":") > 0) |
| url = url.substring(6); |
| else |
| url = url.substring(5); |
| } // otherwise assume that it's a file name |
| return url; |
| } |
| |
| // XmlSerializer api's - delegate to SaxSerializer |
| public void startDocument() { |
| serializer.startDocument(); |
| } |
| |
| public void beginStartTag(String uri, String localName, String prefix) { |
| serializer.beginStartTag(uri, localName, prefix); |
| } |
| |
| public void writeAttribute(String uri, String localName, String prefix, StringBuilder value) { |
| serializer.writeAttribute(uri, localName, prefix, value); |
| } |
| |
| public void writeXmlns(String prefix, String uri) { |
| serializer.writeXmlns(prefix, uri); |
| } |
| |
| public void endStartTag(String uri, String localName, String prefix) { |
| serializer.endStartTag(uri, localName, prefix); |
| } |
| |
| public void endTag() { |
| serializer.endTag(); |
| } |
| |
| public void text(StringBuilder text) { |
| serializer.text(text); |
| } |
| |
| public void cdata(StringBuilder text) { |
| serializer.cdata(text); |
| } |
| |
| public void comment(StringBuilder comment) { |
| serializer.comment(comment); |
| } |
| |
| public void endDocument() { |
| serializer.endDocument(); |
| } |
| |
| public void flush() { |
| serializer.flush(); |
| try { |
| writer.flush(); |
| } catch (IOException e) { |
| throw new TxwException(e); |
| } |
| } |
| |
| // other supporting code |
| private static XMLWriter createWriter(Writer w) { |
| // buffering improves the performance |
| DataWriter dw = new DataWriter(new BufferedWriter(w)); |
| dw.setIndentStep(" "); |
| return dw; |
| } |
| |
| private static XMLWriter createWriter(OutputStream os, String encoding) throws UnsupportedEncodingException { |
| XMLWriter writer = createWriter(new OutputStreamWriter(os,encoding)); |
| writer.setEncoding(encoding); |
| return writer; |
| } |
| |
| private static XMLWriter createWriter(OutputStream os) { |
| try { |
| return createWriter(os,"UTF-8"); |
| } catch (UnsupportedEncodingException e) { |
| // UTF-8 is supported on all platforms. |
| throw new Error(e); |
| } |
| } |
| |
| } |