org.ektorp.support.AttachmentsInOrderParser Maven / Gradle / Ivy

package org.ektorp.support;


import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import com.fasterxml.jackson.core.JsonParseException;
import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.core.JsonToken;

/**
 * A document's MIME multipart/related representation produced by CouchDB
 * uses the order of the attachments in the JSON _attachments object as the
 * order of the attachments in the multipart/related. Thus, the order must
 * be preserved in order to parse a CouchDB multipart/related message.
 *
 * This class parses a document and returns the order of the attachments.
 *
 * @author David Venable
 */
public class AttachmentsInOrderParser
{
    /**
     * Parses a CouchDB document in the form of a JsonParser to get the
     * attachments order. It is important that the JsonParser come straight
     * from the source document and not from an object, or the order will
     * be incorrect.
     * @param documentJsonParser a JsonParser which is at the very root of a JSON CouchDB document
     * @return the list of attachment names in the order provided in the document
     * @throws IOException
     */
    public static List parseAttachmentNames(JsonParser documentJsonParser) throws IOException
    {
        documentJsonParser.nextToken();

        JsonToken jsonToken;
        while((jsonToken = documentJsonParser.nextToken()) != JsonToken.END_OBJECT)
        {
            if(CouchDbDocument.ATTACHMENTS_NAME.equals(documentJsonParser.getCurrentName()))
            {
                return readAttachments(documentJsonParser);
            }
            else if(jsonToken == JsonToken.START_OBJECT)
            {
                readIgnoreObject(documentJsonParser);
            }
        }
        return null;
    }

    private static List readAttachments(JsonParser jsonParser) throws IOException
    {
        jsonParser.nextToken();
        return readAttachmentsObject(jsonParser);
    }

    private static List readAttachmentsObject(JsonParser jsonParser) throws IOException
    {
        List attachmentNameList = new ArrayList();
        while(jsonParser.nextToken() != JsonToken.END_OBJECT)
        {
            String attachmentName = jsonParser.getCurrentName();

            jsonParser.nextToken();
            if(jsonParser.getCurrentToken() != JsonToken.START_OBJECT)
            {
                String message = CouchDbDocument.ATTACHMENTS_NAME + " contains an invalid object.";
                throw new JsonParseException(message, jsonParser.getCurrentLocation());
            }

            readIgnoreObject(jsonParser);

            attachmentNameList.add(attachmentName);
        }

        return attachmentNameList;
    }

    private static void readIgnoreObject(JsonParser jsonParser) throws IOException
    {
        while(jsonParser.nextToken() != JsonToken.END_OBJECT)
        {
            if(jsonParser.getCurrentToken() == JsonToken.START_OBJECT)
            {
                readIgnoreObject(jsonParser);
            }
        }
    }
}