/*
    * Copyright 2012-2025 CodeLibs Project and the Others.
    *
    * 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 org.codelibs.fess.query;
  
  import org.apache.logging.log4j.LogManager;
  import org.apache.logging.log4j.Logger;
  import org.apache.lucene.search.BooleanClause;
  import org.apache.lucene.search.BooleanQuery;
  import org.apache.lucene.search.Query;
  import org.codelibs.fess.entity.QueryContext;
  import org.codelibs.fess.exception.InvalidQueryException;
  import org.lastaflute.core.message.UserMessages;
  import org.opensearch.index.query.BoolQueryBuilder;
  import org.opensearch.index.query.QueryBuilder;
  import org.opensearch.index.query.QueryBuilders;
  
  /**
   * Query command implementation for handling Boolean queries.
   * Converts Lucene BooleanQuery objects to OpenSearch BoolQueryBuilder objects.
   */
  public class BooleanQueryCommand extends QueryCommand {
      /**
       * Default constructor for BooleanQueryCommand.
       */
      public BooleanQueryCommand() {
          super();
      }
  
      private static final Logger logger = LogManager.getLogger(BooleanQueryCommand.class);
  
      @Override
      protected String getQueryClassName() {
          return BooleanQuery.class.getSimpleName();
      }
  
      @Override
      public QueryBuilder execute(final QueryContext context, final Query query, final float boost) {
          if (query instanceof final BooleanQuery booleanQuery) {
              if (logger.isDebugEnabled()) {
                  logger.debug("BooleanQuery: query={}, boost={}", query, boost);
              }
              return convertBooleanQuery(context, booleanQuery, boost);
          }
          throw new InvalidQueryException(messages -> messages.addErrorsInvalidQueryUnknown(UserMessages.GLOBAL_PROPERTY_KEY),
                  "Unknown q: " + query.getClass() + " => " + query);
      }
  
      /**
       * Converts a Lucene BooleanQuery to an OpenSearch BoolQueryBuilder.
       * @param context The query context.
       * @param booleanQuery The boolean query to convert.
       * @param boost The boost factor to apply.
       * @return The converted BoolQueryBuilder.
       */
      protected QueryBuilder convertBooleanQuery(final QueryContext context, final BooleanQuery booleanQuery, final float boost) {
          final BoolQueryBuilder boolQuery = QueryBuilders.boolQuery();
          for (final BooleanClause clause : booleanQuery.clauses()) {
              final QueryBuilder queryBuilder = getQueryProcessor().execute(context, clause.query(), boost);
              if (queryBuilder != null) {
                  switch (clause.occur()) {
                  case MUST:
                      boolQuery.must(queryBuilder);
                      break;
                  case SHOULD:
                      boolQuery.should(queryBuilder);
                      break;
                  case MUST_NOT:
                      boolQuery.mustNot(queryBuilder);
                      break;
                  default:
                      break;
                  }
              }
          }
          if (boolQuery.hasClauses()) {
              return boolQuery;
          }
          return null;
      }
  }