/**
 * $Id$
 * *****************************************************************************
 *    Copyright (C) 2005 - 2007 somewhere in .Net ltd.
 *    All Rights Reserved.  No use, copying or distribution of this
 *    work may be made except in accordance with a valid license
 *    agreement from somewhere in .Net LTD.  This notice must be included on
 *    all copies, modifications and derivatives of this work.
 * *****************************************************************************
 * $LastChangedBy$
 * $LastChangedDate$
 * $LastChangedRevision$
 * *****************************************************************************
 */

package com.ideabase.repository.core.index;

import org.apache.lucene.queryParser.QueryParser;
import org.apache.lucene.queryParser.ParseException;
import org.apache.lucene.analysis.Analyzer;
import org.apache.lucene.search.*;
import org.apache.lucene.document.NumberTools;
import org.apache.log4j.Logger;
import org.apache.log4j.LogManager;

/**
 * Extended query parser supports Number to pad
 * @author <a href="http://hasan.we4tech.com">nhm tanveer...(hasan)</a>
 */
public class ExtendedQueryParser extends QueryParser {

  private final Logger mLog = LogManager.getLogger(getClass());
  private TermValueEmbedFunctionExecutor mEmbedFunctionExecutor;


  public ExtendedQueryParser(final String pField,
                             final Analyzer pAnalyzer,
                             final TermValueEmbedFunctionExecutor pExecutor) {
    super(pField, pAnalyzer);
    mEmbedFunctionExecutor = pExecutor;
  }

  @Override
  protected Query getFuzzyQuery(final String pField,
                                final String pTermString,
                                final float pMinimumSimilarity)
      throws ParseException {
    return super.getFuzzyQuery(
        pField, executeEmbedFunction(pTermString), pMinimumSimilarity);
  }

  private String executeEmbedFunction(final String pTermString) {
    if (mEmbedFunctionExecutor == null) {
      return pTermString;
    } else {
      final String evaledString = mEmbedFunctionExecutor.eval(pTermString);
      if (mLog.isDebugEnabled()) {
        mLog.debug("Eval'd value - " + evaledString);
      }
      return evaledString;
    }
  }

  @Override
  protected Query getRangeQuery(final String pField,
                                final String pLowerTerm,
                                final String pUpperTerm,
                                final boolean pInclusive)
      throws ParseException {

    final boolean includeLower = pLowerTerm != null && !"*".equals(pLowerTerm);
    final boolean includeUpper = pUpperTerm != null && !"*".equals(pUpperTerm);
    return new ConstantScoreRangeQuery(
        pField,
        includeLower ? NumberTools.longToString(Long.parseLong(pLowerTerm)) : "",
        includeUpper ? NumberTools.longToString(Long.parseLong(pUpperTerm)) : "*",
        includeLower, includeUpper) {
    };
  }
}