package eu.dnetlib.pace;

import java.io.IOException;
import java.io.StringWriter;
import java.util.ArrayList;
import java.util.List;

import com.google.common.collect.Lists;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.RandomStringUtils;
import org.apache.commons.lang.StringUtils;

import com.google.gson.Gson;

import eu.dnetlib.data.mapreduce.util.OafTest;
import eu.dnetlib.data.proto.FieldTypeProtos.Qualifier;
import eu.dnetlib.data.proto.FieldTypeProtos.StructuredProperty;
import eu.dnetlib.data.proto.FieldTypeProtos.StructuredProperty.Builder;
import eu.dnetlib.data.proto.OafProtos.Oaf;
import eu.dnetlib.data.proto.OafProtos.OafEntity;
import eu.dnetlib.data.proto.OrganizationProtos.Organization;
import eu.dnetlib.data.proto.PersonProtos.Person;
import eu.dnetlib.data.proto.ResultProtos.Result;
import eu.dnetlib.pace.config.Config;
import eu.dnetlib.pace.config.DedupConfig;
import eu.dnetlib.pace.config.Type;
import eu.dnetlib.pace.model.Field;
import eu.dnetlib.pace.model.FieldValueImpl;
import eu.dnetlib.pace.model.MapDocument;
import eu.dnetlib.pace.model.ProtoDocumentBuilder;
import eu.dnetlib.pace.model.gt.GTAuthor;
import eu.dnetlib.pace.model.gt.GTAuthorMapper;

public abstract class AbstractProtoPaceTest extends OafTest {

	protected DedupConfig getResultFullConf() {
		return DedupConfig.load(readFromClasspath("/eu/dnetlib/pace/result.full.pace.conf"));
	}

	protected DedupConfig getResultSimpleConf() {
		return DedupConfig.load(readFromClasspath("/eu/dnetlib/pace/result.simple.pace.conf"));
	}

	protected DedupConfig getResultConf() {
		return DedupConfig.load(readFromClasspath("/eu/dnetlib/pace/result.pace.conf"));
	}

	protected DedupConfig getOrganizationSimpleConf() {
		return DedupConfig.load(readFromClasspath("/eu/dnetlib/pace/organization.pace.conf"));
	}

	protected DedupConfig getResultAuthorsConf() {
		return DedupConfig.load(readFromClasspath("/eu/dnetlib/pace/result.authors.pace.conf"));
	}

	protected DedupConfig getPersonConf() {
		return DedupConfig.load(readFromClasspath("/eu/dnetlib/pace/person.pace.conf"));
	}

	protected DedupConfig getResultProdConf() {
		return DedupConfig.load(readFromClasspath("/eu/dnetlib/pace/result.prod.pace.conf"));
	}

	protected MapDocument person(final Config conf, final String id, final Oaf oaf) {
		return ProtoDocumentBuilder.newInstance(id, oaf.getEntity(), conf.model());
	}

	protected Oaf getPersonGT(final String path) {
		return new GTAuthorMapper().map(getGTAuthor(path));
	}

	protected GTAuthor getGTAuthor(final String path) {

		final Gson gson = new Gson();

		final String json = readFromClasspath(path);

		final GTAuthor gta = gson.fromJson(json, GTAuthor.class);

		return gta;
	}

	private String readFromClasspath(final String filename) {
		final StringWriter sw = new StringWriter();
		try {
			IOUtils.copy(getClass().getResourceAsStream(filename), sw);
			return sw.toString();
		} catch (final IOException e) {
			throw new RuntimeException("cannot load resource from classpath: " + filename);
		}
	}

	protected MapDocument result(final Config config, final String id, final String title) {
		return result(config, id, title, null, new ArrayList<String>(), null);
	}

	protected MapDocument result(final Config config, final String id, final String title, final String date) {
		return result(config, id, title, date, new ArrayList<String>(), null);
	}

	protected MapDocument result(final Config config, final String id, final String title, final String date, final List<String> pid) {
		return result(config, id, title, date, pid, null);
	}

	protected MapDocument result(final Config config, final String id, final String title, final String date, final String pid) {
		return result(config, id, title, date, pid, null);
	}

	protected MapDocument result(final Config config, final String id, final String title, final String date, final String pid, final List<String> authors) {
		return result(config, id, title, date, Lists.newArrayList(pid), authors);
	}

	protected MapDocument result(final Config config, final String id, final String title, final String date, final List<String> pid, final List<String> authors) {
		final Result.Metadata.Builder metadata = Result.Metadata.newBuilder();
		if (!StringUtils.isBlank(title)) {
			metadata.addTitle(getStruct(title, getQualifier("main title", "dnet:titles")));
			metadata.addTitle(getStruct(RandomStringUtils.randomAlphabetic(10), getQualifier("alternative title", "dnet:titles")));
		}
		if (!StringUtils.isBlank(date)) {
			metadata.setDateofacceptance(sf(date));
		}

		final OafEntity.Builder entity = oafEntity(id, eu.dnetlib.data.proto.TypeProtos.Type.result);
		final Result.Builder result = Result.newBuilder().setMetadata(metadata);

		if (authors != null) {
			for (final String author : authors) {
				result.addAuthor(person(author));
			}
		}

		entity.setResult(result);

		if (pid != null) {
			for(String p : pid) {
				if (!StringUtils.isBlank(p)) {
					entity.addPid(sp(p, "doi"));
					//entity.addPid(sp(RandomStringUtils.randomAlphabetic(10), "oai"));
				}
			}
		}

		final OafEntity build = entity.build();
		return ProtoDocumentBuilder.newInstance(id, build, config.model());
	}

	private Person.Builder person(final String author) {
		final Person.Builder person = Person.newBuilder();

		final eu.dnetlib.pace.model.Person p = new eu.dnetlib.pace.model.Person(author, false);
		final Person.Metadata.Builder metadata = Person.Metadata.newBuilder();
		if (p.isAccurate()) {
			metadata.setFirstname(sf(p.getNormalisedFirstName()));
			metadata.addSecondnames(sf(p.getNormalisedSurname()));
			metadata.setFullname(sf(p.getNormalisedFullname()));
		} else {
			metadata.setFullname(sf(p.getOriginal()));
		}

		return person.setMetadata(metadata);
	}

	private OafEntity.Builder oafEntity(final String id, final eu.dnetlib.data.proto.TypeProtos.Type type) {
		final OafEntity.Builder entity = OafEntity.newBuilder().setId(id).setType(type);
		return entity;
	}

	protected MapDocument organization(final Config config, final String id, final String legalName) {
		return organization(config, id, legalName, null);
	}

	protected MapDocument organization(final Config config, final String id, final String legalName, final String legalShortName) {
		final Organization.Metadata.Builder metadata = Organization.Metadata.newBuilder();
		if (legalName != null) {
			metadata.setLegalname(sf(legalName));
		}
		if (legalShortName != null) {
			metadata.setLegalshortname(sf(legalShortName));
		}

		final OafEntity.Builder entity = oafEntity(id, eu.dnetlib.data.proto.TypeProtos.Type.result);
		entity.setOrganization(Organization.newBuilder().setMetadata(metadata));

		return ProtoDocumentBuilder.newInstance(id, entity.build(), config.model());
	}

	private StructuredProperty sp(final String pid, final String type) {
		final Builder pidSp =
				StructuredProperty.newBuilder().setValue(pid)
						.setQualifier(Qualifier.newBuilder().setClassid(type).setClassname(type).setSchemeid("dnet:pid_types").setSchemename("dnet:pid_types"));
		return pidSp.build();
	}

	protected Field title(final String s) {
		return new FieldValueImpl(Type.String, "title", s);
	}

	protected static StructuredProperty.Builder getStruct(final String value, final Qualifier.Builder qualifier) {
		return StructuredProperty.newBuilder().setValue(value).setQualifier(qualifier);
	}

	/*
	 * protected static StringField.Builder sf(final String s) { return StringField.newBuilder().setValue(s); }
	 * 
	 * protected static Qualifier.Builder getQualifier(final String classname, final String schemename) { return
	 * Qualifier.newBuilder().setClassid(classname).setClassname(classname).setSchemeid(schemename).setSchemename(schemename); }
	 */

}