just serve: Implement client-side ServeContent and extend serve API

1 files changed, 22 insertions, 0 deletions

diff --git a/src/buildtool/serve_api/remote/source_tree_client.cpp b/src/buildtool/serve_api/remote/source_tree_client.cpp
index a2ff930f..e3cc0756 100644
--- a/src/buildtool/serve_api/remote/source_tree_client.cpp
+++ b/src/buildtool/serve_api/remote/source_tree_client.cpp
@@ -120,3 +120,25 @@ auto SourceTreeClient::ServeArchiveTree(
     }
     return response.tree();
 }
+
+auto SourceTreeClient::ServeContent(std::string const& content) -> bool {
+    justbuild::just_serve::ServeContentRequest request{};
+    request.set_content(content);
+
+    grpc::ClientContext context;
+    justbuild::just_serve::ServeContentResponse response;
+    grpc::Status status = stub_->ServeContent(&context, request, &response);
+
+    if (not status.ok()) {
+        LogStatus(&logger_, LogLevel::Debug, status);
+        return false;
+    }
+    if (response.status() !=
+        ::justbuild::just_serve::ServeContentResponse::OK) {
+        logger_.Emit(LogLevel::Debug,
+                     "ServeContent response returned with {}",
+                     static_cast<int>(response.status()));
+        return false;
+    }
+    return true;
+}