// 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 com.google.devtools.build.lib.runtime;

import com.google.common.base.MoreObjects;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableList;
import com.google.devtools.build.lib.bugreport.Crash;
import com.google.devtools.build.lib.concurrent.ThreadSafety.Immutable;
import com.google.devtools.build.lib.server.CommandProtos.ExecRequest;
import com.google.devtools.build.lib.server.FailureDetails.FailureDetail;
import com.google.devtools.build.lib.server.IdleTask;
import com.google.devtools.build.lib.util.DetailedExitCode;
import com.google.devtools.build.lib.util.ExitCode;
import com.google.protobuf.Any;
import javax.annotation.Nullable;

/**
 * The result of a Blaze command. It is usually a {@link ExitCode} with optional {@link
 * FailureDetail}, but can be an instruction to the client to execute a particular binary for "blaze
 * run".
 */
@Immutable
public final class BlazeCommandResult {
  private final DetailedExitCode detailedExitCode;

  @Nullable private final ExecRequest execDescription;
  private final ImmutableList<Any> responseExtensions;
  private final boolean shutdown;
  private final boolean stateKeptAfterBuild;
  private final ImmutableList<IdleTask> idleTasks;

  private BlazeCommandResult(
      DetailedExitCode detailedExitCode,
      @Nullable ExecRequest execDescription,
      boolean shutdown,
      ImmutableList<Any> responseExtensions,
      boolean stateKeptAfterBuild,
      ImmutableList<IdleTask> idleTasks) {
    this.detailedExitCode = Preconditions.checkNotNull(detailedExitCode);
    this.execDescription = execDescription;
    this.shutdown = shutdown;
    this.responseExtensions = responseExtensions;
    this.stateKeptAfterBuild = stateKeptAfterBuild;
    this.idleTasks = idleTasks;
  }

  private BlazeCommandResult(
      DetailedExitCode detailedExitCode, @Nullable ExecRequest execDescription, boolean shutdown) {
    this(detailedExitCode, execDescription, shutdown, ImmutableList.of(), true, ImmutableList.of());
  }

  public ExitCode getExitCode() {
    return detailedExitCode.getExitCode();
  }

  public DetailedExitCode getDetailedExitCode() {
    return detailedExitCode;
  }

  @Nullable
  public FailureDetail getFailureDetail() {
    return detailedExitCode.getFailureDetail();
  }

  public boolean shutdown() {
    return shutdown;
  }

  @Nullable
  public ExecRequest getExecRequest() {
    return execDescription;
  }

  public boolean isSuccess() {
    return detailedExitCode.isSuccess();
  }

  public ImmutableList<Any> getResponseExtensions() {
    return responseExtensions;
  }

  public ImmutableList<IdleTask> getIdleTasks() {
    return idleTasks;
  }

  /** Reflects the value of {@link CommonCommandOptions#keepStateAfterBuild}. */
  public boolean stateKeptAfterBuild() {
    return stateKeptAfterBuild;
  }

  @Override
  public String toString() {
    return MoreObjects.toStringHelper(this)
        .add("exitCode", getExitCode())
        .add("failureDetail", getFailureDetail())
        .add("execDescription", execDescription)
        .add("shutdown", shutdown)
        .add("responseExtensions", responseExtensions)
        .add("stateKeptAfterBuild", stateKeptAfterBuild)
        .toString();
  }

  public static BlazeCommandResult shutdownOnSuccess() {
    return new BlazeCommandResult(DetailedExitCode.success(), null, true);
  }

  static BlazeCommandResult createShutdown(Crash crash) {
    return new BlazeCommandResult(crash.getDetailedExitCode(), null, true);
  }

  public static BlazeCommandResult success() {
    return new BlazeCommandResult(DetailedExitCode.success(), null, false);
  }

  public static BlazeCommandResult failureDetail(FailureDetail failureDetail) {
    return new BlazeCommandResult(DetailedExitCode.of(failureDetail), null, false);
  }

  public static BlazeCommandResult detailedExitCode(DetailedExitCode detailedExitCode) {
    return new BlazeCommandResult(detailedExitCode, null, false);
  }

  public static BlazeCommandResult withResponseExtensions(
      BlazeCommandResult result,
      ImmutableList<Any> responseExtensions,
      boolean stateKeptAfterBuild) {
    return new BlazeCommandResult(
        result.detailedExitCode,
        result.execDescription,
        result.shutdown,
        responseExtensions,
        stateKeptAfterBuild,
        result.idleTasks);
  }

  public static BlazeCommandResult withIdleTasks(
      BlazeCommandResult result, ImmutableList<IdleTask> idleTasks) {
    return new BlazeCommandResult(
        result.detailedExitCode,
        result.execDescription,
        result.shutdown,
        result.responseExtensions,
        result.stateKeptAfterBuild,
        idleTasks);
  }

  public static BlazeCommandResult execute(ExecRequest execDescription) {
    return new BlazeCommandResult(
        DetailedExitCode.success(), Preconditions.checkNotNull(execDescription), false);
  }
}
