just install{,-cas}: offer local CAS as preferred CAS

... to avoid unnecessary downloads and hence speed things up.

4 files changed, 10 insertions, 4 deletions

diff --git a/src/buildtool/graph_traverser/graph_traverser.hpp b/src/buildtool/graph_traverser/graph_traverser.hpp
index e34df5cf..1d5b821d 100644
--- a/src/buildtool/graph_traverser/graph_traverser.hpp
+++ b/src/buildtool/graph_traverser/graph_traverser.hpp
@@ -571,7 +571,8 @@ class GraphTraverser {
         auto output_paths = PrepareOutputPaths(rel_paths);
 
         if (not output_paths or
-            not remote_api_->RetrieveToPaths(object_infos, *output_paths)) {
+            not remote_api_->RetrieveToPaths(
+                object_infos, *output_paths, GetLocalApi())) {
             Logger::Log(LogLevel::Error, "Could not retrieve outputs.");
             return std::nullopt;
         }
diff --git a/src/buildtool/main/install_cas.cpp b/src/buildtool/main/install_cas.cpp
index 03c57d71..b56a1069 100644
--- a/src/buildtool/main/install_cas.cpp
+++ b/src/buildtool/main/install_cas.cpp
@@ -61,8 +61,10 @@ namespace {
 }
 
 #ifndef BOOTSTRAP_BUILD_TOOL
-auto FetchAndInstallArtifacts(gsl::not_null<IExecutionApi*> const& api,
-                              FetchArguments const& clargs) -> bool {
+auto FetchAndInstallArtifacts(
+    gsl::not_null<IExecutionApi*> const& api,
+    gsl::not_null<IExecutionApi*> const& alternative_api,
+    FetchArguments const& clargs) -> bool {
     auto object_info = ObjectInfoFromLiberalString(clargs.object_id);
 
     if (clargs.output_path) {
@@ -72,7 +74,8 @@ auto FetchAndInstallArtifacts(gsl::not_null<IExecutionApi*> const& api,
         }
 
         if (not FileSystemManager::CreateDirectory(output_path.parent_path()) or
-            not api->RetrieveToPaths({object_info}, {output_path})) {
+            not api->RetrieveToPaths(
+                {object_info}, {output_path}, alternative_api)) {
             Logger::Log(LogLevel::Error, "failed to retrieve artifact.");
             return false;
         }
diff --git a/src/buildtool/main/install_cas.hpp b/src/buildtool/main/install_cas.hpp
index 562c7bae..e083776e 100644
--- a/src/buildtool/main/install_cas.hpp
+++ b/src/buildtool/main/install_cas.hpp
@@ -31,6 +31,7 @@
 #ifndef BOOTSTRAP_BUILD_TOOL
 [[nodiscard]] auto FetchAndInstallArtifacts(
     gsl::not_null<IExecutionApi*> const& api,
+    gsl::not_null<IExecutionApi*> const& alternative_api,
     FetchArguments const& clargs) -> bool;
 #endif
 
diff --git a/src/buildtool/main/main.cpp b/src/buildtool/main/main.cpp
index 16385ade..35137d77 100644
--- a/src/buildtool/main/main.cpp
+++ b/src/buildtool/main/main.cpp
@@ -1299,6 +1299,7 @@ auto main(int argc, char* argv[]) -> int {
 
         if (arguments.cmd == SubCommand::kInstallCas) {
             return FetchAndInstallArtifacts(traverser.GetRemoteApi(),
+                                            traverser.GetLocalApi(),
                                             arguments.fetch)
                        ? kExitSuccess
                        : kExitFailure;