just-mr: Implement 'absent' roots

...via an 'absent' pragma in repository descriptions.

For 'git'-type repositories, first interrogates a 'just serve'
remote, if given, before reverting to fetching from the network.

1 files changed, 22 insertions, 0 deletions

diff --git a/src/other_tools/just_mr/setup_utils.cpp b/src/other_tools/just_mr/setup_utils.cpp
index 28355e89..b325a71d 100644
--- a/src/other_tools/just_mr/setup_utils.cpp
+++ b/src/other_tools/just_mr/setup_utils.cpp
@@ -15,6 +15,9 @@
 #include "src/other_tools/just_mr/setup_utils.hpp"
 
 #include <fstream>
+#include <memory>
+#include <optional>
+#include <string>
 #include <unordered_set>
 
 #include "nlohmann/json.hpp"
@@ -24,6 +27,7 @@
 #include "src/buildtool/execution_api/remote/bazel/bazel_api.hpp"
 #include "src/buildtool/file_system/file_system_manager.hpp"
 #include "src/buildtool/logging/logger.hpp"
+#include "src/buildtool/serve_api/remote/config.hpp"
 #include "src/other_tools/just_mr/exit_codes.hpp"
 
 namespace {
@@ -202,4 +206,22 @@ auto SetupRemoteApi(std::optional<std::string> const& remote_exec_addr,
     return nullptr;
 }
 
+auto SetupServeApi(std::optional<std::string> const& remote_serve_addr,
+                   MultiRepoRemoteAuthArguments const& auth) -> ServeApi::Ptr {
+    if (remote_serve_addr) {
+        // setup authentication
+        SetupAuthConfig(auth);
+        // setup remote
+        if (not RemoteServeConfig::SetRemoteAddress(*remote_serve_addr)) {
+            Logger::Log(LogLevel::Error,
+                        "setting remote serve service address '{}' failed.",
+                        *remote_serve_addr);
+            std::exit(kExitConfigError);
+        }
+        auto address = RemoteServeConfig::RemoteAddress();
+        return std::make_unique<ServeApi>(address->host, address->port);
+    }
+    return nullptr;
+}
+
 }  // namespace JustMR::Utils