D-Net

package eu.dnetlib.resolver.parser;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import com.ximpleware.AutoPilot;

import com.ximpleware.VTDGen;

import com.ximpleware.VTDNav;

import eu.dnetlib.data.transform.VtdUtilityParser;

import eu.dnetlib.dli.resolver.model.*;

import eu.dnetlib.pid.resolver.model.ObjectProvenance;

import eu.dnetlib.pid.resolver.model.ObjectRelation;

import eu.dnetlib.pid.resolver.model.ObjectType;

import eu.dnetlib.pid.resolver.model.PID;

import eu.dnetlib.pid.resolver.parser.AbstractResolverParser;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

/**

 * Created by sandro on 9/29/16.

 */

public class ScholixResolverParser extends AbstractResolverParser {

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

	@Override

    public DLIResolvedObject parseObject(final String record) {

		try {

            final DLIResolvedObject parsedObject = new DLIResolvedObject();

            final VTDGen vg = new VTDGen();

			vg.setDoc(record.getBytes());

			vg.parse(true);

			final VTDNav vn = vg.getNav();

			final AutoPilot ap = new AutoPilot(vn);

            ap.declareXPathNameSpace("dri", "http://www.driver-repository.eu/namespace/dri");

            final String resolvedDate = VtdUtilityParser.getSingleValue(ap, vn, "//dri:resolvedDate");

            parsedObject.setResolvedDate(resolvedDate);

            //final String datasourcePrefix = VtdUtilityParser.getSingleValue(ap, vn, "//dri:datasourceprefix");

            ap.declareXPathNameSpace("oaf", "http://namespace.dnet.eu/oaf");

			final String completionStatus = VtdUtilityParser.getSingleValue(ap, vn, "//oaf:completionStatus");

            //final String provisionMode = VtdUtilityParser.getSingleValue(ap, vn, "//oaf:provisionMode");

			List<VtdUtilityParser.Node> provenances =

					VtdUtilityParser.getTextValuesWithAttributes(ap, vn, "//oaf:collectedFrom", Arrays.asList("completionStatus", "id", "name"));

            ap.declareXPathNameSpace("scholix", "http://www.scholix.org");

            final String sourcePid = VtdUtilityParser.getSingleValue(ap, vn, "//scholix:source/identifier/identifier");

            final String sourcePidTye = VtdUtilityParser.getSingleValue(ap, vn, "//scholix:source/identifier/schema");

            final String sourceType = VtdUtilityParser.getSingleValue(ap, vn, "//scholix:source/objectType/type");

            final String sourcePublisherId = VtdUtilityParser.getSingleValue(ap, vn, "//scholix:source/publisher//identifier");

            final String sourcePublisher = VtdUtilityParser.getSingleValue(ap, vn, "//scholix:source/publisher/name");

            if (StringUtils.isEmpty(sourcePid) || StringUtils.isEmpty(sourcePidTye)) {

                return null;

            }

            final List<ObjectProvenance> provenanceList = new ArrayList<>();

            if (provenances != null) {

				provenances.forEach(it -> {

                    final DLIObjectProvenance provenance = new DLIObjectProvenance();

                    provenance.setDatasourceId(it.getAttributes().get("id"));

					provenance.setDatasource(it.getAttributes().get("name"));

					provenance.setCompletionStatus(it.getAttributes().get("completionStatus"));

                    provenance.setPublisher(sourcePublisher);

                    provenance.setPublisherId(sourcePublisherId);

                    provenance.setProvisionMode("collected");

					provenanceList.add(provenance);

				});

			}

			parsedObject.setDatasourceProvenance(provenanceList);

			parsedObject.setCompletionStatus(completionStatus);

            parsedObject.setPid(sourcePid);

            parsedObject.setPidType(sourcePidTye);

            if (!StringUtils.isEmpty(sourceType))

                parsedObject.setType(ObjectType.valueOf(sourceType));

            final String sourceTitle = VtdUtilityParser.getSingleValue(ap, vn, "//scholix:source/title");

            final List<String> creators = VtdUtilityParser.getTextValue(ap, vn, "//scholix:source//scholix:creatorName");

            if (!StringUtils.isEmpty(sourceTitle))

                parsedObject.setTitles(Arrays.asList(sourceTitle));

            if (creators != null) {

                final List<String> sourceCreators = new ArrayList<>();

                creators.forEach(

                        agent -> {

                            if (!StringUtils.isEmpty(agent)) {

                                sourceCreators.add(agent);

                            }

                        }

                );

                parsedObject.setAuthors(sourceCreators);

            }

            parsedObject.setDate(VtdUtilityParser.getSingleValue(ap, vn, "//scholix:link/scholix:publicationDate"));

            final String targetPid = VtdUtilityParser.getSingleValue(ap, vn, "//scholix:target/identifier/identifier");

            final String targetPidTye = VtdUtilityParser.getSingleValue(ap, vn, "//scholix:target/identifier/schema");

            final String targetType = VtdUtilityParser.getSingleValue(ap, vn, "//scholix:target/objectType/type");

            if (StringUtils.isEmpty(targetPid) || StringUtils.isEmpty(targetPidTye)) {

                return null;

			}

            final String relationType = VtdUtilityParser.getSingleValue(ap, vn, "//scholix:relashionship/name");

            final String inverseRelationSemantic = VtdUtilityParser.getSingleValue(ap, vn, "//scholix:relashionship/inverseRelashionship");

            final List<ObjectRelation> relations = new ArrayList<>();

            DLIObjectRelation relation = new DLIObjectRelation();

            relation.setCompletionStatus(CompletionStatus.incomplete.toString());

			relation.setRelationProvenance(parsedObject.getDatasourceProvenance());

            relation.setRelationSemantics(relationType);

            relation.setInverseRelation(inverseRelationSemantic);

            relation.setTargetPID(new PID(targetPid, targetPidTye));

            if (!StringUtils.isEmpty(targetType))

                relation.setTargetType(ObjectType.valueOf(targetType));

            relations.add(relation);

			parsedObject.setRelations(relations);

			return parsedObject;

		} catch (Throwable e) {

			log.error("Error on parsing parser");

		}

		return null;

	}

}