blob: bd90a48dc0071f428d9232e6635fc1379300fc93 [file] [log] [blame]
// Copyright 2023 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.bazel.commands;
import static com.google.common.collect.ImmutableList.toImmutableList;
import static java.nio.charset.StandardCharsets.UTF_8;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import com.google.devtools.build.lib.analysis.NoBuildEvent;
import com.google.devtools.build.lib.analysis.NoBuildRequestFinishedEvent;
import com.google.devtools.build.lib.bazel.bzlmod.BazelFetchAllValue;
import com.google.devtools.build.lib.bazel.commands.RepositoryFetcher.RepositoryFetcherException;
import com.google.devtools.build.lib.bazel.repository.RepositoryOptions;
import com.google.devtools.build.lib.cmdline.LabelConstants;
import com.google.devtools.build.lib.cmdline.RepositoryName;
import com.google.devtools.build.lib.events.Event;
import com.google.devtools.build.lib.events.Reporter;
import com.google.devtools.build.lib.packages.semantics.BuildLanguageOptions;
import com.google.devtools.build.lib.pkgcache.PackageOptions;
import com.google.devtools.build.lib.rules.repository.RepositoryDelegatorFunction;
import com.google.devtools.build.lib.rules.repository.RepositoryDirectoryValue;
import com.google.devtools.build.lib.runtime.BlazeCommand;
import com.google.devtools.build.lib.runtime.BlazeCommandResult;
import com.google.devtools.build.lib.runtime.Command;
import com.google.devtools.build.lib.runtime.CommandEnvironment;
import com.google.devtools.build.lib.runtime.KeepGoingOption;
import com.google.devtools.build.lib.runtime.LoadingPhaseThreadsOption;
import com.google.devtools.build.lib.server.FailureDetails;
import com.google.devtools.build.lib.server.FailureDetails.FailureDetail;
import com.google.devtools.build.lib.server.FailureDetails.FetchCommand.Code;
import com.google.devtools.build.lib.skyframe.PrecomputedValue;
import com.google.devtools.build.lib.skyframe.RepositoryMappingValue.RepositoryMappingResolutionException;
import com.google.devtools.build.lib.util.AbruptExitException;
import com.google.devtools.build.lib.util.DetailedExitCode;
import com.google.devtools.build.lib.util.InterruptedFailureDetails;
import com.google.devtools.build.lib.vfs.FileSystemUtils;
import com.google.devtools.build.lib.vfs.Path;
import com.google.devtools.build.lib.vfs.PathFragment;
import com.google.devtools.build.lib.vfs.Symlinks;
import com.google.devtools.build.skyframe.EvaluationContext;
import com.google.devtools.build.skyframe.EvaluationResult;
import com.google.devtools.build.skyframe.SkyKey;
import com.google.devtools.build.skyframe.SkyValue;
import com.google.devtools.common.options.OptionsParsingResult;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map.Entry;
import java.util.Objects;
import javax.annotation.Nullable;
/** Fetches external repositories into a specified directory. */
@Command(
name = VendorCommand.NAME,
options = {
VendorOptions.class,
PackageOptions.class,
KeepGoingOption.class,
LoadingPhaseThreadsOption.class
},
help = "resource:vendor.txt",
shortDescription =
"Fetches external repositories into a specific folder specified by the flag "
+ "--vendor_dir.")
public final class VendorCommand implements BlazeCommand {
public static final String NAME = "vendor";
// TODO(salmasamy) decide on name and format
private static final String VENDOR_IGNORE = ".vendorignore";
@Override
public BlazeCommandResult exec(CommandEnvironment env, OptionsParsingResult options) {
BlazeCommandResult invalidResult = validateOptions(env, options);
if (invalidResult != null) {
return invalidResult;
}
env.getEventBus()
.post(
new NoBuildEvent(
env.getCommandName(),
env.getCommandStartTime(),
/* separateFinishedEvent= */ true,
/* showProgress= */ true,
env.getCommandId().toString()));
// IS_VENDOR_COMMAND & VENDOR_DIR is already injected in "BazelRepositoryModule", we just need
// to update this value for the delegator function to recognize this call is from VendorCommand
env.getSkyframeExecutor()
.injectExtraPrecomputedValues(
ImmutableList.of(
PrecomputedValue.injected(RepositoryDelegatorFunction.IS_VENDOR_COMMAND, true)));
BlazeCommandResult result;
VendorOptions vendorOptions = options.getOptions(VendorOptions.class);
PathFragment vendorDirectory = options.getOptions(RepositoryOptions.class).vendorDirectory;
LoadingPhaseThreadsOption threadsOption = options.getOptions(LoadingPhaseThreadsOption.class);
try {
env.syncPackageLoading(options);
if (!vendorOptions.repos.isEmpty()) {
result = vendorRepos(env, threadsOption, vendorOptions.repos, vendorDirectory);
} else {
result = vendorAll(env, threadsOption, vendorDirectory);
}
} catch (AbruptExitException e) {
return createFailedBlazeCommandResult(
env.getReporter(), e.getMessage(), e.getDetailedExitCode());
} catch (InterruptedException e) {
return createFailedBlazeCommandResult(
env.getReporter(), "Vendor interrupted: " + e.getMessage());
} catch (IOException e) {
return createFailedBlazeCommandResult(
env.getReporter(), "Error while vendoring repos: " + e.getMessage());
}
env.getEventBus()
.post(
new NoBuildRequestFinishedEvent(
result.getExitCode(), env.getRuntime().getClock().currentTimeMillis()));
return result;
}
@Nullable
private BlazeCommandResult validateOptions(CommandEnvironment env, OptionsParsingResult options) {
if (!options.getOptions(BuildLanguageOptions.class).enableBzlmod) {
return createFailedBlazeCommandResult(
env.getReporter(),
"Bzlmod has to be enabled for vendoring to work, run with --enable_bzlmod");
}
if (options.getOptions(RepositoryOptions.class).vendorDirectory == null) {
return createFailedBlazeCommandResult(
env.getReporter(),
Code.OPTIONS_INVALID,
"You cannot run vendor without specifying --vendor_dir");
}
if (!options.getOptions(PackageOptions.class).fetch) {
return createFailedBlazeCommandResult(
env.getReporter(), Code.OPTIONS_INVALID, "You cannot run vendor with --nofetch");
}
return null;
}
private BlazeCommandResult vendorAll(
CommandEnvironment env, LoadingPhaseThreadsOption threadsOption, PathFragment vendorDirectory)
throws InterruptedException, IOException {
EvaluationContext evaluationContext =
EvaluationContext.newBuilder()
.setParallelism(threadsOption.threads)
.setEventHandler(env.getReporter())
.build();
SkyKey fetchKey = BazelFetchAllValue.key(/* configureEnabled= */ false);
EvaluationResult<SkyValue> evaluationResult =
env.getSkyframeExecutor().prepareAndGet(ImmutableSet.of(fetchKey), evaluationContext);
if (evaluationResult.hasError()) {
Exception e = evaluationResult.getError().getException();
return createFailedBlazeCommandResult(
env.getReporter(),
e != null ? e.getMessage() : "Unexpected error during fetching all external deps.");
}
BazelFetchAllValue fetchAllValue = (BazelFetchAllValue) evaluationResult.get(fetchKey);
vendor(env, vendorDirectory, fetchAllValue.getReposToVendor());
return BlazeCommandResult.success();
}
private BlazeCommandResult vendorRepos(
CommandEnvironment env,
LoadingPhaseThreadsOption threadsOption,
List<String> repos,
PathFragment vendorDirectory)
throws InterruptedException, IOException {
ImmutableMap<RepositoryName, RepositoryDirectoryValue> repositoryNamesAndValues;
try {
repositoryNamesAndValues = RepositoryFetcher.fetchRepos(repos, env, threadsOption);
} catch (RepositoryMappingResolutionException e) {
return createFailedBlazeCommandResult(
env.getReporter(), "Invalid repo name: " + e.getMessage(), e.getDetailedExitCode());
} catch (RepositoryFetcherException e) {
return createFailedBlazeCommandResult(env.getReporter(), e.getMessage());
}
// Split repos to found and not found, vendor found ones and report others
ImmutableList.Builder<RepositoryName> reposToVendor = ImmutableList.builder();
List<String> notFoundRepoErrors = new ArrayList<>();
for (Entry<RepositoryName, RepositoryDirectoryValue> entry :
repositoryNamesAndValues.entrySet()) {
if (entry.getValue().repositoryExists()) {
if (!entry.getValue().excludeFromVendoring()) {
reposToVendor.add(entry.getKey());
}
} else {
notFoundRepoErrors.add(entry.getValue().getErrorMsg());
}
}
vendor(env, vendorDirectory, reposToVendor.build());
if (!notFoundRepoErrors.isEmpty()) {
return createFailedBlazeCommandResult(
env.getReporter(), "Vendoring some repos failed with errors: " + notFoundRepoErrors);
}
return BlazeCommandResult.success();
}
/**
* Copies the fetched repos from the external cache into the vendor directory, unless the repo is
* ignored or was already vendored and up-to-date
*/
private void vendor(
CommandEnvironment env,
PathFragment vendorDirectory,
ImmutableList<RepositoryName> reposToVendor)
throws IOException {
Path vendorPath =
vendorDirectory.isAbsolute()
? env.getRuntime().getFileSystem().getPath(vendorDirectory)
: env.getWorkspace().getRelative(vendorDirectory);
Path externalPath =
env.getDirectories()
.getOutputBase()
.getRelative(LabelConstants.EXTERNAL_REPOSITORY_LOCATION);
Path vendorIgnore = vendorPath.getRelative(VENDOR_IGNORE);
if (!vendorPath.exists()) {
vendorPath.createDirectory();
}
// exclude any ignored repo under .vendorignore
if (vendorIgnore.exists()) {
ImmutableSet<String> ignoredRepos =
ImmutableSet.copyOf(FileSystemUtils.readLines(vendorIgnore, UTF_8));
reposToVendor =
reposToVendor.stream()
.filter(repo -> !ignoredRepos.contains(repo.getName()))
.collect(toImmutableList());
} else {
FileSystemUtils.createEmptyFile(vendorIgnore);
}
// Update "out-of-date" repos under the vendor directory
for (RepositoryName repo : reposToVendor) {
if (!isRepoUpToDate(repo.getName(), vendorPath, externalPath)) {
Path repoUnderVendor = vendorPath.getRelative(repo.getName());
if (!repoUnderVendor.exists()) {
repoUnderVendor.createDirectory();
}
FileSystemUtils.copyTreesBelow(
externalPath.getRelative(repo.getName()), repoUnderVendor, Symlinks.NOFOLLOW);
FileSystemUtils.copyFile(
externalPath.getChild("@" + repo.getName() + ".marker"),
vendorPath.getChild("@" + repo.getName() + ".marker"));
}
}
}
/**
* Returns whether the repo under vendor needs to be updated by comparing its marker file with the
* one under /external
*/
private boolean isRepoUpToDate(String repoName, Path vendorPath, Path externalPath)
throws IOException {
Path vendorMarkerFile = vendorPath.getChild("@" + repoName + ".marker");
if (!vendorMarkerFile.exists()) {
return false;
}
// Since this runs after BazelFetchAllFunction, its guaranteed that the marker files
// under $OUTPUT_BASE/external are up-to-date. We just need to compare it against the marker
// under vendor.
Path externalMarkerFile = externalPath.getChild("@" + repoName + ".marker");
String vendorMarkerContent = FileSystemUtils.readContent(vendorMarkerFile, UTF_8);
String externalMarkerContent = FileSystemUtils.readContent(externalMarkerFile, UTF_8);
return Objects.equals(vendorMarkerContent, externalMarkerContent);
}
private static BlazeCommandResult createFailedBlazeCommandResult(
Reporter reporter, Code fetchCommandCode, String message) {
return createFailedBlazeCommandResult(
reporter,
message,
DetailedExitCode.of(
FailureDetail.newBuilder()
.setMessage(message)
.setFetchCommand(
FailureDetails.FetchCommand.newBuilder().setCode(fetchCommandCode).build())
.build()));
}
private static BlazeCommandResult createFailedBlazeCommandResult(
Reporter reporter, String errorMessage) {
return createFailedBlazeCommandResult(
reporter, errorMessage, InterruptedFailureDetails.detailedExitCode(errorMessage));
}
private static BlazeCommandResult createFailedBlazeCommandResult(
Reporter reporter, String message, DetailedExitCode exitCode) {
reporter.handle(Event.error(message));
return BlazeCommandResult.detailedExitCode(exitCode);
}
}