D-Net

package eu.dnetlib.resolver.parser;

import java.util.Arrays;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.xml.stream.XMLStreamReader;

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

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

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

/**

 * Created by sandro on 9/13/16.

 */

public abstract class AbstractResolverParser {

	protected static final Log log = LogFactory.getLog(AbstractResolverParser.class);

    private List<String> datasetSubTypes = Arrays.asList("dataset", "software", "film", "sound", "physicalobject", "audiovisual", "collection", "other", "study", "metadata");

    public abstract DLIResolvedObject parseObject(final String record);

	protected Map<String, String> getAttributes(final XMLStreamReader parser) {

		final Map<String, String> attributesMap = new HashMap<>();

		for (int i = 0; i < parser.getAttributeCount(); i++) {

			attributesMap.put(parser.getAttributeLocalName(i), parser.getAttributeValue(i));

		}

		return attributesMap;

	}

    protected void setType(final DLIResolvedObject object, final String type) {

        if (!StringUtils.isBlank(type)) {

            if (datasetSubTypes.contains(type.toLowerCase())) {

                object.setType(ObjectType.dataset);

				return;

			} else if (type.toLowerCase().contains("publication")) {

				object.setType(ObjectType.publication);

				return;

			} else {

				object.setType(ObjectType.unknown);

			}

		}

	}

}