/*

 * 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);

        }

    }

}