Subversion Repositories SmartDukaan

package in.shop2020.serving.controllers;

import in.shop2020.datalogger.EventType;

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

import in.shop2020.model.v1.order.HotspotStore;

//import in.shop2020.serving.cache.EhcacheWrapper.CacheKeys;

//import in.shop2020.serving.cache.SnippetCacheWrapper;

import in.shop2020.serving.cache.EhcacheWrapper.CacheKeys;

import in.shop2020.serving.cache.SnippetCacheWrapper;

import in.shop2020.serving.service.SearchService;

import in.shop2020.serving.utils.Utils;

import in.shop2020.thrift.clients.TransactionClient;

import in.shop2020.utils.CategoryManager;

import in.shop2020.utils.DataLogger;

import java.io.UnsupportedEncodingException;

import java.net.URLEncoder;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.HashMap;

import java.util.LinkedHashMap;

import java.util.List;

import java.util.Map;

import java.util.Map.Entry;

import org.apache.commons.lang.StringUtils;

import org.apache.log4j.Logger;

import org.apache.struts2.convention.annotation.Result;

import com.google.gson.Gson;

/**

 * @author rajveer

 *

 */

@Result(name="redirect", location="${location}", type="redirect")

public class SearchController extends BaseController {

    private static final long serialVersionUID = -8392433517042806559L;

    private static Logger log = Logger.getLogger(Class.class);

    private List<String> results;

    private Map<String, Double> dynamicSearchMap;

    /**

     * 

     */

    private Map<String, String> snippets;

    /**

     * 

     */

    private Map<String, List<String[]>> facets = null;

    /**

     * 

     */

    private Map<String, List<List<String>>> crumbs;

    private String  query;

    private String  q;

    private String sortOrder = null;

    private Double minPrice = null;

    private Double maxPrice = null;

    private long windowSize = 20;

    private long page = 1;

    private long totalResults;

    private long beginIndex = 0;

    private String url;

    private String priceUrl = "?";

    private String sortUrl;

    private String location;

    private static HashMap<String, List<String>> categoriesChildren = null;

    private static Map<Long, String> categoryLabelMap = null;

    /**

     * 

     * @return

     * @throws UnsupportedEncodingException

     */

    // GET /query

    public String index() {

        String loginStatus = (String) request.getSession().getAttribute("LOGGED_IN");

        if(loginStatus == null || !loginStatus.equals("TRUE")){

            return "authfail";

        }

        storeId = Long.parseLong((String) request.getSession().getAttribute("STORE_ID"));

        if(!hotspotStores.containsKey(storeId)){

            try{

                HotspotStore hotSpotStore = (new TransactionClient()).getClient().getHotspotStore(storeId, "");

                hotspotStores.put(storeId, hotSpotStore);

            } catch (Exception e) {

                log.error("Unable to get store", e);

            }

        }

        log.info("this.request=" + this.request);

        //url = this.request.getRequestURL();

        query = this.request.getParameter("q");

        q = this.query;

        try {

            url = "q="+ URLEncoder.encode(query, "UTF-8");

            if(query.trim().isEmpty()){

                    location = "/";

                    return "redirect";

            }

            sortUrl = url;

            if(this.request.getParameter("sort") != null){

                url= url + "&sort=" + this.request.getParameter("sort");

                sortOrder = this.request.getParameter("sort");

            }

            priceUrl += url;

            if(this.request.getParameter("page") != null){

                this.page = Long.parseLong(this.request.getParameter("page"));

                this.beginIndex = this.windowSize * (this.page-1);

            }

            if(this.request.getParameter("minPrice") != null){

                this.minPrice = (new Double(this.request.getParameter("minPrice")));

                url= url + "&minPrice=" + this.request.getParameter("minPrice");

            }

            if(this.request.getParameter("maxPrice") != null){

                this.maxPrice = (new Double(this.request.getParameter("maxPrice")));

                url= url + "&maxPrice=" + this.request.getParameter("maxPrice");

            }       

            String[] fqrys = this.request.getParameterValues("fq");

            this.crumbs = new HashMap<String, List<List<String>>>();

            String urlCrumb = "";

            if(query != null) {

                urlCrumb = url; 

            }else {

            }

            if(fqrys != null) {

                log.info("fqrys=" + Arrays.toString(fqrys));

                String filterUrl = "";

                for(int i=0; i<fqrys.length; i++){

                    urlCrumb += "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8");

                    priceUrl += "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8");

                    sortUrl += "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8");

                    String facetName = StringUtils.split(fqrys[i], ":")[0];

                    String facetValue = StringUtils.split(fqrys[i], ":")[1];

                    String facetLabel = Utils.FACET_LABEL_MAP.get(facetName);

                    filterUrl = "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8"); 

                    List<String> acrumb = Arrays.asList(facetValue, filterUrl);

                    if(!crumbs.containsKey(facetLabel)) {

                        crumbs.put(facetLabel, new ArrayList<List<String>>());

                    }

                    this.crumbs.get(facetLabel).add(acrumb);

                }

            }

            url = urlCrumb;

            SearchService search = new SearchService(query, fqrys, (page-1)*windowSize, windowSize, minPrice, maxPrice, sortOrder, sourceId);

            this.results =  search.getResultMap(); 

            this.dynamicSearchMap = search.getDynamicPriceMap();

            // Facets

            List<String> fqs = null;

            if (fqrys != null) {

                fqs = Arrays.asList(fqrys);

            }else {

                fqs = new ArrayList<String>();

            }

            setFacet(fqs, search);

            //If 

            Map<String, Double> priceMap = search.getPriceStatsMap();

            if(priceMap != null){

                if (this.minPrice == null){

                    this.minPrice = priceMap.get("min");

                }

                if(this.maxPrice == null) {

                    this.maxPrice = priceMap.get("max");

                }

            } else{

                this.minPrice = 0.0;

                this.maxPrice = 0.0;

            }

            this.totalResults = search.getTotalResults();

        } catch (Exception e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        return "index";

    }

    private void setFacet(List<String> fqs, SearchService search) throws Exception{

        this.facets = new LinkedHashMap<String, List<String[]>>();

        List<String> filtrableFacets = search.getFilterableFacets();

        for (String filtrableFacet : filtrableFacets) {

            Map<String, Integer> facetDetailMap = search.getFacetDetails(filtrableFacet);

            if(facetDetailMap==null)

                continue;

            List<String[]> values = new ArrayList<String[]>();

            for(Entry<String, Integer> facetEntry : facetDetailMap.entrySet()){

                String selected = "";

                if(fqs.contains(filtrableFacet + ":" + facetEntry.getKey())) {

                    selected = "checked=\"checked\"";

                }

                String facet = facetEntry.getKey();

                String drilldownURL = "&fq=" +  

                    URLEncoder.encode( filtrableFacet + ":" + facet, "UTF-8"); 

                String[] afacet = new String[] { facet, 

                      facetEntry.getValue().toString(), drilldownURL,  selected};

                values.add(afacet);    

            }

            this.facets.put(Utils.FACET_LABEL_MAP.get(filtrableFacet), values);

        }

    }

    public Map<String, String> getSnippets() throws Exception {

        if(results != null){

            snippets = new HashMap<String, String>();   

            for(String docId: results){

                String snippet = (String)SnippetCacheWrapper.getSnippet(CacheKeys.CATEGORY_SNIPPET_CACHE_KEY, docId, sourceId);

                if (snippet != null) {

                    snippets.put(docId, snippet);

                }

            }

        }

        return snippets;

    }

    /**

     * 

     * @return

     */

    public String getQuery() {

        return this.query;

    }

    /**

     * 

     * @return

     */

    public List<String> getResults() {

        return this.results;

    }

    /**

     * 

     * @return

     */

    public Map<String, List<String[]>> getFacets() {

        return this.facets;

    }

    public long getTotalResults(){

        return totalResults;

    }

    public String getUrl(){

        return this.url;

    }

    public String getPriceUrl(){

        return this.priceUrl;

    }

    public String getSortUrl(){

        return this.sortUrl;

    }

    public long getBeginIndex(){

        if(totalResults>0)

            return beginIndex+1;

        return beginIndex;

    }

    public long getTotalPages() {

        return 1 + (totalResults-1)/windowSize;

    }

    public long getCurrentPage() {

        return this.page;

    }

    public Double getMinPrice() {

        return this.minPrice;

    }

    public Double getMaxPrice() {

        return this.maxPrice;

    }

    public Map<String, List<List<String>>> getCrumbs() {

        return this.crumbs;

    }

    public String getSortOrder(){

        return this.sortOrder;

    }

    public String getLocation(){

        return this.location;

    }

    public List<String> getChildren(String categoryLabel) {

        return getCategoriesChildren().get(categoryLabel);

    }

    public static Map<String, List<String>> getCategoriesChildren(){

        Map<Long,String> idLabelMap = new HashMap<Long, String>(); 

        if (categoriesChildren == null) {

            categoriesChildren = new HashMap<String, List<String>>();

            for (Category category: CategoryManager.getCategoryManager().getCategories().values()){

                idLabelMap.put(category.getId(), category.getLabel());

            }

            for (Category category: CategoryManager.getCategoryManager().getCategories().values()){

                if(category.getParent_category_id() != 10000 && category.getParent_category_id() != 0 &&

                        category.getParent_category_id() != Utils.MOBILE_PHONES_CATEGORY){

                    String parentLabel = idLabelMap.get(category.getParent_category_id());

                    if(!categoriesChildren.containsKey(parentLabel)){

                        categoriesChildren.put(parentLabel, new ArrayList<String>());

                    }

                    categoriesChildren.get(parentLabel).add(category.getLabel());

                }

            }

        }

        return categoriesChildren;

    }

    public static String getCategoryLabel(Long id){

        if(categoryLabelMap == null){

            Map<Long,String> idLabelMap = new HashMap<Long, String>(); 

            for (Category category: CategoryManager.getCategoryManager().getCategories().values()){

                idLabelMap.put(category.getId(), category.getLabel());

            }

            categoryLabelMap = idLabelMap;

        } 

        return categoryLabelMap.get(id);

    }

    public String getDynamicSearchMap(){

        if (this.dynamicSearchMap == null) {

            return "{}";

        } else {

            return new Gson().toJson(this.dynamicSearchMap);

        }

    }

	public void setQ(String q) {

		this.q = q;

	}

	public String getQ() {

		return q;

	}

}