Subversion Repositories SmartDukaan

/**

 * 

 */

package in.shop2020.inventory.controllers;

import in.shop2020.model.v1.catalog.InventoryServiceException;

import in.shop2020.model.v1.catalog.Vendor;

import in.shop2020.purchase.PurchaseOrder;

import in.shop2020.purchase.PurchaseService.Client;

import in.shop2020.purchase.PurchaseServiceException;

import in.shop2020.purchase.Supplier;

import in.shop2020.thrift.clients.CatalogClient;

import in.shop2020.thrift.clients.PurchaseClient;

import java.io.BufferedInputStream;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.util.Calendar;

import java.util.List;

import javax.servlet.ServletOutputStream;

import javax.servlet.http.HttpServletResponse;

import org.apache.struts2.convention.annotation.Result;

import org.apache.struts2.convention.annotation.Results;

import org.apache.thrift.TException;

import org.apache.thrift.transport.TTransportException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

@SuppressWarnings("serial")

@Results({ @Result(name = "index", location = "raisePO-index.vm"),

        @Result(name = "result", location = "raisePO-result.vm") })

public class RaisePOController extends BaseController {

    private static Logger log = LoggerFactory

            .getLogger(RaisePOController.class);

    private File purchaseOrderDetails;

    private String vendorId;

    private String result;

    private List<Vendor> vendors;

    private HttpServletResponse response;

    public String index() {

        CatalogClient csc;

        try {

            csc = new CatalogClient();

            in.shop2020.model.v1.catalog.InventoryService.Client catalogClient = csc

                    .getClient();

            setVendors(catalogClient.getAllVendors());

        } catch (TTransportException e) {

            e.printStackTrace();

            setResult(e.getMessage());

        } catch (TException e) {

            e.printStackTrace();

            setResult(e.getMessage());

        }

        return "index";

    }

    /*

     * MAIN METHOD TO TEST THE CODE

     * 

     * public static void main(String[] args) { RaisePOController rpc = new

     * RaisePOController(); rpc.setPurchaseOrderDetails(new

     * File("/home/anupam/Desktop/po.xls")); rpc.setVendorId("3"); rpc.create();

     * System.out.println(rpc.getResult()); }

     */

    public String create() {

        try {

            /*

             * Save the uploaded file to disk

             */

            FileInputStream is = new FileInputStream(purchaseOrderDetails);

            String filename = "/tmp/po-"

                    + Calendar.getInstance().getTime().toString() + ".xls";

            File f = new File(filename);

            FileOutputStream fos = null;

            fos = new FileOutputStream(f);

            byte[] buf = new byte[4096];

            int bytesRead;

            while ((bytesRead = is.read(buf)) != -1) {

                fos.write(buf, 0, bytesRead);

            }

            fos.close();

            int vendor = Integer.parseInt(vendorId);

            CatalogClient csc = new CatalogClient();

            in.shop2020.model.v1.catalog.InventoryService.Client catalogClient = csc

                    .getClient();

            setVendors(catalogClient.getAllVendors());

            if (vendor < 1 || vendor > vendors.size()) {

                setResult("Select a valid Vendor");

                return "result";

            }

            /*

             * Try to create the purchase order for given vendor

             */

            Long purchaseOrderId = null;

            String output = catalogClient

                    .processPurchaseOrder(filename, vendor);

            /*

             * If PO is generated then create the PDF receipt for it

             */

            if (output.contains("POId:")) {

                String[] arr = output.split(":");

                purchaseOrderId = Long.parseLong(arr[1]);

                PurchaseClient warehouseServiceClient = new PurchaseClient();

                Client client = warehouseServiceClient.getClient();

                PurchaseOrder purchaseOrder = client

                        .getPurchaseOrder(purchaseOrderId);

                Supplier supplier = client.getSupplier(purchaseOrder

                        .getSupplierId());

                String pdfFilename = in.shop2020.inventory.service.PdfPoSheetGenerator

                        .generatePdfSheet(purchaseOrder, supplier);

                /*

                 * Put pdf file in a buffer to send it in response

                 */

                File file = new File(pdfFilename);

                byte[] buffer = new byte[(int) file.length()];

                try {

                    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 bytesRead1 = input.read(buffer, totalBytesRead,

                                    bytesRemaining);

                            if (bytesRead1 > 0) {

                                totalBytesRead = totalBytesRead + bytesRead1;

                            }

                        }

                        /*

                         * 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();

                    }

                } catch (FileNotFoundException ex) {

                    log.info("FILE NOT FOUND : " + pdfFilename, ex);

                    setResult(ex.getMessage());

                } catch (IOException ex) {

                    log.info("FILE NOT FOUND : " + pdfFilename);

                    setResult(ex.getMessage());

                }

                response.setHeader("Content-disposition",

                        "attachment; filename=" + pdfFilename);

                /*

                 * Send the pdf file in response

                 */

                ServletOutputStream sos;

                sos = response.getOutputStream();

                sos.write(buffer);

                sos.flush();

            }

        } catch (TTransportException e) {

            e.printStackTrace();

            setResult(e.getMessage());

        } catch (NumberFormatException e) {

            e.printStackTrace();

            setResult(e.getMessage());

        } catch (TException e) {

            e.printStackTrace();

            setResult(e.getMessage());

        } catch (InventoryServiceException e) {

            setResult(e.getMessage());

        } catch (IOException e) {

            log.error("Error occured");

            setResult(e.getMessage());

        } catch (PurchaseServiceException e) {

            setResult(e.getMessage());

        }

        return "result";

    }

    public File getPurchaseOrderDetails() {

        return purchaseOrderDetails;

    }

    public void setPurchaseOrderDetails(File purchaseOrderDetails) {

        this.purchaseOrderDetails = purchaseOrderDetails;

    }

    public String getVendorId() {

        return vendorId;

    }

    public void setVendorId(String vendorId) {

        this.vendorId = vendorId;

    }

    public void setResult(String result) {

        this.result = result;

    }

    public String getResult() {

        return result;

    }

    public void setServletResponse(HttpServletResponse response) {

        this.response = response;

    }

    public List<Vendor> getVendors() {

        return vendors;

    }

    public void setVendors(List<Vendor> vendors) {

        this.vendors = vendors;

    }

}