Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

import in.shop2020.support.services.PaymentDetailsGenerator;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.text.DateFormat;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;

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;

public class PaymentDetailsController implements ServletResponseAware, ServletRequestAware{

        private HttpServletRequest request;
        private HttpServletResponse response;

        private String errorMsg = "";

        public PaymentDetailsController(){

        }

        @Override
        public void setServletRequest(HttpServletRequest req) {
                this.request = req;
        }

        @Override
        public void setServletResponse(HttpServletResponse res) {
                this.response = res;
        }

        public String index()   {
                return "report";
        }

        public String show(){
                return "report";
        }

        // Handles the POST request (Form Submission)
        public String create(){
                //Formatting Form input parameters
                String startDateStr = request.getParameter("startDate");
                String endDateStr = request.getParameter("endDate");
                DateFormat df = new SimpleDateFormat("MM/dd/yyyy");
                Date startDate = null, endDate = null;
                try {
                        startDate = df.parse(startDateStr);
                        endDate = df.parse(endDateStr);
                        Calendar cal = Calendar.getInstance();
                        cal.setTime(endDate);
                        cal.add(Calendar.DATE, 1);
                        endDate.setTime(cal.getTimeInMillis());
                } catch(ParseException pe) {
                        errorMsg = "Please enter start and end dates in format MM/dd/yyyy";
                        return "report";
                }

                PaymentDetailsGenerator paymentDetailGenerator = new PaymentDetailsGenerator();
                ByteArrayOutputStream baos = paymentDetailGenerator.generatePaymentDetailsReport(startDate, endDate);

                if(baos == null) {
                        errorMsg = "No output for given date range";
                        return "report";
                } else {
                        errorMsg = "";
                }

                // Preparing XLS file for output
                response.setContentType("application/vnd.ms-excel");
                response.setHeader("Content-disposition", "inline; filename=payments-report" + ".xls");
                ServletOutputStream sos;
                try {
                        sos = response.getOutputStream();
                        baos.writeTo(sos);
                        sos.flush();
                } catch (IOException e) {
                        errorMsg = "Failed to write to response.";
                        e.printStackTrace();
                }
                return "report";
        }

        public String getErrorMsg() {
                return errorMsg;
        }
}