Subversion Repositories SmartDukaan

package com.spice.profitmandi.web.view.xls;

import java.time.LocalDateTime;

import java.util.List;

import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

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

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

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

import org.springframework.stereotype.Component;

import org.springframework.web.servlet.view.document.AbstractXlsxView;

import com.spice.profitmandi.common.util.FormattingUtils;

import com.spice.profitmandi.dao.entity.transaction.UserWalletHistory;

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

@Component("walletStatement")

public class WalletStatementView extends AbstractXlsxView {

	@Override

	protected void buildExcelDocument(Map<String, Object> model, Workbook workbook, HttpServletRequest request,

			HttpServletResponse response) throws Exception {

		// TODO Auto-generated method stub

	      List<UserWalletHistory> history = (List<UserWalletHistory>) model.get("history");

	      LocalDateTime startDate = (LocalDateTime)model.get("startDate");

	      LocalDateTime endDate = (LocalDateTime)model.get("endDate");

	      Float opening = (float)model.get("opening");

	      //Float closing = (float)model.get("closing");

	      Sheet sheet = workbook.createSheet("Wallet Statement");

	      UserWalletHistory uwhOpening = getOpeningEntry(opening, startDate);

	      //UserWalletHistory closingEntry = getClosingEntry(closing, endDate);

	      history.add(0, uwhOpening);

	      //history.add(closingEntry);

	      sheet.setFitToPage(true);

	      int rowCount = 0;

	      Row header = sheet.createRow(rowCount++);

	      header.createCell(0).setCellValue("Date");

	      header.createCell(1).setCellValue("Transact Type");

	      header.createCell(2).setCellValue("Transact Reference");

	      header.createCell(3).setCellValue("Transact Description");

	      header.createCell(4).setCellValue("Amount Dr");

	      header.createCell(5).setCellValue("Amount Cr");

	      header.createCell(6).setCellValue("Balance");

	      int calculatedClosing = 0;

	      for (UserWalletHistory walletEntry : history) {

	          Row walletEntryRow = sheet.createRow(rowCount++);

	          walletEntryRow.createCell(0).setCellValue(walletEntry.getFormattedDate());

	          if(walletEntry.getReferenceType()==null) {

	        	  walletEntryRow.createCell(1).setCellValue("Closing Balance");

	        	  walletEntryRow.createCell(6).setCellValue(walletEntry.getAmount());

	          } else {

	        	  walletEntryRow.createCell(1).setCellValue(walletEntry.getReferenceType().name());

	          }

	          walletEntryRow.createCell(2).setCellValue(walletEntry.getReference());

	          walletEntryRow.createCell(3).setCellValue(walletEntry.getDescription());

	          if(walletEntry.getAmount() < 0) {

	        	  walletEntryRow.createCell(4).setCellValue(-walletEntry.getAmount());

	        	  walletEntryRow.createCell(5).setCellValue("");

	          } else if (walletEntry.getAmount() > 0) {

	        	  walletEntryRow.createCell(4).setCellValue("");

	        	  walletEntryRow.createCell(5).setCellValue(walletEntry.getAmount());

	          }

	          calculatedClosing = calculatedClosing + walletEntry.getAmount();

	          walletEntryRow.createCell(6).setCellValue(calculatedClosing);

	      }

	      String fileName = "wallet-" + FormattingUtils.formatDate(startDate) + " " + FormattingUtils.formatDate(endDate) + ".xlsx";

	      response.setHeader("Content-Disposition", "attachment; filename=" + fileName);

	}

	private UserWalletHistory getClosingEntry(Float closing, LocalDateTime endDate) {

		UserWalletHistory uwh = new UserWalletHistory();

		uwh.setAmount(0);

		uwh.setDescription("Final Closing balance");

		uwh.setReferenceType(null);

		uwh.setTimestamp(endDate);

		uwh.setReference(0);

		return uwh;

	}

	private UserWalletHistory getOpeningEntry(Float opening, LocalDateTime startDate) {

		UserWalletHistory uwh = new UserWalletHistory();

		uwh.setAmount((int)opening.floatValue());

		uwh.setDescription("Opening Balance Credit");

		uwh.setReferenceType(WalletReferenceType.OPENING_BALANCE);

		uwh.setTimestamp(startDate);

		uwh.setReference(0);

		return uwh;

	}

}