org.mozilla.zest.core.v1.ZestJSON.java Source code

Introduction

Here is the source code for org.mozilla.zest.core.v1.ZestJSON.java

Source

/* 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.zest.core.v1;

import java.lang.reflect.Type;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonDeserializationContext;
import com.google.gson.JsonDeserializer;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParseException;
import com.google.gson.JsonSerializationContext;
import com.google.gson.JsonSerializer;

// TODO: Auto-generated Javadoc
/**
 * The Class ZestJSON.
 */
public class ZestJSON implements JsonDeserializer<ZestElement>, JsonSerializer<ZestElement> {

    /** The gson. */
    private static Gson gson = null;

    /**
     * To string.
     *
     * @param element the element
     * @return the string
     */
    public static String toString(ZestElement element) {
        return getGson().toJson(element);
    }

    /**
     * From string.
     *
     * @param str the str
     * @return the zest element
     */
    public static ZestElement fromString(String str) {
        ZestElement ze = getGson().fromJson(str, ZestElement.class);
        if (ze != null && ze instanceof ZestStatement) {
            ((ZestStatement) ze).init();
        }
        return ze;
    }

    /**
     * Gets the gson.
     *
     * @return the gson
     */
    private static Gson getGson() {
        if (gson == null) {
            // Need to add all of the abstract classes
            gson = new GsonBuilder().registerTypeAdapter(ZestAction.class, new ZestJSON())
                    .registerTypeAdapter(ZestAssignment.class, new ZestJSON())
                    .registerTypeAdapter(ZestAuthentication.class, new ZestJSON())
                    .registerTypeAdapter(ZestElement.class, new ZestJSON())
                    .registerTypeAdapter(ZestStatement.class, new ZestJSON())
                    .registerTypeAdapter(ZestExpressionElement.class, new ZestJSON())
                    .registerTypeAdapter(ZestLoop.class, new ZestJSON())
                    .registerTypeAdapter(ZestLoopState.class, new ZestJSON())
                    .registerTypeAdapter(ZestLoopTokenSet.class, new ZestJSON()).setPrettyPrinting().create();

        }
        return gson;
    }

    @Override
    public ZestElement deserialize(JsonElement element, Type rawType, JsonDeserializationContext arg2)
            throws JsonParseException {
        if (element instanceof JsonObject) {
            String elementType = ((JsonObject) element).get("elementType").getAsString();

            if (elementType.startsWith("Zest")) {
                try {
                    //Class<?> c = Class.forName(this.getClass().getPackage().getName()"org.mozilla.zest.core.v1." + elementType);
                    Class<?> c = Class.forName(this.getClass().getPackage().getName() + "." + elementType);
                    return (ZestElement) getGson().fromJson(element, c);

                } catch (ClassNotFoundException e) {
                    throw new JsonParseException(e);
                }
            }
        }
        return null;
    }

    @Override
    public JsonElement serialize(ZestElement element, Type rawType, JsonSerializationContext context) {
        return new GsonBuilder().setPrettyPrinting().create().toJsonTree(element);
    }

}