Subversion Repositories SmartDukaan

package in.shop2020.recharge.controllers;


import in.shop2020.model.v1.order.RechargeOrderStatus;
import in.shop2020.model.v1.order.RechargeTransaction;
import in.shop2020.thrift.clients.TransactionClient;
import in.shop2020.warehouse.InventoryAvailability;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.List;
import java.util.Map;

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.CreationHelper;
import org.apache.poi.ss.usermodel.DataValidationConstraint.OperatorType;
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.interceptor.ServletRequestAware;
import org.apache.struts2.interceptor.ServletResponseAware;
import org.apache.struts2.util.ServletContextAware;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.sun.mail.iap.Response;



public class ReportController implements ServletRequestAware, ServletContextAware, ServletResponseAware {

    private static Logger logger = LoggerFactory.getLogger(ReportController.class);
    
    protected HttpServletRequest request;
    protected HttpSession session;
    protected HttpServletResponse response;
    private ServletContext context;
    private String startDate;
        private String endDate;
        private String status;
        private String dateselector;
        
        private TransactionClient tsc;
    private in.shop2020.model.v1.order.TransactionService.Client tClient;
    
    private final DateFormat formatter = new SimpleDateFormat("EEE, dd-MMM-yyyy hh:mm a");
    private final DateFormat dateFormatter = new SimpleDateFormat("dd/MM/yyyy");
        
    private List<RechargeTransaction> txns = null;
    public ReportController(){
            try {
            tsc = new TransactionClient();
            tClient = tsc.getClient();
        } catch (Exception e) {
            logger.error("Error connecting to one of the user, order or payment service", e);
        }
        }
                
    
        public String index() throws Exception{
            String loginStatus = (String) request.getSession().getAttribute("LOGGED_IN");
        if(loginStatus == null || !loginStatus.equals("TRUE")){
            return "authfail";
        }
        txns = tClient.getRechargeTransactions(Long.parseLong((String) request.getSession().getAttribute("STORE_ID")));
        return "index";
        }
        
        public String create() throws Exception{
                long sDate = -1;
                long eDate = -1;
                RechargeOrderStatus st = null;
                String loginStatus = (String) request.getSession().getAttribute("LOGGED_IN");
                if(loginStatus == null || !loginStatus.equals("TRUE")){
                        return "authfail";
                }
                if(dateselector.equals("1")) {
                        sDate = new Date().getTime();
                        eDate = sDate;
                }else if (dateselector.equals("2")) {
                        sDate = new Date().getTime() - 86400*1000;
                        eDate = sDate;
                }else {
                        if(!(startDate.equals(""))) {
                                sDate = dateFormatter.parse(startDate).getTime();
                                startDate = dateFormatter.format(sDate);
                        }
                        if(!endDate.equals("")) {
                                eDate = dateFormatter.parse(endDate).getTime();
                                endDate = dateFormatter.format(sDate);
                        }
                }
                if(!status.equals("-1")){
                        st = RechargeOrderStatus.findByValue(Integer.parseInt("0"));
                }

                ByteArrayOutputStream baos = generateReport(tClient.getRechargeTrans(Long.parseLong((String) request.getSession().getAttribute("STORE_ID")), sDate, eDate, st));
        
                response.setContentType("application/vnd.ms-excel");
                String st1 = "ALL";
                if(st!=null){
                        st1 = st.name();
                }
                String fileName = st1 + "-recharge-report";
                if (!startDate.equals("")) {
                        fileName = fileName + "-" + startDate.replaceAll("\\\\", "") ;
                }
                if (!endDate.equals("")) {
                        fileName = fileName + "-" + endDate.replaceAll("\\\\", "") ;
                }
                fileName = fileName + ".xls";
                response.setHeader("Content-disposition", "inline; filename=" + fileName);
        ServletOutputStream sos;
        try {
            sos = response.getOutputStream();
            baos.writeTo(sos);
            sos.flush();
        } catch (IOException e) {
            e.printStackTrace();
        }
                return "index";
        }
        
            
    private ByteArrayOutputStream generateReport(
                        List<RechargeTransaction> rechargeTrans) {
ByteArrayOutputStream baosXLS = new ByteArrayOutputStream();
                
            Workbook wb = new HSSFWorkbook();
            
            Sheet reportSheet = wb.createSheet("Recharge Report");
            
            Row reportSheetHeader = reportSheet.createRow((short)0);
            reportSheetHeader.createCell(0).setCellValue("Order ID");
            reportSheetHeader.createCell(1).setCellValue("Order Date");
            reportSheetHeader.createCell(2).setCellValue("Operator");
            reportSheetHeader.createCell(3).setCellValue("Device");
            reportSheetHeader.createCell(4).setCellValue("Status");
            reportSheetHeader.createCell(5).setCellValue("Amount");
            reportSheetHeader.createCell(6).setCellValue("Discount");
            
            int serialNo = 0;
            
            for(RechargeTransaction transaction : rechargeTrans) {
                serialNo++;
                Row contentRow = reportSheet.createRow((short)serialNo);
                contentRow.createCell(0).setCellValue(transaction.getId());
                contentRow.createCell(1).setCellValue(formatter.format(new Date(transaction.getTransactionTime())));
                contentRow.createCell(2).setCellValue(getOperatorName(transaction.getOperatorId()));
                contentRow.createCell(3).setCellValue(transaction.getDeviceNum());
                contentRow.createCell(4).setCellValue(transaction.getStatus().name());
                contentRow.createCell(5).setCellValue(transaction.getAmount());
                contentRow.createCell(6).setCellValue(transaction.getDiscount());
            }
            try {
                        wb.write(baosXLS);
                } catch (IOException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }
            return baosXLS;
        }


        public String getDateTime(long milliseconds) {
        Calendar cal = Calendar.getInstance();
        cal.setTimeInMillis(milliseconds);
        return formatter.format(cal.getTime());
    }

    public List<RechargeTransaction> getTxns(){
        return txns;
    }
    
    @Override
    public void setServletRequest(HttpServletRequest req) {
        this.request = req;
        this.session = req.getSession();        
    }
    
    @Override
    public void setServletContext(ServletContext context) {
        this.context = context;
    }
    
    @Override
    public void setServletResponse(HttpServletResponse response) {
        this.response = response;
    }

    public String getServletContextPath() {
        return context.getContextPath();
    }
    
    public String getOperatorName(long operatorId) {
        String operatorName = "";
        if (HomeController.getMobileProvidersMap().containsKey(operatorId)) {
            operatorName = HomeController.getMobileProvidersMap().get(operatorId);
        } else if(HomeController.getDthProvidersMap().containsKey(operatorId)) {
            operatorName = HomeController.getDthProvidersMap().get(operatorId);
        } else {
            operatorName = "N/A";
        }
        return operatorName;
    }
    
        public String getStartDate() {
                return startDate;
        }
        
        
        public void setStartDate(String startDate) {
                this.startDate = startDate;
        }

        public String getEndDate() {
                return endDate;
        }


        public void setEndDate(String endDate) {
                this.endDate = endDate;
        }
        
        public String getStatus() {
                return status;
        }


        public void setStatus(String status) {
                this.status = status;
        }


        public String getDateselector() {
                return dateselector;
        }


        public void setDateselector(String dateselector) {
                this.dateselector = dateselector;
        }
        
        

}