Subversion Repositories SmartDukaan

package in.shop2020.inventory.service;

import in.shop2020.thrift.clients.WarehouseClient;

import in.shop2020.utils.GmailUtils;

import in.shop2020.warehouse.InventoryAvailability;

import in.shop2020.warehouse.WarehouseService.Client;

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileReader;

import java.io.FileWriter;

import java.io.IOException;

import java.util.ArrayList;

import java.util.Collections;

import java.util.Comparator;

import java.util.Date;

import java.util.List;

import org.apache.commons.lang.StringUtils;

import org.apache.thrift.TException;

import org.slf4j.LoggerFactory;

public class StockSheetSender{

	public static org.slf4j.Logger logger = LoggerFactory.getLogger(StockSheetSender.class);

	public static void main(String[] args) {

		List<InventoryAvailability> serializedInventoryAvailability = new ArrayList<InventoryAvailability>();

		List<InventoryAvailability> nonSerializedInventoryAvailability = new ArrayList<InventoryAvailability>();

		List<InventoryAvailability> inventoryAvailability = new ArrayList<InventoryAvailability>();

		String emailFromAddress = "build-staging@shop2020.in";

		String password = "shop2020";

		GmailUtils mailer = new GmailUtils();

		//String sendTo[] = new String[]{"manish.sharma@shop2020.in"};

		String sendTo[] = new String[]{ "rajneesh.arora@saholic.com","amit.sirohi@shop2020.in","chaitnaya.vats@saholic.com","venky864@gmail.com",

				"ritesh.chauhan@shop2020.in","khushal.bhatia@saholic.com","himanshu.pandey@shop2020.in", "shailesh.kumar@shop2020.in"};

		File file = null;

		try {

			Client warehouseClient = new WarehouseClient().getClient();

			serializedInventoryAvailability = warehouseClient.getCurrentSerializedInventoryByScans(3298L);

			nonSerializedInventoryAvailability  = warehouseClient.getCurrentNonSerializedInventoryByScans(3298L);

		} catch(TException ex) {

			logger.error(ex.getMessage(), ex);

		}

		inventoryAvailability.addAll(serializedInventoryAvailability);

		inventoryAvailability.addAll(nonSerializedInventoryAvailability);

		int quantity= 0;

		for(InventoryAvailability inva : inventoryAvailability){

			quantity += inva.getQuantity();

		}

		try {

			file = createInventoryAvailabilityFile(inventoryAvailability);

			String htmlFileName = getHtmlFileGenrator(file, "Stock Sheet - Hyderabad, Total Quantity-" + quantity );

			List<File> files  = new ArrayList<File>();

			files.add(file);

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

				mailer.sendSSLMessage(sendTo, "Stock Sheet - Hyderabad", " ", emailFromAddress, password, files, htmlFileName);

			}else {

				mailer.sendSSLMessage(sendTo, "Stock Sheet - Hyderabad", " ", emailFromAddress, password, files);

			}

		}

		catch(Exception e) {

			logger.error(e.getMessage(), e);

		}

	}

	private static File createInventoryAvailabilityFile(List<InventoryAvailability> inventoryAvailability) throws IOException {

		File file = new File("/tmp/SaholicInventory-Hyderabad.xls");

		BufferedWriter bufferedWriter = new BufferedWriter(new FileWriter(file));

		bufferedWriter.write(StringUtils.join(new String[] {

				"Item Id",

				"Brand",

				"Model Name",

				"Model Number",

				"Color",

		"Quantity" }, '\t'));

		Collections.sort(inventoryAvailability, new Comparator<InventoryAvailability>() {

			public int compare(InventoryAvailability arg0,InventoryAvailability arg1) {				

			    int brandComp = arg0.getBrand().compareTo(arg1.getBrand());

			    if(brandComp == 0){

			    	return arg0.getModelNumber().compareTo(arg1.getModelNumber());

			    }

			    return brandComp;

			}

		});

		for(InventoryAvailability invAvailability : inventoryAvailability) {

			if((invAvailability.getModelNumber().equals("Q50")) || (invAvailability.getItemId() == 1457)) {

				continue;

			}

			bufferedWriter.newLine();

			bufferedWriter.write(StringUtils.join(

					new String[] {

							String.valueOf(invAvailability.getItemId()),

							invAvailability.getBrand(),

							invAvailability.getModelName(),

							invAvailability.getModelNumber(),

							invAvailability.getColor(),

							String.valueOf(invAvailability.getQuantity())}, '\t'));

		}

		bufferedWriter.close();

		return file;

	}

	public static String getHtmlFileGenrator(File file, String tableTitle){

		String htmlFilePath = "/tmp/"+tableTitle+"-file.htm";

		StringBuffer sb = new StringBuffer();

		sb.append("");

		try{

			BufferedReader br = new BufferedReader(new FileReader(file));

			String line = "";

			sb.append("<html><table border=\"1\" align=\"center\">"

					+ "<caption><b>" + tableTitle + "</b></caption>");

			int lineCount = 1;

			while((line = br.readLine())!= null){

				String[] values = line.split("\t");

				sb.append("<tr>");

				for(String value : values){

					if(lineCount ==1){

						sb.append("<td><b>"+ value.toUpperCase()+"</b></td>");

					}else{

						sb.append("<td>"+ value+"</td>");

					}

				}

				sb.append("</tr>");

				lineCount++;

			}

			sb.append("</table></html>");

			BufferedWriter out = new BufferedWriter(new FileWriter(htmlFilePath));

			out.write(sb.toString());

			out.flush();

			out.close();

		}

		catch(Exception e){

			e.printStackTrace();

			return "";

		}

		return htmlFilePath;

	}

}