D-Net

package eu.dnetlib.data.claims.migration.handler;

import eu.dnetlib.data.claims.migration.JsonldBuilder;

import eu.dnetlib.data.claims.migration.entity.Claim;

import eu.dnetlib.data.claims.migration.entity.Project;

import eu.dnetlib.data.claimsDemo.QueryGenerator;

import eu.dnetlib.data.claimsDemo.SQLStoreException;

import eu.dnetlib.data.claimsDemo.SqlDAO;

import org.apache.log4j.Logger;

import java.sql.Array;

import java.sql.ResultSet;

import java.text.DateFormat;

import java.text.SimpleDateFormat;

import java.util.ArrayList;

import java.util.Date;

import java.util.List;

import java.util.UUID;

/**

 * Created by konstantina on 10/4/2017.

 */

public class FetchProjectHandler {

    private Logger log = Logger.getLogger(this.getClass());

    SqlDAO sqlDAO = null;

    QueryGenerator queryGenerator = null;

    public FetchProjectHandler(){

    }

    public String fetchProjectIdByTokenAndEmail(String token, String email) throws Exception, SQLStoreException {

        String openaireId = null;

        ArrayList<Object> params = new ArrayList<>();

        String query = queryGenerator.generateSelectProjectIdByTokenAndEmailQuery(token, email, params);

        ResultSet rs = sqlDAO.executePreparedQuery(query, params);

        if(rs.next()) {

            openaireId = rs.getString(1);

        }

        rs.close();

        return openaireId;

    }

    public String fetchProjectIdByToken(String token) throws Exception, SQLStoreException {

        String openaireId = null;

        ArrayList<Object> params = new ArrayList<>();

        String query = queryGenerator.generateSelectProjectIdByTokenQuery(token, params);

        ResultSet rs = sqlDAO.executePreparedQuery(query, params);

        if(rs.next()) {

            openaireId = rs.getString(1);

        }

        rs.close();

        return openaireId;

    }

    public Project fetchProjectById(String projectId) throws Exception, SQLStoreException {

        ArrayList<Object> params = new ArrayList<>();

        String query = queryGenerator.generateSelectProjectByIdQuery(projectId, params);

        ResultSet rs = sqlDAO.executePreparedQuery(query, params);

        Project project = fetchProjectByResultSet(rs);

        if(project == null) {

            log.info("No project with id : "+ projectId+"\n");

        }

        return project;

    }

    public List<String> fetchContactEmailsByProjectId(String projectId) throws Exception, SQLStoreException {

        ArrayList<Object> params = new ArrayList<>();

        String query = queryGenerator.generateSelectContactEmailsByProjectIdQuery(projectId, params);

        ResultSet rs = sqlDAO.executePreparedQuery(query, params);

        if(rs.next()) {

//            Array array = ((String[])rs.getArray(1));

            Array array = rs.getArray(1);

            if(array != null) {

                String[] str_array = (String[]) array.getArray();

                List<String> mails = new ArrayList<String>();

                for (int i = 0; i < str_array.length; i++) {

                    mails.add(str_array[i]);

                }

                return mails;

            }

        }

        return null;

    }

    public Project fetchProjectByResultSet(ResultSet rs) throws Exception {

        Project project = null;

        while(rs.next()) {

            project = new Project();

            project.setOpenaireId(rs.getString(1));

            project.setName(rs.getString(2));

            project.setAcronym(rs.getString(3));

            project.setFunderId(rs.getString(4));

            project.setFunderName(rs.getString(5));

            project.setFunderShortName(rs.getString(6));

            //project.setContactEmails((List<String>)rs.getArray(7));

        }

        return project;

    }

    public SqlDAO getSqlDAO() {

        return sqlDAO;

    }

    public void setSqlDAO(SqlDAO sqlDAO) {

        this.sqlDAO = sqlDAO;

    }

    public QueryGenerator getQueryGenerator() {

        return queryGenerator;

    }

    public void setQueryGenerator(QueryGenerator queryGenerator) {

        this.queryGenerator = queryGenerator;

    }

}