DarkflameServer/dChatServer/ChatWeb.cpp

#include "ChatWeb.h"

#include "Logger.h"
#include "Game.h"
#include "json.hpp"
#include "dCommonVars.h"
#include "MessageType/Chat.h"
#include "dServer.h"
#include "dConfig.h"
#include "PlayerContainer.h"
#include "GeneralUtils.h"
#include "eHTTPMethod.h"
#include "magic_enum.hpp"
#include "ChatPackets.h"
#include "StringifiedEnum.h"
#include "Database.h"
#include "ChatJSONUtils.h"
#include "JSONUtils.h"

using json = nlohmann::json;

void HandleHTTPPlayersRequest(HTTPReply& reply, std::string body) {
	const json data = Game::playerContainer;
	reply.status = data.empty() ? eHTTPStatusCode::NO_CONTENT : eHTTPStatusCode::OK;
	reply.message = data.empty() ? "{\"error\":\"No Players Online\"}" : data.dump();
}

void HandleHTTPTeamsRequest(HTTPReply& reply, std::string body) {
	const json data = Game::playerContainer.GetTeamContainer();
	reply.status = data.empty() ? eHTTPStatusCode::NO_CONTENT : eHTTPStatusCode::OK;
	reply.message = data.empty() ? "{\"error\":\"No Teams Online\"}" : data.dump();
}

void HandleHTTPAnnounceRequest(HTTPReply& reply, std::string body) {
	auto data = GeneralUtils::TryParse<json>(body);
	if (!data) {
		reply.status = eHTTPStatusCode::BAD_REQUEST;
		reply.message = "{\"error\":\"Invalid JSON\"}";
		return;
	}

	const auto& good_data = data.value();
	auto check = JSONUtils::CheckRequiredData(good_data, { "title", "message" });
	if (!check.empty()) {
		reply.status = eHTTPStatusCode::BAD_REQUEST;
		reply.message = check;
	} else {

		ChatPackets::Announcement announcement;
		announcement.title = good_data["title"];
		announcement.message = good_data["message"];
		announcement.Send();

		reply.status = eHTTPStatusCode::OK;
		reply.message = "{\"status\":\"Announcement Sent\"}";
	}
}


void HandleWSChat(mg_connection* connection, json data) {
	auto check = JSONUtils::CheckRequiredData(data, { "user", "message" });
	if (!check.empty()) {
		LOG_DEBUG("Received invalid websocket message: %s", check.c_str());
	} else {
		const auto user = data["user"].get<std::string>();
		const auto message = data["message"].get<std::string>();
		LOG_DEBUG("EXTERNAL Chat message from %s: %s", user.c_str(), message.c_str());
		//TODO: Send chat message to  corret world server to broadcast to players
	}
}

void HandleWSSubscribe(mg_connection* connection, json data) {
	auto check = JSONUtils::CheckRequiredData(data, { "type" });
	if (!check.empty()) {
		LOG_DEBUG("Received invalid websocket message: %s", check.c_str());
	} else {
		const auto type = data["type"].get<std::string>();
		LOG_DEBUG("type %s subscribed", type.c_str());
		const auto sub =  magic_enum::enum_cast<eWSSubscription>(type).value_or(eWSSubscription::INVALID);
		if (sub != eWSSubscription::INVALID) {
			connection->data[GeneralUtils::ToUnderlying(sub)] = 1;
			mg_ws_send(connection, "{\"status\":\"subscribed\"}", 18, WEBSOCKET_OP_TEXT);
		}
	}
}

void HandleWSUnsubscribe(mg_connection* connection, json data) {
	auto check = JSONUtils::CheckRequiredData(data, { "type" });
	if (!check.empty()) {
		LOG_DEBUG("Received invalid websocket message: %s", check.c_str());
	} else {
		const auto type = data["type"].get<std::string>();
		LOG_DEBUG("type %s unsubscribed", type.c_str());
		const auto sub =  magic_enum::enum_cast<eWSSubscription>(type).value_or(eWSSubscription::INVALID);
		if (sub != eWSSubscription::INVALID) {
			connection->data[GeneralUtils::ToUnderlying(sub)] = 0;
			mg_ws_send(connection, "{\"status\":\"unsubscribed\"}", 18, WEBSOCKET_OP_TEXT);
		}
	}
}

void ChatWeb::RegisterRoutes() {
	// REST API v1 routes
	std::string v1_route = "/api/v1/";
	Game::web.RegisterHTTPRoute({
		.path = v1_route + "players",
		.method = eHTTPMethod::GET,
		.handle = HandleHTTPPlayersRequest
	});

	Game::web.RegisterHTTPRoute({
		.path = v1_route + "teams",
		.method = eHTTPMethod::GET,
		.handle = HandleHTTPTeamsRequest
	});

	Game::web.RegisterHTTPRoute({
		.path = v1_route + "announce",
		.method = eHTTPMethod::POST,
		.handle = HandleHTTPAnnounceRequest
	});

	// WebSocket Actions
	Game::web.RegisterWSAction({
		.action = "subscribe",
		.handle = HandleWSSubscribe
	});

	Game::web.RegisterWSAction({
		.action = "unsubscribe",
		.handle = HandleWSUnsubscribe
	});

	Game::web.RegisterWSAction({
		.action = "chat",
		.handle = HandleWSChat
	});
}