package eu.dnetlib.data.actionmanager.actions;

import java.lang.reflect.Type;

import com.google.gson.*;
import com.google.protobuf.InvalidProtocolBufferException;
import com.googlecode.protobuf.format.JsonFormat;
import eu.dnetlib.data.proto.DNGFProtos.DNGF;
import eu.dnetlib.rmi.data.hadoop.actionmanager.Agent;
import eu.dnetlib.rmi.data.hadoop.actionmanager.actions.AbstractActionSerializer;
import eu.dnetlib.rmi.data.hadoop.actionmanager.actions.AtomicAction;
import org.apache.commons.codec.binary.Base64;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * Created by claudio on 30/09/15.
 */
public class AtomicActionSerialiser extends AbstractActionSerializer implements JsonSerializer<AtomicAction> {

	private static final Log log = LogFactory.getLog(AtomicActionSerialiser.class);

	public static String toJSON(final AtomicAction action) {

		final GsonBuilder gson = new GsonBuilder();

		gson.registerTypeAdapter(AtomicAction.class, new AtomicActionSerialiser());

		return gson.create().toJson(action);
	}

	@Override
	public JsonElement serialize(final AtomicAction aa, final Type typeOfSrc, final JsonSerializationContext context) {

		JsonObject json = new JsonObject();
		json.addProperty(targetColumn, aa.getTargetColumn());
		json.addProperty(targetColumnFamily, aa.getTargetColumnFamily());
		json.addProperty(targetRowKey, aa.getTargetRowKey());
		json.addProperty(rawSet, aa.getRawSet());
		json.addProperty(rowKey, aa.getRowKey());
		json.addProperty(actionType, aa.getActionType().toString());

		json.add(agent, getAgentJSON(aa));

		//json.addProperty(targetValue, getTargetValueJSON(aa));

		final String base64 = new String(Base64.encodeBase64(getTargetValueJSON(aa).getBytes()));
		json.addProperty(targetValue, base64);

		return json;
	}

	private JsonElement getAgentJSON(final AtomicAction aa) {

		final Agent a = aa.getAgent();

		JsonObject json = new JsonObject();
		json.addProperty(agent_id, a.getId());
		json.addProperty(agent_name, a.getName());
		json.addProperty(agent_type, a.getType().toString());

		return json;
	}

	private String getTargetValueJSON(final AtomicAction aa) {
		try {
			return aa.getTargetValue() != null && aa.getTargetValue().length > 0 ? JsonFormat.printToString(DNGF.parseFrom(aa.getTargetValue())) : "";
		} catch (InvalidProtocolBufferException e) {
			log.error("unable to parse proto", e);
			return null;
		}
	}
}