Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

import in.shop2020.support.services.CourierDetailsGenerator;
import in.shop2020.support.services.CourierDetailsReportMerger;
import in.shop2020.support.services.ManifestGenerator;

import java.io.ByteArrayOutputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Calendar;
import java.util.GregorianCalendar;
import java.util.HashMap;
import java.util.Map;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.struts2.interceptor.ServletRequestAware;
import org.apache.struts2.interceptor.ServletResponseAware;
import org.apache.struts2.rest.DefaultHttpHeaders;
import org.apache.struts2.rest.HttpHeaders;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ManifestController implements ServletResponseAware, ServletRequestAware {
        
    private static Logger logger = LoggerFactory.getLogger(ManifestController.class);
    
        //FIXME: Read this configuration from the config client
        private String courierDetailsPath = "/CourierDetailReports";
        private long warehouseId;
        private long providerId;
        private boolean isCod;
        
        private HttpServletRequest request;
        private HttpServletResponse response;

        public HttpHeaders index(){
                this.warehouseId = Long.parseLong(request.getParameter("warehouseID"));
                this.providerId = Long.parseLong(request.getParameter("providerID"));
                try {
                    this.isCod = Boolean.parseBoolean(request.getParameter("isCod"));
                } catch (Exception e) {
                    this.isCod = false;
                }
                
                logger.debug("Warehouse Id is:  " + warehouseId);
                logger.debug("Provider Id is: " + providerId);
                logger.debug("Cod is: " + isCod);
                
                Calendar date = new GregorianCalendar();
                int year = date.get(Calendar.YEAR);
                int month = date.get(Calendar.MONTH) +1;
                int day = date.get(Calendar.DAY_OF_MONTH);
                
                String fileNameSuffix = "-" + warehouseId + "-"+ providerId + "-" + year + "-" + month + "-" + day;
                String mergedFileNameSuffix = "-" + 0 + "-"+ providerId + "-" + year + "-" + month + "-" + day;
                if(isCod){
                    fileNameSuffix = "cod" + fileNameSuffix ;
                        mergedFileNameSuffix = "cod" + mergedFileNameSuffix;
                }
                else{
                    fileNameSuffix = "prepaid" + fileNameSuffix;
                        mergedFileNameSuffix = "prepaid" + mergedFileNameSuffix;
                }
                
                ManifestGenerator manifestGenerator = new ManifestGenerator();
                ByteArrayOutputStream baos = manifestGenerator.generateManifestFile(warehouseId, providerId, isCod);
                response.setContentType("application/pdf");
                
                CourierDetailsGenerator courierDetailsGenerator = new CourierDetailsGenerator();
                ByteArrayOutputStream baosXLS = courierDetailsGenerator.generateCourierDetails(warehouseId, providerId, isCod);
                try {
                        String fileName = courierDetailsPath + "/courier-details-" + fileNameSuffix + ".xls";
                        FileOutputStream f = new FileOutputStream(fileName);
                        baosXLS.writeTo(f);
                        f.close();
                } catch (FileNotFoundException e) {
                        logger.error("Unable to create the courier details file", e);
                } catch (IOException e) {
                    logger.error("Unable to create the courier details file", e);
                }
                
                // FIXME This is not a useful way. We need to fix it asap. This is done just to give ease to courier company.
                CourierDetailsReportMerger merger = new CourierDetailsReportMerger();
                try {
                        FileOutputStream f = new FileOutputStream(courierDetailsPath + "/courier-details-" + mergedFileNameSuffix + ".xls");
                        Map<Long, String> warehouseIdFileNames = new HashMap<Long, String>();
                        
                        String p1;
                        if(isCod){
                            p1 = "cod";
                        }
                        else{
                            p1 = "prepaid";
                        }
                        String fName;
                        fName = courierDetailsPath + "/courier-details-" + p1 + "-" + 1 + "-"+ providerId + "-" + year + "-" + month + "-" + day + ".xls";
                        warehouseIdFileNames.put(1L, fName);
                        fName = courierDetailsPath + "/courier-details-" + p1 + "-" + 2 + "-"+ providerId + "-" + year + "-" + month + "-" + day + ".xls";
                        warehouseIdFileNames.put(2L, fName);
                        fName = courierDetailsPath + "/courier-details-" + p1 + "-" + 5 + "-"+ providerId + "-" + year + "-" + month + "-" + day + ".xls";
                        warehouseIdFileNames.put(5L, fName);
                        ByteArrayOutputStream binXLS = merger.mergeCourierDetailsReports(warehouseIdFileNames, 1, true);
                        binXLS.writeTo(f);
                        f.close();
                } catch (FileNotFoundException e) {
                        logger.error("Error while creating the Courier Details report", e);
                } catch (IOException e) {
                        logger.error("IO error while writing the Courier Details report", e);
                }
                
                
                response.setHeader("Content-disposition", "inline; filename=manifest-" + fileNameSuffix + ".pdf" );
                
                ServletOutputStream sos;
                try {
                        sos = response.getOutputStream();
                        baos.writeTo(sos);
                        sos.flush();
                } catch (IOException e) {
                    logger.error("Unable to stream the manifest file", e);
                }       
                return new DefaultHttpHeaders("lsuccess");
        }
        
        @Override
        public void setServletResponse(HttpServletResponse response) {
                this.response  = response;
        }

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