D-Net

package eu.dnetlib.server;

import eu.dnetlib.goldoa.domain.*;

import eu.dnetlib.goldoa.service.RequestManager;

import org.apache.commons.fileupload.FileItem;

import org.apache.commons.fileupload.FileItemFactory;

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.ByteArrayInputStream;

import java.io.IOException;

import java.text.SimpleDateFormat;

import java.util.Date;

import java.util.List;

/**

 * Created by antleb on 7/9/15.

 */

public class RequestServlet extends HttpServlet {

	private RequestManager requestManager;

	@Override

	public void init(ServletConfig config) throws ServletException {

		super.init(config);

		ApplicationContext context = WebApplicationContextUtils

				.getWebApplicationContext(getServletContext());

		this.requestManager = (RequestManager) context.getBean("requestManager");

	}

	@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);

			String requestId = null;

			float apcPaid = 0;

			float transferCost = 0;

			float otherCost = 0;

			Date datePaid = null;

			String comment = null;

            String personId = null;

            String action = null;

			for (FileItem item:items) {

				if (item.isFormField()) {

					String fieldName = item.getFieldName();

					String value = item.getString();

					System.out.println(value);

					if (fieldName.equals("apcPaid") && value != null && !value.equals("")) {

						apcPaid = Float.parseFloat(value);

					} else if (fieldName.equals("transferCosts") && value != null && !value.equals("")) {

						transferCost = Float.parseFloat(value);

					} else if (fieldName.equals("otherExpenses") && value != null && !value.equals("")) {

						otherCost = Float.parseFloat(value);

					} else if (fieldName.equals("comment") && value!=null && !value.trim().equals("")) {

						comment = value;

                    } else if (fieldName.equals("personId") && value!=null && !value.trim().equals("")) {

                        personId = value;

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

						requestId = value;

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

						datePaid = new SimpleDateFormat("yyyy/MM/dd").parse(value);

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

                        action = value;

                    }

				}

			}

            if(action!=null && action.equals("edit")) {

                Request r = requestManager.getById(requestId);

                if(r!=null) {

                    /*Request savedRequest = request.getRequest();

                    savedRequest.setApc_paid(apcPaid);

                    savedRequest.setTransfer_cost(transferCost);

                    savedRequest.setOther_cost(otherCost);

                    savedRequest.setDate_paid(datePaid);*/

                    requestManager.saveRequest(r);

                }

            } else {

                requestManager.paidRequest(requestId, personId, comment, null, apcPaid, transferCost, otherCost, datePaid);

            }

			for (FileItem item:items) {

				if (!item.isFormField()) {

					String contentType = item.getContentType();

					requestManager.uploadTransferReceipt(requestId, contentType, item.getInputStream());

				}

			}

			response.setStatus(200);

		}

		catch(Exception e){

			throw new RuntimeException(e);

		}

	}

	@Override

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

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

		try {

			BankTransferReceipt file = requestManager.downloadBankTransferReceipt(requestId);

			resp.setContentType(file.getContentType());

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

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

		} catch (RequestManagerException e) {

			switch (e.getErrorCause()) {

				case NOT_EXISTS:

					resp.setStatus(404);

					resp.getOutputStream().println("Bank transfer receipt for request " + requestId + " not found");

					break;

				case UNKNOWN:

				default:

					resp.setStatus(500);

					resp.getOutputStream().println("Error getting transfer receipt for request " + requestId);

					break;

			}

		}

	}

}