/*
 * (c) 2020 Pluto Inc.
 */
package tv.pluto.mongo.utils.json;

import com.google.gson.JsonDeserializationContext;
import com.google.gson.JsonDeserializer;
import com.google.gson.JsonElement;
import com.google.gson.JsonParseException;
import com.google.gson.JsonPrimitive;
import com.google.gson.JsonSerializationContext;
import com.google.gson.JsonSerializer;
import java.lang.reflect.Type;
import org.bson.types.ObjectId;

public class ObjectIdAdapter implements JsonDeserializer<ObjectId>, JsonSerializer<ObjectId> {
  @Override
  public JsonElement serialize(ObjectId src, Type typeOfSrc, JsonSerializationContext context) {
    return new JsonPrimitive(src.toHexString());
  }

  @Override
  public ObjectId deserialize(JsonElement json, Type typeOfT, JsonDeserializationContext context)
      throws JsonParseException {
    if (json.getAsString() == null || json.getAsString().isEmpty()) {
      return null;
    } else {
      return new ObjectId(json.getAsString());
    }
  }
}