Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.util.Collection;

import java.util.List;

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

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

import in.shop2020.model.v1.inventory.InventoryService.Client;

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

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

import in.shop2020.model.v1.inventory.VendorItemPricing;

import in.shop2020.support.utils.ReportsUtils;

import in.shop2020.thrift.clients.CatalogClient;

import in.shop2020.thrift.clients.InventoryClient;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpSession;

import org.apache.commons.io.FileUtils;

import org.apache.poi.hssf.usermodel.HSSFWorkbook;

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.convention.annotation.InterceptorRef;

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

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

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

import org.apache.struts2.interceptor.ServletRequestAware;

import org.apache.thrift.TException;

import org.apache.thrift.transport.TTransportException;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import com.opensymphony.xwork2.ActionSupport;

@SuppressWarnings("serial")

@InterceptorRefs({

	@InterceptorRef("defaultStack"),

	@InterceptorRef("login")

})

@Results({

	@Result(name="authfail", type="redirectAction", params = {"actionName" , "reports"})

})

public class VendorPriceController extends ActionSupport implements ServletRequestAware {

	private static Logger logger = LoggerFactory.getLogger(VendorPriceController.class);

	private HttpServletRequest request;

	private HttpSession session;

	private String id;

	private String errorMsg = "";

	long vendorId;

	private File pricingFile;

    private String pricingFileContentType;

    private String pricingFileFileName;

	public String index() {

		if(!ReportsUtils.canAccessReport((Long)session.getAttribute(ReportsUtils.ROLE), request.getServletPath()))

			return "authfail";

		return "authsuccess";

	}

	public String show() {

		return "authsuccess";

	}

	public String create(){

		File fileToCreate = null;

        try {

            fileToCreate = new File("/tmp/", this.pricingFileFileName);

            FileUtils.copyFile(this.pricingFile, fileToCreate);

        } catch (Exception e) {

           logger.error("Error while writing pricing file to the local file system", e);

           addActionError("Error while writing  pricing file report to the local file system");

        }

        Workbook wb = null;

        try {

            wb = new HSSFWorkbook(new FileInputStream(fileToCreate));

        } catch (FileNotFoundException e) {

            logger.error("Unable to open the pricing file", e);

            addActionError("Unable to open the pricing file. Please check the report format.");

        } catch (IOException e) {

        	logger.error("Unable to open the pricing file", e);

            addActionError("Unable to open the pricing file. Please check the report format.");

        }

        InventoryClient isc = null;

        CatalogClient csc = null;

        try {

            isc = new InventoryClient();

            csc = new CatalogClient("catalog_service_server_host_amazon","catalog_service_server_port");

        } catch (TTransportException e) {

            logger.error("Unable to establish connection to the transaction service", e);

            addActionError("Unable to establish connection to the transaction service");

        }

        Client invClient = isc.getClient();

        in.shop2020.model.v1.catalog.CatalogService.Client catClient = csc.getClient(); 

        Sheet sheet = wb.getSheetAt(0);

        Row firstRow = sheet.getRow(0);

        logger.info("Last row number is:" + sheet.getLastRowNum());

        int successful = 0, failed = 0;

        for (Row row : sheet) {

            if(row.equals(firstRow))

                continue;

            logger.info("Row no. " + row.getRowNum());

            long itemId = (long)row.getCell(0).getNumericCellValue();

            double mrp = Math.round(row.getCell(1).getNumericCellValue());

            double sellingPrice = Math.round(row.getCell(2).getNumericCellValue());

            double mop = Math.round(row.getCell(3).getNumericCellValue());

            double dealerPrice = Math.round(row.getCell(4).getNumericCellValue());

            double transferPrice = Math.round(row.getCell(5).getNumericCellValue());

            double nlc = Math.round(row.getCell(6).getNumericCellValue());

            try{

            	Item item = catClient.getItem(itemId);

            	if(sellingPrice != item.getSellingPrice() || mrp != item.getMrp()){

            		item.setSellingPrice(sellingPrice);

            		item.setMrp(mrp);

            		catClient.updateItem(item);

            	}

            	VendorItemPricing pricing = invClient.getItemPricing(itemId, vendorId);

            	pricing.setNlc(nlc);

            	pricing.setTransferPrice(transferPrice);

            	pricing.setDealerPrice(dealerPrice);

            	pricing.setMop(mop);

            	invClient.addVendorItemPricing(pricing);

            	successful++;

            } catch (Exception e) {

            	failed++;

            	addActionError(e.getMessage());

			}

        }

        addActionMessage("Updated pricing for " + successful + " items");

        addActionError("Unable to update " + failed + " items and received following errors.");

        return "authsuccess";

	}

	public File getPricingFile() {

        return pricingFile;

    }

    public void setPricingFile(File pricingFile) {

        this.pricingFile = pricingFile;

    }

	public String getPricingFileFileName() {

        return pricingFileFileName;

    }

    public void setPricingFileFileName(String pricingFileFileName) {

        this.pricingFileFileName = pricingFileFileName;

    }

	@Override

	public void setServletRequest(HttpServletRequest request) {

		this.request = request;

		this.session = request.getSession();

	}

	public String getErrorMsg(){

		Collection<String> actionErrors = getActionErrors();

        if(actionErrors != null && !actionErrors.isEmpty()){

            for (String str : actionErrors) {

                errorMsg += "<BR/>" + str;

            }   

        }

        return this.errorMsg;

	}

	public String getSuccessMsg(){

		String successMsg = "";

		Collection<String> actionMessages = getActionMessages();

        if(actionMessages != null && !actionMessages.isEmpty()){

            for (String str : actionMessages) {

            	successMsg += "<BR/>" + str;

            }   

        }

        return successMsg;

	}

	public long getVendorId() {

		return vendorId;

	}

	public void setVendorId(long vendorId) {

		this.vendorId = vendorId;

	}

	public void setId(String id) {

		this.id = id;

	}

	public String getId() {

		return id;

	}

	public List<Vendor> getAllVendors() throws TException{

		InventoryClient isc = null;

        try {

            isc = new InventoryClient();

        } catch (TTransportException e) {

            logger.error("Unable to establish connection to the transaction service", e);

            addActionError("Unable to establish connection to the transaction service");

        }

        Client invClient = isc.getClient();

        return invClient.getAllVendors();

	}

	public void setPricingFileContentType(String pricingFileContentType) {

		this.pricingFileContentType = pricingFileContentType;

	}

	public String getPricingFileContentType() {

		return pricingFileContentType;

	}

}