Subversion Repositories SmartDukaan

package in.shop2020.util;

import java.io.File;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.commons.lang.StringEscapeUtils;
import org.apache.commons.lang.StringUtils;

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 in.shop2020.model.v1.catalog.Item;
import in.shop2020.model.v1.catalog.status;
import in.shop2020.model.v1.catalog.CatalogService.Client;
import in.shop2020.thrift.clients.CatalogClient;

public class MobileSiteDataXMLGenerator {
        
        private List<Entity> entities;
        private List<Item> items;
        private Map<Long, Item> entityItemMap = new HashMap<Long, Item>();
        private String[] xmlIndentation = {"", "    ", "        ", "            ","                "};
        
        private DefinitionsContainer defContainer = Catalog.getInstance().getDefinitionsContainer();
    private CatalogClient csc;
    private Client client;
        
        public MobileSiteDataXMLGenerator(List<Entity> entities) throws Exception {
                this.entities = entities;
        csc = new CatalogClient();
        client = csc.getClient();

        items = client.getAllItemsByStatus(status.ACTIVE);
        items.addAll(client.getAllItemsByStatus(status.COMING_SOON));
        items.addAll(client.getAllItemsByStatus(status.PAUSED));
        
        for (Item item: items)  {
                if(!entityItemMap.containsKey(item.getCatalogItemId())) {
                        
                        entityItemMap.put(item.getCatalogItemId(), item);
                }
        }
        }
        
        void generate() throws Exception        {
        List<String> productXMLSnippets = new ArrayList<String>();
                productXMLSnippets.add("<Products>");
                
                for (Entity entity: entities)   {
                        try     {
                                long categoryId = defContainer.getCategory(entity.getCategoryID()).getParentCategory().getID();
                                
                                if(categoryId != Utils.MOBILE_PHONES_CATAGORY && categoryId != Utils.TABLETS_CATEGORY && categoryId != Utils.LAPTOPS_CATEGORY)  {
                                        continue;
                                }
                        } catch (NullPointerException e) {
                                continue;
                        }
                        
                        String description = "";
                        Slide slide = entity.getSlide(130054);
                        
                        if (slide == null)      continue;
                        
                        List<Feature> features = slide.getFeatures();
                        
                        if (features == null)   continue;
                        
                        for(Feature feature: slide.getFeatures())       {
                        
                                if (feature.getFeatureDefinitionID() == 120081) {
                        List<Bullet> bullets = feature.getBullets();
                        
                        if (bullets == null)    continue;
                        
                        int counter = 0;
                        
                        for (Bullet bullet : bullets) {
                                
                            PrimitiveDataObject dataObject = (PrimitiveDataObject) bullet.getDataObject();
                            
                            description += dataObject.getValue();
                            if(counter < 2)     description += ", ";
                            counter ++;
                        }
                    }
                        }
                        if (! entityItemMap.containsKey(entity.getID()))        continue;
                        
                        Item item = entityItemMap.get(entity.getID());
                        String productType = "";
                        String url = getProductURL(entity.getID(), item);
                        String imageUrl = "http://www.saholic.com/images/website" + File.separator + entity.getID() + File.separator + "thumbnail.jpg";
                        
                        if (entity.getCategoryID() == 10006)    {
                                productType = "Mobile Phone";
                        
                        } else if (entity.getCategoryID() == 10010)     {
                                productType = "Tablet";
                                
                        } else if (entity.getCategoryID() == 10050)     {
                                productType = "Laptop";
                        }
                        productXMLSnippets.add(this.xmlIndentation[1] + "<Product>");   
                        
                        productXMLSnippets.add(this.xmlIndentation[2] + "<ID>" + entity.getID() + "</ID>");
                        productXMLSnippets.add(this.xmlIndentation[2] + "<Type>" + productType + "</Type>");
                        productXMLSnippets.add(this.xmlIndentation[2] + "<Brand>" + item.getBrand() + "</Brand>");
                        productXMLSnippets.add(this.xmlIndentation[2] + "<ModelName>" + item.getModelName() + "</ModelName>");
                        productXMLSnippets.add(this.xmlIndentation[2] + "<ModelNumber>" + item.getModelNumber() + "</ModelNumber>");
                        productXMLSnippets.add(this.xmlIndentation[2] + "<URL>" + url + "</URL>");
                        productXMLSnippets.add(this.xmlIndentation[2] + "<ImageURL>" + imageUrl + "</ImageURL>");
                        productXMLSnippets.add(this.xmlIndentation[2] + "<ShortDesc>" + StringEscapeUtils.escapeXml(description) + "</ShortDesc>");
                        productXMLSnippets.add(this.xmlIndentation[2] + "<MRP>" + item.getMrp() + "</MRP>");
                        productXMLSnippets.add(this.xmlIndentation[2] + "<SellingPrice>" + (int)item.getSellingPrice() + "</SellingPrice>");
                        
                        productXMLSnippets.add(this.xmlIndentation[1] + "</Product>");
                }
                productXMLSnippets.add("</Products>");
                String productDataXML = StringUtils.join(productXMLSnippets, "\n");
                
                Utils.info(productDataXML);
                
                // Write it to file
                String productXMLFilename = Utils.EXPORT_PARTNERS_CONTENT_PATH + "msitedata.xml";
                DBUtils.store(productDataXML, productXMLFilename);
        }
        
        private String getProductURL(long entityId, Item item){
                //long rootCategoryId = catm.getCategory(item.getCategory()).getParent_category_id();
                in.shop2020.metamodel.definitions.Category parentCategory = defContainer.getCategory(item.getCategory()).getParentCategory();
                
                String url = "http://www.saholic.com/" + parentCategory.getLabel().toLowerCase().replace(' ', '-') + "/";
                String productUrl = ((item.getBrand() != null) ? item.getBrand().trim() + " " : "").toLowerCase().replace(' ', '-')
        + "-" + ((item.getModelName() != null) ? item.getModelName().trim() + " " : "").toLowerCase().replace(' ', '-') 
            + "-" + (( item.getModelNumber() != null ) ? item.getModelNumber().trim() + " ": "" ).toLowerCase().replace(' ', '-')
        + "-" + entityId;
                productUrl = productUrl.replaceAll("/", "-");
                url = url + productUrl;
                url = url.replaceAll("-+", "-");
                return url;
        }
        
        public static void main(String[] args) {
                try {
                        List<Entity> entities = new ArrayList<Entity>(CreationUtils.getEntities().values());
                        MobileSiteDataXMLGenerator generator = new MobileSiteDataXMLGenerator(entities);
                        generator.generate();
                } catch (Exception e) {
                        e.printStackTrace();
                }
        }
}