Subversion Repositories SmartDukaan

/**
 * 
 */
package in.shop2020.support.controllers;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;

import org.apache.commons.io.FileUtils;
import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.mime.MultipartEntity;
import org.apache.http.entity.mime.content.FileBody;
import org.apache.http.entity.mime.content.StringBody;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
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.dispatcher.StreamResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.opensymphony.xwork2.ActionSupport;

/**
 * @author mandeep
 * 
 */
public class FileArchiveController extends ActionSupport {
    /**
     * 
     */
    private static final String ARCHIVE_DIR = "/InventoryReports/";
    private static Logger logger = LoggerFactory.getLogger(FileArchiveController.class);
    
    private static enum FILE_TYPES {
        SARVOTTAM_INVENTORY_TSV_FILE(2, 175),
        SNCA_INVENTORY_XLS_FILE(1, 193);

        int type;
        long warehouseId;

        FILE_TYPES(int type, long warehouseId) {
            this.type = type;
            this.warehouseId = warehouseId;
        }

        int getType() {
            return type;
        }
        
        long getWarehouseId() {
            return warehouseId;
        }
    };

    private File file;
    private int fileType;
    private SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd-HH:mm:ss");

    public StreamResult create() {
        try {
            logger.info("Trying to archive file with type: " + fileType);
            File newFile = new File(ARCHIVE_DIR + fileType + "-"
                    + sdf.format(new Date()) + ".xls");
            FileUtils.copyFile(file, newFile);

            // For Sarvottam, convert to xls
            if (fileType == FILE_TYPES.SARVOTTAM_INVENTORY_TSV_FILE.getType()) {
                File file = new File(convertToXls(newFile));
                uploadInventoryFile(file, FILE_TYPES.SARVOTTAM_INVENTORY_TSV_FILE.getWarehouseId());
            }

            return new StreamResult(new ByteArrayInputStream(
                    ("File uploaded successfully at " + sdf.format(new Date()))
                            .getBytes()));

        } catch (IOException e) {
            logger.error("Error archiving file", e);
            return new StreamResult(new ByteArrayInputStream(
                    ("Could not upload file at " + sdf.format(new Date()))
                            .getBytes()));
        }
    }

    /**
     * @param file
     */
    private void uploadInventoryFile(File file, long warehouseId) {
        try {
            String inventoryUploadUrl = "http://localhost:8080/inventory/supplier-inventory";
            HttpPost inventoryUploadPost = new HttpPost(inventoryUploadUrl);
            MultipartEntity entity = new MultipartEntity();
            entity.addPart("warehouseId", new StringBody(String.valueOf(warehouseId)));
            entity.addPart("file", new FileBody(file));
            inventoryUploadPost.setEntity(entity);

            HttpClient client = new DefaultHttpClient();
            HttpResponse response = client.execute(inventoryUploadPost);
            byte[] b = new byte[1000];
            response.getEntity().getContent().read(b);
            System.out.println(new String(b));
            logger.info("Response: " + new String(b));
        } catch (Exception e) {
            logger.error("Could not upload inventory: " + e);
        }
    }

    /**
     * @param tsvFile
     */
    private String convertToXls(File tsvFile) {
        try {
            Workbook w = new HSSFWorkbook();
            String convertedFileName = ARCHIVE_DIR + "Converted-"
                    + tsvFile.getName();
            FileOutputStream fileOut = new FileOutputStream(convertedFileName);
            Sheet sheet = w.createSheet();
            BufferedReader br = new BufferedReader(new FileReader(tsvFile));
            String line = br.readLine();
            int i = 0;
            while (line != null) {
                Row row = sheet.createRow(i++);
                int j = 0;
                for (String s : line.split("\t")) {
                    Cell cell = row.createCell(j++);
                    System.out.println("Writing: " + s);
                    cell.setCellValue(s);
                }
                line = br.readLine();
            }
            w.write(fileOut);
            return convertedFileName;
        } catch (Exception e) {
            logger.error("Error converting file", e);
            return tsvFile.getName();
        }
    }

    public File getFile() {
        return file;
    }

    public void setFile(File file) {
        this.file = file;
    }

    public int getFileType() {
        return fileType;
    }

    public void setFileType(int fileType) {
        this.fileType = fileType;
    }

    public static void main(String[] args) {
        File file = new File("/home/mandeep/2-2012-08-11-14:58:00.xls");
        new FileArchiveController().uploadInventoryFile(file, 175);
        System.out.println("done");
    }
}