APICallDecoder.java

/*
 * APICallDecoder.java
 * Copyright (C) 2011,2012 Wannes De Smet
 * 
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.xenmaster.web;

import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;

import org.xenmaster.web.Hook.APICall;

import com.google.gson.Gson;
import com.google.gson.JsonArray;
import com.google.gson.JsonDeserializationContext;
import com.google.gson.JsonDeserializer;
import com.google.gson.JsonElement;
import com.google.gson.JsonNull;
import com.google.gson.JsonObject;
import com.google.gson.JsonParseException;

/**
 * 
 * @created Nov 7, 2011
 * @author double-u
 */
public class APICallDecoder implements JsonDeserializer<Hook.APICall> {
    
    protected Gson gson = new Gson();

    @Override
    public APICall deserialize(JsonElement json, Type typeOfT, JsonDeserializationContext context) throws JsonParseException {
        if (!json.isJsonObject()) {
            throw new IllegalArgumentException("Invalid JSON Apic object");
        }
        JsonObject obj = (JsonObject) json;
        APICall apic = new APICall();
        for (Entry<String, JsonElement> e : obj.entrySet()) {
            switch (e.getKey()) {
                case "ref":
                    if (e.getValue() != null && !(e.getValue() instanceof JsonNull)) {
                        apic.ref = e.getValue().getAsString();
                    }
                    break;
                case "args":
                    if (e.getValue().isJsonArray()) {
                        JsonArray arr = e.getValue().getAsJsonArray();
                        ArrayList<Object> args = new ArrayList<>(arr.size());
                        for (Iterator<JsonElement> it = arr.iterator(); it.hasNext();) {
                            args.add(deserialize(it.next()));
                        }
                        apic.args = args.toArray();
                    } else {
                        apic.args = null;
                    }
                    break;
            }
        }
        return apic;
    }

    protected Object deserialize(JsonElement json) {
        if (json.isJsonPrimitive()) {
            return json.getAsString();
        } else if (json.isJsonObject()) {
            return deserializeToMap(json.getAsJsonObject());
        } else if (json.isJsonArray()) {
            return deserializeToArray(json.getAsJsonArray());
        }

        return null;
    }

    protected Object[] deserializeToArray(JsonArray arr) {
        ArrayList array = new ArrayList();
        for (Iterator<JsonElement> it = arr.iterator(); it.hasNext();) {
            array.add(deserialize(it.next()));
        }
        return array.toArray();
    }

    protected Map<String, Object> deserializeToMap(JsonObject obj) {
        HashMap<String, Object> map = new HashMap<>();
        for (Entry<String, JsonElement> e : obj.entrySet()) {
            Object value = deserialize(e.getValue());
            map.put(e.getKey(), value);
        }
        return map;
    }
}