D-Net

	package eu.dnetlib.openaire.thrift;

import eu.dnetlib.actionmanager.common.Agent;

import eu.dnetlib.actionmanager.common.Agent.AGENT_TYPE;

import eu.dnetlib.actionmanager.common.Operation;

import eu.dnetlib.actionmanager.common.Provenance;

import eu.dnetlib.actionmanager.set.ActionManagerSet;

import eu.dnetlib.api.enabling.ActionManagerService;

import eu.dnetlib.api.enabling.ISLookUpService;

import eu.dnetlib.api.enabling.ISLookUpServiceException;

import eu.dnetlib.openaire.action.ActionUtils;

import gr.uoa.di.driver.enabling.ISLookUpException;

import gr.uoa.di.driver.util.ServiceLocator;

import java.io.FileWriter;

import java.sql.Connection;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.util.ArrayList;

import java.util.Date;

import java.util.List;

import javax.sql.DataSource;

import org.apache.log4j.Logger;

import org.apache.velocity.exception.ParseErrorException;

import org.apache.velocity.exception.ResourceNotFoundException;

import org.springframework.transaction.annotation.Transactional;

import org.springframework.jdbc.datasource.DataSourceUtils;

public class ClaimConnector {

	private static Logger logger = Logger.getLogger(ClaimConnector.class);

	private DataSource datasource = null;

	private ServiceLocator<ActionManagerService> actionManagerLocator = null;

	//Using an isLookUp Impl just to make an xQuery to IS. Maybe move all the 

	//functionality to commons?

	private ServiceLocator<ISLookUpService> lookupServiceLocator = null;   

	public List<Claim> getClaimedPublications(final String user) throws SQLException {

		Connection con = DataSourceUtils.getConnection(datasource);

		PreparedStatement stmt = con.prepareStatement("select \"resultid\", \"date\" , \"type\", \"xml\" from claims where \"agent\" = ?");

		stmt.setString(1, user);

		ResultSet rs = stmt.executeQuery();

		List<Claim> res = new ArrayList<Claim>();

		while (rs.next()) {

            res.add(new Claim(rs.getString(1), rs.getDate(2).getTime(), rs.getString(3), rs.getString(4)));

		}

        rs.close();

        stmt.close();

	con.close();

		return res;

	}

    public List<Claim> getClaimedPublications(Date from, Date to) throws SQLException {

	logger.info("Returning all claimed publications from " + from + " to " + to);

        Connection con = DataSourceUtils.getConnection(datasource);

        PreparedStatement stmt = con.prepareStatement("select \"resultid\", \"date\" , \"type\", \"xml\" from claims where \"agent\" != 'inference' and \"date\" >= ?::date and \"date\" <= ?::date ");

        stmt.setDate(1, new java.sql.Date(from.getTime()));

        stmt.setDate(2, new java.sql.Date(to.getTime()));

        ResultSet rs = stmt.executeQuery();

        List<Claim> res = new ArrayList<Claim>();

        while (rs.next()) {

            res.add(new Claim(rs.getString(1), rs.getDate(2).getTime(), rs.getString(3), rs.getString(4)));

        }

        rs.close();

        stmt.close();

	con.close();

	logger.debug("number of claims: " + res.size());

        return res;

    }

    public void deleteClaim(String userEmail, String resultId) throws SQLException {

        Connection con = DataSourceUtils.getConnection(datasource);

        PreparedStatement stmt = con.prepareStatement("delete from claims where \"agent\" = ? and resultid =?");

        stmt.setString(1, userEmail);

        stmt.setString(2, resultId);

        stmt.executeUpdate();

        stmt.close();

	con.close();

    }

    /**

     *

     * @param id {openaireId, publication doi, ORCIDId (personId-ORCIDPublicationId), data doi}

     * @param source {openaire, doi (crossref), orcid, datacite}

     * @param title

     * @param description

     * @param access_mode

     * @param embargoEndDate

     * @param authors

     * @param url

     * @param dcSource

     * @param userName

     * @param userEmail

     * @param projects

     * @param subjects

     * @param publicationDate

     * @param publisher

     * @param language

     * @param category

     * @param doi

     * @throws ResourceNotFoundException

     * @throws ParseErrorException

     * @throws Exception

     */

	@Transactional

	public void insertAndClaimDocument(

			String id, 

			String source, 

			String title, 

			String description, 

			String access_mode, 

			String embargoEndDate, 

			List<Author> authors, 

			String url,

			String dcSource,

			String userName, 

			String userEmail, 

			List<Project> projects,

			List<String> subjects,

            List<String> concepts,

			String publicationDate,

			String publisher, 

			String language, 

			String category, 

			String doi) throws Exception {

		Connection con = DataSourceUtils.getConnection(datasource);

		PreparedStatement stmt = null;

		String dmf = null;

		String resultId = ActionUtils.getIdentifier(ActionUtils.getPrefix(source), id);

		Date now = new Date();

		if (!source.toLowerCase().equals("openaire") && title != null) {

			stmt = con.prepareStatement("insert into claims (\"agent\", \"resultid\", \"date\", type, xml, set, provenance, nsprefix) values (?, ?, ?, ?, ?, ?, ?, ?)");

			dmf = ActionUtils.createDMF(id, source, title, description, access_mode, embargoEndDate, authors, url, dcSource, doi, new ArrayList<Project>(), subjects, concepts, publicationDate, publisher, language, category);

			stmt.setString(1, userEmail);

			stmt.setString(2, resultId);

			stmt.setDate(3, new java.sql.Date(now.getTime()));

			stmt.setString(4, "dmf2actions");

			stmt.setString(5, dmf);

            stmt.setString(6, ActionUtils.getSet(source, "dmf"));

            stmt.setString(7, ActionUtils.getProvenance(source));

            stmt.setString(8, ActionUtils.getPrefix(source));

			stmt.executeUpdate();

			stmt.close();

		}

		for (Project project:projects) {

			String xml = ActionUtils.createResultProjectRelation("openaire".equals(source)?null:source, id, project.getType(), project.getProjectId());

			stmt = con.prepareStatement("insert into claims (\"agent\", \"resultid\", \"date\", type, xml, set, provenance, nsprefix) values (?, ?, ?, ?, ?, ?, ?, ?)");

			stmt.setString(1, userEmail);

			stmt.setString(2, resultId);

			stmt.setDate(3, new java.sql.Date(now.getTime()));

			stmt.setString(4, "rels2actions");

			stmt.setString(5, xml);

            stmt.setString(6, ActionUtils.getSet(source, "relation"));

            stmt.setString(7, ActionUtils.getProvenance(source));

            stmt.setString(8, ActionUtils.getPrefix(source));

			stmt.executeUpdate();

			stmt.close();

		}

		con.close();

	}	

	@Deprecated

	public void insertInferredDocument(

			String id, 

			String source, 

			String title,

			String description, 

			String access_mode, 

			String embargoEndDate,

			List<Author> authors, 

			String url, 

			String dcSource,

			List<Project> projects,

			List<String> subjects,

            List<String> concepts,

			String publicationDate, 

			String publisher, 

			String language, 

			String category, 

			String doi,

            String setName) throws Exception {

        System.out.println("Adding doc " + id);

        try {

            actionManagerLocator.getService().createSet(new ActionManagerSet(setName, setName));

        } catch (Exception e) {

            if (e.getMessage() != null && e.getMessage().contains("already registered")) {

                logger.debug("Set already exists");

            } else {

                throw e;

            }

        }

        if (title != null) {

            String dmf = ActionUtils.createDMF(id, source, title, description, access_mode, embargoEndDate, authors, url, dcSource, doi, projects, subjects, new ArrayList<String>(), publicationDate, publisher, language, category);

            FileWriter fw = new FileWriter("/tmp/concepts/" + id + "-insert.xml");

            fw.append(dmf);

            fw.close();

           actionManagerLocator.getService().createAction("dmf2actions", setName, new Agent("inference", "inference",  AGENT_TYPE.service), Operation.INSERT, dmf, Provenance.sysimport_mining_repository, "0.9", ActionUtils.getPrefix(source));

        }

        String dmf = ActionUtils.createDMF(id, source, null, null, null, null, new ArrayList<Author>(), null, null, null, new ArrayList<Project>(), new ArrayList<String>(), concepts, null, null, null, null);

        FileWriter fw = new FileWriter("/tmp/concepts/" + id + "-update.xml");

        fw.append(dmf);

        fw.close();

        actionManagerLocator.getService().createAction("dmf2updateActions", setName, new Agent("inference", "inference",  AGENT_TYPE.service), Operation.INSERT, dmf, Provenance.sysimport_mining_repository, "0.9", ActionUtils.getPrefix(source));

	}

	@Deprecated

	public void insertInferredRelation(String resultSource, String resultId,

			String projectType, String projectId, String setName) throws Exception {

        String relation = ActionUtils.createResultProjectRelation(resultSource, resultId, projectType, projectId);

		try {

			actionManagerLocator.getService().createSet(new ActionManagerSet(setName, setName));

		} catch (Exception e) {

			logger.warn("Set already exists");

		}

		//actionManagerLocator.getService().createAction("rels2actions", setName, new Agent("inference", "inference",  AGENT_TYPE.service), Operation.INSERT, relation, Provenance.sysimport_mining_repository, "0.9", "openaire____");

	}

	public ServiceLocator<ActionManagerService> getActionManagerLocator() {

		return actionManagerLocator;

	}

	public void setActionManagerLocator(

			ServiceLocator<ActionManagerService> actionManagerLocator) {

		this.actionManagerLocator = actionManagerLocator;

	}

	public DataSource getDatasource() {

		return datasource;

	}

	public void setDatasource(DataSource datasource) {

		this.datasource = datasource;

	}

	public ServiceLocator<ISLookUpService> getLookupServiceLocator() {

		return lookupServiceLocator;

	}

	public void setLookupServiceLocator(

			ServiceLocator<ISLookUpService> lookupServiceLocator) {

		this.lookupServiceLocator = lookupServiceLocator;

	}

	public List<String> getContexts() throws ISLookUpException, ISLookUpServiceException {

		return lookupServiceLocator.getService().quickSearchProfile("for $x in collection('/db/DRIVER/ContextDSResources/ContextDSResourceType') return $x");

	}

}