Subversion Repositories SmartDukaan

package com.hotspotstore.controllers;

import java.io.IOException;

import java.io.InputStream;

import java.io.UnsupportedEncodingException;

import java.net.MalformedURLException;

import java.net.URISyntaxException;

import java.net.URL;

import java.net.URLEncoder;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Collections;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.StringTokenizer;

import org.apache.commons.io.IOUtils;

import org.apache.commons.lang.StringUtils;

import org.apache.http.client.utils.URIBuilder;

import org.apache.log4j.Logger;

import org.json.JSONException;

import org.json.JSONObject;

import com.google.gson.Gson;

import com.hotspotstore.model.SolrResultPojo;

public class SearchController extends BaseController{

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

	private String id;

	private SolrResultPojo ssr;

	private long windowSize = 24;

	private String q;

	public String getQ() {

		return q;

	}

	public void setQ(String q) {

		this.q = q;

	}

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

	public static final Map<String, String> FACET_LABEL_MAP = Collections.unmodifiableMap(

			new HashMap<String, String>(){

				/**

				 * 

				 */

				private static final long serialVersionUID = 1L;

				{

					put("F_50010","Category");

					put("F_50011","Sub Category");

					put("F_50002","Price");

					put("F_50001","Brand");

					put("F_50006","Data Connectivity");

					put("F_50007","Camera Resolution");

					put("F_50039","");

					put("F_50012","Display");

					put("F_50013","Operating System");

					put("F_50014","RAM");

					put("F_50015","Storage Capacity");

					put("F_50017","Processor");

					put("F_50018","Capacity");

					put("F_50019","Class");

					put("F_50020","Capacity");

					put("F_50021","Capacity");

					put("F_50022","Type");

					put("F_50023","Interface");

					//Resolution for Composite

					put("F_50024","Resolution");

					put("F_50025","Optical Zoom");

					put("F_50026","Display Size");

					//Resolution for DSLR

					put("F_50027","Resolution");

					put("F_50028","Availability");

					put("F_50031","Operating System");

					put("F_50032","Screen Size");	//Screen size for mobiles

					put("F_50033","Screen Size");	//Screen size for laptop

					put("F_50034","Screen Size"); 	//Screen size for tablets

					put("F_50035","Operating System"); 	//Screen size for tablets

					put("F_50036","Camera Resolution"); 	//Screen size for tablets

					put("F_50037","Voice Call Facility"); 	//Screen size for tablets

					put("F_50038","Mobile Type"); 	//Mobile Type

					put("F_50028","Availability");

				}

		});

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

		return crumbs;

	}

	public void setCrumbs(Map<String, List<List<String>>> crumbs) {

		this.crumbs = crumbs;

	}

	public long getWindowSize() {

		return windowSize;

	}

	public void setWindowSize(long windowSize) {

		this.windowSize = windowSize;

	}

	public long getPage() {

		return page;

	}

	public void setPage(long page) {

		this.page = page;

	}

	public long getBeginIndex(){

		if(totalResults>0)

			return beginIndex+1;

		return beginIndex;

	}

	public long getTotalPages() {

		return 1 + (totalResults-1)/windowSize;

	}

	public void setBeginIndex(long beginIndex) {

		this.beginIndex = beginIndex;

	}

	private long page = 1;

	private long totalResults;

	private long beginIndex = 0;

	private String url;

	public String getUrl() {

		return url;

	}

	public void setUrl(String url) {

		this.url = url;

	}

	public long getTotalResults() {

		return totalResults;

	}

	public void setTotalResults(long totalResults) {

		this.totalResults = totalResults;

	}

	public long getCurrentPage() {

		return this.page;

	}

	private static String mobileApiUrl = "http://127.0.0.1:8080/mobileapi";

	public String index() throws URISyntaxException, IOException, JSONException{

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

		String query = q;

		int length = 1;

		if(fqrys!= null){

			length += fqrys.length;

		}

		url = "?";

		String[] newfqrys = new String[length];

		String urlCrumb = url;

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

		if(query != null) {

			urlCrumb = url; 

		}

		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");

				newfqrys[i+1] = fqrys[i];

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

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

				String facetLabel = 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);

			}

		}

		newfqrys[0] ="H_00001:true";

		url = urlCrumb;

		log.info("crumbs "+crumbs);

		String category_url = mobileApiUrl+"/search/";

		URIBuilder generalSearchUrl = new URIBuilder(category_url);

		generalSearchUrl.addParameter("q", query);

		for (String s : newfqrys){

			if (s==null || s.isEmpty()){

				continue;

			}

			generalSearchUrl.addParameter("fq", s.trim());  

		}

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

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

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

		}

		generalSearchUrl.addParameter("index", String.valueOf(beginIndex));

		generalSearchUrl.addParameter("offset", String.valueOf(windowSize));

		URL url = generalSearchUrl.build().toURL();

		log.info("Search Url Search Results"+url.toString());

		InputStream is = url.openStream();

		String jsonString;

		try{

			jsonString = IOUtils.toString(is, "UTF-8");

		}

		finally{

			is.close();

		}

		JSONObject jsonObj = new JSONObject(jsonString);

		Gson gson = new Gson();

		ssr = gson.fromJson(jsonObj.getString("result"), SolrResultPojo.class);

		this.totalResults = ssr.getTotalCount();

		log.info("ssr label "+ssr.getLabel());

		return "index";

	}

	public String getEncodedCompontent(String input) throws UnsupportedEncodingException{

		return URLEncoder.encode(input, "UTF-8");

	}

	public String getId() {

		return id;

	}

	public SolrResultPojo getSsr() {

		return ssr;

	}

	public void setSsr(SolrResultPojo ssr) {

		this.ssr = ssr;

	}

	public void setId(String id) {

		StringTokenizer tokenizer = new StringTokenizer(id,"-");

		while(tokenizer.hasMoreTokens()){

			this.id = tokenizer.nextToken();

		}

	}

}