Propogate BAZEL_VERBOSE_FAILURES and BAZEL_SUBCOMMANDS to the execution environments of runtime tools.

--
MOS_MIGRATED_REVID=112149571

Author: calpeyser

src/main/java/com/google/devtools/build/lib/analysis/actions/SpawnAction.java

@@ -83,6 +83,9 @@ void extend(ExtraActionInfo.Builder builder) {
 
   private static final String GUID = "ebd6fce3-093e-45ee-adb6-bf513b602f0d";
 
+  private static final String VERBOSE_FAILURES_KEY = "BAZEL_VERBOSE_FAILURES";
+  private static final String SUBCOMMANDS_KEY = "BAZEL_SUBCOMMANDS";
+
   private final CommandLine argv;
 
   private final boolean executeUnconditionally;
@@ -92,11 +95,13 @@ void extend(ExtraActionInfo.Builder builder) {
   private final ImmutableMap<PathFragment, Artifact> inputManifests;
 
   private final ResourceSet resourceSet;
-  private final ImmutableMap<String, String> environment;
+  private ImmutableMap<String, String> environment;
   private final ImmutableMap<String, String> executionInfo;
 
   private final ExtraActionInfoSupplier<?> extraActionInfoSupplier;
 
+  private final boolean verboseFailuresAndSubcommandsInEnv;
+
   /**
    * Constructs a SpawnAction using direct initialization arguments.
    * <p>
@@ -140,7 +145,8 @@ public SpawnAction(
         ImmutableMap.<PathFragment, Artifact>of(),
         mnemonic,
         false,
-        null);
+        null,
+        false);
   }
 
   /**
@@ -165,6 +171,9 @@ public SpawnAction(
    * @param inputManifests entries in inputs that are symlink manifest files.
    *        These are passed to remote execution in the environment rather than as inputs.
    * @param mnemonic the mnemonic that is reported in the master log.
+   * @param verboseFailuresAndSubcommandsInEnv if the presense of "--verbose_failures" and/or
+   *        "--subcommands" in the execution should be propogated to the environment of the
+   *        action.
    */
   public SpawnAction(
       ActionOwner owner,
@@ -179,7 +188,8 @@ public SpawnAction(
       ImmutableMap<PathFragment, Artifact> inputManifests,
       String mnemonic,
       boolean executeUnconditionally,
-      ExtraActionInfoSupplier<?> extraActionInfoSupplier) {
+      ExtraActionInfoSupplier<?> extraActionInfoSupplier,
+      boolean verboseFailuresAndSubcommandsInEnv) {
     super(owner, tools, inputs, outputs);
     this.resourceSet = resourceSet;
     this.executionInfo = executionInfo;
@@ -190,6 +200,7 @@ public SpawnAction(
     this.mnemonic = mnemonic;
     this.executeUnconditionally = executeUnconditionally;
     this.extraActionInfoSupplier = extraActionInfoSupplier;
+    this.verboseFailuresAndSubcommandsInEnv = verboseFailuresAndSubcommandsInEnv;
   }
 
   /**
@@ -246,6 +257,22 @@ protected void internalExecute(
   public void execute(ActionExecutionContext actionExecutionContext)
       throws ActionExecutionException, InterruptedException {
     Executor executor = actionExecutionContext.getExecutor();
+
+    // Reconstruct environment to communicate if verbose_failures and/or subcommands is set.
+    if (verboseFailuresAndSubcommandsInEnv) {
+      ImmutableMap.Builder<String, String> environmentBuilder =
+          new ImmutableMap.Builder<String, String>().putAll(environment);
+
+      if (executor.getVerboseFailures()) {
+        environmentBuilder.put(VERBOSE_FAILURES_KEY, "true");
+      }
+      if (executor.reportsSubcommands()) {
+        environmentBuilder.put(SUBCOMMANDS_KEY, "true");
+      }
+
+      this.environment = environmentBuilder.build();
+    }
+
     try {
       internalExecute(actionExecutionContext);
     } catch (ExecException e) {
@@ -356,6 +383,11 @@ public ExtraActionInfo.Builder getExtraActionInfo() {
 
   /**
    * Returns the environment in which to run this action.
+   *
+   * <p>Note that if setVerboseFailuresAndSubcommandsInEnv() is called on the builder, and either
+   * verbose_failures or subcommands is specified in the execution context, corresponding variables
+   * will be added to the environment.  These variables will not be known until execution time,
+   * however, and so are not returned by getEnvironment().
    */
   public Map<String, String> getEnvironment() {
     return environment;
@@ -464,6 +496,8 @@ public static class Builder {
     private String mnemonic = "Unknown";
     private ExtraActionInfoSupplier<?> extraActionInfoSupplier = null;
 
+    private boolean verboseFailuresAndSubcommandsInEnv = false;
+
     /**
      * Creates a SpawnAction builder.
      */
@@ -492,6 +526,7 @@ public Builder(Builder other) {
       this.progressMessage = other.progressMessage;
       this.paramFileInfo = other.paramFileInfo;
       this.mnemonic = other.mnemonic;
+      this.verboseFailuresAndSubcommandsInEnv = other.verboseFailuresAndSubcommandsInEnv;
     }
 
     /**
@@ -578,7 +613,8 @@ public Action[] build(ActionOwner owner, AnalysisEnvironment analysisEnvironment
               ImmutableMap.copyOf(inputAndToolManifests),
               mnemonic,
               executeUnconditionally,
-              extraActionInfoSupplier));
+              extraActionInfoSupplier,
+              verboseFailuresAndSubcommandsInEnv));
       return actions.toArray(new Action[actions.size()]);
     }
 
@@ -677,6 +713,17 @@ public Builder useDefaultShellEnvironment() {
       return this;
     }
 
+    /**
+     * Sets the environment variable "BAZEL_VERBOSE_FAILURES" to "true" if --verbose_failures is
+     * set in the execution context. Sets the environment variable "BAZEL_SUBCOMMANDS" to "true"
+     * if --subcommands is set in the execution context.
+     *
+     */
+    public Builder setVerboseFailuresAndSubcommandsInEnv() {
+      this.verboseFailuresAndSubcommandsInEnv = true;
+      return this;
+    }
+
     /**
      * Makes the action always execute, even if none of its inputs have changed.
      *

src/main/java/com/google/devtools/build/lib/bazel/rules/genrule/GenRuleAction.java

@@ -60,7 +60,8 @@ public GenRuleAction(
         runfilesManifests,
         "Genrule",
         false,
-        null);
+        null,
+        false);
   }
 
   @Override

src/main/java/com/google/devtools/build/lib/rules/extra/ExtraAction.java

@@ -99,7 +99,8 @@ public ExtraAction(
         getManifests(shadowedAction),
         mnemonic,
         false,
-        null);
+        null,
+        false);
     this.shadowedAction = shadowedAction;
     this.runfilesManifests = ImmutableMap.copyOf(runfilesManifests);
     this.createDummyOutput = createDummyOutput;

src/main/java/com/google/devtools/build/lib/rules/objc/BundleSupport.java

@@ -242,6 +242,7 @@ private void registerInterfaceBuilderActions(ObjcProvider objcProvider) {
               // https://github.com/bazelbuild/bazel/issues/285 is fixed.
               .addInput(attributes.realpath())
               .addInput(CompilationSupport.xcrunwrapper(ruleContext).getExecutable())
+              .setVerboseFailuresAndSubcommandsInEnv()
               .build(ruleContext));
     }
   }
@@ -285,6 +286,7 @@ private void registerMomczipActions(ObjcProvider objcProvider) {
               // https://github.com/google/bazel/issues/285 is fixed.
               .addInput(attributes.realpath())
               .addInput(CompilationSupport.xcrunwrapper(ruleContext).getExecutable())
+              .setVerboseFailuresAndSubcommandsInEnv()
              .setCommandLine(CustomCommandLine.builder()
                   .addPath(outputZip.getExecPath())
                   .add(datamodel.archiveRootForMomczip())
@@ -318,6 +320,7 @@ private void registerConvertXibsActions(ObjcProvider objcProvider) {
               // https://github.com/bazelbuild/bazel/issues/285 is fixed.
               .addInput(attributes.realpath())
               .addInput(CompilationSupport.xcrunwrapper(ruleContext).getExecutable())
+              .setVerboseFailuresAndSubcommandsInEnv()
               .build(ruleContext));
     }
   }
@@ -372,6 +375,7 @@ private void registerMergeInfoplistAction(
             .addInputArgument(plMergeControlArtifact)
             .addTransitiveInputs(mergingContentArtifacts)
             .addOutput(ObjcRuleClasses.intermediateArtifacts(ruleContext).mergedInfoplist())
+            .setVerboseFailuresAndSubcommandsInEnv()
             .build(ruleContext));
   }
 
@@ -400,6 +404,7 @@ private void registerActoolActionIfNecessary(ObjcProvider objcProvider) {
             // https://github.com/google/bazel/issues/285 is fixed.
             .addInput(attributes.realpath())
             .addInput(CompilationSupport.xcrunwrapper(ruleContext).getExecutable())
+            .setVerboseFailuresAndSubcommandsInEnv()
             .setCommandLine(actoolzipCommandLine(
                 objcProvider,
                 zipOutput,

src/main/java/com/google/devtools/build/lib/rules/objc/ReleaseBundlingSupport.java

@@ -747,6 +747,7 @@ private void registerBundleMergeActions(Artifact ipaUnsigned,
         .addInputArgument(bundleMergeControlArtifact)
         .addTransitiveInputs(bundleContentArtifacts)
         .addOutput(ipaUnsigned)
+        .setVerboseFailuresAndSubcommandsInEnv()
         .build(ruleContext));
   }