JavaLibraryBuilder.java

/*
 * Copyright 2014-present Facebook, Inc.
 *
 * 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.facebook.buck.jvm.java;

import static com.facebook.buck.jvm.java.JavaCompilationConstants.DEFAULT_JAVAC_OPTIONS;
import static com.facebook.buck.jvm.java.JavaCompilationConstants.DEFAULT_JAVA_CONFIG;

import com.facebook.buck.io.ProjectFilesystem;
import com.facebook.buck.model.BuildTarget;
import com.facebook.buck.model.BuildTargetFactory;
import com.facebook.buck.model.Either;
import com.facebook.buck.rules.AbstractNodeBuilderWithImmutableArg;
import com.facebook.buck.rules.DefaultBuildTargetSourcePath;
import com.facebook.buck.rules.PathSourcePath;
import com.facebook.buck.rules.SourcePath;
import com.facebook.buck.testutil.FakeProjectFilesystem;
import com.google.common.collect.ImmutableSet;
import com.google.common.hash.HashCode;
import java.nio.file.Path;
import java.util.Optional;

public class JavaLibraryBuilder
    extends AbstractNodeBuilderWithImmutableArg<
        JavaLibraryDescriptionArg.Builder, JavaLibraryDescriptionArg, JavaLibraryDescription,
        DefaultJavaLibrary> {

  private final ProjectFilesystem projectFilesystem;

  protected JavaLibraryBuilder(
      BuildTarget target, ProjectFilesystem projectFilesystem, HashCode hashCode) {
    super(
        new JavaLibraryDescription(DEFAULT_JAVA_CONFIG, DEFAULT_JAVAC_OPTIONS),
        target,
        projectFilesystem,
        hashCode);
    this.projectFilesystem = projectFilesystem;
  }

  protected JavaLibraryBuilder(
      BuildTarget target,
      JavaBuckConfig javaBuckConfig,
      ProjectFilesystem projectFilesystem,
      HashCode hashCode) {
    super(
        new JavaLibraryDescription(javaBuckConfig, DEFAULT_JAVAC_OPTIONS),
        target,
        projectFilesystem,
        hashCode);
    this.projectFilesystem = projectFilesystem;
  }

  public static JavaLibraryBuilder createBuilder(String qualifiedTarget) {
    return createBuilder(BuildTargetFactory.newInstance(qualifiedTarget));
  }

  public static JavaLibraryBuilder createBuilder(BuildTarget target) {
    return new JavaLibraryBuilder(target, new FakeProjectFilesystem(), null);
  }

  public static JavaLibraryBuilder createBuilder(
      BuildTarget target, JavaBuckConfig javaBuckConfig) {
    return new JavaLibraryBuilder(target, javaBuckConfig, new FakeProjectFilesystem(), null);
  }

  public static JavaLibraryBuilder createBuilder(
      BuildTarget target, ProjectFilesystem projectFilesystem) {
    return new JavaLibraryBuilder(target, projectFilesystem, null);
  }

  public static JavaLibraryBuilder createBuilder(
      BuildTarget target, JavaBuckConfig javaBuckConfig, ProjectFilesystem projectFilesystem) {
    return new JavaLibraryBuilder(target, javaBuckConfig, projectFilesystem, null);
  }

  public static JavaLibraryBuilder createBuilder(BuildTarget target, HashCode hashCode) {
    return new JavaLibraryBuilder(target, new FakeProjectFilesystem(), hashCode);
  }

  public JavaLibraryBuilder addDep(BuildTarget rule) {
    getArgForPopulating().addDeps(rule);
    return this;
  }

  public JavaLibraryBuilder addAnnotationProcessorDep(BuildTarget rule) {
    getArgForPopulating().addAnnotationProcessorDeps(rule);
    return this;
  }

  public JavaLibraryBuilder addExportedDep(BuildTarget rule) {
    getArgForPopulating().addExportedDeps(rule);
    return this;
  }

  public JavaLibraryBuilder addProvidedDep(BuildTarget rule) {
    getArgForPopulating().addProvidedDeps(rule);
    return this;
  }

  public JavaLibraryBuilder addResource(SourcePath sourcePath) {
    getArgForPopulating().addResources(sourcePath);
    return this;
  }

  public JavaLibraryBuilder setResourcesRoot(Path root) {
    getArgForPopulating().setResourcesRoot(Optional.of(root));
    return this;
  }

  public JavaLibraryBuilder setMavenCoords(String mavenCoords) {
    getArgForPopulating().setMavenCoords(Optional.of(mavenCoords));
    return this;
  }

  public JavaLibraryBuilder addSrc(SourcePath path) {
    getArgForPopulating().addSrcs(path);
    return this;
  }

  public JavaLibraryBuilder addSrc(Path path) {
    return addSrc(new PathSourcePath(projectFilesystem, path));
  }

  public JavaLibraryBuilder addSrcTarget(BuildTarget target) {
    return addSrc(new DefaultBuildTargetSourcePath(target));
  }

  public JavaLibraryBuilder setProguardConfig(SourcePath proguardConfig) {
    getArgForPopulating().setProguardConfig(Optional.of(proguardConfig));
    return this;
  }

  public JavaLibraryBuilder setCompiler(SourcePath javac) {
    Either<BuiltInJavac, SourcePath> value = Either.ofRight(javac);
    getArgForPopulating().setCompiler(Optional.of(value));
    return this;
  }

  public JavaLibraryBuilder setSourceLevel(String sourceLevel) {
    getArgForPopulating().setSource(Optional.of(sourceLevel));
    return this;
  }

  public JavaLibraryBuilder setTargetLevel(String targetLevel) {
    getArgForPopulating().setTarget(Optional.of(targetLevel));
    return this;
  }

  public JavaLibraryBuilder setAnnotationProcessors(ImmutableSet<String> annotationProcessors) {
    getArgForPopulating().setAnnotationProcessors(annotationProcessors);
    return this;
  }

  public JavaLibraryBuilder addTest(BuildTarget test) {
    getArgForPopulating().addTests(test);
    return this;
  }
}