Subversion Repositories SmartDukaan

package in.shop2020.inventory.controllers;

import in.shop2020.model.v1.order.Order;
import in.shop2020.thrift.clients.InventoryClient;
import in.shop2020.thrift.clients.TransactionClient;

import java.io.BufferedInputStream;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.io.InputStream;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.servlet.ServletOutputStream;

import org.apache.commons.lang.StringUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

public class PriceDropController extends BaseController{
        
        private TransactionClient tsc;
        private in.shop2020.model.v1.order.TransactionService.Client tClient;
        
        private InventoryClient ic;
        private in.shop2020.model.v1.inventory.InventoryService.Client iClient;
        
        private static Log logger = LogFactory.getLog(PurchaseReportController.class);
        private String closingDate;
        private int itemId;
    public int getItemId() {
                return itemId;
        }

        public void setItemId(int itemId) {
                this.itemId = itemId;
        }

        private final DateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
    
    
    public String index() {
        return INDEX;
    }
    
    public String downloadImeis() throws Exception {
        long closingDateLong;
                try {
                        closingDateLong = sdf.parse(this.closingDate).getTime();
                } catch (Exception e) {
                        this.addActionError("Invalid date format");
                        return INDEX;
                }
                tsc = new TransactionClient();
                tClient = tsc.getClient();
                
                List<Order> inTransitOrders;

                inTransitOrders = tClient.getInTransitOrdersOnDateByItemId(closingDateLong, this.itemId);
                
                Set<Long> warehouseSet = new HashSet<Long>();
                for(Order o: inTransitOrders) {
                        warehouseSet.add(o.getFulfilmentWarehouseId());
                }
                
                ic = new InventoryClient();
                iClient = ic.getClient();
                
                Map<Long, String> warehouseNameMap = new HashMap<Long, String>(); 
                for (Long warehouseId : warehouseSet) {
                        warehouseNameMap.put(warehouseId, iClient.getWarehouseName(warehouseId));
                }

                byte[] buffer = null;
        File file = createFile(inTransitOrders, warehouseNameMap);
        buffer = new byte[(int) file.length()];
        InputStream input = null;
        try {
            int totalBytesRead = 0;
            input = new BufferedInputStream(new FileInputStream(file));
            while (totalBytesRead < buffer.length) {
                int bytesRemaining = buffer.length - totalBytesRead;
                // input.read() returns -1, 0, or more :
                int bytesRead = input.read(buffer, totalBytesRead,
                        bytesRemaining);
                if (bytesRead > 0) {
                    totalBytesRead = totalBytesRead + bytesRead;
                }
            }
            /*
             * the above style is a bit tricky: it places bytes into the
             * 'buffer' array; 'buffer' is an output parameter; the while
             * loop usually has a single iteration only.
             */
        } finally {
            input.close();
        }

        response.setContentType("application/vnd.ms-excel");
        response.setHeader("Content-disposition", "inline; filename="
                + file.getName());

        ServletOutputStream sos = response.getOutputStream();
        sos.write(buffer);
        sos.flush();
                
        return null;
    }

        private File createFile(List<Order> inTransitOrders, Map<Long, String> warehouseNameMap) {
                try {
                String tmpDir = System.getProperty("java.io.tmpdir");
            File file = new File(tmpDir + "/IntransitIMEIS-" + this.closingDate + this.itemId
                    + ".xls");
            BufferedWriter bufferedWriter = new BufferedWriter(new FileWriter(
                    file));
            bufferedWriter.write(StringUtils.join(new String[] { "Order Id",
                    "Vendor Name", "Brand", "ModelName", "ModelNumber", "Color", "UnitPrice", "IMEI"},
                    '\t'));

            for (Order order : inTransitOrders) {
                bufferedWriter.newLine();
                in.shop2020.model.v1.order.LineItem lineitem = order.getLineitems().get(0);
                for(String serialNumber : lineitem.getSerial_number().split(",")){
                        bufferedWriter.write(StringUtils.join(
                                        new String[] {
                                                        String.valueOf(order.getId()),
                                                        warehouseNameMap.get(order.getFulfilmentWarehouseId()),
                                                        lineitem.getBrand(),
                                                        lineitem.getModel_name(),
                                                        lineitem.getModel_number(),
                                                        lineitem.getColor(),
                                                        String.valueOf(lineitem.getUnit_price()),
                                                        serialNumber,
                                                        order.getStatusDescription() }, '\t'));
                }
            }

            bufferedWriter.close();
            return file;
        } catch (Exception e) {
            return null;
        }
        }

        public void setClosingDate(String closingDate) {
                this.closingDate = closingDate;
        }

        public String getClosingDate() {
                return closingDate;
        }
}