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 org.apache.struts2.convention.annotation.InterceptorRef;

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

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

@InterceptorRefs({

    @InterceptorRef("myDefault"),

    @InterceptorRef("login")

})

public class PaymentDetailsController extends ReportsController {

    private String errorMsg = "";

    private final String authsuccess = "authsuccess";

    private final String authfail = "authfail";

    public PaymentDetailsController() {

    }

    public String index() {

        if(!canAccessReport()) {

            return "exception";

        }

        return authsuccess;

    }

    // Handles the POST request (Form Submission)

    public String create() {

        // Formatting Form input parameters

        String startDateStr = getServletRequest().getParameter("startDate");

        String endDateStr = getServletRequest().getParameter("endDate");

        String statusStr = getServletRequest().getParameter("status");

        String filenameStr = "";

        int status = 1;

        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());

            status = Integer.parseInt(statusStr);

            filenameStr = (status == 0 ? "success" : "failed-pending");

        } catch (ParseException pe) {

            errorMsg = "Please enter start and end dates in format MM/dd/yyyy";

            return authsuccess;

        } catch (NumberFormatException nfe) {

            errorMsg = "Please select payment status";

            return authsuccess;

        }

        PaymentDetailsGenerator paymentDetailGenerator = new PaymentDetailsGenerator();

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

        if (baos == null) {

            errorMsg = "No output for given date range";

            return authsuccess;

        } else {

            errorMsg = "Generating report...";

        }

        // Preparing XLS file for output

        getServletResponse().setContentType("application/vnd.ms-excel");

        getServletResponse().setHeader("Content-disposition", "inline; filename=payments-report-" + filenameStr + ".xls");

        ServletOutputStream sos;

        try {

            sos = getServletResponse().getOutputStream();

            baos.writeTo(sos);

            sos.flush();

            errorMsg = "Report generated";

        } catch (IOException e) {

            errorMsg = "Failed to write to response.";

            e.printStackTrace();

        }

        return authsuccess;

    }

    public String getErrorMsg() {

        return errorMsg;

    }

}