gerrit-server/src/main/java/com/google/gerrit/server/query/change/ParentProjectPredicate.java

// Copyright (C) 2013 The Android Open Source Project
//
// 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 com.google.gerrit.server.query.change;

import com.google.gerrit.extensions.common.ProjectInfo;
import com.google.gerrit.reviewdb.client.Project;
import com.google.gerrit.server.CurrentUser;
import com.google.gerrit.server.project.ListChildProjects;
import com.google.gerrit.server.project.ProjectCache;
import com.google.gerrit.server.project.ProjectResource;
import com.google.gerrit.server.project.ProjectState;
import com.google.gerrit.server.query.OrPredicate;
import com.google.gerrit.server.query.Predicate;
import com.google.inject.Provider;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

class ParentProjectPredicate extends OrPredicate<ChangeData> {
  private final String value;

  ParentProjectPredicate(
      ProjectCache projectCache,
      Provider<ListChildProjects> listChildProjects,
      Provider<CurrentUser> self,
      String value) {
    super(predicates(projectCache, listChildProjects, self, value));
    this.value = value;
  }

  private static List<Predicate<ChangeData>> predicates(
      ProjectCache projectCache,
      Provider<ListChildProjects> listChildProjects,
      Provider<CurrentUser> self,
      String value) {
    ProjectState projectState = projectCache.get(new Project.NameKey(value));
    if (projectState == null) {
      return Collections.emptyList();
    }

    List<Predicate<ChangeData>> r = new ArrayList<>();
    r.add(new ProjectPredicate(projectState.getProject().getName()));
    ListChildProjects children = listChildProjects.get();
    children.setRecursive(true);
    for (ProjectInfo p : children.apply(new ProjectResource(projectState.controlFor(self.get())))) {
      r.add(new ProjectPredicate(p.name));
    }
    return r;
  }

  @Override
  public String toString() {
    return ChangeQueryBuilder.FIELD_PARENTPROJECT + ":" + value;
  }
}