/*
 * Copyright 2018 The Bazel Authors. All rights reserved.
 *
 * 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 io.bazel.kotlin.builder;

import com.google.auto.value.AutoValue;
import com.google.common.collect.ImmutableSet;
import io.bazel.kotlin.builder.utils.BazelRunFiles;

import java.util.Optional;
import java.util.Set;
import java.util.stream.Stream;

public final class Deps {
  @SuppressWarnings({"WeakerAccess", "unused"})
  @AutoValue
  public abstract static class Dep {
    public abstract String label();

    public abstract String moduleName();

    public abstract Set<String> runtimeDeps();

    public abstract Set<String> compileJars();

    public static Builder builder() {
      return new AutoValue_Deps_Dep.Builder().runtimeDeps(ImmutableSet.of());
    }

    @SuppressWarnings("UnusedReturnValue")
    @AutoValue.Builder
    public abstract static class Builder {
      public abstract Builder compileJars(Set<String> compileJars);

      public abstract Builder label(String label);

      public abstract Builder runtimeDeps(Set<String> runtimeDeps);

      public abstract Builder moduleName(String moduleName);

      abstract String label();

      abstract Optional<String> moduleName();

      abstract Dep autoBuild();

      public Dep build() {
        if (!moduleName().isPresent()) {
          moduleName(label());
        }
        return autoBuild();
      }
    }

    /** Collect all of the compile jars of all the dependencies. */
    public static Stream<String> classpathOf(Dep... dependencies) {
      return Stream.of(dependencies).flatMap(it -> it.compileJars().stream());
    }

    /** Import a single dep. Similar to a `kt_jvm_import` or a `kt_js_import`. */
    public static Dep importJar(String label, String compileJar) {
      return Dep.builder()
          .label(label)
          .compileJars(ImmutableSet.of(BazelRunFiles.resolveVerified(compileJar).getAbsolutePath()))
          .build();
    }
  }

  @AutoValue
  public abstract static class AnnotationProcessor {
    abstract String processClass();

    abstract Set<String> processorPath();

    public static Builder builder() {
      return new AutoValue_Deps_AnnotationProcessor.Builder();
    }

    @AutoValue.Builder
    public abstract static class Builder {
      public abstract Builder processClass(String processClass);

      public abstract Builder processorPath(Set<String> processorPath);

      public abstract AnnotationProcessor build();
    }
  }
}