com.almende.eve.rpc.jsonrpc.JSONRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of eve-core Show documentation
There is a newer version: 3.1.1
package com.almende.eve.rpc.jsonrpc;

import java.io.IOException;
import java.io.Serializable;

import com.almende.eve.rpc.jsonrpc.jackson.JOM;
import com.fasterxml.jackson.annotation.JsonIgnore;
import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonParseException;
import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.databind.JsonMappingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ObjectNode;

public class JSONRequest implements Serializable {
	private static final long serialVersionUID = 1970046457233622444L;

	protected ObjectNode req = JOM.createObjectNode();

	public enum VERSION {
		ONE, TWO
	};

	public JSONRequest() {
		init(null, null, null);
	}

	public JSONRequest(String json) throws JSONRPCException,
			JsonParseException, JsonMappingException, IOException {
		ObjectMapper mapper = JOM.getInstance();
		init(mapper.readValue(json, ObjectNode.class));
	}

	public JSONRequest(ObjectNode request) throws JSONRPCException {
		init(request);
	}

	public void init(ObjectNode request) throws JSONRPCException {
		if (request == null || request.isNull()) {
			throw new JSONRPCException(JSONRPCException.CODE.INVALID_REQUEST,
					"Request is null");
		}
		if (request.has("jsonrpc") && request.get("jsonrpc").isTextual()) {
			if (!request.get("jsonrpc").asText().equals("2.0")) {
				throw new JSONRPCException(
						JSONRPCException.CODE.INVALID_REQUEST,
						"Value of member 'jsonrpc' is not equal to '2.0'");
			}
		}
		if (!request.has("method")) {
			throw new JSONRPCException(JSONRPCException.CODE.INVALID_REQUEST,
					"Member 'method' missing in request");
		}
		if (!(request.get("method").isTextual())) {
			throw new JSONRPCException(JSONRPCException.CODE.INVALID_REQUEST,
					"Member 'method' is no String");
		}
		/*
		 * TODO: cleanup if (!request.has("params")) { throw new
		 * JSONRPCException(JSONRPCException.CODE.INVALID_REQUEST,
		 * "Member 'params' missing in request"); }
		 */
		if (request.has("params") && !(request.get("params").isObject())) {
			throw new JSONRPCException(JSONRPCException.CODE.INVALID_REQUEST,
					"Member 'params' is no ObjectNode");
		}

		init(request.get("id"), request.get("method").asText(),
				(ObjectNode) request.get("params"));
	}

	/*
	 * public JSONRequest (String method) { init (null, method, null); }
	 */

	public JSONRequest(String method, ObjectNode params) {
		init(null, method, params);
	}

	public JSONRequest(Object id, String method, ObjectNode params) {
		init(id, method, params);
	}

	public JSONRequest(String method, ObjectNode params, String callbackUrl,
			String callbackMethod) {
		init(null, method, params);
		setCallback(callbackUrl, callbackMethod);
	}

	public JSONRequest(Object id, String method, ObjectNode params,
			String callbackUrl, String callbackMethod) {
		init(id, method, params);
		setCallback(callbackUrl, callbackMethod);
	}

	private void init(Object id, String method, ObjectNode params) {
		setVersion();
		setId(id);
		setMethod(method);
		setParams(params);
	}

	public void setId(Object id) {
		ObjectMapper mapper = JOM.getInstance();
		req.put("id", mapper.convertValue(id, JsonNode.class));
	}

	public Object getId() {
		ObjectMapper mapper = JOM.getInstance();
		return mapper.convertValue(req.get("id"), Object.class);
	}

	public void setMethod(String method) {
		req.put("method", method);
	}

	public String getMethod() {
		if (req.has("method")) {
			return req.get("method").asText();
		}
		return null;
	}

	public void setParams(ObjectNode params) {
		req.put("params", params != null ? params : JOM.createObjectNode());
	}

	public ObjectNode getParams() {
		return (ObjectNode) req.get("params");
	}

	public void putParam(String name, Object value) {
		ObjectMapper mapper = JOM.getInstance();
		req.with("params")
				.put(name, mapper.convertValue(value, JsonNode.class));
	}

	public Object getParam(String name) {
		ObjectMapper mapper = JOM.getInstance();
		ObjectNode params = req.with("params");
		if (params.has(name)) {
			return mapper.convertValue(params.get(name), Object.class);
		}
		return null;
	}

	public Object hasParam(String name) {
		return req.get("params").has(name);
	}

	private void setVersion() {
		req.put("jsonrpc", "2.0");
	}

	public void setCallback(String url, String method) {
		ObjectNode callback = JOM.createObjectNode();
		callback.put("url", url);
		callback.put("method", method);
		req.put("callback", callback);
	}

	public String getCallbackUrl() {
		JsonNode callback = req.get("callback");
		if (callback != null && callback.isObject() && callback.has("url")
				&& callback.get("url").isTextual()) {
			return callback.get("url").asText();
		}
		return null;
	}

	public String getCallbackMethod() {
		JsonNode callback = req.get("callback");
		if (callback != null && callback.isObject() && callback.has("method")
				&& callback.get("method").isTextual()) {
			return callback.get("method").asText();
		}
		return null;
	}

	public boolean hasCallback() {
		return req.has("callback");
	}

	@JsonIgnore
	public ObjectNode getObjectNode() {
		return req;
	}

	@Override
	public String toString() {
		ObjectMapper mapper = JOM.getInstance();
		try {
			return mapper.writeValueAsString(req);
		} catch (Exception e) {
			e.printStackTrace();
		}
		return null;
	}

	private void writeObject(java.io.ObjectOutputStream out) throws IOException {
		ObjectMapper mapper = new ObjectMapper();
		mapper.configure(JsonGenerator.Feature.AUTO_CLOSE_TARGET, false);
		mapper.writeValue(out, req);
	}

	private void readObject(java.io.ObjectInputStream in) throws IOException,
			ClassNotFoundException {
		ObjectMapper mapper = new ObjectMapper();
		mapper.configure(JsonParser.Feature.AUTO_CLOSE_SOURCE, false);
		req = mapper.readValue(in, ObjectNode.class);
	}

}