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.ArrayList;

import java.util.Collection;

import java.util.List;

import in.shop2020.support.utils.ReportsUtils;

import in.shop2020.thrift.clients.LogisticsClient;

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.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 AwbUploadController extends ActionSupport implements ServletRequestAware {

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

    private HttpServletRequest request;

    private HttpSession session;

    private File awbFile;

    private String awbFileContentType;

    private String awbFileFileName;

    private String errorMsg = "";

    public String index() {

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

            return "authfail";

        checkForErrors();

        return "authsuccess";

    }

    public String create() {

        File fileToCreate = null;

        try {

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

            FileUtils.copyFile(this.awbFile, fileToCreate);

        } catch (Exception e) {

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

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

        }

        long providerId = Long.parseLong(this.request.getParameter("providerId"));

        boolean isCod = Boolean.parseBoolean(this.request.getParameter("awbType"));

        if(checkForErrors())

            return "authsuccess";

        //Parse the file and submit the data for update to the logistics service

        Workbook wb = null;

        try {

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

        } catch (FileNotFoundException e) {

            logger.error("Unable to open the Awb report", e);

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

        } catch (IOException e) {

            logger.error("Unable to open the Awb report", e);

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

        }

        if(checkForErrors())

            return "authsuccess";

        List<String> awbs = new ArrayList<String>();

        Sheet sheet = wb.getSheetAt(0);

        Row firstRow = sheet.getRow(0);

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

        for (Row row : sheet) {

            if(row.equals(firstRow))

                continue;

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

            String awb = row.getCell(0).getStringCellValue();

            awbs.add(awb);

        }

        if(checkForErrors())

            return "authsuccess";

        try {

            LogisticsClient logisticsClient = new LogisticsClient();

            in.shop2020.logistics.LogisticsService.Client lsc = logisticsClient.getClient();

            lsc.addNewAwbs(providerId, isCod, awbs);

        } catch (TException e) {

        	logger.error("Unable to add AWBs", e);

            addActionError(e.getMessage());

        }

        checkForErrors();

        return "authsuccess";

    }

    @Override

    public void setServletRequest(HttpServletRequest request) {

        this.request = request;

        this.session = request.getSession();

    }

    public File getAwbFile() {

        return awbFile;

    }

    public void setAwbFile(File awbFile) {

        this.awbFile = awbFile;

    }

    public String getAwbFileContentType() {

        return awbFileContentType;

    }

    public void setAwbFileContentType(String awbFileContentType) {

        this.awbFileContentType = awbFileContentType;

    }

    public String getAwbFileFileName() {

        return awbFileFileName;

    }

    public void setAwbFileFileName(String awbFileFileName) {

        this.awbFileFileName = awbFileFileName;

    }

    public String getErrorMsg(){

        return this.errorMsg;

    }

    private boolean checkForErrors(){

        Collection<String> actionErrors = getActionErrors();

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

            for (String str : actionErrors) {

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

            }

            return true;

        }

        return false;

    }

}