Subversion Repositories SmartDukaan

package in.shop2020.support.services;

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.PaymentException;

import in.shop2020.payments.PaymentStatus;

import in.shop2020.thrift.clients.PaymentServiceClient;

import in.shop2020.thrift.clients.TransactionServiceClient;

import java.io.ByteArrayOutputStream;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

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 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.thrift.TException;

public class PaymentDetailsGenerator {

	TransactionServiceClient tsc;

	in.shop2020.model.v1.order.TransactionService.Client tClient;

	PaymentServiceClient psc;

	in.shop2020.payments.PaymentService.Client pClient;

	public PaymentDetailsGenerator() {

		try {

			tsc = new TransactionServiceClient();

			tClient = tsc.getClient();

			psc = new PaymentServiceClient();

			pClient = psc.getClient();

		} catch (Exception e) {

			e.printStackTrace();

		}

	}

	public ByteArrayOutputStream generatePaymentDetailsReport(Date startDate, Date endDate) {

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

		List<Payment> payments = null;

		List<Payment> pendingPayments = null;

		try {

			payments = pClient.getPayments(startDate.getTime(), endDate.getTime(), PaymentStatus.FAILED, 1);

			pendingPayments = pClient.getPayments(startDate.getTime(), endDate.getTime(), PaymentStatus.INIT, 1);

		} catch (PaymentException e) {

			e.printStackTrace();

		} catch (TException e) {

			e.printStackTrace();

		}

		if(payments != null && pendingPayments != null) {

			payments.addAll(pendingPayments);	

		} else {

			payments = pendingPayments;

		}

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

			return null;

		}

		// Preparing XLS file for output

		return getSpreadSheetData(payments);

	}

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

	private ByteArrayOutputStream getSpreadSheetData(List<Payment> payments)	{

		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 = tClient.getTransaction(payment.getMerchantTxnId()).getId();

				orders = tClient.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 = tClient.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 static void main(String[] args) {

		String startDateStr = "01/01/2011";

		String endDateStr = "04/30/2011";

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

			pe.printStackTrace();

		}

		PaymentDetailsGenerator pdg = new PaymentDetailsGenerator();

		try {

			String userHome = System.getProperty("user.home");

			FileOutputStream f = new FileOutputStream(userHome + "/payment-details-report.xls");

			ByteArrayOutputStream baosXLS = pdg.generatePaymentDetailsReport(startDate, endDate);

			baosXLS.writeTo(f);

			f.close();

		} catch (FileNotFoundException e) {

			e.printStackTrace();

		} catch (IOException e) {

			e.printStackTrace();

		}

		System.out.println("Successfully generated the payment details report");

	}

}