/*
    * Copyright 2012-2021 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.es.config.cbean.bs;
  
  import java.util.ArrayList;
  import java.util.List;
  import java.util.Map;
  
  import org.codelibs.fesen.action.search.SearchRequestBuilder;
  import org.codelibs.fesen.index.query.QueryBuilder;
  import org.codelibs.fess.es.config.allcommon.EsAbstractConditionBean;
  import org.codelibs.fess.es.config.bsentity.dbmeta.RelatedQueryDbm;
  import org.codelibs.fess.es.config.cbean.RelatedQueryCB;
  import org.codelibs.fess.es.config.cbean.ca.RelatedQueryCA;
  import org.codelibs.fess.es.config.cbean.ca.bs.BsRelatedQueryCA;
  import org.codelibs.fess.es.config.cbean.cq.RelatedQueryCQ;
  import org.codelibs.fess.es.config.cbean.cq.bs.BsRelatedQueryCQ;
  import org.dbflute.cbean.ConditionQuery;
  
  /**
   * @author ESFlute (using FreeGen)
   */
  public class BsRelatedQueryCB extends EsAbstractConditionBean {
  
      // ===================================================================================
      //                                                                           Attribute
      //                                                                           =========
      protected BsRelatedQueryCQ _conditionQuery;
      protected BsRelatedQueryCA _conditionAggregation;
      protected HpSpecification _specification;
  
      // ===================================================================================
      //                                                                             Control
      //                                                                             =======
      @Override
      public RelatedQueryDbm asDBMeta() {
          return RelatedQueryDbm.getInstance();
      }
  
      @Override
      public String asTableDbName() {
          return "related_query";
      }
  
      @Override
      public boolean hasSpecifiedColumn() {
          return _specification != null;
      }
  
      @Override
      public ConditionQuery localCQ() {
          return doGetConditionQuery();
      }
  
      // ===================================================================================
      //                                                                         Primary Key
      //                                                                         ===========
      public RelatedQueryCB acceptPK(String id) {
          assertObjectNotNull("id", id);
          BsRelatedQueryCB cb = this;
          cb.query().docMeta().setId_Equal(id);
          return (RelatedQueryCB) this;
      }
  
      @Override
      public void acceptPrimaryKeyMap(Map<String, ? extends Object> primaryKeyMap) {
          acceptPK((String) primaryKeyMap.get("_id"));
      }
  
      // ===================================================================================
      //                                                                               Build
      //                                                                               =====
  
      @Override
      public SearchRequestBuilder build(SearchRequestBuilder builder) {
          if (_conditionQuery != null) {
              QueryBuilder queryBuilder = _conditionQuery.getQuery();
              if (queryBuilder != null) {
                  builder.setQuery(queryBuilder);
              }
              _conditionQuery.getFieldSortBuilderList().forEach(sort -> {
                  builder.addSort(sort);
              });
          }
  
          if (_conditionAggregation != null) {
             _conditionAggregation.getAggregationBuilderList().forEach(builder::addAggregation);
         }
 
         if (_specification != null) {
             builder.setFetchSource(_specification.columnList.toArray(new String[_specification.columnList.size()]), null);
         }
 
         return builder;
     }
 
     // ===================================================================================
     //                                                                               Query
     //                                                                               =====
     public BsRelatedQueryCQ query() {
         assertQueryPurpose();
         return doGetConditionQuery();
     }
 
     protected BsRelatedQueryCQ doGetConditionQuery() {
         if (_conditionQuery == null) {
             _conditionQuery = createLocalCQ();
         }
         return _conditionQuery;
     }
 
     protected BsRelatedQueryCQ createLocalCQ() {
         return new RelatedQueryCQ();
     }
 
     // ===================================================================================
     //                                                                         Aggregation
     //                                                                         ===========
     public BsRelatedQueryCA aggregation() {
         assertAggregationPurpose();
         return doGetConditionAggregation();
     }
 
     protected BsRelatedQueryCA doGetConditionAggregation() {
         if (_conditionAggregation == null) {
             _conditionAggregation = createLocalCA();
         }
         return _conditionAggregation;
     }
 
     protected BsRelatedQueryCA createLocalCA() {
         return new RelatedQueryCA();
     }
 
     // ===================================================================================
     //                                                                             Specify
     //                                                                             =======
     public HpSpecification specify() {
         assertSpecifyPurpose();
         if (_specification == null) {
             _specification = new HpSpecification();
         }
         return _specification;
     }
 
     protected void assertQueryPurpose() {
     }
 
     protected void assertAggregationPurpose() {
     }
 
     protected void assertSpecifyPurpose() {
     }
 
     public static class HpSpecification {
         protected List<String> columnList = new ArrayList<>();
 
         private void doColumn(String name) {
             columnList.add(name);
         }
 
         public void columnId() {
             doColumn("_id");
         }
 
         public void columnCreatedBy() {
             doColumn("createdBy");
         }
 
         public void columnCreatedTime() {
             doColumn("createdTime");
         }
 
         public void columnQueries() {
             doColumn("queries");
         }
 
         public void columnTerm() {
             doColumn("term");
         }
 
         public void columnUpdatedBy() {
             doColumn("updatedBy");
         }
 
         public void columnUpdatedTime() {
             doColumn("updatedTime");
         }
 
         public void columnVirtualHost() {
             doColumn("virtualHost");
         }
     }
 }