Subversion Repositories SmartDukaan

package in.shop2020.util;

import in.shop2020.metamodel.core.Entity;

import in.shop2020.metamodel.core.EntityState;

import in.shop2020.metamodel.core.EntityStatus;

import in.shop2020.metamodel.util.CreationUtils;

import in.shop2020.metamodel.util.ExpandedEntity;

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

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

import in.shop2020.thrift.clients.CatalogServiceClient;

import java.util.ArrayList;

import java.util.Date;

import java.util.List;

import java.util.Map;

import org.apache.commons.lang.StringUtils;

/**

 * Command line utility to generate xml file for partners

 * 

 * @author rajveer

 *

 */

public class ProductListGenerator {

	private String[] xmlIndentation = {"", "    ", "        ", "            ","                "};

	CatalogServiceClient catalogServiceClient = null;

	in.shop2020.model.v1.catalog.InventoryService.Client client = null;

	public static long MOBILE_PHONES = 10001;

	public static void main(String[] args) throws Exception {

		/*

		String[] options = new String[] {"domain", "partner"};

		String[] values = new String[] {"saholic.com", "phonecurry"};

		String usage = "Usage: ProductListGenerator ["+ StringUtils.join(options, "|") +"] ["+ StringUtils.join(values, "|") + "] ";

		if(args.length < 1) {

			System.out.println(usage);

			System.exit(-1);

		}

		String inputCommand = args[0];

		String inputID = null;

		if(args.length == 2){

			inputID = args[1];

		}

		*/

		ProductListGenerator generator = new ProductListGenerator();

		generator.generateXMLList();

	}

	public ProductListGenerator() throws Exception {

		catalogServiceClient = new CatalogServiceClient();

		client = catalogServiceClient.getClient();

	}

	/**

	 * 

	 * @param expEntity

	 * @return url

	 */

	private String getEntityURL(ExpandedEntity expEntity){

		String url = "http://www.saholic.com/" + expEntity.getCategory().getParentCategory().getLabel().toLowerCase().replace(' ', '-') + "/";

		String productUrl = expEntity.getBrand().toLowerCase().replace(' ', '-')

        + "-" + expEntity.getModelName().toLowerCase().replace(' ', '-') 

	    + "-" + expEntity.getModelNumber().toLowerCase().replace(' ', '-')

        + "-" + expEntity.getID();

		productUrl = productUrl.replaceAll("/", "-");

		url = url + productUrl;

		url = url.replaceAll("--", "-");

		return url;

	}

	/**

	 * get xml feed for partner sites

	 * @param irDataXMLSnippets

	 * @throws Exception

	 */

	private void getProductsXML(ArrayList<String> irDataXMLSnippets) throws Exception{ 	

		Map<Long, Entity> entities =  CreationUtils.getEntities();

		for(Entity entity: entities.values()){

			EntityState state = CreationUtils.getEntityState(entity.getID());

			ExpandedEntity expEntity = new ExpandedEntity(entity);

			if(expEntity.getCategoryID() == -1 || expEntity.getCategory().getParentCategory().getID() != MOBILE_PHONES){

				continue;

			}

			if(state.getStatus() != EntityStatus.READY){

				continue;

			}

			double itemPrice = 0;

			List<Item> items = client.getItemsByCatalogId(entity.getID());

			if(items == null){

				continue;

			}

			Date todate = new Date();

			Boolean hasActiveItem = false;

			for(Item item: items){

				if(item.getItemStatus() == status.ACTIVE && todate.getTime() > item.getStartDate()){

						hasActiveItem = true;

				}

				if(itemPrice == 0){

					itemPrice = item.getSellingPrice();

				}else if(itemPrice>item.getSellingPrice()){

					itemPrice = item.getSellingPrice();

				}

			}

			if(!hasActiveItem){

				continue;

			}

			long categoryID = expEntity.getCategoryID();

			if(categoryID == -1){

				continue;

			}

			irDataXMLSnippets.add(this.xmlIndentation[1] + "<products>");	

			irDataXMLSnippets.add(this.xmlIndentation[2] + "<ProductSKU>" + expEntity.getID() + "</ProductSKU>");

			String title = expEntity.getBrand() + " " + expEntity.getModelName() + " " + expEntity.getModelNumber();

			irDataXMLSnippets.add(this.xmlIndentation[2] + "<ProductName>" + title + "</ProductName>");

			String url = getEntityURL(expEntity);

			irDataXMLSnippets.add(this.xmlIndentation[2] + "<ProductURL>" + url + "</ProductURL>");

			irDataXMLSnippets.add(this.xmlIndentation[2] + "<ProductPrice>" + itemPrice + "</ProductPrice>");

			irDataXMLSnippets.add(this.xmlIndentation[1] + "</products>");		

		}

	}

	/**

	 * Generate the xml list of all the active phones and store in a file

	 * @throws Exception

	 */

	public void generateXMLList() throws Exception {

		ArrayList<String> productXMLSnippets = new ArrayList<String>();

		productXMLSnippets.add("<saholic.com>");

		getProductsXML(productXMLSnippets);

		productXMLSnippets.add("</saholic.com>");

		String productDataXML = StringUtils.join(productXMLSnippets, "\n");

		Utils.info(productDataXML);

		// Write it to file

		String productXMLFilename = Utils.EXPORT_PARTNERS_CONTENT_PATH + "phonecurry.xml";

		DBUtils.store(productDataXML, productXMLFilename);

	}

}