hub deprovision all (#938)

* implement "deprovision all" for hub

2 files changed, 81 insertions, 0 deletions

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);