base/src/com/google/idea/blaze/base/command/buildresult/BuildResultHelperBep.java

/*
 * Copyright 2017 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.idea.blaze.base.command.buildresult;

import com.google.common.collect.ImmutableList;
import com.google.idea.blaze.base.async.process.LineProcessingOutputStream;
import com.google.idea.blaze.base.async.process.LineProcessingOutputStream.LineProcessor;
import com.google.idea.blaze.base.model.primitives.Label;
import com.intellij.openapi.diagnostic.Logger;
import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.List;
import java.util.Optional;
import java.util.function.Predicate;

/**
 * Build event protocol implementation to get build results.
 *
 * <p>The build even protocol (BEP for short) is a proto-based protocol used by bazel to communicate
 * build events.
 */
class BuildResultHelperBep implements BuildResultHelper {

  private static final Logger logger = Logger.getInstance(BuildResultHelperBep.class);
  private final File outputFile;
  private final Predicate<String> fileFilter;
  private ImmutableList<File> result;

  BuildResultHelperBep(Predicate<String> fileFilter) {
    this.fileFilter = fileFilter;
    outputFile = BuildEventProtocolUtils.createTempOutputFile();
  }

  @Override
  public List<String> getBuildFlags() {
    return BuildEventProtocolUtils.getBuildFlags(outputFile);
  }

  @Override
  public OutputStream stderr(LineProcessor... lineProcessors) {
    return LineProcessingOutputStream.of(ImmutableList.copyOf(lineProcessors));
  }

  @Override
  public ImmutableList<File> getBuildArtifacts() {
    if (result == null) {
      result =
          readResult(
                  input ->
                      BuildEventProtocolOutputReader.parseAllOutputFilenames(input, fileFilter))
              .orElse(ImmutableList.of());
    }
    return result;
  }

  @Override
  public ImmutableList<File> getBuildArtifactsForTarget(Label target) {
    if (result == null) {
      result =
          readResult(
                  input ->
                      BuildEventProtocolOutputReader.parseArtifactsForTarget(
                          input, target, fileFilter))
              .orElse(ImmutableList.of());
    }
    return result;
  }

  private <V> Optional<V> readResult(BepReader<V> readAction) {
    try (InputStream inputStream = new BufferedInputStream(new FileInputStream(outputFile))) {
      return Optional.of(readAction.read(inputStream));
    } catch (IOException e) {
      logger.error(e);
      return Optional.empty();
    } finally {
      close();
    }
  }

  @Override
  public void close() {
    if (!outputFile.delete()) {
      logger.warn("Could not delete BEP output file: " + outputFile);
    }
  }

  private interface BepReader<V> {
    V read(InputStream inputStream) throws IOException;
  }
}