Add a list-container subcommand to zen builds command

1 files changed, 52 insertions, 0 deletions

diff --git a/src/zenutil/jupiter/jupiterbuildstorage.cpp b/src/zenutil/jupiter/jupiterbuildstorage.cpp
index 24e062c7b..b2fabb43f 100644
--- a/src/zenutil/jupiter/jupiterbuildstorage.cpp
+++ b/src/zenutil/jupiter/jupiterbuildstorage.cpp
@@ -35,6 +35,58 @@ public:
 	}
 	virtual ~JupiterBuildStorage() {}
 
+	virtual CbObject ListContainers() override
+	{
+		ZEN_TRACE_CPU("Jupiter::ListContainers");
+
+		Stopwatch	  ExecutionTimer;
+		auto		  _			 = MakeGuard([&]() { m_Stats.TotalExecutionTimeUs += ExecutionTimer.GetElapsedTimeUs(); });
+		JupiterResult ListResult = m_Session.ListBuildNamespaces();
+		AddStatistic(ListResult);
+		if (!ListResult.Success)
+		{
+			throw std::runtime_error(fmt::format("Failed listing containers: {} ({})", ListResult.Reason, ListResult.ErrorCode));
+		}
+		CbObject NamespaceResponse = PayloadToCbObject("Failed listing containers"sv, ListResult.Response);
+
+		CbObjectWriter Response;
+		Response.BeginArray("results"sv);
+		for (CbFieldView NamespaceField : NamespaceResponse["namespaces"])
+		{
+			std::string_view Namespace = NamespaceField.AsString();
+			if (!Namespace.empty())
+			{
+				Response.BeginObject();
+				Response.AddString("name", Namespace);
+
+				JupiterResult BucketsResult = m_Session.ListBuildBuckets(Namespace);
+				AddStatistic(BucketsResult);
+				if (!BucketsResult.Success)
+				{
+					throw std::runtime_error(
+						fmt::format("Failed listing containers: {} ({})", BucketsResult.Reason, BucketsResult.ErrorCode));
+				}
+				CbObject BucketResponse = PayloadToCbObject("Failed listing containers"sv, BucketsResult.Response);
+
+				Response.BeginArray("items");
+				for (CbFieldView BucketField : BucketResponse["buckets"])
+				{
+					std::string_view Bucket = BucketField.AsString();
+					if (!Bucket.empty())
+					{
+						Response.AddString(Bucket);
+					}
+				}
+				Response.EndArray();
+
+				Response.EndObject();
+			}
+		}
+		Response.EndArray();
+
+		return Response.Save();
+	}
+
 	virtual CbObject ListBuilds(CbObject Query) override
 	{
 		ZEN_TRACE_CPU("Jupiter::ListBuilds");