D-Net

package eu.dnetlib.server;

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;

				if (item.isFormField()) {

					String fieldName = item.getFieldName();

					String value = item.getString();

					System.out.println(value);

                        personId = value;

						requestId = value;

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

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

                        action = value;

                    }

			}

                if(r!=null) {

                    savedRequest.setTransfer_cost(transferCost);

                    savedRequest.setOther_cost(otherCost);

            } else {

				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;

			}

		}

	}

}