DarkflameServer/dDashboardServer/routes/ServerState.h

#pragma once

#include <chrono>
#include <ctime>
#include <map>
#include <mutex>
#include <vector>
#include <string>
#include <cstdint>
#include "json.hpp"

struct ServerStatus {
	bool online{false};
	uint32_t players{0};
	std::string version{};
	std::chrono::steady_clock::time_point lastSeen{};
	int64_t since{}; // unix time the dashboard first saw it up this time (0: down)

	// Online or not, keeping `since` for as long as it stays up
	void Set(bool up) {
		if (up && !online) since = static_cast<int64_t>(std::time(nullptr));
		if (!up) since = 0;
		online = up;
		if (up) lastSeen = std::chrono::steady_clock::now();
	}
};

struct WorldInstanceInfo {
	uint32_t mapID{0};
	uint32_t instanceID{0};
	uint32_t cloneID{0};
	uint32_t players{0};
	std::string ip{};
	uint32_t port{0};
	bool isPrivate{false};
	std::string zoneName{};
	// For property instances (cloneID != 0), looked up once when the world is added
	std::string propertyId{};
	std::string propertyName{};
	std::string ownerId{};
	std::string ownerName{};
	// "starting" (launched, not connected yet) or "stopping" (in g_PendingWorlds), or "draining" for a world that is up
	// but whose players are being moved to a new instance (in g_WorldInstances); empty for one that is just up
	std::string state{};
};

namespace ServerState {
	extern ServerStatus g_AuthStatus;
	extern ServerStatus g_ChatStatus;
	// The UGC server (docs/UgcServer.md): enabled is master's enable_ugc_server, pid the process master last started
	extern ServerStatus g_UgcStatus;
	extern bool g_UgcEnabled;
	extern uint32_t g_UgcPid;
	extern std::vector<WorldInstanceInfo> g_WorldInstances;
	// Worlds master launched that aren't connected yet, or that are shutting down (kept apart: not running)
	extern std::vector<WorldInstanceInfo> g_PendingWorlds;
	extern std::mutex g_StatusMutex;

	inline nlohmann::json GetServerStateJson() {
		std::lock_guard lock(g_StatusMutex);

		nlohmann::json data;
		data["auth"]["online"] = g_AuthStatus.online;
		data["auth"]["players"] = g_AuthStatus.players;
		data["chat"]["online"] = g_ChatStatus.online;
		data["chat"]["players"] = g_ChatStatus.players;
		data["auth"]["since"] = g_AuthStatus.since;
		data["chat"]["since"] = g_ChatStatus.since;
		data["ugc"] = { {"enabled", g_UgcEnabled}, {"online", g_UgcStatus.online}, {"since", g_UgcStatus.since} };

		uint32_t totalOnlinePlayers = 0;
		data["worlds"] = nlohmann::json::array();
		for (const auto& world : g_WorldInstances) {
			totalOnlinePlayers += world.players;
			data["worlds"].push_back({
				{"mapID", world.mapID},
				{"instanceID", world.instanceID},
				{"cloneID", world.cloneID},
				{"players", world.players},
				{"isPrivate", world.isPrivate},
				{"zoneName", world.zoneName},
				{"state", world.state}
			});
		}

		data["startingWorlds"] = nlohmann::json::array();
		for (const auto& world : g_PendingWorlds) {
			data["startingWorlds"].push_back({ {"mapID", world.mapID}, {"instanceID", world.instanceID}, {"cloneID", world.cloneID},
				{"zoneName", world.zoneName}, {"state", world.state} });
		}

		data["stats"]["onlinePlayers"] = totalOnlinePlayers;
		data["stats"]["worlds"] = g_WorldInstances.size();
		return data;
	}

	/**
	 * The state as anyone signed in may see it (without players_view): no clone IDs or private flags, and property
	 * instances merged into one row per zone ({properties: n}), so nobody can tell who is on which property.
	 */
	inline nlohmann::json PlayerSafe(nlohmann::json state) {
		nlohmann::json worlds = nlohmann::json::array();
		std::map<uint32_t, size_t> propertyRows; // mapID -> index in worlds
		for (const auto& world : state.value("worlds", nlohmann::json::array())) {
			const auto mapID = world.value("mapID", 0u);
			if (world.value("cloneID", 0u) == 0) {
				worlds.push_back({ {"mapID", mapID}, {"instanceID", world.value("instanceID", 0u)}, {"players", world.value("players", 0u)},
					{"zoneName", world.value("zoneName", "")}, {"state", world.value("state", "")} });
				continue;
			}
			const auto it = propertyRows.find(mapID);
			if (it == propertyRows.end()) {
				propertyRows[mapID] = worlds.size();
				worlds.push_back({ {"mapID", mapID}, {"instanceID", 0}, {"players", world.value("players", 0u)},
					{"zoneName", world.value("zoneName", "")}, {"properties", 1} });
			} else {
				auto& row = worlds[it->second];
				row["players"] = row.value("players", 0u) + world.value("players", 0u);
				row["properties"] = row.value("properties", 0u) + 1;
			}
		}
		state["worlds"] = std::move(worlds);
		// Starting and stopping worlds without clone IDs (a property instance only says it's a property)
		nlohmann::json pending = nlohmann::json::array();
		for (const auto& world : state.value("startingWorlds", nlohmann::json::array())) {
			pending.push_back({ {"mapID", world.value("mapID", 0u)}, {"instanceID", world.value("instanceID", 0u)}, {"zoneName", world.value("zoneName", "")},
				{"state", world.value("state", "")}, {"property", world.value("cloneID", 0u) != 0} });
		}
		state["startingWorlds"] = std::move(pending);
		return state;
	}
}