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 

	             */

	            Awb awb = logisticsClient.getAwbForNumber(awbNumber);

	            Provider provider = logisticsClient.getProvider(awb.getProviderId());

	            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;

    }

}