Subversion Repositories SmartDukaan

package in.shop2020.dtrapi.controllers;

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.InputStream;

import java.io.OutputStream;

import javax.servlet.http.HttpServletResponse;

import org.apache.struts2.convention.annotation.Result;

import org.apache.struts2.interceptor.ServletResponseAware;

import com.opensymphony.xwork2.Action;

import com.opensymphony.xwork2.ActionSupport;

import in.shop2020.dtrapi.utils.PdfGenerator;

@Result(

    name = "success", 

    type = "stream", 

    params = { 

        "contentType", "${type}", 

        "inputName", "${stream}", 

        "bufferSize", "1024", 

        "contentDisposition", "attachment;filename=\"${filename}\""

    }

)

public class FileDisplayController extends ActionSupport implements ServletResponseAware{

    /**

	 * 

	 */

	private static final long serialVersionUID = 1L;

	//private String type = "application/pdf";

    private String filename;

    private InputStream stream;

    //private HttpServletRequest request;

    private HttpServletResponse response;

	private int id;

    //@org.apache.struts2.convention.annotation.Action(value = "/fofoDetailPdf", params="fofoId")

    public String show() throws Exception {

    	//int fofoId = Integer.parseInt(request.getParameter("fofoId"));

        //filename = "myimage.jpg";

        //File img = new File("/path/to/image/image.jpg");

    	OutputStream outputStream = response.getOutputStream();

    	PdfGenerator.generateAndWrite(outputStream, id);

        stream = this.toInputStream(outputStream);

        response.setHeader("Cache-Control", "no-cache");

        response.setHeader("Content-Disposition", "attachment;filename=\"fofoDetail.pdf\"");

        response.setContentType("application/pdf");

        response.setContentLength(((ByteArrayOutputStream)outputStream).size());

        return Action.SUCCESS;

    }

    private ByteArrayInputStream toInputStream(OutputStream outputStream){

    	ByteArrayOutputStream buffer = (ByteArrayOutputStream) outputStream;

    	byte[] bytes = buffer.toByteArray();

    	ByteArrayInputStream inputStream = new ByteArrayInputStream(bytes);

    	return inputStream;

    }

    /*private String getType() {

        return this.type;

    }*/

    private String getFilename() {

        return this.filename;

    }

    private InputStream getStream() {

        return this.stream;

    }

	/*@Override

	public void setServletRequest(HttpServletRequest arg0) {

		this.request = arg0;

	}*/

	@Override

	public void setServletResponse(HttpServletResponse arg0) {

		this.response = arg0;

	}

	public int getId() {

		return id;

	}

	public void setId(int id) {

		this.id = id;

	}

}