D-Net

package unitest;

import eu.dnetlib.goldoa.domain.*;

import eu.dnetlib.goldoa.service.AlternativeFundingBidManager;

import eu.dnetlib.goldoa.service.InvoiceManager;

import eu.dnetlib.goldoa.service.OrganizationManager;

import org.apache.commons.io.IOUtils;

import org.junit.Test;

import org.junit.runner.RunWith;

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

import org.springframework.test.annotation.Rollback;

import org.springframework.test.context.ContextConfiguration;

import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import javax.transaction.Transactional;

import java.io.*;

import java.io.File;

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.sql.Timestamp;

import java.text.ParseException;

import java.text.SimpleDateFormat;

import java.util.Date;

import java.util.List;

import static eu.dnetlib.goldoa.domain.Constants.*;

@RunWith( SpringJUnit4ClassRunner.class )

@ContextConfiguration(locations = "classpath:applicationContext-goldoa-service-test.xml")

public class AlternativeFundingBidsTest {

    @Autowired

    OrganizationManager organizationManager;

    @Autowired

    InvoiceManager invoiceManager;

    @Autowired

    AlternativeFundingBidManager alternativeFundingBidManager;

    @Test

    @Transactional

    @Rollback(true)

    public void importAlternativeFundingBids(){

        alternativeFundingBidManager.importAlternativeFundingBids();

        /*String path = "/home/panagiotis/Desktop/uploadFiles/";

        BufferedReader br = null;

        String s = "";

        String cvsSplitBy = ",";

        AlternativeFundingBid bid = null;

        try {

            br = new BufferedReader(new FileReader(path + "Alternative Funding Bids - First round.csv"));

            s = br.readLine();

            while ((s = br.readLine()) != null) {

                System.out.println("Line -> " + s);

                String[] line = s.split(cvsSplitBy);

                String id = !line[0].equals("")?line[0].trim():null;

                String funding_name = !line[1].equals("")?line[1].trim():null;

                String organization = !line[2].equals("")?line[2].trim():null;

                String country = !line[3].equals("")?line[3].trim():null;

                String totalAmount = !line[4].equals("")?line[4].trim():null;

                String contractStartDate = !line[5].equals("")?line[5].trim():null;

                String contractEndDate = !line[6].equals("")?line[6].trim():null;

                String firstInvoiceDate = !line[7].equals("")?line[7].trim():null;

                String firstInvoiceNumber = !line[8].equals("")?line[8].trim():null;

                String secondInvoiceDate = !line[9].equals("")?line[9].trim():null;

                String secondInvoiceNumber = !line[10].equals("")?line[10].trim():null;

                String firstPaymentAmount = !line[11].equals("")?line[11].trim():null;

                String firstPaymentDate = !line[12].equals("")?line[12].trim():null;

                String firstPaymentOtherCosts = !line[13].equals("")?line[13].trim():null;

                String firstPaymentTransferCost = !line[14].equals("")?line[14].trim():null;

                String secondPaymentAmount = !line[15].equals("")?line[15].trim():null;

                String secondPaymentDate = !line[16].equals("")?line[16].trim():null;

                String secondPaymentOtherCosts = !line[17].equals("")?line[17].trim():null;

                String secondPaymentTransferCost = !line[18].equals("")?line[18].trim():null;

                List<Object> orgs =  organizationManager.search(organization);

                if(orgs == null)

                    System.out.println("Organization " + organization +  "  not exists!");

                Organization org = (Organization) organizationManager.search(organization).get(0);

                bid = this.initializeBid(funding_name,id,country,totalAmount,contractStartDate,contractEndDate);

                System.out.println("Bid initialized!");

                bid.setFirstInvoice(this.uploadInvoiceFile(path,funding_name,FIRST_INVOICE_PDF,

                        firstInvoiceDate,firstInvoiceNumber));

                System.out.println("First invoice ready!");

                bid.setSecondInvoice(this.uploadInvoiceFile(path,funding_name,SECOND_INVOICE_PDF,

                        secondInvoiceDate,secondInvoiceNumber));

                System.out.println("Second invoice ready!");

                bid.setFirstPaymentDocument(this.uploadReceiptFile(path,funding_name,FIRST_PAYMENT_PDF,

                        firstPaymentAmount,firstPaymentDate,firstPaymentOtherCosts,firstPaymentTransferCost));

                System.out.println("First payment ready!");

                bid.setSecondPaymentDocument(this.uploadReceiptFile(path,funding_name,SECOND_PAYMENT_PDF,

                        secondPaymentAmount,secondPaymentDate,secondPaymentOtherCosts,secondPaymentTransferCost));

                System.out.println("Second payment ready!");

                bid.setContract(this.uploadContract(path,funding_name,CONTRACT_PDF));

                System.out.println("Contract ready!");

                //alternativeFundingBidManager.save(bid);

            }

        } catch (IOException | ParseException e) {

            e.printStackTrace();

        }*/

    }

    private eu.dnetlib.goldoa.domain.File uploadContract(String basePath, String funding_name, String contract) {

        String fullPath = basePath + funding_name + "/" + contract;

        File file = new File(basePath + funding_name + "/" + contract);

        InputStream inputStream = null;

        Path source = Paths.get(fullPath);

        String contentType = null;

        eu.dnetlib.goldoa.domain.File contractFile = null;

        try {

            inputStream = new FileInputStream(file);

            contentType = Files.probeContentType(source);

            final ByteArrayOutputStream baos = new ByteArrayOutputStream();

            IOUtils.copy(inputStream, baos);

            IOUtils.closeQuietly(baos);

            contractFile = new eu.dnetlib.goldoa.domain.File();

            contractFile.setId(invoiceManager.getFileId());

            contractFile.setMimetype(contentType);

            contractFile.setFile(baos.toByteArray());

        } catch (IOException e) {

            e.printStackTrace();

        }

        return contractFile;

    }

    private BankTransferReceipt uploadReceiptFile(String basePath, String funding_name, String receipt,

                                                  String paymentAmount, String paymentDate,

                                                  String paymentOtherCosts, String paymentTransferCost) throws ParseException {

        String fullPath = basePath + funding_name + "/" + receipt;

        File file = new File(basePath + funding_name + "/" + receipt);

        InputStream inputStream = null;

        Path source = Paths.get(fullPath);

        String contentType = null;

        Invoice invoice = null;

        BankTransferReceipt bankTransferReceipt = null;

        try {

            inputStream = new FileInputStream(file);

            contentType = Files.probeContentType(source);

            final ByteArrayOutputStream baos = new ByteArrayOutputStream();

            IOUtils.copy(inputStream, baos);

            IOUtils.closeQuietly(baos);

            bankTransferReceipt = new BankTransferReceipt();

            bankTransferReceipt.setContent(baos.toByteArray());

            bankTransferReceipt.setContentType(contentType);

            bankTransferReceipt.setAmount(Double.valueOf(paymentAmount));

            Date parsedDate = new SimpleDateFormat("yyyy-MM-dd").parse(paymentDate);

            Timestamp st = new java.sql.Timestamp(parsedDate.getTime());

            bankTransferReceipt.setDate(st);

            bankTransferReceipt.setOtherCosts(Double.valueOf(paymentOtherCosts));

            bankTransferReceipt.setTransferCost(Double.valueOf(paymentTransferCost));

        } catch (IOException e) {

            e.printStackTrace();

        }

        return bankTransferReceipt;

    }

    private AlternativeFundingBid initializeBid(String funding_name, String id, String country,

                                                String totalAmount, String contractStartDate,

                                                String contractEndDate) {

        AlternativeFundingBid bid = new AlternativeFundingBid();

        bid.setBidName(funding_name);

        bid.setId(id);

        bid.setCountry(alternativeFundingBidManager.getCountry(country));

        bid.setBidNo(Integer.valueOf(id));

        bid.setTotalAmountRequested(Double.parseDouble(totalAmount));

        bid.setFirstPayment(Double.parseDouble(totalAmount)/2);

        bid.setSecondPayment(Double.parseDouble(totalAmount)/2);

        bid.setRound(FIRST_ROUND);

        try {

            if(contractStartDate != null){

                Date parsedStartDate = new SimpleDateFormat("yyyy-MM-dd").parse(contractStartDate);

                Timestamp start = new java.sql.Timestamp(parsedStartDate.getTime());

                bid.setContractStartDate(start);

            }

            if(contractEndDate != null){

                Date parsedEndDate = new SimpleDateFormat("yyyy-MM-dd").parse(contractEndDate);

                Timestamp end = new java.sql.Timestamp(parsedEndDate.getTime());

                bid.setContractEndDate(end);

            }

        } catch (ParseException e) {

            System.out.println("Start or end date null! ");

        }

        return bid;

    }

    private Invoice uploadInvoiceFile(String basePath, String funding_name, String invoicePdf,

                                      String invoiceDate, String invoiceNumber) throws ParseException {

        String fullPath = basePath + funding_name + "/" + invoicePdf;

        File file = new File(basePath + funding_name + "/" + invoicePdf);

        InputStream inputStream = null;

        Path source = Paths.get(fullPath);

        String contentType = null;

        Invoice invoice = null;

        try {

            inputStream = new FileInputStream(file);

            contentType = Files.probeContentType(source);

            invoice = new Invoice();

            invoice.setSource("portal");

            invoice.setNumber(invoiceNumber);

            Date parsedStartDate = new SimpleDateFormat("yyyy-MM-dd").parse(invoiceDate);

            Timestamp st = new java.sql.Timestamp(parsedStartDate.getTime());

            invoice.setDate(st);

            invoice = invoiceManager.saveInvoice(invoice);

            invoice = invoiceManager.uploadInvoice(invoice.getId(), contentType, inputStream);

            invoiceManager.saveInvoice(invoice);

        } catch (IOException | ManagerException e) {

            e.printStackTrace();

        }

        return invoice;

    }

}