/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
package org.mozilla.gecko.sync;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.io.StringReader;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import org.json.simple.JSONArray;
import org.json.simple.JSONObject;
import org.json.simple.parser.JSONParser;
import org.json.simple.parser.ParseException;
/**
* Extend JSONObject to do little things, like, y'know, accessing members.
*
* @author rnewman
*
*/
public class ExtendedJSONObject {
public JSONObject object;
private static Object processParseOutput(Object parseOutput) {
if (parseOutput instanceof JSONObject) {
return new ExtendedJSONObject((JSONObject) parseOutput);
} else {
return parseOutput;
}
}
public static Object parse(String string) throws IOException, ParseException {
return processParseOutput(new JSONParser().parse(string));
}
public static Object parse(InputStreamReader reader) throws IOException, ParseException {
return processParseOutput(new JSONParser().parse(reader));
}
public static Object parse(InputStream stream) throws IOException, ParseException {
InputStreamReader reader = new InputStreamReader(stream, "UTF-8");
return ExtendedJSONObject.parse(reader);
}
/**
* Helper method to get a JSONObject from a String. Input: String containing
* JSON. Output: Extracted JSONObject. Throws: Exception if JSON is invalid.
*
* @throws ParseException
* @throws IOException
* @throws NonObjectJSONException
* If the object is valid JSON, but not an object.
*/
public static ExtendedJSONObject parseJSONObject(String jsonString)
throws IOException,
ParseException,
NonObjectJSONException {
return new ExtendedJSONObject(jsonString);
}
public ExtendedJSONObject() {
this.object = new JSONObject();
}
public ExtendedJSONObject(JSONObject o) {
this.object = o;
}
public ExtendedJSONObject(String jsonString) throws IOException, ParseException, NonObjectJSONException {
if (jsonString == null) {
this.object = new JSONObject();
return;
}
Reader in = new StringReader(jsonString);
Object obj = new JSONParser().parse(in);
if (obj instanceof JSONObject) {
this.object = ((JSONObject) obj);
} else {
throw new NonObjectJSONException(obj);
}
}
// Passthrough methods.
public Object get(String key) {
return this.object.get(key);
}
public Long getLong(String key) {
return (Long) this.get(key);
}
public String getString(String key) {
return (String) this.get(key);
}
/**
* Return an Integer if the value for this key is an Integer, Long, or String
* that can be parsed as a base 10 Integer.
* Passes through null.
*
* @throws NumberFormatException
*/
public Integer getIntegerSafely(String key) throws NumberFormatException {
Object val = this.object.get(key);
if (val == null) {
return null;
}
if (val instanceof Integer) {
return (Integer) val;
}
if (val instanceof Long) {
return new Integer(((Long) val).intValue());
}
if (val instanceof String) {
return Integer.parseInt((String) val, 10);
}
throw new NumberFormatException("Expecting Integer, got " + val.getClass());
}
/**
* Return a server timestamp value as milliseconds since epoch.
*
* @param key
* @return A Long, or null if the value is non-numeric or doesn't exist.
*/
public Long getTimestamp(String key) {
Object val = this.object.get(key);
// This is absurd.
if (val instanceof Double) {
double millis = ((Double) val).doubleValue() * 1000;
return new Double(millis).longValue();
}
if (val instanceof Float) {
double millis = ((Float) val).doubleValue() * 1000;
return new Double(millis).longValue();
}
if (val instanceof Number) {
// Must be an integral number.
return ((Number) val).longValue() * 1000;
}
return null;
}
public boolean containsKey(String key) {
return this.object.containsKey(key);
}
public String toJSONString() {
return this.object.toJSONString();
}
public String toString() {
return this.object.toString();
}
public void put(String key, Object value) {
@SuppressWarnings("unchecked")
Map<Object, Object> map = this.object;
map.put(key, value);
}
/**
* Remove key-value pair from JSONObject.
*
* @param key
* to be removed.
* @return true if key exists and was removed, false otherwise.
*/
public boolean remove(String key) {
Object res = this.object.remove(key);
return (res != null);
}
public ExtendedJSONObject getObject(String key) throws NonObjectJSONException {
Object o = this.object.get(key);
if (o == null) {
return null;
}
if (o instanceof ExtendedJSONObject) {
return (ExtendedJSONObject) o;
}
if (o instanceof JSONObject) {
return new ExtendedJSONObject((JSONObject) o);
}
throw new NonObjectJSONException(o);
}
public ExtendedJSONObject clone() {
return new ExtendedJSONObject((JSONObject) this.object.clone());
}
/**
* Helper method for extracting a JSONObject from its string encoding within
* another JSONObject.
*
* Input: JSONObject and key. Output: JSONObject extracted. Throws: Exception
* if JSON is invalid.
*
* @throws NonObjectJSONException
* @throws ParseException
* @throws IOException
*/
public ExtendedJSONObject getJSONObject(String key) throws IOException,
ParseException,
NonObjectJSONException {
String val = (String) this.object.get(key);
return ExtendedJSONObject.parseJSONObject(val);
}
@SuppressWarnings("unchecked")
public Iterable<Entry<String, Object>> entryIterable() {
return this.object.entrySet();
}
@SuppressWarnings("unchecked")
public Set<String> keySet() {
return this.object.keySet();
}
public org.json.simple.JSONArray getArray(String key) throws NonArrayJSONException {
Object o = this.object.get(key);
if (o == null) {
return null;
}
if (o instanceof JSONArray) {
return (JSONArray) o;
}
throw new NonArrayJSONException(o);
}
public int size() {
return this.object.size();
}
}