Subversion Repositories SmartDukaan

package in.shop2020.util;

import in.shop2020.metamodel.core.Entity;

import in.shop2020.metamodel.util.CreationUtils;

import in.shop2020.model.v1.catalog.CatalogService.Client;

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

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

import in.shop2020.thrift.clients.CatalogClient;

import in.shop2020.ui.util.CatalogUploderToGAE;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

public class WeeklyContentCreation {

	private static Log log = LogFactory.getLog(ContentGenerationUtility.class);

	private static List<Item> items = null;

	private static Map<Long,List<Item>> entityIdItemMap = new HashMap<Long, List<Item>>();

	private static List<Long> allValidEntityIds = null;

	private static List<Entity> allValidEntities = null;

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

		//Boolean.parseBoolean("true");

		Client client = new CatalogClient().getClient();

		items = client.getAllItemsByStatus(status.ACTIVE);

		items.addAll(client.getAllItemsByStatus(status.PAUSED));

		items.addAll(client.getAllItemsByStatus(status.COMING_SOON));

		populateEntityIdItemMap();

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

		allValidEntities = new ArrayList<Entity>();

		allValidEntityIds = new ArrayList<Long>(entityIdItemMap.keySet());

		for(Long entityId : entityIdItemMap.keySet()) {

			if(allEntities.containsKey(entityId)){

				allValidEntities.add(allEntities.get(entityId));

			}

		}

		try {

		//Temporarily added

		AmazonSCDataGenerator asg = new AmazonSCDataGenerator(allValidEntities, "ALL");

		asg.generateSCProdData();

		}catch(Exception e){

			System.out.println("Could not generate Amazon feed.");

		}

		// Upload catalog to Google App Engine.

		log.info("Uploading Catalog to Google app engine");

		CatalogUploderToGAE catalogUploaderToGAE = new CatalogUploderToGAE();

		catalogUploaderToGAE.uploadItems(items);

		// FIXME Avoiding the finding of accesories, as list of categories for

		// which we need to find accessories is hardocoded in code.

		// We need to make that configurable. Also creating ticket to improve

		// it.

		try {

			log.info("Finding accessories");

			AccessoriesFinder af = new AccessoriesFinder(new HashSet<Long>(

					allValidEntityIds));

			Map<Long, Map<Long, List<Long>>> relatedAccessories = af

					.findAccessories();

			CreationUtils.storeRelatedAccessories(relatedAccessories);

		} catch (Exception e) {

			log.error("Error while generating accessories" + e);

		}

        // Generate synonyms list. This will be used in PriceComparisonTool to

        // resolve the product names.

        log.info("Generating synonyms");

        SynonymExporter sx = new SynonymExporter();

        sx.storeSynonyms(allValidEntities);

        //Generation for website SEO      

        log.info("Generating HTML for Site Index");

        ProductIndexGenerator indexGenerator = new ProductIndexGenerator(allValidEntities);

        indexGenerator.generate();

        log.info("Generating HTML for Site for Product Documents");

        ProductDocumentsGenerator asGenerator = new ProductDocumentsGenerator(allValidEntities);

        asGenerator.generate();

        log.info("Generating HTML for Accessories Compatibility Index");	

        CompatibleAccessoriesIndexGenerator generator = new CompatibleAccessoriesIndexGenerator(allValidEntities);

        generator.generate();

        log.info("Generating HTML for Most Frequently searched keywords");

        MostFrequentlySearchedKeywords mfsk = new MostFrequentlySearchedKeywords();

        mfsk.generate();

        log.info("Generating HTML for Most Compared Index");

        MostComparedIndexGenerator mostCompGenerator = new MostComparedIndexGenerator(allValidEntityIds);

        mostCompGenerator.generate();

        log.info("Generating XML for Mobile Site XML feed");

        MobileSiteDataXMLGenerator mSiteXMLGenerator = new MobileSiteDataXMLGenerator(allValidEntities);

        mSiteXMLGenerator.generate();

	}

    private static void populateEntityIdItemMap() {

        Utils.info("Processing " + items.size() + " items");

        for (Item item : items) {

            Utils.info(item.getId() + " Item is adding");

            List<Item> itemList = entityIdItemMap.get(item.getCatalogItemId());

            if (itemList == null) {

                itemList = new ArrayList<Item>();

            } 

            itemList.add(item);

            entityIdItemMap.put(item.getCatalogItemId(), itemList);

        }

    }

}