AtomStatement.java

package org.swordapp.server;

import org.apache.abdera.Abdera;
import org.apache.abdera.i18n.iri.IRI;
import org.apache.abdera.model.Category;
import org.apache.abdera.model.Entry;
import org.apache.abdera.model.Feed;
import org.apache.abdera.model.Link;

import javax.xml.namespace.QName;
import java.io.IOException;
import java.io.Writer;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Map;

public class AtomStatement extends Statement {
    private final String author;
    private final String feedUri;
    private final String title;
    private final String updated;

    public AtomStatement(final String feedUri, final String author, final String title, final String updated) {
        this.contentType = "application/atom+xml;type=feed";
        this.author = author != null && !"".equals(author) ? author : "Unknown";
        this.feedUri = feedUri;
        this.title = title != null ? title : "Untitled";
        this.updated = updated;
    }

    @Override
    public void writeTo(final Writer out) throws IOException {
        Abdera abdera = new Abdera();
        Feed feed = abdera.newFeed();

        // id
        // link@rel="self" -> point to id
        // title
        // updated
        feed.setId(this.feedUri);
        feed.addLink(this.feedUri, "self");
        feed.setTitle(this.title);
        feed.addAuthor(this.author);

        if (this.updated != null) {
            feed.setUpdated(this.updated);
        } else {
            feed.setUpdated(new Date());
        }

        // create an entry for each Resource Part
        for (ResourcePart resource : this.resources) {
            Entry entry = feed.addEntry();

            // id
            // summary
            // title
            // updated
            entry.setContent(new IRI(resource.getUri()), resource.getMediaType());
            entry.setId(resource.getUri());
            entry.setTitle("Resource " + resource.getUri());
            entry.setSummary("Resource Part");
            entry.setUpdated(new Date());

            for (String linkHref : resource.getSelfLinks()) {
                entry.addLink(linkHref, Link.REL_SELF);
            }
        }

        // create an entry for each original deposit
        for (OriginalDeposit deposit : this.originalDeposits) {
            Entry entry = feed.addEntry();

            // id
            // summary
            // title
            // updated
            entry.setId(deposit.getUri());
            entry.setTitle("Original Deposit " + deposit.getUri());
            entry.setSummary("Original Deposit");
            entry.setUpdated(new Date());

            for (String linkHref : deposit.getSelfLinks()) {
                entry.addLink(linkHref, Link.REL_SELF);
            }
    
            if (deposit.getMediaType() != null) {
                entry.setContent(new IRI(deposit.getUri()), deposit.getMediaType());
            }
            entry.addCategory(UriRegistry.SWORD_TERMS_NAMESPACE, UriRegistry.SWORD_ORIGINAL_DEPOSIT, "Original Deposit");
            if (deposit.getDepositedOn() != null) {
                SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss");
                entry.addSimpleExtension(new QName(UriRegistry.SWORD_DEPOSITED_ON), sdf.format(deposit.getDepositedOn()));
            }

            if (deposit.getDepositedOnBehalfOf() != null) {
                entry.addSimpleExtension(new QName(UriRegistry.SWORD_DEPOSITED_ON_BEHALF_OF), deposit.getDepositedOnBehalfOf());
            }

            if (deposit.getDepositedBy() != null) {
                entry.addSimpleExtension(new QName(UriRegistry.SWORD_DEPOSITED_BY), deposit.getDepositedBy());
            }

            if (deposit.getPackaging() != null) {
                for (String packaging : deposit.getPackaging()) {
                    entry.addSimpleExtension(UriRegistry.SWORD_PACKAGING, packaging);
                }
            }
        }
    
        // now at the state as a categories
        for (Map.Entry<String, String> state : this.states.entrySet()) {
            Category cat = feed.addCategory(UriRegistry.SWORD_STATE, state.getKey(), "State");
            if (state.getValue() != null) {
                cat.setText(state.getValue());
            }
        }

        // now write the feed
        feed.writeTo(out);
    }
}