Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import javax.servlet.ServletOutputStream;

import javax.servlet.http.HttpServletResponse;

import in.shop2020.support.services.InvoiceGenerationService;

import org.apache.struts2.interceptor.ServletResponseAware;

import org.apache.struts2.rest.DefaultHttpHeaders;

import org.apache.struts2.rest.HttpHeaders;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class InvoiceController implements ServletResponseAware{

	private static Logger logger = LoggerFactory.getLogger(InvoiceController.class);

	private int errorCode = 0;

	private String errorMessage;

	private String id;

	private long orderId;

	private HttpServletResponse response;

	public InvoiceController(){

	}

	public HttpHeaders show(){

		logger.info("Printing invoice for order id: " + getId());

		this.orderId = Long.parseLong(getId());

		InvoiceGenerationService invoiceGenerationService = new InvoiceGenerationService();

		ByteArrayOutputStream baos = invoiceGenerationService.generateInvoice(orderId);

		response.setContentType("application/pdf");

		response.setHeader("Content-disposition", "inline; filename=invoice-"+orderId+".pdf" );

		ServletOutputStream sos;

		try {

			sos = response.getOutputStream();

			baos.writeTo(sos);

			sos.flush();

		} catch (IOException e) {

			logger.error("Encountered error while sending invoice response: ", e);

		}

		return new DefaultHttpHeaders("lsuccess");

	}

	public int getErrorCode() {

		return errorCode;

	}

	public String getErrorMessage() {

		return errorMessage;

	}

	public String getId(){

		return id;

	}

	public void setId(String id){

		this.id = id;

	}

	@Override

	public void setServletResponse(HttpServletResponse response) {

		this.response  = response;

	}

}