D-Net

package eu.dnetlib.data.utils;

import java.io.IOException;

import java.io.StringWriter;

import java.io.Writer;

import java.util.Collection;

import java.util.stream.Collectors;

import org.apache.commons.lang3.StringUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.jbibtex.BibTeXDatabase;

import org.jbibtex.BibTeXEntry;

import org.jbibtex.BibTeXFormatter;

import org.jbibtex.Key;

import org.jbibtex.StringValue;

import org.jbibtex.StringValue.Style;

import eu.dnetlib.data.mdstore.plugins.objects.MdRecord;

public class BibTexConverter {

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

	public static String asBibTex(final MdRecord record) {

		try {

			final BibTeXEntry dbEntry = new BibTeXEntry(findBibTexType(record.getType()), new Key(record.getId()));

			addSimpleStringValue(dbEntry, BibTeXEntry.KEY_TITLE, record.getTitle());

			addMultipleStringValue(dbEntry, BibTeXEntry.KEY_AUTHOR, record.getCreators());

			addSimpleStringValue(dbEntry, BibTeXEntry.KEY_PUBLISHER, record.getPublisher());

			addSimpleStringValue(dbEntry, BibTeXEntry.KEY_YEAR, record.getDate());

			addMultipleStringValue(dbEntry, BibTeXEntry.KEY_DOI, record.getDois());

			final BibTeXFormatter formatter = new BibTeXFormatter();

			final Writer res = new StringWriter();

			final BibTeXDatabase db = new BibTeXDatabase();

			db.addObject(dbEntry);

			formatter.format(db, res);

			return res.toString();

		} catch (final IOException e) {

			log.warn("Error generating bibtex", e);

			return "";

		}

	}

	private static void addSimpleStringValue(final BibTeXEntry dbEntry, final Key key, final Object value) {

		if ((value != null) && StringUtils.isNotBlank(value.toString())) {

			dbEntry.addField(key, new StringValue(value.toString(), Style.BRACED));

		}

	}

	private static void addMultipleStringValue(final BibTeXEntry dbEntry, final Key key, final Collection<String> list) {

		final String val = list.stream()

				.map(String::trim)

				.filter(StringUtils::isNotBlank)

				.collect(Collectors.joining(" and "));

		if (StringUtils.isNotBlank(val)) {

			dbEntry.addField(key, new StringValue(val, Style.BRACED));

		}

	}

	private static Key findBibTexType(final String resourceType) {

		switch (resourceType) {

		case "Conference object":

			return BibTeXEntry.TYPE_INPROCEEDINGS;

		case "Report":

			return BibTeXEntry.TYPE_TECHREPORT;

		case "Article":

			return BibTeXEntry.TYPE_ARTICLE;

		case "Other":

			return BibTeXEntry.TYPE_MISC;

		case "Part of book or chapter of book":

			return BibTeXEntry.TYPE_INBOOK;

		case "Book":

			return BibTeXEntry.TYPE_BOOK;

		case "Bachelor thesis":

			return BibTeXEntry.TYPE_MASTERSTHESIS;

		case "Doctoral thesis":

			return BibTeXEntry.TYPE_PHDTHESIS;

		case "Patent":

			return BibTeXEntry.TYPE_MISC;

		case "Master thesis":

			return BibTeXEntry.TYPE_MASTERSTHESIS;

		default:

			return BibTeXEntry.TYPE_MISC;

		}

	}

}