com.denimgroup.threadfix.importer.util.JsonUtils.java Source code

Introduction

Here is the source code for com.denimgroup.threadfix.importer.util.JsonUtils.java

Source

////////////////////////////////////////////////////////////////////////
//
//     Copyright (c) 2009-2015 Denim Group, Ltd.
//
//     The contents of this file are subject to the Mozilla Public License
//     Version 2.0 (the "License"); you may not use this file except in
//     compliance with the License. You may obtain a copy of the License at
//     http://www.mozilla.org/MPL/
//
//     Software distributed under the License is distributed on an "AS IS"
//     basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
//     License for the specific language governing rights and limitations
//     under the License.
//
//     The Original Code is ThreadFix.
//
//     The Initial Developer of the Original Code is Denim Group, Ltd.
//     Portions created by Denim Group, Ltd. are Copyright (C)
//     Denim Group, Ltd. All Rights Reserved.
//
//     Contributor(s): Denim Group, Ltd.
//
////////////////////////////////////////////////////////////////////////

package com.denimgroup.threadfix.importer.util;

import com.denimgroup.threadfix.CollectionUtils;
import com.denimgroup.threadfix.logging.SanitizedLogger;
import com.google.gson.*;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;
import java.lang.reflect.Type;
import java.util.*;

/**
 * Created by mac on 4/4/14.
 */
public class JsonUtils {

    private static final SanitizedLogger LOG = new SanitizedLogger(JsonUtils.class);

    /**
     * Convenience method to wrap the exception catching.
     * TODO validate to save generating an exception on invalid input
     * @param responseContents
     * @return JSON array object or null
     */
    @Nullable
    public static JSONArray getJSONArray(String responseContents) {
        try {
            return new JSONArray(responseContents);
        } catch (JSONException e) {
            LOG.warn("JSON Parsing failed.", e);
            return null;
        }
    }

    @Nullable
    public static String getStringProperty(String jsonString, String propertyName) {

        if (jsonString == null || !jsonString.contains(propertyName)) {
            LOG.warn("JSON string doesn't contain " + propertyName);
            return null;
        }

        JSONObject object = getJSONObject(jsonString);

        if (object != null) {
            try {
                return object.getString(propertyName);
            } catch (JSONException e) {
                LOG.warn("JSON Parsing failed.", e);
            }
        }

        return null;
    }

    @Nullable
    public static Long getLongProperty(String jsonString, String propertyName) {

        if (jsonString == null || !jsonString.contains(propertyName)) {
            LOG.warn("JSON string doesn't contain " + propertyName);
            return null;
        }

        JSONObject object = getJSONObject(jsonString);

        if (object != null) {
            try {
                return object.getLong(propertyName);
            } catch (JSONException e) {
                LOG.warn("JSON Parsing failed.", e);
            }
        }

        return null;
    }

    /**
     * Convenience method to wrap the exception catching.
     * TODO validate to save generating an exception on invalid input
     * @param responseContents
     * @return JSON object or null
     */
    @Nullable
    public static JSONObject getJSONObject(String responseContents) {
        try {
            return new JSONObject(responseContents);
        } catch (JSONException e) {
            LOG.warn("JSON Parsing failed.", e);
            return null;
        }
    }

    @Nonnull
    public static Iterable<JSONObject> toJSONObjectIterable(final String jsonString) throws JSONException {
        if (jsonString == null) {
            return toJSONObjectIterable(new JSONArray());
        }
        return toJSONObjectIterable(new JSONArray(jsonString));
    }

    @Nonnull
    public static Iterable<JSONObject> toJSONObjectIterable(final JSONArray array) throws JSONException {
        return new Iterable<JSONObject>() {

            @Override
            public Iterator<JSONObject> iterator() {
                return new Iterator<JSONObject>() {

                    int index = 0;

                    @Override
                    public boolean hasNext() {
                        return index < array.length();
                    }

                    @Override
                    public JSONObject next() {
                        try {
                            return array.getJSONObject(index++);
                        } catch (JSONException e) {
                            throw new ArrayIndexOutOfBoundsException();
                        }
                    }

                    @Override
                    public void remove() {
                        index++;
                    }
                };
            }
        };
    }

    @Nonnull
    public static <T> T toObject(final String jsonString, Class<T> tClass) {
        return getGson().fromJson(jsonString, tClass);
    }

    @Nonnull
    public static <T> List<T> toObjectList(final String jsonString, Class<T> tClass) throws JSONException {
        List<T> result = CollectionUtils.list();
        for (JSONObject jsonObject : toJSONObjectIterable(jsonString)) {
            result.add(getGson().fromJson(String.valueOf(jsonObject), tClass));
        }
        return result;
    }

    public static Map<String, String> jsonToMap(String jsonStr) {
        Map<String, String> map = CollectionUtils.map();
        JSONObject jObject;
        if (jsonStr != null)
            try {
                jObject = new JSONObject(jsonStr);
                Iterator<?> keys = jObject.keys();
                while (keys.hasNext()) {
                    String key = (String) keys.next();
                    String value = jObject.getString(key);
                    map.put(key, value);
                }
            } catch (JSONException e) {
                LOG.warn("JSON Parsing failed.", e);
            }
        return map;
    }

    static class DateSerializer implements JsonSerializer<Date>, JsonDeserializer<Date> {

        @Override
        public JsonElement serialize(Date src, Type typeOfSrc, JsonSerializationContext context) {
            return new JsonPrimitive(src.getTime());
        }

        @Override
        public Date deserialize(JsonElement json, Type typeOfT, JsonDeserializationContext context)
                throws JsonParseException {
            Calendar cal = Calendar.getInstance();
            cal.setTimeInMillis(json.getAsJsonPrimitive().getAsLong());
            return cal.getTime();
        }
    }

    private static Gson getGson() {
        GsonBuilder gsonBuilder = new GsonBuilder();
        gsonBuilder.registerTypeAdapter(Date.class, new DateSerializer());
        return gsonBuilder.create();
    }

}