hub deprovision all (#938)

* implement "deprovision all" for hub

3 files changed, 109 insertions, 5 deletions

diff --git a/src/zenserver-test/hub-tests.cpp b/src/zenserver-test/hub-tests.cpp
index 487e22b4b..35a840e5d 100644
--- a/src/zenserver-test/hub-tests.cpp
+++ b/src/zenserver-test/hub-tests.cpp
@@ -329,17 +329,36 @@ TEST_CASE("hub.lifecycle.children")
 		CHECK_EQ(Result.AsText(), "GhijklmNop"sv);
 	}
 
-	Result = Client.Post("modules/abc/deprovision");
+	// Deprovision all modules at once
+	Result = Client.Post("deprovision");
 	REQUIRE(Result);
+	CHECK_EQ(Result.StatusCode, HttpResponseCode::Accepted);
+	{
+		CbObject	Body		= Result.AsObject();
+		CbArrayView AcceptedArr = Body["Accepted"].AsArrayView();
+		CHECK_EQ(AcceptedArr.Num(), 2u);
+		bool FoundAbc = false;
+		bool FoundDef = false;
+		for (CbFieldView F : AcceptedArr)
+		{
+			if (F.AsString() == "abc"sv)
+			{
+				FoundAbc = true;
+			}
+			else if (F.AsString() == "def"sv)
+			{
+				FoundDef = true;
+			}
+		}
+		CHECK(FoundAbc);
+		CHECK(FoundDef);
+	}
 	REQUIRE(WaitForModuleGone(Client, "abc"));
+	REQUIRE(WaitForModuleGone(Client, "def"));
 	{
 		HttpClient ModClient(fmt::format("http://localhost:{}", AbcPort), kFastTimeout);
 		CHECK(WaitForPortUnreachable(ModClient));
 	}
-
-	Result = Client.Post("modules/def/deprovision");
-	REQUIRE(Result);
-	REQUIRE(WaitForModuleGone(Client, "def"));
 	{
 		HttpClient ModClient(fmt::format("http://localhost:{}", DefPort), kFastTimeout);
 		CHECK(WaitForPortUnreachable(ModClient));
@@ -349,6 +368,10 @@ TEST_CASE("hub.lifecycle.children")
 	Result = Client.Get("status");
 	REQUIRE(Result);
 	CHECK_EQ(Result.AsObject()["modules"].AsArrayView().Num(), 0u);
+
+	// Deprovision-all with no modules
+	Result = Client.Post("deprovision");
+	CHECK(Result);
 }
 
 static bool
diff --git a/src/zenserver/hub/httphubservice.cpp b/src/zenserver/hub/httphubservice.cpp
index e6a900066..e4b0c28d0 100644
--- a/src/zenserver/hub/httphubservice.cpp
+++ b/src/zenserver/hub/httphubservice.cpp
@@ -121,6 +121,11 @@ HttpHubService::HttpHubService(Hub& Hub, HttpProxyHandler& Proxy, HttpStatsServi
 		HttpVerb::kGet);
 
 	m_Router.RegisterRoute(
+		"deprovision",
+		[this](HttpRouterRequest& Req) { HandleDeprovisionAll(Req.ServerRequest()); },
+		HttpVerb::kPost);
+
+	m_Router.RegisterRoute(
 		"modules/{moduleid}",
 		[this](HttpRouterRequest& Req) {
 			std::string_view ModuleId = Req.GetCapture(1);
@@ -371,6 +376,81 @@ HttpHubService::GetActivityCounter()
 }
 
 void
+HttpHubService::HandleDeprovisionAll(HttpServerRequest& Request)
+{
+	std::vector<std::string> ModulesToDeprovision;
+	m_Hub.EnumerateModules([&ModulesToDeprovision](std::string_view ModuleId, const Hub::InstanceInfo& InstanceInfo) {
+		if (InstanceInfo.State == HubInstanceState::Provisioned || InstanceInfo.State == HubInstanceState::Hibernated)
+		{
+			ModulesToDeprovision.push_back(std::string(ModuleId));
+		}
+	});
+
+	if (ModulesToDeprovision.empty())
+	{
+		return Request.WriteResponse(HttpResponseCode::OK);
+	}
+	std::vector<std::string> Rejected;
+	std::vector<std::string> Accepted;
+	std::vector<std::string> Completed;
+	for (const std::string& ModuleId : ModulesToDeprovision)
+	{
+		Hub::Response Response = m_Hub.Deprovision(ModuleId);
+		switch (Response.ResponseCode)
+		{
+			case Hub::EResponseCode::NotFound:
+				// Ignore
+				break;
+			case Hub::EResponseCode::Rejected:
+				Rejected.push_back(ModuleId);
+				break;
+			case Hub::EResponseCode::Accepted:
+				Accepted.push_back(ModuleId);
+				break;
+			case Hub::EResponseCode::Completed:
+				Completed.push_back(ModuleId);
+				break;
+		}
+	}
+	if (Rejected.empty() && Accepted.empty() && Completed.empty())
+	{
+		return Request.WriteResponse(HttpResponseCode::OK);
+	}
+	HttpResponseCode Response = HttpResponseCode::OK;
+	CbObjectWriter	 Writer;
+	if (!Completed.empty())
+	{
+		Writer.BeginArray("Completed");
+		for (const std::string& ModuleId : Completed)
+		{
+			Writer.AddString(ModuleId);
+		}
+		Writer.EndArray();	// Completed
+	}
+	if (!Accepted.empty())
+	{
+		Writer.BeginArray("Accepted");
+		for (const std::string& ModuleId : Accepted)
+		{
+			Writer.AddString(ModuleId);
+		}
+		Writer.EndArray();	// Accepted
+		Response = HttpResponseCode::Accepted;
+	}
+	if (!Rejected.empty())
+	{
+		Writer.BeginArray("Rejected");
+		for (const std::string& ModuleId : Rejected)
+		{
+			Writer.AddString(ModuleId);
+		}
+		Writer.EndArray();	// Rejected
+		Response = HttpResponseCode::Conflict;
+	}
+	Request.WriteResponse(Response, Writer.Save());
+}
+
+void
 HttpHubService::HandleModuleGet(HttpServerRequest& Request, std::string_view ModuleId)
 {
 	Hub::InstanceInfo InstanceInfo;
diff --git a/src/zenserver/hub/httphubservice.h b/src/zenserver/hub/httphubservice.h
index ff2cb0029..f4d1b0b89 100644
--- a/src/zenserver/hub/httphubservice.h
+++ b/src/zenserver/hub/httphubservice.h
@@ -53,6 +53,7 @@ private:
 	HttpStatsService&  m_StatsService;
 	HttpStatusService& m_StatusService;
 
+	void HandleDeprovisionAll(HttpServerRequest& Request);
 	void HandleModuleGet(HttpServerRequest& Request, std::string_view ModuleId);
 	void HandleModuleDelete(HttpServerRequest& Request, std::string_view ModuleId);