src/main/java/com/googlesource/gerrit/plugins/depends/on/Resolver.java

// Copyright (C) 2021 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.googlesource.gerrit.plugins.depends.on;

import com.google.gerrit.entities.BranchNameKey;
import com.google.gerrit.server.query.change.ChangeData;
import com.google.gerrit.server.query.change.InternalChangeQuery;
import com.google.inject.Inject;
import com.google.inject.Provider;
import java.util.HashSet;
import java.util.Set;

public class Resolver {
  private final Provider<InternalChangeQuery> queryProvider;

  @Inject
  public Resolver(Provider<InternalChangeQuery> queryProvider) {
    this.queryProvider = queryProvider;
  }

  /** Are all the deps resolved to a specific change? */
  protected static boolean isResolved(Set<DependsOn> deps) {
    for (DependsOn dep : deps) {
      if (dep.id() == null) {
        return false;
      }
    }
    return true;
  }

  protected Set<DependsOn> resolve(Set<DependsOn> deps, Set<Set<BranchNameKey>> deliverables) {
    /* ToDo: optimize and query all changes with a given Key up front */
    Set<DependsOn> current = new HashSet<>();
    for (DependsOn dep : deps) {
      if (dep.id() == null) {
        for (Set<BranchNameKey> deliverable : deliverables) {
          Set<DependsOn> resolved = resolve(dep, deliverable);
          if (resolved.isEmpty()) {
            current.add(dep);
          } else {
            current.addAll(resolved);
          }
        }
      } else {
        current.add(dep);
      }
    }
    return current;
  }

  protected Set<DependsOn> resolve(DependsOn dep, Set<BranchNameKey> deliverable) {
    Set<DependsOn> found = new HashSet<>();
    // Although we expect at most one change for each deliverable,
    // it is possible that someone re-used a change id across different projects,
    // therefore return all changes and let the caller decide what to do with them.
    for (BranchNameKey branch : deliverable) {
      for (ChangeData change : queryProvider.get().byBranchKey(branch, dep.key())) {
        found.add(DependsOn.create(String.valueOf(change.getId())));
      }
    }
    return found;
  }
}