/* Copyright (c) 2009 Google Inc.
 *
 * Licensed under the Apache 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.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.google.wave.api.impl;

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;
import com.google.gson.reflect.TypeToken;
import com.google.wave.api.JsonRpcResponse;
import com.google.wave.api.JsonRpcConstant.ParamsProperty;
import com.google.wave.api.JsonRpcConstant.ResponseProperty;

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

/**
 * Custom serializer and deserializer that serializes and deserializes a
 * {@link JsonRpcResponse}.
 *
 * @author mprasetya@google.com (Marcel Prasetya)
 */
public class JsonRpcResponseGsonAdaptor implements JsonDeserializer<JsonRpcResponse>,
    JsonSerializer<JsonRpcResponse>{

  @Override
  public JsonRpcResponse deserialize(JsonElement json, Type type,
      JsonDeserializationContext context) throws JsonParseException {
    JsonObject jsonObject = json.getAsJsonObject();

    String id = jsonObject.get(ResponseProperty.ID.key()).getAsString();
    if (jsonObject.has(ResponseProperty.ERROR.key())) {
      JsonElement errorObject = jsonObject.get(ResponseProperty.ERROR.key());
      String errorMessage = errorObject.getAsJsonObject().get("message").getAsString();
      return JsonRpcResponse.error(id, errorMessage);
    }

    // Deserialize the data.
    Map<ParamsProperty, Object> properties = new HashMap<ParamsProperty, Object>();
    JsonElement data = jsonObject.get(ResponseProperty.DATA.key());
    if (data != null && data.isJsonObject()) {
      for (Entry<String, JsonElement> parameter : data.getAsJsonObject().entrySet()) {
        ParamsProperty parameterType = ParamsProperty.fromKey(parameter.getKey());
        if (parameterType == null) {
          // Skip this unknown parameter.
          continue;
        }
        Object object = null;
        if (parameterType == ParamsProperty.BLIPS) {
          object = context.deserialize(parameter.getValue(), GsonFactory.BLIP_MAP_TYPE);
        } else if (parameterType == ParamsProperty.PARTICIPANTS_ADDED || 
            parameterType == ParamsProperty.PARTICIPANTS_REMOVED) {
          object = context.deserialize(parameter.getValue(), GsonFactory.PARTICIPANT_LIST_TYPE);
        } else if (parameterType == ParamsProperty.THREADS) {
          object = context.deserialize(parameter.getValue(), GsonFactory.THREAD_MAP_TYPE);
        } else {
          object = context.deserialize(parameter.getValue(), parameterType.clazz());
        }
        properties.put(parameterType, object);
      }
    }

    return JsonRpcResponse.result(id, properties);
  }

  @Override
  public JsonElement serialize(JsonRpcResponse src, Type type, JsonSerializationContext context) {
    JsonObject result = new JsonObject();
    result.addProperty(ResponseProperty.ID.key(), src.getId());
    if (src.isError()) {
      JsonObject error = new JsonObject();
      error.addProperty(ParamsProperty.MESSAGE.key(), src.getErrorMessage());
      result.add(ResponseProperty.ERROR.key(), error);
    } else {
      JsonObject data = new JsonObject();
      for (Entry<ParamsProperty, Object> properties : src.getData().entrySet()) {
        data.add(properties.getKey().key(), context.serialize(properties.getValue()));
      }
      result.add(ResponseProperty.DATA.key(), data);
    }
    return result;
  }
}