Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

import in.shop2020.support.services.ManifestGenerator;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.util.Calendar;

import java.util.GregorianCalendar;

import javax.servlet.ServletOutputStream;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import org.apache.struts2.interceptor.ServletRequestAware;

import org.apache.struts2.interceptor.ServletResponseAware;

import org.apache.struts2.rest.DefaultHttpHeaders;

import org.apache.struts2.rest.HttpHeaders;

public class ManifestController implements ServletResponseAware, ServletRequestAware {

	private String id;

	private long warehouseId;

	private long providerId;

	private HttpServletRequest request;

	private HttpServletResponse response;

	public HttpHeaders index(){

		this.warehouseId = Long.parseLong(request.getParameter("warehouseID"));

		this.providerId = Long.parseLong(request.getParameter("providerID"));

		System.out.println("Warehouse Id is:  " + warehouseId);

		System.out.println("Provider Id is: " + providerId);

		ManifestGenerator manifestGenerator = new ManifestGenerator();

		ByteArrayOutputStream baos = manifestGenerator.generateManifestFile(warehouseId, providerId);

		response.setContentType("application/pdf");

		Calendar date = new GregorianCalendar();

		int year = date.get(Calendar.YEAR);

		int month = date.get(Calendar.MONTH) +1;

		int day = date.get(Calendar.DAY_OF_MONTH);

		response.setHeader("Content-disposition", "inline; filename=manifest-"+warehouseId+"-"+ providerId + "-"+year+"-"+ month+"-" + day +".pdf" );

		ServletOutputStream sos;

		try {

			sos = response.getOutputStream();

			baos.writeTo(sos);

			sos.flush();

		} catch (IOException e) {

			// TODO Auto-generated catch block

			e.printStackTrace();

		}	

		return new DefaultHttpHeaders("lsuccess");

	}

//	public String getId(){

//		return id;

//	}

//	

//	public void setId(String id){

//		this.id = id;

//	}

	@Override

	public void setServletResponse(HttpServletResponse response) {

		this.response  = response;

	}

	@Override

	public void setServletRequest(HttpServletRequest request) {

		this.request = request;

	}

}