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 InvoiceServlet 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("alternativeId")) {
                        invoice.setAlternativeId(value);
                    } else if (fieldName.equals("dateIssued")) {
                        invoice.setDate(new SimpleDateFormat("yyyy/MM/dd").parse(value));
                    }
                } else {
                    String contentType = item.getContentType();
                    invoice = invoiceManager.uploadInvoice(invoice.getId(), contentType, item.getInputStream());
                }
            }

            invoiceManager.saveInvoice(invoice);

            response.setContentType("text/plain");
            response.setContentLength(invoice.getId().length()+6);

            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;
            }
        }
    }
}