com.github.stephenc.javaisotools.sabre.impl.XMLAtomHandler.java Source code

Introduction

Here is the source code for com.github.stephenc.javaisotools.sabre.impl.XMLAtomHandler.java

Source

/*
 * Copyright (c) 2010. Stephen Connolly.
 * Copyright (c) 2006. Michael Hartle <mhartle@rbg.informatik.tu-darmstadt.de>.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */

package com.github.stephenc.javaisotools.sabre.impl;

import java.io.IOException;
import java.io.InputStream;
import java.util.Stack;

import com.github.stephenc.javaisotools.sabre.DataReference;
import com.github.stephenc.javaisotools.sabre.Element;
import com.github.stephenc.javaisotools.sabre.Fixup;
import com.github.stephenc.javaisotools.sabre.HandlerException;
import com.github.stephenc.javaisotools.sabre.StreamHandler;
import org.apache.commons.lang.StringEscapeUtils;
import org.xml.sax.ContentHandler;
import org.xml.sax.SAXException;

public class XMLAtomHandler implements StreamHandler {

    private ContentHandler contentHandler = null;
    private String namespace = null;
    private String prefix = null;
    private Stack elements = null;

    public XMLAtomHandler(ContentHandler contentHandler, String namespace, String prefix) {
        this.contentHandler = contentHandler;
        this.namespace = namespace;
        this.prefix = prefix;
        this.elements = new Stack();
    }

    public void startDocument() throws HandlerException {
        try {
            this.contentHandler.startElement(this.namespace, "root", this.prefix + ":root", null);
        } catch (SAXException e) {
            throw new HandlerException(e);
        }
    }

    public void startElement(Element element) throws HandlerException {
        try {
            this.elements.push(element);
            this.contentHandler.startElement(this.namespace, element.toString(),
                    this.prefix + ":" + element.toString(), null);
        } catch (SAXException e) {
            throw new HandlerException(e);
        }
    }

    public void data(DataReference reference) throws HandlerException {
        long showLength = 0;

        try {
            InputStream inputStream = null;
            byte[] buffer = null;
            String data = null;
            long length = reference.getLength();

            if (length > 32) {
                showLength = 32;
            } else {
                showLength = length;
            }

            buffer = new byte[(int) showLength];
            inputStream = reference.createInputStream();
            inputStream.read(buffer);
            inputStream.close();

            data = new String(buffer);
            data = StringEscapeUtils.escapeJava(data);
            this.contentHandler.characters(data.toCharArray(), 0, data.length());
        } catch (SAXException e) {
            throw new HandlerException(e);
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    public Fixup fixup(DataReference reference) throws HandlerException {
        throw new HandlerException("Cannot handle fixup in SAX.");
    }

    public long mark() throws HandlerException {
        throw new HandlerException("Cannot mark position in SAX.");
    }

    public void endElement() throws HandlerException {
        Element element = null;

        try {
            element = (Element) this.elements.pop();
            this.contentHandler.endElement(this.namespace, element.toString(),
                    this.prefix + ":" + element.toString());
        } catch (SAXException e) {
            throw new HandlerException(e);
        }
    }

    public void endDocument() throws HandlerException {
        try {
            this.contentHandler.endElement(this.namespace, "root", this.prefix + ":root");
        } catch (SAXException e) {
            throw new HandlerException(e);
        }
    }
}