Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

import in.shop2020.logistics.Awb;
import in.shop2020.logistics.DeliveryType;
import in.shop2020.logistics.LogisticsServiceException;
import in.shop2020.logistics.Provider;
import in.shop2020.model.v1.catalog.InventoryServiceException;
import in.shop2020.model.v1.catalog.Warehouse;
import in.shop2020.model.v1.order.LineItem;
import in.shop2020.model.v1.order.Order;
import in.shop2020.model.v1.order.OrderStatus;
import in.shop2020.model.v1.order.TransactionServiceException;
import in.shop2020.support.utils.FileUtils;
import in.shop2020.thrift.clients.CatalogClient;
import in.shop2020.thrift.clients.HelperClient;
import in.shop2020.thrift.clients.LogisticsClient;
import in.shop2020.thrift.clients.TransactionClient;
import in.shop2020.utils.LogisticsUser;
import in.shop2020.support.models.AwbDetails;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.PrintStream;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.Collection;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.ListIterator;
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.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.CreationHelper;
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.apache.thrift.TException;
import org.apache.thrift.transport.TTransportException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Allows executives of courier companies to login and download courier details
 * report which they then use to upload into their database.
 * 
 * @author Chandranshu
 * 
 */
public class CourierDetailsController implements ServletResponseAware,
                ServletRequestAware, ServletContextAware {
    
    private static Logger logger = LoggerFactory.getLogger(CourierDetailsController.class);
    
        private String id;
        private int daysToSubtract;
        
        //FIXME: Read this configuration from the config client
        private String courierDetailsPath = "/CourierDetailReports";
        
        private ServletContext context;
        private HttpServletRequest request;
        private HttpServletResponse response;
        private HttpSession session;

    private String awbNumbers;
    private List<AwbDetails> detailedAWBs;

    private String errorMsg = "";

        public String index(){
                if(getSessionUserName()==null)
                        return "authfail";
                else
                        return "authsuccess";
        }
        
        // Handler for POST /courier-details
        public String create(){
                String username = request.getParameter("username");
                String password = request.getParameter("password");
                try{
                        HelperClient helperServiceClient = new HelperClient();
                        in.shop2020.utils.HelperService.Client client = helperServiceClient.getClient();
                        LogisticsUser user = client.authenticateLogisticsUser(username, password);
                        session.setAttribute("username", user.getUsername());
                        session.setAttribute("providerId", Long.valueOf(user.getProviderId()));
                }catch(Exception e){
                        logger.error("Error authenticating the user " + username, e);
                        return "authfail";
                }
                return "authsuccess";
        }
        
        // Handler for GET /courier-details/<warehouseId>
        public String show(){
                try {
                        long warehouseId = Long.parseLong(getId());
                        if(warehouseId == 1){
                                warehouseId = 0;
                        }
                        long providerId = ((Long)session.getAttribute("providerId")).longValue();
                        boolean isCod;
                        try {
                    isCod = Boolean.parseBoolean(request.getParameter("isCod"));
                } catch (Exception e) {
                    isCod = false;
                }
                        logger.info("Download request for " + (isCod ? "COD" : "Prepaid") + " Courier Details report of warehouse Id: " + warehouseId + " and provider Id:" + providerId);
                        
                        String deliveryType = "prepaid";
                        if(isCod)
                            deliveryType = "cod";
                        
                        response.setContentType("application/vnd.ms-excel");
                        
                        Calendar date = new GregorianCalendar();
                        date.add(Calendar.DAY_OF_MONTH, daysToSubtract);
                        int year = date.get(Calendar.YEAR);
                        int month = date.get(Calendar.MONTH) +1;
                        int day = date.get(Calendar.DAY_OF_MONTH);
                        String fileName = courierDetailsPath + "/courier-details-" + deliveryType + "-" + warehouseId + "-" + providerId + "-" + year + "-" + month + "-" + day +".xls";
                        response.setHeader("Content-disposition", "inline; filename=courier-details-" + deliveryType + "-" + warehouseId + "-" + providerId + "-" + year + "-"+ month + "-" + day +".xls" );
                        
                        ServletOutputStream sos;
                        try {
                                ByteArrayOutputStream baos = new ByteArrayOutputStream();
                                baos.write(FileUtils.getBytesFromFile(new File(fileName)));
                                sos = response.getOutputStream();
                                baos.writeTo(sos);
                                sos.flush();
                        } catch (IOException e) {
                                logger.error("Unable to stream the courier details report", e);
                        }       
                        return "authsuccess";
                }catch(NumberFormatException nfe){
                        logger.error("Unable to parse the warehouse id", nfe);
                }
                return "authfail";
        }
        
        public static void main(String[] args) {
            CourierDetailsController cdc = new CourierDetailsController();
            cdc.setAwbNumbers("4340987735");
            String msg = cdc.getAwbDetails();
            System.out.println(msg);
            //58539182004
            //43726980393
    }
        
        /**
         * Use this method to get details of a given awb number
         */
        public String getAwbDetails() {
            Order order = null;
            if(awbNumbers.isEmpty()) {
                setErrorMsg("Field cannot be empty");
                return "info";
            }
            List<AwbDetails> tempList = new ArrayList<AwbDetails>() ;
            String [] awbArray = awbNumbers.split(",");
            for(String awbNumber : awbArray) {
                try {
                    LogisticsClient lsc = new LogisticsClient();
                    TransactionClient tsc = new TransactionClient();
                    CatalogClient csc = new CatalogClient();
                    in.shop2020.model.v1.catalog.InventoryService.Client inventoryClient = csc.getClient();
                    in.shop2020.model.v1.order.TransactionService.Client txnClient = tsc.getClient();
                    in.shop2020.logistics.LogisticsService.Client logisticsClient = lsc.getClient();
                    
                    /*
                     * Get required stuff
                     * FIXME: Reduce service calls 
                     */
                    Provider provider = logisticsClient.getProvider(((Long)session.getAttribute("providerId")).longValue());
                    order = txnClient.getOrderForAwb(awbNumber);
                    Warehouse warehouse = inventoryClient.getWarehouse(order.getWarehouse_id());
                   
                    String accountNo;
                    if(order.isCod())
                         accountNo = provider.getDetails().get(DeliveryType.COD).getAccountNo();
                    else
                        accountNo = provider.getDetails().get(DeliveryType.PREPAID).getAccountNo();
                    
                    AwbDetails detailedAwb = new AwbDetails();
                    
                    detailedAwb.setAwbNumber(awbNumber);
                    detailedAwb.setAccountCode(accountNo);
                    detailedAwb.setAddress1(order.getCustomer_address1());
                    detailedAwb.setAddress2(order.getCustomer_address2());
                    detailedAwb.setAmountToCollect("" + order.getTotal_amount());
                    Date date = new Date(order.getPickup_timestamp());
                    detailedAwb.setAwbDate(date.toString());
                    detailedAwb.setCity(order.getCustomer_city());
                    detailedAwb.setCustomerName(order.getCustomer_name());
                    detailedAwb.setItemId("" + order.getLineitems().get(0).getId());
                    detailedAwb.setOrderId("" + order.getId());
                    detailedAwb.setPacketWeight("" + order.getTotal_weight());
                    if (order.isCod()) {
                        detailedAwb.setPaymentMode("COD");
                    } else {
                        detailedAwb.setPaymentMode("Prepaid");
                    }
                    
                    detailedAwb.setPhoneNumber("" + order.getCustomer_mobilenumber());
                    detailedAwb.setPickupLocation(warehouse.getLocation());
                    detailedAwb.setPinCode(order.getCustomer_pincode());
                    LineItem lineitem = order.getLineitems().get(0);
                    detailedAwb.setProductName(lineitem.getBrand() + " " + lineitem.getModel_name() + " " 
                                                 + lineitem.getItem_number() + " " + lineitem.getColor());
                    detailedAwb.setShipmentValue("" + order.getTotal_amount());
                    detailedAwb.setState(order.getCustomer_state());
                    
                    tempList.add(detailedAwb);

                } catch (TTransportException e) {
                    setErrorMsg("Your request cannot be processed due to technical error. Please try later.");
                } catch (TException e) {
                    setErrorMsg(e.getMessage());
                } catch (TransactionServiceException e) {
                    setErrorMsg(e.getMessage());
                } catch (LogisticsServiceException e) {
                    setErrorMsg(e.getMessage());
            } catch (InventoryServiceException e) {
                setErrorMsg(e.getMessage());
            }
        setDetailedAWBs(tempList);
            }
            return "info";
        }

    /**
         * Sets the daysToSubtract to -2 and then invokes the standard show() handler.
         * Should be used to view day before yesterday's courier details report.
         * 
     * @return the same string tokens as show
     */
        public String dayBefore(){
                daysToSubtract = -2;
                return show();
        }

    /**
     * Sets the daysToSubtract to -1 and then invokes the standard show()
     * handler. Should be used to view yesterday's courier details report.
     * 
     * @return the same string tokens as show
     */
        public String yesterday(){
                daysToSubtract = -1;
                return show();
        }

    /**
     * Sets the daysToSubtract to 0 and then invokes the standard show()
     * handler. Should be used to view today's courier details report.
     * 
     * @return the same string tokens as show
     */
        public String today(){
                daysToSubtract = 0;
                return show();
        }
                
        @Override
        public void setServletContext(ServletContext context) {
                this.context = context;
        }
        
        @Override
        public void setServletResponse(HttpServletResponse response) {
                this.response  = response;
        }

        @Override
        public void setServletRequest(HttpServletRequest request) {
                this.request = request;
                this.session = request.getSession();
        }

        public String getId(){
                return id;
        }
        
        public void setId(String id){
                this.id = id;
        }
        
        public String getSessionUserName(){
                return (String) session.getAttribute("username");
        }

    /**
     * Gets the list of all warehouses and maps the warehouse ids to their
     * display name.
     * 
     * @return the mapping of warehouse if to its display namee
     */
        public Map<Long, String> getWarehouses(){
                Map<Long, String> warehouseMap = new HashMap<Long, String>();
                try{
                        CatalogClient csc = new CatalogClient();
                        in.shop2020.model.v1.catalog.InventoryService.Client catalogClient= csc.getClient();
                        List<Warehouse> warehouses = catalogClient.getAllWarehouses(true);
                        for(Warehouse warehouse : warehouses){
                                // FIXME Remove the hardcoding
                                if(warehouse.getId() == 2 || warehouse.getId() == 5){
                                        continue;
                                }
                                warehouseMap.put(warehouse.getId(), warehouse.getDisplayName());
                        }
                }catch(Exception e){
                        logger.error("Error getting the list of warehouses", e);
                }
                return warehouseMap;
        }

        public String getServletContextPath(){
                return context.getContextPath();
        }

    public String getAwbNumbers() {
        return awbNumbers;
    }

    public void setAwbNumbers(String awbNumbers) {
        this.awbNumbers = awbNumbers;
    }

    public List<AwbDetails> getDetailedAWBs() {
        return detailedAWBs;
    }

    public void setDetailedAWBs(List<AwbDetails> detailedAWBs) {
        this.detailedAWBs = detailedAWBs;
    }

    public String getErrorMsg() {
        return errorMsg;
    }

    public void setErrorMsg(String errorMsg) {
        this.errorMsg = errorMsg;
    }
}