content: /**
   * Licensed to the Apache Software Foundation (ASF) under one
   * or more contributor license agreements.  See the NOTICE file
   * distributed with this work for additional information
   * regarding copyright ownership.  The ASF licenses this file
   * to you 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
   * <p>
   * http://www.apache.org/licenses/LICENSE-2.0
   * <p>
   * 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.apache.atlas.discovery;
  
  import org.apache.atlas.repository.Constants;
  import org.apache.atlas.type.AtlasStructType;
  import org.apache.commons.collections.CollectionUtils;
  import org.apache.commons.lang3.StringUtils;
  
  import static org.apache.atlas.discovery.SearchContext.MATCH_ALL_NOT_CLASSIFIED;
  import static org.apache.atlas.discovery.SearchProcessor.INDEX_SEARCH_PREFIX;
  import static org.apache.atlas.repository.Constants.CLASSIFICATION_NAMES_KEY;
  import static org.apache.atlas.repository.Constants.PROPAGATED_CLASSIFICATION_NAMES_KEY;
  import static org.apache.atlas.repository.Constants.STATE_PROPERTY_KEY;
  
  public class GraphIndexQueryBuilder {
      SearchContext context;
  
      GraphIndexQueryBuilder(SearchContext context) {
          this.context = context;
      }
  
      void addClassificationTypeFilter(StringBuilder indexQuery) {
          if (indexQuery != null && CollectionUtils.isNotEmpty(context.getClassificationNames())) {
              String classificationNames = AtlasStructType.AtlasAttribute.escapeIndexQueryValue(context.getClassificationNames(), true);
  
              if (indexQuery.length() != 0) {
                  indexQuery.append(" AND ");
              }
  
              indexQuery.append("(").append(INDEX_SEARCH_PREFIX).append('\"').append(CLASSIFICATION_NAMES_KEY).append('\"').append(':').append(classificationNames)
                      .append(" OR ").append(INDEX_SEARCH_PREFIX).append('\"').append(PROPAGATED_CLASSIFICATION_NAMES_KEY)
                      .append('\"').append(':').append(classificationNames).append(")");
          }
      }
  
      void addClassificationAndSubTypesQueryFilter(StringBuilder indexQuery) {
          if (indexQuery != null && CollectionUtils.isNotEmpty(context.getClassificationTypes())) {
              String classificationTypesQryStr = context.getClassificationTypesQryStr();
  
              if (indexQuery.length() != 0) {
                  indexQuery.append(" AND ");
              }
  
              indexQuery.append("(").append(INDEX_SEARCH_PREFIX).append("\"").append(CLASSIFICATION_NAMES_KEY)
                      .append("\"").append(":").append(classificationTypesQryStr).append(" OR ").append(INDEX_SEARCH_PREFIX)
                      .append("\"").append(PROPAGATED_CLASSIFICATION_NAMES_KEY).append("\"").append(":").append(classificationTypesQryStr).append(")");
          }
      }
  
      void addClassificationFilterForBuiltInTypes(StringBuilder indexQuery) {
          if (indexQuery != null && CollectionUtils.isNotEmpty(context.getClassificationTypes())) {
              if (context.getClassificationTypes().iterator().next() == MATCH_ALL_NOT_CLASSIFIED) {
                  if (indexQuery.length() != 0) {
                      indexQuery.append(" AND ");
                  }
  
                  indexQuery.append("( *:* -").append(INDEX_SEARCH_PREFIX).append("\"").append(CLASSIFICATION_NAMES_KEY).append("\":[* TO *] AND -")
                          .append(INDEX_SEARCH_PREFIX).append("\"").append(PROPAGATED_CLASSIFICATION_NAMES_KEY).append("\":[* TO *])");
              }
          }
      }
  
      void addActiveStateQueryFilter(StringBuilder indexQuery) {
          if (context.getSearchParameters().getExcludeDeletedEntities() && indexQuery != null) {
              if (indexQuery.length() != 0) {
                  indexQuery.append(" AND ");
              }
  
              indexQuery.append("(").append(INDEX_SEARCH_PREFIX).append("\"").append(STATE_PROPERTY_KEY).append("\"").append(":" + "ACTIVE").append(")");
          }
      }
  
      void addTypeAndSubTypesQueryFilter(StringBuilder indexQuery, String typeAndAllSubTypesQryStr) {
          if (indexQuery != null && StringUtils.isNotEmpty(typeAndAllSubTypesQryStr)) {
              if (indexQuery.length() > 0) {
                  indexQuery.append(" AND ");
              }
  
              indexQuery.append("(").append(INDEX_SEARCH_PREFIX).append("\"").append(Constants.TYPE_NAME_PROPERTY_KEY)
                      .append("\":").append(typeAndAllSubTypesQryStr).append(")");
          }
      }
  
      void addTypeAndSubTypesQueryFilterEdges(StringBuilder indexQuery, String typeAndAllSubTypesQryStr) {
          if (indexQuery != null && StringUtils.isNotEmpty(typeAndAllSubTypesQryStr)) {
              if (indexQuery.length() > 0) {
                  indexQuery.append(" AND ");
              }
  
              indexQuery.append("(").append(INDEX_SEARCH_PREFIX).append("\"").append(Constants.RELATIONSHIP_TYPE_PROPERTY_KEY).append("\":").append(typeAndAllSubTypesQryStr).append(")");
          }
      }
  }
  