Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

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

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

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

import in.shop2020.payments.Payment;

import in.shop2020.payments.PaymentStatus;

import in.shop2020.thrift.clients.PaymentServiceClient;

import in.shop2020.thrift.clients.TransactionServiceClient;

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 java.util.List;

import javax.servlet.ServletOutputStream;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;

import org.apache.poi.ss.usermodel.Cell;

import org.apache.poi.ss.usermodel.CellStyle;

import org.apache.poi.ss.usermodel.Font;

import org.apache.poi.ss.usermodel.Row;

import org.apache.poi.ss.usermodel.Sheet;

import org.apache.poi.ss.usermodel.Workbook;

import org.apache.poi.ss.util.CellRangeAddress;

import org.apache.struts2.interceptor.ServletRequestAware;

import org.apache.struts2.interceptor.ServletResponseAware;

import org.apache.thrift.TException;

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

		try	{

			//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;

			TransactionServiceClient tsc = new TransactionServiceClient();

			in.shop2020.model.v1.order.TransactionService.Client tclient = tsc.getClient();

			PaymentServiceClient psc = new PaymentServiceClient();

			in.shop2020.payments.PaymentService.Client pClient = psc.getClient();

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

				errorMsg = "Date Range: " + startDate + " to " + endDate;

			} catch(ParseException pe) {

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

				return "report";

			}

			// Retrieving all the payments between start and end dates with status as failed and for gateway Id = 1 (HDFC)

			List<Payment> payments = pClient.getPayments(startDate.getTime(), endDate.getTime(), PaymentStatus.FAILED, 1);

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

				errorMsg = "There is no FAILED payment within given date range";

				return "report";

			}

			// Preparing XLS file for output

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

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

			ServletOutputStream sos;

			try {

				ByteArrayOutputStream baos = getSpreadSheetData(payments, tclient);

				sos = response.getOutputStream();

				baos.writeTo(sos);

				sos.flush();

			} catch (IOException e) {

				errorMsg = "Failed to write to response.";

				e.printStackTrace();

			}

		} catch (ParseException e)	{

			errorMsg = e.getMessage();

			e.printStackTrace();

		} catch (TransactionServiceException e)	{

			errorMsg = e.getMessage();

			e.printStackTrace();

		} catch (Exception e)	{

			errorMsg = e.getMessage();

			e.printStackTrace();

		}

		return "report";

	}

	// Prepares the XLS worksheet object and fills in the data with proper formatting

	private ByteArrayOutputStream getSpreadSheetData(List<Payment> payments, in.shop2020.model.v1.order.TransactionService.Client tsc)	{

		ByteArrayOutputStream baosXLS = new ByteArrayOutputStream();

	    Workbook wb = new HSSFWorkbook();

	    Font font = wb.createFont();

	    font.setBoldweight(Font.BOLDWEIGHT_BOLD);

	    CellStyle style = wb.createCellStyle();

	    style.setFont(font);

	    Sheet paymentSheet = wb.createSheet("Payment");

	    short paymentSerialNo = 0;

	    Row titleRow = paymentSheet.createRow(paymentSerialNo ++);

	    Cell titleCell = titleRow.createCell(0);

	    titleCell.setCellValue("Payment Details");

	    titleCell.setCellStyle(style);

	    paymentSheet.addMergedRegion(new CellRangeAddress(0, 0, 0, 6));

	    paymentSheet.createRow(paymentSerialNo ++);

	    Row headerRow = paymentSheet.createRow(paymentSerialNo ++);

	    headerRow.createCell(0).setCellValue("Payment Id");

	    headerRow.createCell(1).setCellValue("Txn Id");

	    headerRow.createCell(2).setCellValue("Amount");

	    headerRow.createCell(3).setCellValue("Payment Status");

	    headerRow.createCell(4).setCellValue("Payment Status Description");

	    headerRow.createCell(5).setCellValue("Reference Code");

	    headerRow.createCell(6).setCellValue("Transaction Time");

	    headerRow.createCell(7).setCellValue("Customer Id");

	    headerRow.createCell(8).setCellValue("Name");

	    headerRow.createCell(9).setCellValue("MobileNo");

	    headerRow.createCell(10).setCellValue("Pincode");

	    headerRow.createCell(11).setCellValue("City");

	    headerRow.createCell(12).setCellValue("State");

	    headerRow.createCell(13).setCellValue("Email");

	    headerRow.createCell(14).setCellValue("Order Id");

	    headerRow.createCell(15).setCellValue("Order Status");

	    headerRow.createCell(16).setCellValue("Order Status Description");

	    headerRow.createCell(17).setCellValue("Item Id");

	    headerRow.createCell(18).setCellValue("Product Group");

	    headerRow.createCell(19).setCellValue("Brand");

	    headerRow.createCell(20).setCellValue("Model Name");

	    headerRow.createCell(21).setCellValue("Model Number");

	    headerRow.createCell(22).setCellValue("Quantity");

	    List<Order> orders;

		long txnId;

		Row contentRow;

		Calendar calendar = Calendar.getInstance();

		DateFormat formatter = new SimpleDateFormat("EEE, dd-MMM-yyyy hh:mm a");

		for(Payment payment : payments) {

			try {

				paymentSerialNo++;

				contentRow = paymentSheet.createRow(paymentSerialNo);

				contentRow.createCell(0).setCellValue(payment.getPaymentId());

				contentRow.createCell(1).setCellValue(payment.getMerchantTxnId());

				contentRow.createCell(2).setCellValue(payment.getAmount());

				contentRow.createCell(3).setCellValue(payment.getStatus().name());

				contentRow.createCell(4).setCellValue(payment.getDescription());

				contentRow.createCell(5).setCellValue(payment.getReferenceCode());

				calendar.setTimeInMillis(payment.getInitTimestamp());

				contentRow.createCell(6).setCellValue(formatter.format(calendar.getTime()));

				txnId = tsc.getTransaction(payment.getMerchantTxnId()).getId();

				orders = tsc.getOrdersForTransaction(txnId, payment.getUserId());

				List<LineItem> lineItems;

				if(orders != null && !orders.isEmpty()) {

					contentRow.createCell(7).setCellValue(orders.get(0).getCustomer_id());

					contentRow.createCell(8).setCellValue(orders.get(0).getCustomer_name());

					contentRow.createCell(9).setCellValue(orders.get(0).getCustomer_mobilenumber());

					contentRow.createCell(10).setCellValue(orders.get(0).getCustomer_pincode());

					contentRow.createCell(11).setCellValue(orders.get(0).getCustomer_city());

					contentRow.createCell(12).setCellValue(orders.get(0).getCustomer_state());

					contentRow.createCell(13).setCellValue(orders.get(0).getCustomer_email());		

					for(Order o : orders) {

						paymentSerialNo++;

						contentRow = paymentSheet.createRow(paymentSerialNo);

						contentRow.createCell(14).setCellValue(o.getId());

						contentRow.createCell(15).setCellValue(o.getStatus().name());

						contentRow.createCell(16).setCellValue(o.getStatusDescription());

						lineItems = tsc.getLineItemsForOrder(o.getId());

						for(LineItem i : lineItems) {

							/*Right now there can be only one line item in an order.

						  So putting line item details in the same row as order details. Commenting below 2 lines for this.*/

							//paymentSerialNo++;

							//contentRow = paymentSheet.createRow(paymentSerialNo);

							contentRow.createCell(17).setCellValue(i.getId());

							contentRow.createCell(18).setCellValue(i.getProductGroup());

							contentRow.createCell(19).setCellValue(i.getBrand());

							contentRow.createCell(20).setCellValue(i.getModel_name());

							contentRow.createCell(21).setCellValue(i.getModel_number());

							contentRow.createCell(22).setCellValue(i.getQuantity());

						}

					}

				}

			} catch (TransactionServiceException e) {

				e.printStackTrace();

			} catch (TException e) {

				e.printStackTrace();

			}

		}

		paymentSheet.autoSizeColumn(0);

		paymentSheet.autoSizeColumn(1);

		paymentSheet.autoSizeColumn(2);

		paymentSheet.autoSizeColumn(3);

		paymentSheet.autoSizeColumn(4);

		paymentSheet.autoSizeColumn(5);

		paymentSheet.autoSizeColumn(6);

		paymentSheet.autoSizeColumn(7);

		paymentSheet.autoSizeColumn(8);

		paymentSheet.autoSizeColumn(9);

		paymentSheet.autoSizeColumn(10);

		paymentSheet.autoSizeColumn(11);

		paymentSheet.autoSizeColumn(12);

		paymentSheet.autoSizeColumn(13);

		paymentSheet.autoSizeColumn(14);

		paymentSheet.autoSizeColumn(15);

		paymentSheet.autoSizeColumn(16);

		paymentSheet.autoSizeColumn(17);

		paymentSheet.autoSizeColumn(18);

		paymentSheet.autoSizeColumn(19);

		paymentSheet.autoSizeColumn(20);

		paymentSheet.autoSizeColumn(21);

		paymentSheet.autoSizeColumn(22);

		// Write the workbook to the output stream

		try {

			wb.write(baosXLS);

			baosXLS.close();

		} catch (IOException e) {

			e.printStackTrace();

		}		

		return baosXLS;

	}

	public String getErrorMsg() {

		return errorMsg;

	}

}