Adding an info command to display a top-level summary of disk space etc (#602)

this also adds a central, shared folder for storing information which may be found by any instance on the host. The directory is currently located alongside the default install and state directory.

Initially this is used to store a collection of known `root_manifest` locations and a copy of the latest manifest version which allow us to find all known locations where zen state is present.

1 files changed, 71 insertions, 4 deletions

diff --git a/src/zenserver/config.cpp b/src/zenserver/config.cpp
index 5f2c3351e..e3286bfb8 100644
--- a/src/zenserver/config.cpp
+++ b/src/zenserver/config.cpp
@@ -5,6 +5,8 @@
 #include "config/luaconfig.h"
 #include "diag/logging.h"
 
+#include <zencore/compactbinarybuilder.h>
+#include <zencore/compactbinaryvalidation.h>
 #include <zencore/crypto.h>
 #include <zencore/except.h>
 #include <zencore/fmtutils.h>
@@ -41,7 +43,7 @@ ZEN_THIRD_PARTY_INCLUDES_END
 namespace zen {
 
 std::filesystem::path
-PickDefaultStateDirectory()
+PickDefaultSystemRootDirectory()
 {
 	// Pick sensible default
 	PWSTR	ProgramDataDir = nullptr;
@@ -50,7 +52,7 @@ PickDefaultStateDirectory()
 	if (SUCCEEDED(hRes))
 	{
 		std::filesystem::path FinalPath(ProgramDataDir);
-		FinalPath /= L"Epic\\Zen\\Data";
+		FinalPath /= L"Epic\\Zen";
 		::CoTaskMemFree(ProgramDataDir);
 
 		return FinalPath;
@@ -66,7 +68,7 @@ PickDefaultStateDirectory()
 namespace zen {
 
 std::filesystem::path
-PickDefaultStateDirectory()
+PickDefaultSystemRootDirectory()
 {
 	int			  UserId = getuid();
 	const passwd* Passwd = getpwuid(UserId);
@@ -79,6 +81,62 @@ PickDefaultStateDirectory()
 
 namespace zen {
 
+std::filesystem::path
+PickDefaultStateDirectory(std::filesystem::path SystemRoot)
+{
+	if (SystemRoot.empty())
+		return SystemRoot;
+
+	return SystemRoot / "Data";
+}
+
+void
+EmitCentralManifest(const std::filesystem::path& SystemRoot, Oid Identifier, CbObject Manifest, std::filesystem::path ManifestPath)
+{
+	CbObjectWriter Cbo;
+	Cbo << "path" << ManifestPath.generic_wstring();
+	Cbo << "manifest" << Manifest;
+
+	const std::filesystem::path StatesPath = SystemRoot / "States";
+
+	CreateDirectories(StatesPath);
+	WriteFile(StatesPath / fmt::format("{}", Identifier), Cbo.Save().GetBuffer().AsIoBuffer());
+}
+
+std::vector<CbObject>
+ReadAllCentralManifests(const std::filesystem::path& SystemRoot)
+{
+	std::vector<CbObject> Manifests;
+
+	DirectoryContent Content;
+	GetDirectoryContent(SystemRoot / "States", DirectoryContent::IncludeFilesFlag, Content);
+
+	for (std::filesystem::path& File : Content.Files)
+	{
+		try
+		{
+			FileContents	FileData	  = ReadFile(File);
+			IoBuffer		DataBuffer	  = FileData.Flatten();
+			CbValidateError ValidateError = ValidateCompactBinary(DataBuffer, CbValidateMode::All);
+
+			if (ValidateError == CbValidateError::None)
+			{
+				Manifests.push_back(LoadCompactBinaryObject(DataBuffer));
+			}
+			else
+			{
+				ZEN_WARN("failed to load manifest '{}': {}", File, ToString(ValidateError));
+			}
+		}
+		catch (std::exception& Ex)
+		{
+			ZEN_WARN("failed to load manifest '{}': {}", File, Ex.what());
+		}
+	}
+
+	return Manifests;
+}
+
 void
 ValidateOptions(ZenServerOptions& ServerOptions)
 {
@@ -343,6 +401,7 @@ ParseConfigFile(const std::filesystem::path& Path,
 	LuaOptions.AddOption("server.logid"sv, ServerOptions.LogId, "log-id"sv);
 	LuaOptions.AddOption("server.sentry.disable"sv, ServerOptions.NoSentry, "no-sentry"sv);
 	LuaOptions.AddOption("server.sentry.allowpersonalinfo"sv, ServerOptions.SentryAllowPII, "sentry-allow-personal-info"sv);
+	LuaOptions.AddOption("server.systemrootdir"sv, ServerOptions.SystemRootDir, "system-dir"sv);
 	LuaOptions.AddOption("server.datadir"sv, ServerOptions.DataDir, "data-dir"sv);
 	LuaOptions.AddOption("server.contentdir"sv, ServerOptions.ContentDir, "content-dir"sv);
 	LuaOptions.AddOption("server.abslog"sv, ServerOptions.AbsLogFile, "abslog"sv);
@@ -503,6 +562,7 @@ ParseCliOptions(int argc, char* argv[], ZenServerOptions& ServerOptions)
 	// stream operator to convert argv value into the options type. std::fs::path
 	// expects paths in streams to be quoted but argv paths are unquoted. By
 	// going into a std::string first, paths with whitespace parse correctly.
+	std::string SystemRootDir;
 	std::string DataDir;
 	std::string ContentDir;
 	std::string AbsLogFile;
@@ -525,6 +585,7 @@ ParseCliOptions(int argc, char* argv[], ZenServerOptions& ServerOptions)
 	options.add_options()("help", "Show command line help");
 	options.add_options()("t, test", "Enable test mode", cxxopts::value<bool>(ServerOptions.IsTest)->default_value("false"));
 	options.add_options()("data-dir", "Specify persistence root", cxxopts::value<std::string>(DataDir));
+	options.add_options()("system-dir", "Specify system root", cxxopts::value<std::string>(SystemRootDir));
 	options.add_options()("snapshot-dir",
 						  "Specify a snapshot of server state to mirror into the persistence root at startup",
 						  cxxopts::value<std::string>(BaseSnapshotDir));
@@ -975,6 +1036,7 @@ ParseCliOptions(int argc, char* argv[], ZenServerOptions& ServerOptions)
 		}
 		logging::RefreshLogLevels();
 
+		ServerOptions.SystemRootDir					  = MakeSafePath(SystemRootDir);
 		ServerOptions.DataDir						  = MakeSafePath(DataDir);
 		ServerOptions.BaseSnapshotDir				  = MakeSafePath(BaseSnapshotDir);
 		ServerOptions.ContentDir					  = MakeSafePath(ContentDir);
@@ -1022,9 +1084,14 @@ ParseCliOptions(int argc, char* argv[], ZenServerOptions& ServerOptions)
 		throw;
 	}
 
+	if (ServerOptions.SystemRootDir.empty())
+	{
+		ServerOptions.SystemRootDir = PickDefaultSystemRootDirectory();
+	}
+
 	if (ServerOptions.DataDir.empty())
 	{
-		ServerOptions.DataDir = PickDefaultStateDirectory();
+		ServerOptions.DataDir = PickDefaultStateDirectory(ServerOptions.SystemRootDir);
 	}
 
 	if (ServerOptions.AbsLogFile.empty())