D-Net

package eu.dnetlib.server;

import eu.dnetlib.goldoa.domain.Invoice;

import eu.dnetlib.goldoa.domain.InvoiceFile;

import eu.dnetlib.goldoa.domain.ManagerException;

import eu.dnetlib.goldoa.service.EligibilityManager;

import eu.dnetlib.goldoa.service.InvoiceManager;

import org.apache.commons.fileupload.FileItem;

import org.apache.commons.fileupload.FileItemFactory;

import org.apache.commons.fileupload.FileItemIterator;

import org.apache.commons.fileupload.FileItemStream;

import org.apache.commons.fileupload.disk.DiskFileItemFactory;

import org.apache.commons.fileupload.servlet.ServletFileUpload;

import org.apache.commons.io.IOUtils;

import org.springframework.context.ApplicationContext;

import org.springframework.web.context.support.WebApplicationContextUtils;

import javax.servlet.ServletConfig;

import javax.servlet.ServletException;

import javax.servlet.http.HttpServlet;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import java.io.*;

import java.text.SimpleDateFormat;

import java.util.Date;

import java.util.List;

/**

 * Created by stefania on 3/19/15.

 */

public class FileUpload extends HttpServlet {

    private InvoiceManager invoiceManager;

    @Override

    public void init(ServletConfig config) throws ServletException {

        super.init(config);

        ApplicationContext context = WebApplicationContextUtils

                .getWebApplicationContext(getServletContext());

        this.invoiceManager = (InvoiceManager) context.getBean("invoiceManager");

    }

    @Override

    public void doPost(HttpServletRequest request, HttpServletResponse response)  throws ServletException, IOException {

        try {

            FileItemFactory factory = new DiskFileItemFactory();

            ServletFileUpload upload = new ServletFileUpload(factory);

            List<FileItem> items = upload.parseRequest(request);

            Invoice invoice = new Invoice();

            invoice.setSource("portal");

            invoice = invoiceManager.saveInvoice(invoice);

            for (FileItem item:items) {

                if (item.isFormField()) {

                    String fieldName = item.getFieldName();

                    String value = item.getString();

                    if (fieldName.equals("invoiceNumber")) {

                        invoice.setNumber(value);

                    } else if (fieldName.equals("dateIssued")) {

                        invoice.setDate(new SimpleDateFormat("yyyy/MM/dd").parse(value));

                    }

                } else {

                    String contentType = item.getContentType();

                    invoiceManager.uploadInvoice(invoice.getId(), contentType, item.getInputStream());

                }

            }

            invoiceManager.saveInvoice(invoice);

            response.setContentType("text/plain");

            response.setContentLength(invoice.getId().length());

            response.getOutputStream().print(invoice.getId());

        }

        catch(Exception e){

            throw new RuntimeException(e);

        }

    }

    @Override

    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {

        String invoiceId = req.getParameter("invoiceId");

        try {

            InvoiceFile file = invoiceManager.downloadInvoice(invoiceId);

            resp.setContentType(file.getMimetype());

            resp.setContentLength(file.getFile().length);

            IOUtils.copy(new ByteArrayInputStream(file.getFile()), resp.getOutputStream());

        } catch (ManagerException e) {

            switch (e.getErrorCause()) {

                case NOT_EXISTS:

                    resp.setStatus(404);

                    resp.getOutputStream().println("Invoice with id " + invoiceId + " not found");

                    break;

                case UNKNOWN:

                    resp.setStatus(500);

                    resp.getOutputStream().println("Error getting invoice with id " + invoiceId);

                    break;

            }

        }

    }

}