buildSrc/src/main/java/com/uber/okbuck/core/dependency/DependencyCache.java [29:184]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
public class DependencyCache {

  private static final Logger LOG = LoggerFactory.getLogger(DependencyCache.class);
  private final Project rootProject;
  private final DependencyManager dependencyManager;
  private final boolean skipPrebuilt;
  private final Map<VersionlessDependency, OExternalDependency> forcedDeps = new HashMap<>();

  public DependencyCache(
      Project project,
      DependencyManager dependencyManager,
      boolean skipPrebuilt,
      @Nullable String forcedConfiguration) {
    this.rootProject = project.getRootProject();
    this.dependencyManager = dependencyManager;
    this.skipPrebuilt = skipPrebuilt;

    if (forcedConfiguration != null) {
      build(forcedConfiguration)
          .forEach(
              dependency -> {
                forcedDeps.put(dependency.getVersionless(), dependency);
              });
    }
  }

  public DependencyCache(
      Project project, DependencyManager dependencyManager, @Nullable String forcedConfiguration) {
    this(project, dependencyManager, false, forcedConfiguration);
  }

  public DependencyCache(Project project, DependencyManager dependencyManager) {
    this(project, dependencyManager, false);
  }

  public DependencyCache(
      Project project, DependencyManager dependencyManager, boolean skipPrebuilt) {
    this(project, dependencyManager, skipPrebuilt, null);
  }

  public final OExternalDependency get(OExternalDependency externalDependency) {
    LOG.info("Requested dependency {}", externalDependency);
    OExternalDependency dependency =
        forcedDeps.getOrDefault(externalDependency.getVersionless(), externalDependency);
    LOG.info("Picked dependency {}", dependency);

    dependencyManager.addDependency(dependency, skipPrebuilt);

    return dependency;
  }

  public final void addDependencies(DependencySet dependencySet) {
    this.dependencyManager.addRawDependencies(
        dependencySet
            .withType(ExternalDependency.class)
            .stream()
            .filter(dependency -> dependency.getGroup() != null && dependency.getVersion() != null)
            .collect(Collectors.toSet()));
  }

  /**
   * Get the list of annotation processor classes provided by a dependency.
   *
   * @param externalDependency The dependency
   * @return The list of annotation processor classes available in the manifest
   */
  public Set<String> getAnnotationProcessors(OExternalDependency externalDependency) {
    OExternalDependency dependency =
        forcedDeps.getOrDefault(externalDependency.getVersionless(), externalDependency);

    try {
      String processors =
          getJarFileContent(
              dependency.getRealDependencyFile(),
              "META-INF/services/javax.annotation.processing.Processor");

      return ImmutableSet.copyOf(processors.split(","));

    } catch (IOException e) {
      throw new IllegalStateException(e);
    }
  }

  /**
   * Check if the dependency has an auto value extension.
   *
   * @param externalDependency The dependency
   * @return Whether the dependency has auto value extension.
   */
  public boolean hasAutoValueExtension(OExternalDependency externalDependency) {
    OExternalDependency dependency =
        forcedDeps.getOrDefault(externalDependency.getVersionless(), externalDependency);
    try {
      String extensions =
          getJarFileContent(
              dependency.getRealDependencyFile(),
              "META-INF/services/com.google.auto.value.extension.AutoValueExtension");
      return !extensions.isEmpty();
    } catch (IOException e) {
      throw new IllegalStateException(e);
    }
  }

  private static String getJarFileContent(File dependencyFile, String filePathString)
      throws IOException {
    String content;

    JarFile jarFile = new JarFile(dependencyFile);
    JarEntry jarEntry = (JarEntry) jarFile.getEntry(filePathString);
    if (jarEntry != null) {
      List<String> entries =
          Arrays.stream(
                  IOUtils.toString(jarFile.getInputStream(jarEntry), "UTF-8").trim().split("\\n"))
              .filter(
                  entry -> {
                    return !entry.startsWith("#")
                        && !entry.trim().isEmpty(); // filter out comments and empty lines
                  })
              .collect(Collectors.toList());
      content = String.join(",", entries);
    } else {
      content = "";
    }
    return content;
  }

  /**
   * Use this method to populate dependency caches of tools/languages etc. This is not meant to be
   * used across multiple threads/gradle task executions which can run in parallel. This method is
   * fully synchronous.
   *
   * @param configuration The configuration to materialize into the dependency cache
   */
  public Set<OExternalDependency> build(Configuration configuration) {
    OkBuckExtension okBuckExtension = ProjectUtil.getOkBuckExtension(rootProject);

    ExternalDependenciesExtension externalDependenciesExtension =
        okBuckExtension.getExternalDependenciesExtension();
    JetifierExtension jetifierExtension = okBuckExtension.getJetifierExtension();

    addDependencies(configuration.getAllDependencies());

    return DependencyUtils.resolveExternal(
            rootProject, configuration, externalDependenciesExtension, jetifierExtension)
        .stream()
        .map(this::get)
        .peek(externalDependency -> externalDependency.updateFirstLevel(true))
        .collect(Collectors.toSet());
  }

  private Set<OExternalDependency> build(String configuration) {
    Configuration useful = DependencyUtils.useful(configuration, rootProject);

    Preconditions.checkNotNull(useful);

    return build(useful);
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



plugin/src/main/java/com/uber/okbuck/core/dependency/DependencyCache.java [29:184]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
public class DependencyCache {

  private static final Logger LOG = LoggerFactory.getLogger(DependencyCache.class);
  private final Project rootProject;
  private final DependencyManager dependencyManager;
  private final boolean skipPrebuilt;
  private final Map<VersionlessDependency, OExternalDependency> forcedDeps = new HashMap<>();

  public DependencyCache(
      Project project,
      DependencyManager dependencyManager,
      boolean skipPrebuilt,
      @Nullable String forcedConfiguration) {
    this.rootProject = project.getRootProject();
    this.dependencyManager = dependencyManager;
    this.skipPrebuilt = skipPrebuilt;

    if (forcedConfiguration != null) {
      build(forcedConfiguration)
          .forEach(
              dependency -> {
                forcedDeps.put(dependency.getVersionless(), dependency);
              });
    }
  }

  public DependencyCache(
      Project project, DependencyManager dependencyManager, @Nullable String forcedConfiguration) {
    this(project, dependencyManager, false, forcedConfiguration);
  }

  public DependencyCache(Project project, DependencyManager dependencyManager) {
    this(project, dependencyManager, false);
  }

  public DependencyCache(
      Project project, DependencyManager dependencyManager, boolean skipPrebuilt) {
    this(project, dependencyManager, skipPrebuilt, null);
  }

  public final OExternalDependency get(OExternalDependency externalDependency) {
    LOG.info("Requested dependency {}", externalDependency);
    OExternalDependency dependency =
        forcedDeps.getOrDefault(externalDependency.getVersionless(), externalDependency);
    LOG.info("Picked dependency {}", dependency);

    dependencyManager.addDependency(dependency, skipPrebuilt);

    return dependency;
  }

  public final void addDependencies(DependencySet dependencySet) {
    this.dependencyManager.addRawDependencies(
        dependencySet
            .withType(ExternalDependency.class)
            .stream()
            .filter(dependency -> dependency.getGroup() != null && dependency.getVersion() != null)
            .collect(Collectors.toSet()));
  }

  /**
   * Get the list of annotation processor classes provided by a dependency.
   *
   * @param externalDependency The dependency
   * @return The list of annotation processor classes available in the manifest
   */
  public Set<String> getAnnotationProcessors(OExternalDependency externalDependency) {
    OExternalDependency dependency =
        forcedDeps.getOrDefault(externalDependency.getVersionless(), externalDependency);

    try {
      String processors =
          getJarFileContent(
              dependency.getRealDependencyFile(),
              "META-INF/services/javax.annotation.processing.Processor");

      return ImmutableSet.copyOf(processors.split(","));

    } catch (IOException e) {
      throw new IllegalStateException(e);
    }
  }

  /**
   * Check if the dependency has an auto value extension.
   *
   * @param externalDependency The dependency
   * @return Whether the dependency has auto value extension.
   */
  public boolean hasAutoValueExtension(OExternalDependency externalDependency) {
    OExternalDependency dependency =
        forcedDeps.getOrDefault(externalDependency.getVersionless(), externalDependency);
    try {
      String extensions =
          getJarFileContent(
              dependency.getRealDependencyFile(),
              "META-INF/services/com.google.auto.value.extension.AutoValueExtension");
      return !extensions.isEmpty();
    } catch (IOException e) {
      throw new IllegalStateException(e);
    }
  }

  private static String getJarFileContent(File dependencyFile, String filePathString)
      throws IOException {
    String content;

    JarFile jarFile = new JarFile(dependencyFile);
    JarEntry jarEntry = (JarEntry) jarFile.getEntry(filePathString);
    if (jarEntry != null) {
      List<String> entries =
          Arrays.stream(
                  IOUtils.toString(jarFile.getInputStream(jarEntry), "UTF-8").trim().split("\\n"))
              .filter(
                  entry -> {
                    return !entry.startsWith("#")
                        && !entry.trim().isEmpty(); // filter out comments and empty lines
                  })
              .collect(Collectors.toList());
      content = String.join(",", entries);
    } else {
      content = "";
    }
    return content;
  }

  /**
   * Use this method to populate dependency caches of tools/languages etc. This is not meant to be
   * used across multiple threads/gradle task executions which can run in parallel. This method is
   * fully synchronous.
   *
   * @param configuration The configuration to materialize into the dependency cache
   */
  public Set<OExternalDependency> build(Configuration configuration) {
    OkBuckExtension okBuckExtension = ProjectUtil.getOkBuckExtension(rootProject);

    ExternalDependenciesExtension externalDependenciesExtension =
        okBuckExtension.getExternalDependenciesExtension();
    JetifierExtension jetifierExtension = okBuckExtension.getJetifierExtension();

    addDependencies(configuration.getAllDependencies());

    return DependencyUtils.resolveExternal(
            rootProject, configuration, externalDependenciesExtension, jetifierExtension)
        .stream()
        .map(this::get)
        .peek(externalDependency -> externalDependency.updateFirstLevel(true))
        .collect(Collectors.toSet());
  }

  private Set<OExternalDependency> build(String configuration) {
    Configuration useful = DependencyUtils.useful(configuration, rootProject);

    Preconditions.checkNotNull(useful);

    return build(useful);
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -