package org.openmrs.module.chartsearch.page.controller;

/**
 * Created by Eli on 10/03/14.
 */

import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;

import javax.servlet.http.HttpServletRequest;

import org.apache.solr.client.solrj.response.FacetField;
import org.apache.solr.client.solrj.response.FacetField.Count;
import org.openmrs.Patient;
import org.openmrs.api.context.Context;
import org.openmrs.module.appui.UiSessionContext;
import org.openmrs.module.chartsearch.ChartListItem;
import org.openmrs.module.chartsearch.EncounterItem;
import org.openmrs.module.chartsearch.FormItem;
import org.openmrs.module.chartsearch.ObsItem;
import org.openmrs.module.chartsearch.SearchAPI;
import org.openmrs.module.chartsearch.SearchPhrase;
import org.openmrs.module.chartsearch.solr.ChartSearchIndexer;
import org.openmrs.module.chartsearch.solr.ChartSearchSearcher;
import org.openmrs.module.chartsearch.web.dwr.DWRChartSearchService;
import org.openmrs.module.emrapi.patient.PatientDomainWrapper;
import org.openmrs.ui.framework.annotation.BindParams;
import org.openmrs.ui.framework.annotation.InjectBeans;
import org.openmrs.ui.framework.page.PageModel;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.RequestParam;

public class ChartsearchPageController {

    private static final Logger log = LoggerFactory.getLogger(ChartsearchPageController.class);

    private ChartSearchIndexer chartSearchIndexer = getComponent(ChartSearchIndexer.class);

    private ChartSearchSearcher searcher = getComponent(ChartSearchSearcher.class);

	private List<Count> facets;

	public List<Count> getFacets() {
		return facets;
	}

	public void controller(PageModel model, @BindParams SearchPhrase search_phrase, UiSessionContext sessionContext, @RequestParam("patientId") Patient patient,
                           @InjectBeans PatientDomainWrapper patientDomainWrapper, HttpServletRequest request) {

    	String[] categories = request.getParameterValues("categories");
    	if (categories == null) {
        	categories = new String[0];
        }
    	List<String> selectedCategories = Arrays.asList(categories);


        patientDomainWrapper.setPatient(patient);
        model.addAttribute("patient", patientDomainWrapper);

        SearchAPI searchAPIInstance = SearchAPI.getInstance();

        log.info("getting patient ID :" + patient);
        log.info("trying to index a patient");

        if (chartSearchIndexer != null && patient != null) {
            //chartSearchIndexer.clearIndex(IndexClearStrategies.IDS.toString(), patient.getPatientId()+"", 0, 0);
            chartSearchIndexer.indexPatientData(patient.getPatientId());
        }
        log.info("indexed patient");
        //Searching an empty phrase to get all results to show at start
        SearchPhrase emptyPhrase = new SearchPhrase("");
        List<ChartListItem> items = searchAPIInstance.search(patient.getPatientId(), search_phrase, selectedCategories);
        List<ChartListItem> updatedItems = new ArrayList<ChartListItem>();
        //loop to get full details about observations.
        for (ChartListItem chartListItem : items) {
            if (chartListItem instanceof ObsItem) {
                int itemObsId = ((ObsItem) chartListItem).getObsId();
                ChartListItem updatedObservation = DWRChartSearchService.getObservationDetails(itemObsId);
                updatedItems.add(updatedObservation);
            }
            if (chartListItem instanceof FormItem) {
                updatedItems.add(chartListItem);
            }

            if (chartListItem instanceof EncounterItem) {
                updatedItems.add(chartListItem);
            }

        }

        //setting results to show.
        searchAPIInstance.setResults(updatedItems);

        LinkedList<Count> facetFields = new LinkedList<Count> ();
    	facetFields.addAll(searcher.getFacetFieldValueNamesAndCounts());

    	for (int i = facetFields.indexOf(facetFields.getFirst()); i <= facetFields.indexOf(facetFields.getLast()); i++) {
    		System.out.println("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
    		Count facet = facetFields.get(i);
    		this.facets.add(facet);
    		System.out.println("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
    	}
    	model.addAttribute("facets", getFacets());
    	System.out.println("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");

    }

    private <T> T getComponent(Class<T> clazz) {
        List<T> list = Context.getRegisteredComponents(clazz);
        if (list == null || list.size() == 0)
            throw new RuntimeException("Cannot find component of " + clazz);
        return list.get(0);
    }

}