Subversion Repositories SmartDukaan

/**
 * 
 */
package in.shop2020.serving.controllers;

import in.shop2020.serving.pages.PageContentKeys;
import in.shop2020.serving.pages.PageEnum;
import in.shop2020.serving.pages.PageManager;
import in.shop2020.serving.services.PageLoaderHandler;
import in.shop2020.serving.services.SolrSearchService;
import in.shop2020.serving.utils.FileUtils;
import in.shop2020.util.Utils;

import java.io.File;
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.StringTokenizer;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.lang.StringUtils;
import org.apache.juli.logging.Log;
import org.apache.juli.logging.LogFactory;
import org.apache.struts2.interceptor.ServletRequestAware;
import org.apache.struts2.rest.DefaultHttpHeaders;
import org.apache.struts2.rest.HttpHeaders;

/**
 * @author naveen
 *
 */
public class SearchController extends BaseController {

        /**
         * 
         */
        private static Log log = LogFactory.getLog(SearchController.class);
        
        private Map<String,String> htmlSnippets = new HashMap<String, String>();
        /**
         * 
         */
        private Map<String, String[]> results;

        /**
         * 
         */
        private Map<String, String> snippets;
        
        /**
         * 
         */
        private Map<String, List<String[]>> facets;
        
        /**
         * 
         */
        private List<String[]> crumbs;
        
        private String  query;
        
        private Double minPrice;
        private Double maxPrice;

        /**
         * 
         * @return
         * @throws UnsupportedEncodingException
         */
    // GET /query
    public HttpHeaders index() throws UnsupportedEncodingException {
        
        PageLoaderHandler pageLoader = new PageLoaderHandler();
                
                htmlSnippets.put("HEADER", pageLoader.getHeaderHtml(userinfo.getUserId(), userinfo.isSessionId(), userinfo.getNameOfUser()));
                htmlSnippets.put("MAIN_MENU", pageLoader.getMainMenuHtml());
                htmlSnippets.put("SEARCH_BAR", pageLoader.getSearchBarHtml(userinfo.getTotalItems(), 10000));
                htmlSnippets.put("CUSTOMER_SERVICE", pageLoader.getCustomerServiceHtml());
                htmlSnippets.put("MY_RESEARCH", pageLoader.getMyResearchHtml(userinfo.getUserId(), userinfo.isSessionId()));
                htmlSnippets.put("FOOTER",pageLoader.getFooterHtml());
                htmlSnippets.put("FOOTER",pageLoader.getFooterHtml());
                htmlSnippets.put("SEARCH_HEADER","");
                htmlSnippets.put("SEARCH_DETAILS","");
                
        log.info("this.request=" + this.request);
        
        query = this.request.getParameter("q");
        
        // if multiple words are given, it should do AND for all of them
        if(query.contains(" ")){
                StringTokenizer tokens = new StringTokenizer(query, " ");
                query = "";
                while(tokens.hasMoreTokens()){
                        query = query + " AND " + tokens.nextToken();
                }
                 
        }
        log.info("query=" + query);
        
        String[] fqrys = this.request.getParameterValues("fq");

        this.crumbs = new ArrayList<String[]>();
        
        String strCrumb = "";
        String urlCrumb = "";
        if(query != null) {
                strCrumb = query;
                urlCrumb = "q=" + URLEncoder.encode(query, "UTF-8"); 
                this.crumbs.add(new String[] { strCrumb, urlCrumb } );
                        log.info("acrumb=" + 
                                        Arrays.toString(new String[] { strCrumb, urlCrumb }));
        }else {
                /**
                 * FIXME to handle if qry is null
                 */
        }
        
//      HashMap<String,String> facetqueries = new HashMap<String, String>();
        
        if(fqrys != null) {
                log.info("fqrys=" + Arrays.toString(fqrys));
        
                for(int i=0; i<fqrys.length; i++) {
                        String facetValue = StringUtils.split(fqrys[i], ":")[1];
        //              facetqueries.put(StringUtils.split(fqrys[i], ":")[0], StringUtils.split(fqrys[i], ":")[1]);
                        //strCrumb += " " + facetValue;
                        strCrumb =  facetValue;
                        urlCrumb += "&fq=" + URLEncoder.encode(fqrys[i], "UTF-8");
                        String[] acrumb = new String[] { strCrumb, urlCrumb };
                        log.info("acrumb=" + Arrays.toString(acrumb));
                        
                        this.crumbs.add(acrumb);
                }
        }
        
        /**
                 * FIXME facte definition ids
                 */
        // Hard coded for now
//      long[] facetDefIDs = new long[] {50001, 50003, 50004, 50005, 50006, 
//                      50007, 50008, 50009};
        
        String[] facetDefIDs = new String[] {"Category","F_50002","F_50001",  "F_50006", "F_50007" };
        
        /* All facets. Removing 5 of them
        String[] facetDefIDs = new String[] {"Category","F_50002","F_50001", "F_50003", "F_50004", "F_50005", "F_50006", 
                        "F_50007", "F_50008", "F_50009"};
        */              

        
        /**
                 * FIXME facet labels
                 */
        // Hard-coded for now
        /*
        String[] facetLabels = new String[] {"Category","Price",
                "Brand", "Form Factor", "Carry In Pocket", "Cellular Technologies", 
                "Data Connectivity", "Camera Resolution", "Built-in Memory", 
                "Talk time"
                };
        */
        
        String[] facetLabels = new String[] {"Category","Price",
                "Brand", "Data Connectivity", "Camera Resolution"
                };
        
        SolrSearchService search = new SolrSearchService(query, fqrys, facetDefIDs);
        this.results = new LinkedHashMap<String, String[]>();
        this.results =  search.getResultMap(); 
        
    /*  
        long[] entityIDs = search.getResultEntityIDs();
        log.info("entityIDs=" + Arrays.toString(entityIDs));
        
        String[] entityNames = search.getResultEntityNames();
        log.info("entityNames=" + Arrays.toString(entityNames));
        
        this.results = new LinkedHashMap<String, String[]>();
        
        if(entityIDs != null) {
                for(int i=0; i<entityIDs.length; i++) {
                        String key = (new Long(entityIDs[i])).toString();
                        
                        String[] values = new String[] {entityNames[i]};
                        
                        this.results.put(key, values);
                }
        }
        */
        //log.info("this.results=" + this.results);
        
        // Facets
        this.facets = new LinkedHashMap<String, List<String[]>>();
        String qryString = this.request.getQueryString();
        log.info("qryString=" + qryString);
        
        
        for (int i=0; i<facetDefIDs.length; i++) {
                String facetDefID = facetDefIDs[i];
                String facetLabel = facetLabels[i];
                HashMap<String, Integer> facetDetailMap = search.getFacetDetails(facetDefID);
                if(facetDetailMap==null)
                        continue;
                List<String[]> values = new ArrayList<String[]>();
                for(String facet: facetDetailMap.keySet()){
//                      String extraInfo = "LINK";
//                      if(facetqueries.containsKey(facet)){
//                              extraInfo = "NO_LINK";
//                      }
                        String drilldownURL = qryString;
                        drilldownURL += "&fq=" + facetDefID + ":" + 
                                URLEncoder.encode(facet, "UTF-8"); 
                        String[] afacet = new String[] { facet, 
                                        facetDetailMap.get(facet).toString(), drilldownURL  };
                        values.add(afacet);    
                  
                }
                
                this.facets.put(facetLabel, values);
        }
        
        Map<String, Double> priceMap = search.getPriceStatsMap();
        if(priceMap != null){
                this.minPrice = priceMap.get("min");
                this.maxPrice = priceMap.get("max");
        }else{
                this.minPrice = 0.0;
                this.maxPrice = 0.0;
        }
        /*
        for (int i=0; i<facetDefIDs.length; i++) {
                long facetDefID = facetDefIDs[i];
                String facetLabel = facetLabels[i];
                //log.info("facetLabel=" + facetLabel);
                
                String[] facetValues = search.getFacetValues(facetDefID);
                String[] facetCounts = search.getFacetCounts(facetDefID);
                
                
                List<String[]> values = new ArrayList<String[]>();
        
                
                for(int j=0; j<facetValues.length; j++) {
                        
                        if(facetCounts[j].equals("0")) {
                                continue;
                        }
                        
                        String drilldownURL = qryString;
                        drilldownURL += "&fq=F_" + facetDefID + ":" + 
                                URLEncoder.encode(facetValues[j], "UTF-8"); 
                        //log.info("drilldownURL=" + drilldownURL);
                        
                        String[] afacet = new String[] { facetValues[j], 
                                        facetCounts[j], drilldownURL };
                        //log.info("afacet=" + Arrays.toString(afacet));
                        
                        values.add(afacet);    
                }
                //log.info("values=" + values);
                
                this.facets.put(facetLabel, values);
                
        }
        //log.info("this.facets=" + this.facets);
        */
        
        return new DefaultHttpHeaders("index")
            .disableCaching();
    }


    public Map<String, String> getSnippets() throws Exception {
        if(results != null){
                snippets = new HashMap<String, String>();       
                for(String docId: results.keySet()){
                                snippets.put(docId, FileUtils.read( Utils.EXPORT_ENTITIES_PATH + docId + File.separator +"SearchSnippet.html"));
                        }
        }
                return snippets;
    }
        
    /**
     * 
     * @return
     */
    public String getQuery() {
        return this.query;
    }
    
    
    /**
     * 
     * @return
     */
    public Map<String, String[]> getResults() {
        return this.results;
    }
    
    /**
     * 
     * @return
     */
    public Map<String, List<String[]>> getFacets() {
        return this.facets;
    }
    
    public Double getMinPrice() {
        return this.minPrice;
    }
    
    public Double getMaxPrice() {
        return this.maxPrice;
    }
    
    /**
     * 
     * @return
     */
    public List<String[]> getCrumbs() {
        return this.crumbs;
    }
    
    /**
     * 
     */
    
    public String getHeaderSnippet(){
                return htmlSnippets.get("HEADER");
        }
        
        public String getMainMenuSnippet(){
                return htmlSnippets.get("MAIN_MENU");
        }
        
        public String getSearchBarSnippet(){
                return htmlSnippets.get("SEARCH_BAR");
        }
                        
        public String getCustomerServiceSnippet(){
                return htmlSnippets.get("CUSTOMER_SERVICE");
        }
        
        public String getSearchHeaderSnippet(){
                return htmlSnippets.get("SEARCH_HEADER");
        }
        
        public String getSearchDetailsSnippet(){
                return htmlSnippets.get("SEARCH_DETAILS");
        }
        
        public String getMyResearchSnippet(){
                return htmlSnippets.get("MY_RESEARCH");
        }
                                
        public String getFooterSnippet(){
                return htmlSnippets.get("FOOTER");
        }
        
        public String getJsFileSnippet(){
                return htmlSnippets.get("JS_FILES");
        }
        
        public String getCssFileSnippet(){
                return htmlSnippets.get("CSS_FILES");
        }
}