package eu.dnetlib.pace;

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

import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.RandomStringUtils;
import org.apache.commons.lang.StringUtils;

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.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.DynConf;
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;

public abstract class AbstractProtoPaceTest extends OafTest {

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

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

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

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

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

	private String readFromClasspath(final String filename) {
		StringWriter sw = new StringWriter();
		try {
			IOUtils.copy(getClass().getResourceAsStream(filename), sw);
			return sw.toString();
		} catch (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, null, null);
	}

	protected MapDocument result(final Config config, final String id, final String title, final String date) {
		return result(config, id, title, date, null, 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) {
		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));
		}

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

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

		entity.setResult(result);

		if (!StringUtils.isBlank(pid)) {
			entity.addPid(sp(pid, "doi"));
			entity.addPid(sp(RandomStringUtils.randomAlphabetic(10), "oai"));
		}

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

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

		eu.dnetlib.pace.model.Person p = new eu.dnetlib.pace.model.Person(author, false);
		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) {
		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) {
		Organization.Metadata.Builder metadata = Organization.Metadata.newBuilder();
		if (legalName != null) {
			metadata.setLegalname(sf(legalName));
		}
		if (legalShortName != null) {
			metadata.setLegalshortname(sf(legalShortName));
		}

		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.fields());
	}

	private StructuredProperty sp(final String pid, final String type) {
		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); }
	 */

}