/*
    * 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.CrawlingInfoDbm;
  import org.codelibs.fess.es.config.cbean.CrawlingInfoCB;
  import org.codelibs.fess.es.config.cbean.ca.CrawlingInfoCA;
  import org.codelibs.fess.es.config.cbean.ca.bs.BsCrawlingInfoCA;
  import org.codelibs.fess.es.config.cbean.cq.CrawlingInfoCQ;
  import org.codelibs.fess.es.config.cbean.cq.bs.BsCrawlingInfoCQ;
  import org.dbflute.cbean.ConditionQuery;
  
  /**
   * @author ESFlute (using FreeGen)
   */
  public class BsCrawlingInfoCB extends EsAbstractConditionBean {
  
      // ===================================================================================
      //                                                                           Attribute
      //                                                                           =========
      protected BsCrawlingInfoCQ _conditionQuery;
      protected BsCrawlingInfoCA _conditionAggregation;
      protected HpSpecification _specification;
  
      // ===================================================================================
      //                                                                             Control
      //                                                                             =======
      @Override
      public CrawlingInfoDbm asDBMeta() {
          return CrawlingInfoDbm.getInstance();
      }
  
      @Override
      public String asTableDbName() {
          return "crawling_info";
      }
  
      @Override
      public boolean hasSpecifiedColumn() {
          return _specification != null;
      }
  
      @Override
      public ConditionQuery localCQ() {
          return doGetConditionQuery();
      }
  
      // ===================================================================================
      //                                                                         Primary Key
      //                                                                         ===========
      public CrawlingInfoCB acceptPK(String id) {
          assertObjectNotNull("id", id);
          BsCrawlingInfoCB cb = this;
          cb.query().docMeta().setId_Equal(id);
          return (CrawlingInfoCB) 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 BsCrawlingInfoCQ query() {
         assertQueryPurpose();
         return doGetConditionQuery();
     }
 
     protected BsCrawlingInfoCQ doGetConditionQuery() {
         if (_conditionQuery == null) {
             _conditionQuery = createLocalCQ();
         }
         return _conditionQuery;
     }
 
     protected BsCrawlingInfoCQ createLocalCQ() {
         return new CrawlingInfoCQ();
     }
 
     // ===================================================================================
     //                                                                         Aggregation
     //                                                                         ===========
     public BsCrawlingInfoCA aggregation() {
         assertAggregationPurpose();
         return doGetConditionAggregation();
     }
 
     protected BsCrawlingInfoCA doGetConditionAggregation() {
         if (_conditionAggregation == null) {
             _conditionAggregation = createLocalCA();
         }
         return _conditionAggregation;
     }
 
     protected BsCrawlingInfoCA createLocalCA() {
         return new CrawlingInfoCA();
     }
 
     // ===================================================================================
     //                                                                             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 columnCreatedTime() {
             doColumn("createdTime");
         }
 
         public void columnExpiredTime() {
             doColumn("expiredTime");
         }
 
         public void columnName() {
             doColumn("name");
         }
 
         public void columnSessionId() {
             doColumn("sessionId");
         }
     }
 }