D-Net

package eu.dnetlib.data.mapreduce.util;

import java.util.List;

import com.google.common.base.Function;

import com.google.common.collect.Iterables;

import com.google.common.collect.Lists;

import com.google.protobuf.Descriptors.FieldDescriptor;

import com.google.protobuf.GeneratedMessage;

import eu.dnetlib.data.proto.DNGFProtos.DNGFEntity;

import eu.dnetlib.data.proto.TypeProtos.Type;

import eu.dnetlib.data.transform.AbstractProtoMapper;

public class DNGFEntityDecoder extends AbstractProtoMapper {

	private final DNGFEntity dngfEntity;

	private DNGFEntityDecoder(final DNGFEntity oafEntity) {

		this.dngfEntity = oafEntity;

	}

	public static DNGFEntityDecoder decode(final DNGFEntity oafEntity) {

		return new DNGFEntityDecoder(oafEntity);

	}

	public Type getType() {

		return dngfEntity.getType();

	}

	public String getId() {

		return dngfEntity.getId();

	}

	public GeneratedMessage getDNGFEntity() {

		return dngfEntity;

	}

	public GeneratedMessage getEntity() {

		final FieldDescriptor fd = dngfEntity.getDescriptorForType().findFieldByName(typeName());

		return (GeneratedMessage) dngfEntity.getField(fd);

	}

	public String typeName() {

		return dngfEntity.getType().toString();

	}

	public GeneratedMessage getMetadata() {

		final GeneratedMessage entity = getEntity();

		final FieldDescriptor fd = entity.getDescriptorForType().findFieldByName("metadata");

		final GeneratedMessage field = (GeneratedMessage) entity.getField(fd);

		return field;

	}

	public List<String> getFieldValues(final String path) {

		return Lists.newLinkedList(Iterables.transform(processPath(getDNGFEntity(), path, eu.dnetlib.pace.config.Type.String), new Function<Object, String>() {

			@Override

			public String apply(final Object o) {

				return o.toString();

			}

		}));

	}

	public String getDateOfCollection() {

		return dngfEntity.getDateofcollection();

	}

	public String getDateOfTransformation() {

		return dngfEntity.getDateoftransformation();

	}

}