/*
 *
 *  * Copyright 2014 Orient Technologies.
 *  *
 *  * Licensed under the Apache License, Version 2.0 (the "License");
 *  * you may not use this file except in compliance with the License.
 *  * You may obtain a copy of the License at
 *  *
 *  *      http://www.apache.org/licenses/LICENSE-2.0
 *  *
 *  * Unless required by applicable law or agreed to in writing, software
 *  * distributed under the License is distributed on an "AS IS" BASIS,
 *  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  * See the License for the specific language governing permissions and
 *  * limitations under the License.
 *
 */

package com.orientechnologies.lucene.query;

import com.orientechnologies.lucene.tx.OLuceneTxChanges;
import com.orientechnologies.orient.core.command.OCommandContext;
import org.apache.lucene.facet.FacetsConfig;
import org.apache.lucene.facet.taxonomy.TaxonomyReader;
import org.apache.lucene.index.MultiReader;
import org.apache.lucene.search.Filter;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.Sort;

import java.io.IOException;

/**
 * Created by Enrico Risa on 08/01/15.
 */
public class OLuceneQueryContext {

  public enum QueryContextCFG {
    NO_FILTER_NO_SORT, FILTER_SORT, FILTER, SORT
  }

  public final    OCommandContext context;
  public final    Query           query;
  public final    Filter          filter;
  public final    Sort            sort;
  protected final IndexSearcher   searcher;
  public          QueryContextCFG cfg;
  public boolean facet     = false;
  public boolean drillDown = false;
  public    TaxonomyReader   reader;
  protected OLuceneTxChanges changes;
  private   FacetsConfig     facetConfig;
  private   String           facetField;
  private   String           drillDownQuery;

  public OLuceneQueryContext(OCommandContext context, IndexSearcher searcher, Query query) {
    this(context, searcher, query, null);
  }

  public OLuceneQueryContext(OCommandContext context, IndexSearcher searcher, Query query, Filter filter) {
    this(context, searcher, query, filter, null);
  }

  public OLuceneQueryContext(OCommandContext context, IndexSearcher searcher, Query query, Filter filter, Sort sort) {
    this.context = context;
    this.searcher = searcher;
    this.query = query;
    this.filter = filter;
    this.sort = sort;
    initCFG();
  }

  private void initCFG() {
    if (filter != null && sort != null) {
      cfg = QueryContextCFG.FILTER_SORT;
    } else if (filter == null && sort == null) {
      cfg = QueryContextCFG.NO_FILTER_NO_SORT;
    } else if (filter != null) {
      cfg = QueryContextCFG.FILTER;
    } else {
      cfg = QueryContextCFG.SORT;
    }
  }

  public OLuceneQueryContext setFacet(boolean facet) {
    this.facet = facet;
    return this;
  }

  public OLuceneQueryContext setReader(TaxonomyReader reader) {
    this.reader = reader;
    return this;
  }

  public FacetsConfig getFacetConfig() {
    return facetConfig;
  }

  public void setFacetConfig(FacetsConfig facetConfig) {
    this.facetConfig = facetConfig;
  }

  public String getFacetField() {
    return facetField;
  }

  public void setFacetField(String facetField) {
    this.facetField = facetField;
  }

  public boolean isDrillDown() {
    return drillDown;
  }

  public String getDrillDownQuery() {
    return drillDownQuery;
  }

  public void setDrillDownQuery(String drillDownQuery) {
    this.drillDownQuery = drillDownQuery;
    drillDown = drillDownQuery != null;
  }

  public boolean isInTx() {
    return changes != null;
  }

  public OLuceneQueryContext setChanges(OLuceneTxChanges changes) {
    this.changes = changes;
    return this;
  }

  public OLuceneTxChanges changes() {
    return changes;
  }

  public IndexSearcher getSearcher() throws IOException {

    return changes == null ?
        searcher :
        new IndexSearcher(new MultiReader(searcher.getIndexReader(), changes.searcher().getIndexReader()));
  }

}