Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

import in.shop2020.logistics.PickupStore;

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

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

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

import in.shop2020.support.utils.ReportsUtils;

import in.shop2020.thrift.clients.LogisticsClient;

import in.shop2020.thrift.clients.TransactionClient;

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

import javax.servlet.ServletOutputStream;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import javax.servlet.http.HttpSession;

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

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

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

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

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

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

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

import org.apache.struts2.interceptor.ServletRequestAware;

import org.apache.struts2.interceptor.ServletResponseAware;

import org.apache.struts2.util.ServletContextAware;

import org.apache.thrift.TException;

import org.apache.thrift.transport.TTransportException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

@InterceptorRefs({

    @InterceptorRef("defaultStack"),

    @InterceptorRef("login")

})

@Results({

    @Result(name="authfail", type="redirectAction", params = {"actionName" , "reports"})

})

public class StoreCollectionController implements ServletRequestAware, ServletResponseAware, ServletContextAware {

    private static Logger logger = LoggerFactory.getLogger(StoreCollectionController.class);

    private HttpServletRequest request;

    private HttpSession session;

    private HttpServletResponse response;

    private ServletContext context;

    private String error = "";

    private Client client;

    public StoreCollectionController() throws TTransportException{

    	client = (new TransactionClient()).getClient();

    }

    public String index() {

        if(!ReportsUtils.canAccessReport((Long)session.getAttribute(ReportsUtils.ROLE), request.getServletPath())) {

            return "authfail";

        }

        return "index";

    }

    // Handles the POST request (Form Submission)

    public String create() throws TException {

    	String storeIdString = request.getParameter("storeId");

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

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

        DateFormat df = new SimpleDateFormat("dd/MM/yyyy");

        Date startDate = null, endDate = null;

        long storeId = 0;

        try {

        	storeId = Long.parseLong(storeIdString);

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

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

        } 

        ByteArrayOutputStream baos = generateCollectionReport(startDate, endDate, storeId);

        if (baos == null) {

            error = "No output for given date range";

        } else {

            error = "Generating report...";

        }

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

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

        ServletOutputStream sos;

        try {

            sos = response.getOutputStream();

            baos.writeTo(sos);

            sos.flush();

            error = "Report generated";

        } catch (IOException e) {

            error = "Failed to write to response.";

            logger.error("Unable to stream the colection report", e);

        }

        return "index";

    }

    private ByteArrayOutputStream generateCollectionReport(Date startDate,	Date endDate, long storeId) throws TException {

    	List<Order> orders = client.getOrdersCollectionAtStore(storeId, startDate.getTime(), endDate.getTime(), true);

        ByteArrayOutputStream baosXLS = new ByteArrayOutputStream();

        Workbook wb = new HSSFWorkbook();

        DateFormat formatter = new SimpleDateFormat("dd-MMM-yyyy");

        Font font = wb.createFont();

        font.setBoldweight(Font.BOLDWEIGHT_BOLD);

        CellStyle style = wb.createCellStyle();

        style.setFont(font);

        DataFormat format = wb.createDataFormat();

        CellStyle styleAmount = wb.createCellStyle();

        styleAmount.setDataFormat(format.getFormat("#,##0.00"));

        Sheet collectionSheet = wb.createSheet("Collection");

        short collectionSerialNo = 0;

        Row headerRow = collectionSheet.createRow(collectionSerialNo++);

        headerRow.createCell(0).setCellValue("S.No");

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

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

        headerRow.createCell(3).setCellValue("Order Type");

        headerRow.createCell(4).setCellValue("Collection Date");

        headerRow.createCell(5).setCellValue("Order Amount");

        headerRow.createCell(6).setCellValue("Collected Amount");

        //headerRow.createCell(7).setCellValue("Store Margin");

        for(Order order: orders){

        	Row contentRow = collectionSheet.createRow(collectionSerialNo++);

        	LineItem line = order.getLineitems().get(0);

        	String productName = line.getBrand() + " " + line.getModel_name() + " " + line.getModel_number() + " " + line.getColor();

        	productName = productName.replaceAll("null", "").replaceAll("  ", " ");

        	contentRow.createCell(0).setCellValue(collectionSerialNo-1);

        	contentRow.createCell(1).setCellValue(order.getId());

        	contentRow.createCell(2).setCellValue(productName);

        	contentRow.createCell(3).setCellValue(order.isCod() ? "COD" : "Prepaid");

        	contentRow.createCell(4).setCellValue(formatter.format(order.getDelivery_timestamp()));

        	contentRow.createCell(5).setCellValue(order.getTotal_amount());

        	contentRow.createCell(6).setCellValue(order.isCod() ? order.getTotal_amount() : 0.0);

        	//contentRow.createCell(7).setCellValue(order.isCod() ? .01 * order.getTotal_amount() : .005 * order.getTotal_amount());

        }

        try {

            wb.write(baosXLS);

            baosXLS.close();

        } catch (IOException e) {

            logger.error("Error while streaming payment details report", e);

        }

		return baosXLS;

	}

    public String getErrorMsg() {

        return error;

    }

    @Override

    public void setServletRequest(HttpServletRequest req) {

        this.request = req;

        this.session = req.getSession();        

    }

    @Override

    public void setServletResponse(HttpServletResponse res) {

        this.response = res;

    }

    @Override

    public void setServletContext(ServletContext context) {

        this.context = context;

    }

    public String getServletContextPath() {

        return context.getContextPath();

    }

    public List<PickupStore> getAllStores() throws TTransportException, TException {

    	return (new LogisticsClient()).getClient().getAllPickupStores();

    }

}