Subversion Repositories SmartDukaan

package in.shop2020.util;

import in.shop2020.metamodel.core.Bullet;

import in.shop2020.metamodel.core.Entity;

import in.shop2020.metamodel.core.Feature;

import in.shop2020.metamodel.core.PrimitiveDataObject;

import in.shop2020.metamodel.core.Slide;

import in.shop2020.metamodel.definitions.Catalog;

import in.shop2020.metamodel.definitions.DefinitionsContainer;

import in.shop2020.metamodel.util.CreationUtils;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import org.apache.commons.lang.StringUtils;

public class ProductIndexGenerator {

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

	private List<Entity> entities;

	private DefinitionsContainer defContainer = Catalog.getInstance().getDefinitionsContainer();

	private Map<String, List<String>> brandSynonyms = new HashMap<String, List<String>>();

	public ProductIndexGenerator(List<Entity> entities)	{

		Map<String, String> brSynonyms = new HashMap<String, String>(); 

		try {

			brSynonyms =  CreationUtils.getSynonyms().get("brand");

			for(String brand : brSynonyms.keySet()){

				List<String> brandSyns = Arrays.asList((brand + "," + StringUtils.defaultString(brSynonyms.get(brand))).split(","));

				brandSynonyms.put(brand, brandSyns);

			}

		} catch (Exception e) {

			// TODO: handle exception

		}

		this.entities = entities;

	}

	private String getProductURL(Entity entity){

		in.shop2020.metamodel.definitions.Category parentCategory = defContainer.getCategory(entity.getCategoryID()).getParentCategory();

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

		String productUrl = ((entity.getBrand() != null) ? entity.getBrand().trim() + " " : "").toLowerCase().replace(' ', '-')

        + "-" + ((entity.getModelName() != null) ? entity.getModelName().trim() + " " : "").toLowerCase().replace(' ', '-') 

	    + "-" + ((entity.getModelNumber() != null ) ? entity.getModelNumber().trim() + " ": "" ).toLowerCase().replace(' ', '-')

        + "-" + entity.getID();

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

		url = url + productUrl;

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

		return url;

	}

	public void generate()	{

		StringBuilder sb = new StringBuilder();

		for (Entity entity: entities)	{

			String brand = entity.getBrand();

			try{

			String url = getProductURL(entity);

			Slide slide = entity.getSlide(Utils.SUMMARY_SLIDE_DEFINITION_ID);

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

			modelNameSyn.add(entity.getModelName());

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

			modelNumberSyn.add(entity.getModelNumber());

			for(Feature feature: slide.getFeatures())	{

				if(feature.getFeatureDefinitionID() == 120156)	{

					List<Bullet> bullets =  feature.getBullets();

					if(bullets != null)	{

						for(Bullet bullet: bullets){

							PrimitiveDataObject pdo = (PrimitiveDataObject) bullet.getDataObject();

							modelNameSyn.add(pdo.getValue());

						}

					}

				} else if(feature.getFeatureDefinitionID() == 120157)	{

					List<Bullet> bullets =  feature.getBullets();

					if(bullets != null)	{

						for(Bullet bullet: bullets)	{

							PrimitiveDataObject pdo = (PrimitiveDataObject)bullet.getDataObject();

							modelNumberSyn.add(pdo.getValue());

						}

					}

				}

			}

			for(String br : getBrandSynonyms(brand)) {

				for (String mName : modelNameSyn){

					for (String mNumber : modelNumberSyn){

						sb.append("<div>\n");

						sb.append(indentation[1] + "<a href='" + url + "'>");

						sb.append(br + " " + mName + " " + mNumber);

						sb.append("</div>\n");

						sb.append("</a>\n");

					}

				}

			}

			}catch(Exception e){

				e.printStackTrace();

			}

		}

		String indexFilename = Utils.EXPORT_PARTNERS_CONTENT_PATH + "productindex.html";

		try	{

			DBUtils.store(sb.toString(), indexFilename);

		} catch (Exception e) {

			e.printStackTrace();

		}

	}

	private List<String> getBrandSynonyms(String brand) {

		if(!brandSynonyms.containsKey(brand)){

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

			list.add(brand);

			brandSynonyms.put(brand, list);

		}

		return brandSynonyms.get(brand);

	}

	public static void main(String[] args) {

		try {

			List<Entity> entities = new ArrayList<Entity>(CreationUtils.getEntities().values());

			ProductIndexGenerator generator = new ProductIndexGenerator(entities);

			generator.generate();

		} catch (Exception e) {

			e.printStackTrace();

		}

	}

}