Subversion Repositories SmartDukaan

package in.shop2020.support.controllers;

import in.shop2020.model.v1.catalog.CatalogService.Client;
import in.shop2020.model.v1.catalog.CompetitorPricing;
import in.shop2020.model.v1.catalog.Item;
import in.shop2020.support.utils.ReportsUtils;
import in.shop2020.thrift.clients.CatalogClient;

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.GregorianCalendar;
import java.util.HashMap;
import java.util.List;
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.commons.io.FileUtils;
import org.apache.poi.hssf.usermodel.HSSFRow;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
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.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;

import com.opensymphony.xwork2.ValidationAwareSupport;

@SuppressWarnings({"unused","deprecation"})

@InterceptorRefs({
    @InterceptorRef("defaultStack"),
    @InterceptorRef("login")
})
@Results({
    @Result(name = "redirect", location = "${url}", type = "redirect"),
    @Result(name="authsuccess", type="redirectAction", params = {"actionName" , "reports"})
})

public class CompetitionScrapingController extends ValidationAwareSupport implements ServletRequestAware ,ServletResponseAware, ServletContextAware{

    /**
     * 
     */
    
    private static Logger logger = LoggerFactory.getLogger(CompetitionScrapingController.class);
    
    private HttpServletRequest request;
    private HttpServletResponse response;
    private HttpSession session;
    private ServletContext context;
    private static final long serialVersionUID = -4056242323967836986L;
    private File file;
    private List<CompetitorPricing> previousUploads;
    private String[] snapdeal;
    private String[] flipkart;
    private String[] amazon;
    private String[] itemId;
    private String requestId;
    private List<CompetitorPricing> data;
    
    
    public String processItems() throws TException{
        long uploadId = getTimeInMilliseconds();
        String userEmail = getUserEmail();
        Map<Long,CompetitorPricing> itemScrapingMap = new HashMap<Long,CompetitorPricing>();
        for(int i=0;i<itemId.length;i++){
            CompetitorPricing cp = new CompetitorPricing(); 
            cp.setItem_id(Long.valueOf(itemId[i]));
            cp.setUploadId(uploadId);
            cp.setUserEmail(userEmail);
            cp.setIsSnapdeal(Boolean.valueOf(snapdeal[i]));
            cp.setIsFlipkart(Boolean.valueOf(flipkart[i]));
            cp.setIsAmazon(Boolean.valueOf(amazon[i]));
            itemScrapingMap.put(cp.getItem_id(), cp);
        }
        Client cc = new CatalogClient("catalog_service_server_host_amazon","catalog_service_server_port").getClient();
        cc.addCompetitorScraping(itemScrapingMap);
        return getPreviousData();
    }

    public String index() throws TException{
        if (!ReportsUtils.canAccessReport((Long) session.getAttribute(ReportsUtils.ROLE),request.getServletPath())) {
            return "authfail";
        }
        return "index";
    }
    
    public String getUploadData() throws NumberFormatException, TException{
        Client cc = new CatalogClient("catalog_service_server_host_amazon","catalog_service_server_port").getClient();
        data = cc.getUploadResultById(Long.valueOf(this.requestId));
        return "show";
    }
    
    public String getPreviousData() throws TException{
        Client cc = new CatalogClient("catalog_service_server_host_amazon","catalog_service_server_port").getClient();
        setPreviousUploads(cc.getPreviousCompetitorScraping(10));
        return "previous";
    }
    
    public void downloadReport() throws NumberFormatException, TException, IOException{
        File file = new File("/tmp/"+this.requestId+".xls");
        HSSFWorkbook hwb=new HSSFWorkbook();
        HSSFSheet sheet =  hwb.createSheet(this.requestId);
        HSSFRow rowhead=   sheet.createRow((short)0);
        rowhead.createCell((short) 0).setCellValue("ITEM-ID");
        rowhead.createCell((short) 1).setCellValue("PRODUCT");
        rowhead.createCell((short) 2).setCellValue("Our Snapdeal SP");
        rowhead.createCell((short) 3).setCellValue("Our Snapdeal Offer Price");
        rowhead.createCell((short) 4).setCellValue("Lowest Snapdeal Seller");
        rowhead.createCell((short) 5).setCellValue("Lowest Snapdeal SP");
        rowhead.createCell((short) 6).setCellValue("Lowest Offer Price");
        rowhead.createCell((short) 7).setCellValue("Our Snapdeal Inventory");
        rowhead.createCell((short) 8).setCellValue("Lowest Seller Snapdeal Inventory");
        rowhead.createCell((short) 9).setCellValue("Our Flipkart SP");
        rowhead.createCell((short) 10).setCellValue("Lowest Seller Flipkart");
        rowhead.createCell((short) 11).setCellValue("Lowest Flipkart SP");
        rowhead.createCell((short) 12).setCellValue("Our Flipkart Inventory");
        rowhead.createCell((short) 13).setCellValue("Our Price");
        rowhead.createCell((short) 14).setCellValue("Lowest Amazon Seller");
        rowhead.createCell((short) 15).setCellValue("Lowest Amazon Price");
        Client cc = new CatalogClient("catalog_service_server_host_amazon","catalog_service_server_port").getClient();
        data = cc.getUploadResultById(Long.valueOf(this.requestId));
        int iterator=1;
        for (CompetitorPricing item : data){
            HSSFRow row = sheet.createRow((short)iterator);
            row.createCell((short) 0).setCellValue(item.getItem_id());
            row.createCell((short) 1).setCellValue(item.getProductName());
            row.createCell((short) 2).setCellValue(item.getOurSnapdealPrice());
            row.createCell((short) 3).setCellValue(item.getOurSnapdealOfferPrice());
            row.createCell((short) 4).setCellValue(item.getLowestSnapdealSeller());
            row.createCell((short) 5).setCellValue(item.getLowestSnapdealPrice());
            row.createCell((short) 6).setCellValue(item.getLowestSnapdealOfferPrice());
            row.createCell((short) 7).setCellValue(item.getOurSnapdealInventory());
            row.createCell((short) 8).setCellValue(item.getLowestSnapdealSellerInventory());
            row.createCell((short) 9).setCellValue(item.getOurFlipkartPrice());
            row.createCell((short) 10).setCellValue(item.getLowestFlipkartSeller());
            row.createCell((short) 11).setCellValue(item.getLowestFlipkartPrice());
            row.createCell((short) 12).setCellValue(item.getOurFlipkartInventory());
            row.createCell((short) 13).setCellValue(item.getOurAmazonPrice());
            row.createCell((short) 14).setCellValue(item.getLowestAmazonSeller());
            row.createCell((short) 15).setCellValue(item.getLowestAmazonPrice());
            iterator++;
        }
        FileOutputStream fileOut = null;
        try {
            fileOut = new FileOutputStream(file);
        } catch (FileNotFoundException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        try {
            hwb.write(fileOut);
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        try {
            fileOut.close();
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        byte[] 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;
                int bytesRead = input.read(buffer, totalBytesRead, bytesRemaining); 
                if (bytesRead > 0){
                    totalBytesRead = totalBytesRead + bytesRead;
                }
            }
        }
        finally {
            input.close();
            file.delete();
        }

        response.setHeader("Content-Disposition", "attachment; filename="+this.requestId+".xls");
        response.setContentType("application/octet-stream");
        ServletOutputStream sos;
        try {
            sos = response.getOutputStream();
            sos.write(buffer);
            sos.flush();
        } catch (IOException e) {
            System.out.println("Unable to stream the manifest file");
        }   
    }

    public String uploadSheet(){
        return "upload";
    }
    
    public String addItems(){
        return "addItem";
    }

    public String uploadBulkFile() throws IOException, TException{
        long uploadId = getTimeInMilliseconds();
        String userEmail = getUserEmail();
        File fileToCreate = new File("/temp-uploads", "competition-scraping-"+String.valueOf(uploadId)+".xls");
        FileUtils.copyFile(this.file, fileToCreate);
        FileInputStream iFile = new FileInputStream(new File(file.getAbsolutePath()));
        HSSFWorkbook workbook = new HSSFWorkbook(iFile);
        HSSFSheet sheet = workbook.getSheetAt(0);
        Map<Long,CompetitorPricing> itemScrapingMap = new HashMap<Long,CompetitorPricing>();
        for (int iterator=(sheet.getFirstRowNum()+1);iterator<=sheet.getLastRowNum();iterator++){
            if (sheet.getLastRowNum()-1>100){
                logger.info("Greater than 100, returning");
                addActionError("Sku's cant be greater than 100");
                return "redirect";
            }
            CompetitorPricing cp = new CompetitorPricing(); 
            if (checkEmptyString(sheet.getRow(iterator).getCell(0))){
                continue;
            }
            else {
                cp.setItem_id((long) sheet.getRow(iterator).getCell(0).getNumericCellValue());
            }
            
            if (!checkEmptyString(sheet.getRow(iterator).getCell(1))){
                if ((long)sheet.getRow(iterator).getCell(1).getNumericCellValue()==1){
                    cp.setIsSnapdeal(true);
                }
            }
            
            if (!checkEmptyString(sheet.getRow(iterator).getCell(2))){
                if ((long)sheet.getRow(iterator).getCell(2).getNumericCellValue()==1){
                    cp.setIsFlipkart(true);
                }
            }
            if (!checkEmptyString(sheet.getRow(iterator).getCell(3))){
                if ((long)sheet.getRow(iterator).getCell(3).getNumericCellValue()==1){
                    cp.setIsAmazon(true);
                }
            }
            cp.setUploadId(uploadId);
            cp.setUserEmail(userEmail);
            itemScrapingMap.put(cp.getItem_id(), cp);
        }
        Client cc = new CatalogClient("catalog_service_server_host_amazon","catalog_service_server_port").getClient();
        cc.addCompetitorScraping(itemScrapingMap);
        return getPreviousData();
    }
    
    public boolean checkEmptyString(Cell cell){
        if (cell==null){
            return true;
        }
        return false;
    }

    public long getTimeInMilliseconds(){
        Calendar cal=GregorianCalendar.getInstance();
        return cal.getTimeInMillis();
    }
    
    public String getTimestamp(long timestamp){
        return new java.util.Date(timestamp).toLocaleString();
    }
    
    public String getUserEmail(){
        logger.info(session.getAttribute(ReportsUtils.USER_NAME).toString());
        logger.info(session.getAttribute(ReportsUtils.USER_EMAIL).toString());
        return session.getAttribute(ReportsUtils.USER_EMAIL).toString();
    }


    public File getFile() {
        return file;
    }

    public void setFile(File file) {
        this.file = file;
    }
    
    public void setPreviousUploads(List<CompetitorPricing> previousUploads) {
        this.previousUploads = previousUploads;
    }

    public List<CompetitorPricing> getPreviousUploads() {
        return previousUploads;
    }

    public HttpServletRequest getRequest() {
        logger.info("set request"+request.toString());
        return request;
    }

    public void setRequest(HttpServletRequest request) {
        this.request = request;
    }

    public HttpServletResponse getResponse() {
        return response;
    }

    public void setResponse(HttpServletResponse response) {
        this.response = response;
    }

    public HttpSession getSession() {
        return session;
    }

    public void setSession(HttpSession session) {
        this.session = session;
    }

    public ServletContext getContext() {
        return context;
    }

    public void setContext(ServletContext context) {
        this.context = context;
    }

    public void setServletRequest(HttpServletRequest req) {
        this.request = req;
        this.session = req.getSession();        
    }

    public void setServletContext(ServletContext arg0) {
        // TODO Auto-generated method stub

    }

    public void setServletResponse(HttpServletResponse response) {
        this.response = response;
    }

    public void setSnapdeal(String[] snapdeal) {
        this.snapdeal = snapdeal;
    }

    public String[] getSnapdeal() {
        return snapdeal;
    }

    public void setFlipkart(String[] flipkart) {
        this.flipkart = flipkart;
    }

    public String[] getFlipkart() {
        return flipkart;
    }

    public void setAmazon(String[] amazon) {
        this.amazon = amazon;
    }

    public String[] getAmazon() {
        return amazon;
    }

    public void setItemId(String[] itemId) {
        this.itemId = itemId;
    }

    public String[] getItemId() {
        return itemId;
    }

   
    public List<CompetitorPricing> getData() {
        return data;
    }

    public void setData(List<CompetitorPricing> data) {
        this.data = data;
    }

    public String getRequestId() {
        return requestId;
    }

    public void setRequestId(String requestId) {
        this.requestId = requestId;
    }

    



}