buildSrc/src/main/java/com/uber/okbuck/core/dependency/DependencyFactory.java [22:224]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
public final class DependencyFactory {

  public static final String LOCAL_GROUP = "local";

  private static final String LOCAL_DEP_VERSION = "1.0.0-LOCAL";

  private final Map<ExternalDependency, Set<VersionlessDependency>> unresolvedToVersionless =
      new HashMap<>();

  private final Map<OResolvedDependency, OExternalDependency> externalDependencyCache =
      new HashMap<>();

  public DependencyFactory() {}

  /**
   * Create an External Dependency
   *
   * @param group group of the dependency
   * @param name name of the dependency
   * @param version version of the dependency
   * @param dependencyFile file of the dependency
   * @param externalDependenciesExtension External Dependency Extension
   * @param jetifierExtension Jetifier Extension
   * @return External Dependency
   */
  public synchronized OExternalDependency from(
      String group,
      String name,
      String version,
      File dependencyFile,
      @Nullable File dependencySourceFile,
      ExternalDependenciesExtension externalDependenciesExtension,
      JetifierExtension jetifierExtension) {
    String classifier = DependencyUtils.getModuleClassifier(dependencyFile.getName(), version);

    VersionlessDependency versionlessDependency =
        VersionlessDependency.builder()
            .setGroup(group)
            .setName(name)
            .setClassifier(Optional.ofNullable(Strings.emptyToNull(classifier)))
            .build();

    OResolvedDependency resolvedDependency =
        OResolvedDependency.builder()
            .setVersionless(versionlessDependency)
            .setVersion(version)
            .setIsVersioned(externalDependenciesExtension.isVersioned(versionlessDependency))
            .setRealDependencyFile(dependencyFile)
            .setRealDependencySourceFile(Optional.ofNullable(dependencySourceFile))
            .build();

    if (externalDependencyCache.containsKey(resolvedDependency)) {
      return externalDependencyCache.get(resolvedDependency);
    }

    OExternalDependency externalDependency;

    if (group.equals(LOCAL_GROUP) || isLocalDependency(dependencyFile.getAbsolutePath())) {
      externalDependency =
          new LocalOExternalDependency(
              resolvedDependency, externalDependenciesExtension, jetifierExtension);
    } else {
      externalDependency =
          new OExternalDependency(
              resolvedDependency, externalDependenciesExtension, jetifierExtension);
    }

    externalDependencyCache.put(resolvedDependency, externalDependency);
    return externalDependency;
  }

  /**
   * Create an External Dependency from a local dependency
   *
   * @param localDependency local dependency file
   * @param externalDependenciesExtension External Dependency Extension
   * @param jetifierExtension Jetifier Extension
   * @return External Dependency
   */
  public LocalOExternalDependency fromLocal(
      File localDependency,
      @Nullable File localSourceDependency,
      ExternalDependenciesExtension externalDependenciesExtension,
      JetifierExtension jetifierExtension) {

    String name = FilenameUtils.getBaseName(localDependency.getName());
    return (LocalOExternalDependency)
        from(
            LOCAL_GROUP,
            name,
            LOCAL_DEP_VERSION,
            localDependency,
            localSourceDependency,
            externalDependenciesExtension,
            jetifierExtension);
  }

  public synchronized Set<VersionlessDependency> fromDependency(ExternalDependency dependency) {
    if (unresolvedToVersionless.containsKey(dependency)) {
      return unresolvedToVersionless.get(dependency);
    } else {
      VersionlessDependency.Builder vDependencyBuilder =
          VersionlessDependency.builder().setName(dependency.getName());
      String group = dependency.getGroup();

      if (group == null) {
        vDependencyBuilder.setGroup(LOCAL_GROUP);
      } else {
        vDependencyBuilder.setGroup(group);
      }

      Set<VersionlessDependency> vDeps = new HashSet<>();

      if (dependency.getArtifacts().size() > 0) {
        vDeps.addAll(
            dependency
                .getArtifacts()
                .stream()
                .map(
                    dependencyArtifact ->
                        vDependencyBuilder
                            .setClassifier(Optional.ofNullable(dependencyArtifact.getClassifier()))
                            .build())
                .collect(Collectors.toSet()));
      } else {
        vDeps.add(vDependencyBuilder.build());
      }

      unresolvedToVersionless.put(dependency, vDeps);
      return vDeps;
    }
  }

  /**
   * Returns a set of versionless dependency from the given gradle resolved dependency.
   *
   * @param dependency gradle dependency
   * @return VersionlessDependency object
   */
  public static Set<VersionlessDependency> fromDependency(ResolvedDependency dependency) {
    Set<ResolvedArtifact> selfArtifacts = dependency.getModuleArtifacts();
    return fromResolvedArtifacts(selfArtifacts);
  }

  /**
   * Returns a set of versionless dependency from the given gradle resolved dependency.
   *
   * @param dependency gradle dependency
   * @return VersionlessDependency object
   */
  public static Set<VersionlessDependency> childrenFromDependency(ResolvedDependency dependency) {
    Set<ResolvedArtifact> childArtifacts =
        dependency
            .getChildren()
            .stream()
            .map(child -> child.getParentArtifacts(dependency))
            .flatMap(Collection::stream)
            .collect(Collectors.toSet());

    return fromResolvedArtifacts(childArtifacts);
  }

  private static Set<VersionlessDependency> fromResolvedArtifacts(Set<ResolvedArtifact> artifacts) {
    return artifacts
        .stream()
        .map(
            resolvedArtifact -> {
              if (resolvedArtifact.getId().getComponentIdentifier()
                  instanceof ModuleComponentIdentifier) {
                ModuleComponentIdentifier id =
                    (ModuleComponentIdentifier) resolvedArtifact.getId().getComponentIdentifier();
                return VersionlessDependency.builder()
                    .setName(id.getModule())
                    .setGroup(id.getGroup())
                    .setClassifier(Optional.ofNullable(resolvedArtifact.getClassifier()))
                    .build();
              }
              return null;
            })
        .filter(Objects::nonNull)
        .collect(Collectors.toSet());
  }

  public void finalizeDependencies() {
    unresolvedToVersionless.clear();
    externalDependencyCache.clear();
  }

  /**
   * Returns whether the dependency should be marked local or not. These dependencies can't be
   * downloaded via buck. Snapshot dependencies with version suffixed with `-SNAPSHOT`. Local m2
   * dependencies with version suffixed with `-LOCAL`.
   *
   * <p>Note: we need to pass in the whole dependency file instead of just the version. `-SNAPSHOT`
   * dependencies with specific date and time in version is also not supported by buck and hence to
   * access correctly we need to look at the whole path. eg:
   * com/jakewharton/butterknife/9.0.0-SNAPSHOT/butterknife-9.0.0-20181220.030319-77.jar
   *
   * @param dependencyFilePath Absolute path string of the dependency file
   * @return Whether the dependency should be local or not.
   */
  private static boolean isLocalDependency(String dependencyFilePath) {
    return dependencyFilePath.contains("-SNAPSHOT") || dependencyFilePath.contains("-LOCAL");
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



plugin/src/main/java/com/uber/okbuck/core/dependency/DependencyFactory.java [22:224]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
public final class DependencyFactory {

  public static final String LOCAL_GROUP = "local";

  private static final String LOCAL_DEP_VERSION = "1.0.0-LOCAL";

  private final Map<ExternalDependency, Set<VersionlessDependency>> unresolvedToVersionless =
      new HashMap<>();

  private final Map<OResolvedDependency, OExternalDependency> externalDependencyCache =
      new HashMap<>();

  public DependencyFactory() {}

  /**
   * Create an External Dependency
   *
   * @param group group of the dependency
   * @param name name of the dependency
   * @param version version of the dependency
   * @param dependencyFile file of the dependency
   * @param externalDependenciesExtension External Dependency Extension
   * @param jetifierExtension Jetifier Extension
   * @return External Dependency
   */
  public synchronized OExternalDependency from(
      String group,
      String name,
      String version,
      File dependencyFile,
      @Nullable File dependencySourceFile,
      ExternalDependenciesExtension externalDependenciesExtension,
      JetifierExtension jetifierExtension) {
    String classifier = DependencyUtils.getModuleClassifier(dependencyFile.getName(), version);

    VersionlessDependency versionlessDependency =
        VersionlessDependency.builder()
            .setGroup(group)
            .setName(name)
            .setClassifier(Optional.ofNullable(Strings.emptyToNull(classifier)))
            .build();

    OResolvedDependency resolvedDependency =
        OResolvedDependency.builder()
            .setVersionless(versionlessDependency)
            .setVersion(version)
            .setIsVersioned(externalDependenciesExtension.isVersioned(versionlessDependency))
            .setRealDependencyFile(dependencyFile)
            .setRealDependencySourceFile(Optional.ofNullable(dependencySourceFile))
            .build();

    if (externalDependencyCache.containsKey(resolvedDependency)) {
      return externalDependencyCache.get(resolvedDependency);
    }

    OExternalDependency externalDependency;

    if (group.equals(LOCAL_GROUP) || isLocalDependency(dependencyFile.getAbsolutePath())) {
      externalDependency =
          new LocalOExternalDependency(
              resolvedDependency, externalDependenciesExtension, jetifierExtension);
    } else {
      externalDependency =
          new OExternalDependency(
              resolvedDependency, externalDependenciesExtension, jetifierExtension);
    }

    externalDependencyCache.put(resolvedDependency, externalDependency);
    return externalDependency;
  }

  /**
   * Create an External Dependency from a local dependency
   *
   * @param localDependency local dependency file
   * @param externalDependenciesExtension External Dependency Extension
   * @param jetifierExtension Jetifier Extension
   * @return External Dependency
   */
  public LocalOExternalDependency fromLocal(
      File localDependency,
      @Nullable File localSourceDependency,
      ExternalDependenciesExtension externalDependenciesExtension,
      JetifierExtension jetifierExtension) {

    String name = FilenameUtils.getBaseName(localDependency.getName());
    return (LocalOExternalDependency)
        from(
            LOCAL_GROUP,
            name,
            LOCAL_DEP_VERSION,
            localDependency,
            localSourceDependency,
            externalDependenciesExtension,
            jetifierExtension);
  }

  public synchronized Set<VersionlessDependency> fromDependency(ExternalDependency dependency) {
    if (unresolvedToVersionless.containsKey(dependency)) {
      return unresolvedToVersionless.get(dependency);
    } else {
      VersionlessDependency.Builder vDependencyBuilder =
          VersionlessDependency.builder().setName(dependency.getName());
      String group = dependency.getGroup();

      if (group == null) {
        vDependencyBuilder.setGroup(LOCAL_GROUP);
      } else {
        vDependencyBuilder.setGroup(group);
      }

      Set<VersionlessDependency> vDeps = new HashSet<>();

      if (dependency.getArtifacts().size() > 0) {
        vDeps.addAll(
            dependency
                .getArtifacts()
                .stream()
                .map(
                    dependencyArtifact ->
                        vDependencyBuilder
                            .setClassifier(Optional.ofNullable(dependencyArtifact.getClassifier()))
                            .build())
                .collect(Collectors.toSet()));
      } else {
        vDeps.add(vDependencyBuilder.build());
      }

      unresolvedToVersionless.put(dependency, vDeps);
      return vDeps;
    }
  }

  /**
   * Returns a set of versionless dependency from the given gradle resolved dependency.
   *
   * @param dependency gradle dependency
   * @return VersionlessDependency object
   */
  public static Set<VersionlessDependency> fromDependency(ResolvedDependency dependency) {
    Set<ResolvedArtifact> selfArtifacts = dependency.getModuleArtifacts();
    return fromResolvedArtifacts(selfArtifacts);
  }

  /**
   * Returns a set of versionless dependency from the given gradle resolved dependency.
   *
   * @param dependency gradle dependency
   * @return VersionlessDependency object
   */
  public static Set<VersionlessDependency> childrenFromDependency(ResolvedDependency dependency) {
    Set<ResolvedArtifact> childArtifacts =
        dependency
            .getChildren()
            .stream()
            .map(child -> child.getParentArtifacts(dependency))
            .flatMap(Collection::stream)
            .collect(Collectors.toSet());

    return fromResolvedArtifacts(childArtifacts);
  }

  private static Set<VersionlessDependency> fromResolvedArtifacts(Set<ResolvedArtifact> artifacts) {
    return artifacts
        .stream()
        .map(
            resolvedArtifact -> {
              if (resolvedArtifact.getId().getComponentIdentifier()
                  instanceof ModuleComponentIdentifier) {
                ModuleComponentIdentifier id =
                    (ModuleComponentIdentifier) resolvedArtifact.getId().getComponentIdentifier();
                return VersionlessDependency.builder()
                    .setName(id.getModule())
                    .setGroup(id.getGroup())
                    .setClassifier(Optional.ofNullable(resolvedArtifact.getClassifier()))
                    .build();
              }
              return null;
            })
        .filter(Objects::nonNull)
        .collect(Collectors.toSet());
  }

  public void finalizeDependencies() {
    unresolvedToVersionless.clear();
    externalDependencyCache.clear();
  }

  /**
   * Returns whether the dependency should be marked local or not. These dependencies can't be
   * downloaded via buck. Snapshot dependencies with version suffixed with `-SNAPSHOT`. Local m2
   * dependencies with version suffixed with `-LOCAL`.
   *
   * <p>Note: we need to pass in the whole dependency file instead of just the version. `-SNAPSHOT`
   * dependencies with specific date and time in version is also not supported by buck and hence to
   * access correctly we need to look at the whole path. eg:
   * com/jakewharton/butterknife/9.0.0-SNAPSHOT/butterknife-9.0.0-20181220.030319-77.jar
   *
   * @param dependencyFilePath Absolute path string of the dependency file
   * @return Whether the dependency should be local or not.
   */
  private static boolean isLocalDependency(String dependencyFilePath) {
    return dependencyFilePath.contains("-SNAPSHOT") || dependencyFilePath.contains("-LOCAL");
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -