AtomWriter.java example

Explorer
geotoolkit-master
/*
 *    Geotoolkit - An Open Source Java GIS Toolkit
 *    http://www.geotoolkit.org
 *
 *    (C) 2010, Geomatys
 *
 *    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.
 */
package org.geotoolkit.atom.xml;

import java.net.URI;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;
import org.geotoolkit.atom.model.AtomEmail;
import org.geotoolkit.atom.model.AtomLink;
import org.geotoolkit.atom.model.AtomPersonConstruct;
import org.geotoolkit.xml.StaxStreamWriter;
import static org.geotoolkit.atom.xml.AtomConstants.*;

/**
 *
 * @author Samuel Andrés
 * @module
 */
public class AtomWriter extends StaxStreamWriter {

    public AtomWriter() {
        super();
    }

    public void setWriter(XMLStreamWriter writer) {
        this.writer = writer;
    }

    /**
     *
     * @param author
     * @throws XMLStreamException
     */
    public void writeAuthor(AtomPersonConstruct author) throws XMLStreamException {
        writer.writeStartElement(URI_ATOM, TAG_AUTHOR);
        for (Object param : author.getParams()) {
            if (param instanceof String) {
                this.writeName((String) param);
            } else if (param instanceof URI) {
                this.writeUri((URI) param);
            } else if (param instanceof AtomEmail) {
                this.writeEmail((AtomEmail) param);
            }
        }
        writer.writeEndElement();
    }

    /**
     *
     * @param link
     * @throws XMLStreamException
     */
    public void writeLink(AtomLink link) throws XMLStreamException {
        writer.writeStartElement(URI_ATOM, TAG_LINK);
        if (link.getHref() != null) {
            writer.writeAttribute(ATT_HREF, link.getHref());
        }
        if (link.getRel() != null) {
            writer.writeAttribute(ATT_REL, link.getRel());
        }
        if (link.getType() != null) {
            writer.writeAttribute(ATT_TYPE, link.getType());
        }
        if (link.getHreflang() != null) {
            writer.writeAttribute(ATT_HREFLANG, link.getHreflang());
        }
        if (link.getTitle() != null) {
            writer.writeAttribute(ATT_TITLE, link.getTitle());
        }
        if (link.getLength() != null) {
            writer.writeAttribute(ATT_LENGTH, link.getLength());
        }
        writer.writeEndElement();
    }

    /**
     *
     * @param name
     * @throws XMLStreamException
     */
    private void writeName(String name) throws XMLStreamException {
        writer.writeStartElement(URI_ATOM, TAG_NAME);
        writer.writeCharacters(name);
        writer.writeEndElement();
    }

    /**
     *
     * @param uri
     * @throws XMLStreamException
     */
    private void writeUri(URI uri) throws XMLStreamException {
        writer.writeStartElement(URI_ATOM, TAG_URI);
        writer.writeCharacters(uri.toString());
        writer.writeEndElement();
    }

    /**
     * 
     * @param email
     * @throws XMLStreamException
     */
    private void writeEmail(AtomEmail email) throws XMLStreamException {
        writer.writeStartElement(URI_ATOM, TAG_EMAIL);
        writer.writeCharacters(email.getAddress());
        writer.writeEndElement();
    }
}