Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

import in.shop2020.model.v1.order.TransactionService;

import in.shop2020.model.v1.order.TransactionServiceException;

import in.shop2020.thrift.clients.TransactionClient;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.IOException;

import java.nio.ByteBuffer;

import java.nio.channels.Channels;

import java.nio.channels.WritableByteChannel;

import java.util.ArrayList;

import java.util.Calendar;

import java.util.Date;

import java.util.GregorianCalendar;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.servlet.ServletContext;

import javax.servlet.ServletOutputStream;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import javax.servlet.http.HttpSession;

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

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

import org.apache.struts2.interceptor.ServletRequestAware;

import org.apache.struts2.interceptor.ServletResponseAware;

import org.apache.struts2.util.ServletContextAware;

import org.apache.thrift.TException;

import org.apache.thrift.transport.TTransportException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

@SuppressWarnings("serial")

@InterceptorRefs({

    @InterceptorRef("myDefault"),

    @InterceptorRef("login")

})

/**

 * 

 * @author Rajveer

 * 

 */

public class InvoiceController implements ServletResponseAware,

		ServletRequestAware, ServletContextAware {

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

    private ServletContext context;

	private HttpServletRequest request;

	private HttpServletResponse response;

	private HttpSession session;

	private String id;

	private String orderId;

    private String errorMsg = "";

	public String show(){

		orderId = id;

		response.setContentType("application/pdf");

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

        ByteBuffer buffer = null;

        try {

            if (orderId == null || orderId.isEmpty()) {

                logger.error("Order Id is null or empty");

                return "show";

            }

            TransactionClient tc = new TransactionClient("support_transaction_service_server_host", "transaction_service_server_port");

            TransactionService.Client orderClient = tc.getClient();

            buffer = orderClient.retrieveInvoice(Long.parseLong(orderId));

            if(!buffer.hasArray()) {

                logger.error("The invoice does not found");

            }

        } catch (Exception e) {

            System.out.println(e.getMessage());

        }

        ServletOutputStream sos;

        try {

            sos = response.getOutputStream();

            sos.write(buffer.array());

            sos.flush();

        } catch (Exception e) {

            System.out.println("Unable to stream the invoice file");

        }

        return "show";

    }

	@Override

	public void setServletContext(ServletContext context) {

		this.context = context;

	}

	@Override

	public void setServletResponse(HttpServletResponse response) {

		this.response  = response;

	}

	@Override

	public void setServletRequest(HttpServletRequest request) {

		this.request = request;

		this.session = request.getSession();

	}

	public String getId(){

		return id;

	}

	public void setId(String id){

		this.id = id;

	}

	public String getServletContextPath(){

		return context.getContextPath();

	}

    public String getErrorMsg() {

        return errorMsg;

    }

    public void setErrorMsg(String errorMsg) {

        this.errorMsg = errorMsg;

    }

}