JsonRequestReader.java example

Explorer
org.eclipse.ecr-master
/*
 * Copyright (c) 2006-2011 Nuxeo SA (http://nuxeo.com/) and others.
 *
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     bstefanescu
 */
package org.eclipse.ecr.automation.server.jaxrs.io;

import java.io.IOException;
import java.io.InputStream;
import java.lang.annotation.Annotation;
import java.lang.reflect.Type;
import java.util.Iterator;

import javax.ws.rs.Consumes;
import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.ext.MessageBodyReader;
import javax.ws.rs.ext.Provider;

import net.sf.json.JSONObject;

import org.eclipse.ecr.automation.server.jaxrs.ExecutionRequest;
import org.eclipse.ecr.core.api.DocumentRef;
import org.eclipse.ecr.core.api.DocumentRefList;
import org.eclipse.ecr.core.api.IdRef;
import org.eclipse.ecr.core.api.PathRef;
import org.eclipse.ecr.core.api.impl.DocumentRefListImpl;
import org.nuxeo.common.utils.FileUtils;
import org.nuxeo.common.utils.StringUtils;

/**
 * @author <a href="mailto:bs@nuxeo.com">Bogdan Stefanescu</a>
 */
@Provider
@Consumes("application/json+nxrequest")
public class JsonRequestReader implements MessageBodyReader<ExecutionRequest> {

    public static final MediaType targetMediaType = new MediaType("application", "json+nxrequest");

    public boolean isReadable(Class<?> arg0, Type arg1, Annotation[] arg2,
            MediaType arg3) {
        return (targetMediaType.equals(arg3) && ExecutionRequest.class.isAssignableFrom(arg0));
    }

    public ExecutionRequest readFrom(Class<ExecutionRequest> arg0, Type arg1,
            Annotation[] arg2, MediaType arg3,
            MultivaluedMap<String, String> headers, InputStream in)
            throws IOException, WebApplicationException {
        return readRequest(in,headers);
    }

    public static ExecutionRequest readRequest(InputStream in, MultivaluedMap<String, String> headers)
            throws IOException {
        String content = FileUtils.read(in);
        return readRequest(content, headers);
    }

    @SuppressWarnings("unchecked")
    public static ExecutionRequest readRequest(String content, MultivaluedMap<String, String> headers) {
        JSONObject json = JSONObject.fromObject(content);
        String input = json.optString("input", null);
        JSONObject jsonParams = json.optJSONObject("params");
        JSONObject jsonContext = json.optJSONObject("context");

        String documentProperties = json.optString("documentProperties", null);
        if (documentProperties!=null) {
            headers.putSingle(JsonDocumentListWriter.DOCUMENT_PROPERTIES_HEADER, documentProperties);
        }

        Object inObj = null;
        if (input != null) {
            if (input.startsWith("doc:")) {
                inObj = docRefFromString(input.substring(4).trim());
            } else if (input.startsWith("docs:")) {
                String[] ar = StringUtils.split(input.substring(5).trim(), ',',
                        true);
                DocumentRefList list = new DocumentRefListImpl(ar.length);
                for (String s : ar) {
                    list.add(docRefFromString(s));
                }
                inObj = list;
            }
        }
        ExecutionRequest req = new ExecutionRequest(inObj);

        if (jsonParams != null) {
            Iterator<String> it = jsonParams.keys();
            while (it.hasNext()) {
                String key = it.next();
                String value = jsonParams.getString(key);
                req.setParam(key, value);
            }
        }

        if (jsonContext != null) {
            Iterator<String> it = jsonContext.keys();
            while (it.hasNext()) {
                String key = it.next();
                String value = jsonContext.getString(key);
                req.setContextParam(key, value);
            }
        }

        return req;
    }

    public static DocumentRef docRefFromString(String input) {
        if (input.startsWith("/")) {
            return new PathRef(input);
        } else {
            return new IdRef(input);
        }
    }
}