D-Net

package eu.dnetlib.server;

import eu.dnetlib.goldoa.domain.ManagerException;

import eu.dnetlib.goldoa.service.EligibilityManager;

import eu.dnetlib.goldoa.service.InvoiceManager;

import org.apache.commons.fileupload.FileItemFactory;

import org.apache.commons.fileupload.FileItemStream;

import org.apache.commons.io.IOUtils;

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

import javax.servlet.http.HttpServlet;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import java.util.Date;

import java.util.List;

/**

 * Created by stefania on 3/19/15.

 */

    @Override

    public void init(ServletConfig config) throws ServletException {

        super.init(config);

        ApplicationContext context = WebApplicationContextUtils

                .getWebApplicationContext(getServletContext());

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

    }

    @Override

            FileItemFactory factory = new DiskFileItemFactory();

            ServletFileUpload upload = new ServletFileUpload(factory);

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

            invoice = invoiceManager.saveInvoice(invoice);

                if (item.isFormField()) {

                    String fieldName = item.getFieldName();

                    String value = item.getString();

                        invoice.setNumber(value);

                        invoice.setAlternativeId(value);

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

                    }

                } else {

                    String contentType = item.getContentType();

                }

            invoiceManager.saveInvoice(invoice);

        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;

            }

        }

    }