Subversion Repositories SmartDukaan

package in.shop2020.util;

import in.shop2020.metamodel.core.Entity;
import in.shop2020.metamodel.core.Slide;
import in.shop2020.metamodel.definitions.Catalog;
import in.shop2020.metamodel.definitions.Category;
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 CompatibleAccessoriesIndexGenerator {

        private String[] indentation = {"", "    ", "        ", "            ","                "};
        private DefinitionsContainer defContainer = Catalog.getInstance().getDefinitionsContainer();
        List<Entity> entities;
        private Map<Long, Long> categorySlides = new HashMap<Long, Long>();
        private Map<Long, String> categoryLabels = new HashMap<Long, String>();

        private Map<Long, List<String>> categorySynonyms = new HashMap<Long, List<String>>();
        private Map<String, List<String>> brandSynonyms = new HashMap<String, List<String>>();
        
        public CompatibleAccessoriesIndexGenerator(List<Entity> entities) throws Exception      {
                
                categorySlides.put((long) 10014, (long) 130067);
                categorySlides.put((long) 10018, (long) 130073);
                
                categoryLabels.put((long) 10014, "Battery");
                categoryLabels.put((long) 10018, "Carrying Case");

                Map<String, String> brSynonyms = new HashMap<String, String>(); 
                Map<String, String> catSynonyms = new HashMap<String, String>();
                
                
                try {
                        brSynonyms =  CreationUtils.getSynonyms().get("brand");
                        catSynonyms =  CreationUtils.getSynonyms().get("subcategory");
                        
                        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
                }
                catSynonyms.put("Battery", StringUtils.defaultString("Battery," + StringUtils.defaultString(catSynonyms.get("Battery"))));
                catSynonyms.put("Carrying Case", StringUtils.defaultString("Carrying Case," + StringUtils.defaultString(catSynonyms.get("Carrying Case"))));
                
                categorySynonyms.put((long) 10014, Arrays.asList(catSynonyms.get("Battery").split(",")));
                categorySynonyms.put((long) 10018, Arrays.asList(catSynonyms.get("Carrying Case").split(",")));
                this.entities = entities;
        }
        
        public void generate()  {
                StringBuilder sb = new StringBuilder();
                
                for(Entity entity: entities)    {
                        
                        long categoryId = entity.getCategoryID();
                        String url = getProductURL(entity);
                        
                        if (url != null && categorySlides.keySet().contains(categoryId))        {
                                
                                Slide compatibilitySlide = entity.getSlide(categorySlides.get(categoryId));
                                
                                try {
                                        String text = compatibilitySlide.getFreeformContent().getFreeformText();
                                        
                                        if(text.indexOf("<ul>") > -1)   continue;
                                        for(String name: getHandsetNames(text)) {
                                                for(String categoryLabel : getCategorySynonyms(categoryId)){
                                                        sb.append(indentation[1] + "<div>\n");
                                                        sb.append(indentation[2] + "<a href='" + url + "'>" + name + " " + categoryLabel + "</a>\n");
                                                        sb.append(indentation[1] + "</div>\n");
                                                }
                                        }
                                        
                                } catch (NullPointerException e) {
                                        // TODO: handle exception
                                }
                        }
                }
                
                String indexFilename = Utils.EXPORT_PARTNERS_CONTENT_PATH + "compatible-accessories-index.html";
                
                try     {
                        DBUtils.store(sb.toString(), indexFilename);
                        
                } catch (Exception e) {
                        e.printStackTrace();
                }
        }
        
        private String getProductURL(Entity entity)     {

                Category category = defContainer.getCategory(entity.getCategoryID());
                try {
                        Category parentCategory = category.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;
                        
                } catch (NullPointerException e) {
                        return null;
                }
        }
        
        private List<String> getHandsetNames(String rawText)    {
                
                List<String> names = new ArrayList<String>();
                
                for (String namesWithSameBrand: rawText.trim().split("\n"))     {
                        boolean isFirst = true;
                        List<String> brandSynonyms = null;
                        
                        for(String name: namesWithSameBrand.trim().split(","))  {
                                name = name.trim();
                                String [] brandName;
                                if(isFirst)     {
                                        brandName = name.split(" ", 2);
                                        if(brandName.length==1){
                                                continue;
                                        }
                                        if(brandName[1].startsWith("Ericsson")) {
                                                brandName[0] = "Sony Ericsson";
                                                brandName[1] = brandName[1].substring(9);
                                        }
                                        brandSynonyms = getBrandSynonyms(brandName[0]);
                                        name = brandName[1];
                                        isFirst = false;
                                } 
                                for(String brand : brandSynonyms){
                                        names.add(brand + " " + name);
                                }
                                
                        }
                }
                return names;
        }
        
        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);
        }

        private List<String> getCategorySynonyms(Long categoryId) {
                return categorySynonyms.get(categoryId);
        }
        
        

        public static void main(String[] args) {
                try {
                        List<Entity> entities = new ArrayList<Entity>(CreationUtils.getEntities().values());
                        CompatibleAccessoriesIndexGenerator generator = new CompatibleAccessoriesIndexGenerator(entities);
                        generator.generate();
                } catch (Exception e) {
                        e.printStackTrace();
                }
        }
}