D-Net

package eu.dnetlib.functionality.modular.ui.patcheditor.record;

import com.mongodb.client.MongoCollection;

import eu.dnetlib.data.mdstore.modular.connector.MDStoreManagerInfo;

import eu.dnetlib.data.mdstore.modular.mongodb.MDStoreTransactionManagerImpl;

import eu.dnetlib.enabling.locators.UniqueServiceLocator;

import eu.dnetlib.functionality.modular.ui.patcheditor.exceptions.PatchEditorException;

import eu.dnetlib.miscutils.datetime.DateUtils;

import eu.dnetlib.rmi.data.MDStoreServiceException;

import eu.dnetlib.rmi.enabling.ISLookUpException;

import eu.dnetlib.rmi.enabling.ISLookUpService;

import java.io.StringReader;

import java.util.List;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.bson.Document;

import org.dom4j.Element;

import org.dom4j.io.SAXReader;

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

public class PatchManagerImpl implements PatchManager {

	@Autowired

	private UniqueServiceLocator serviceLocator;

	@Autowired

	private MDStoreTransactionManagerImpl mdstoreTransactionManager;

	private final SAXReader reader = new SAXReader();

	private static final Log log = LogFactory.getLog(PatchManager.class); // NOPMD by marko on 11/24/08 5:02 PM

	@Override

	public void commit(PatchedRecord record) throws PatchEditorException {

		try {

			final String patchMdStoreId = getPatchMDStoreId(record.getRepositoryId());

			final MongoCollection<Document> collection = getPatchMdStoreMongoCollection(patchMdStoreId);

			final Document mongoQuery = new Document().append("id", record.getId());

			final Document store_patch = collection.find(mongoQuery).first();

			if (store_patch == null) {

				final Document patched_document = mongoQuery

						.append("body", record.asXML())

						.append("timestamp", System.currentTimeMillis());

				collection.insertOne(patched_document);

			} else {

				final String oldPatch = store_patch.getString("body");

				collection.findOneAndUpdate(mongoQuery, new Document("$set", new Document("body", mergePatches(record.asXML(), oldPatch))));

			}

			log.info("patchMdStoreId" + patchMdStoreId);

		} catch (final Throwable e) {

			throw new PatchEditorException(e);

		}

	}

	public String mergePatches(String newPatch, final String oldPatch) {

		try {

			final org.dom4j.Document newDoc = reader.read(new StringReader(newPatch));

			final org.dom4j.Document oldDoc = reader.read(new StringReader(oldPatch));

			final Element e = (Element) oldDoc.selectSingleNode("//patches");

			for (final Object o : newDoc.selectNodes("//patch")) {

				e.add(((Element) o).createCopy());

			}

			oldDoc.selectSingleNode("//*[local-name()='dateOfCollection']").setText(DateUtils.now_ISO8601());

			return oldDoc.asXML();

		} catch (final Exception e) {

			return newPatch;

		}

	}

	private String getPatchMDStoreId(final String repositoryId) throws ISLookUpException {

		final String getPatchMdstoreIdQuery = "for $x in collection('/db/DRIVER/WorkflowDSResources/WorkflowDSResourceType') where "

				+ "($x//DATASOURCE/@id/string()='%s') return $x//PARAM[./@name='patchMdstoreId']/text()";

		final ISLookUpService lookUpClient = serviceLocator.getService(ISLookUpService.class);

		final List<String> patchMdStoreIds = lookUpClient.quickSearchProfile(String.format(getPatchMdstoreIdQuery, repositoryId));

		if (patchMdStoreIds == null || patchMdStoreIds.size() != 1) { throw new RuntimeException(

				"Unexpected number of result executing query " + String.format(getPatchMdstoreIdQuery, patchMdStoreIds.get(0)) + " expected: 1"); }

		return patchMdStoreIds.get(0);

	}

	private MongoCollection<Document> getPatchMdStoreMongoCollection(final String mdStoreId) throws MDStoreServiceException {

		final MDStoreManagerInfo info = mdstoreTransactionManager.getInfoForCurrentMdStore(mdStoreId);

		return mdstoreTransactionManager.getDb().getCollection(info.getCurrentId());

	}

}