D-Net

package eu.dnetlib.resolver;

import java.util.Arrays;

import eu.dnetlib.resolver.model.CompletionStatus;

import eu.dnetlib.resolver.model.ObjectProvenance;

import eu.dnetlib.resolver.model.ResolvedObject;

import org.springframework.beans.factory.annotation.Autowired;

/**

 * Created by sandro on 9/26/16.

 */

public class NCBINResolver extends AbstractPIDResolver {

	private final static String baseUrl =

			"https://www.ncbi.nlm.nih.gov/sviewer/viewer.fcgi?db=nuccore&val=%s&page_size=5&fmt_mask=0&report=gbx&retmode=text&page=1&page_size=1";

	private final static String NCBI_NSPREFIX = "r3d100010778";

	@Autowired

	private NCBINParser NCBINParser;

	@Override

	protected boolean canResolvePid(final String pidType) {

		return pidType != null && (pidType.toLowerCase().equals("protein") || pidType.toLowerCase().equals("ncbi-n"));

	}

	@Override

	protected ResolvedObject resolve(final String pid, final String pidType) {

		if ((pid == null))

			return null;

		final ResolvedObject record = NCBINParser.parseRecord(requestURL(String.format(baseUrl, pid)));

		if (record != null) {

			record.setPid(pid);

			record.setPidType(pidType);

			record.setCompletionStatus(CompletionStatus.complete.toString());

			ObjectProvenance provenance = new ObjectProvenance();

			setDatasourceProvenace(provenance, NCBI_NSPREFIX);

			record.setDatasourceProvenance(Arrays.asList(provenance));

		}

		return record;

	}

	public NCBINParser getNCBINParser() {

		return NCBINParser;

	}

	public void setNCBINParser(final NCBINParser NCBINParser) {

		this.NCBINParser = NCBINParser;

	}

}