forked from alterware/iw6-mod
152 lines
3.7 KiB
C++
152 lines
3.7 KiB
C++
#include <std_include.hpp>
|
|
#include "loader/component_loader.hpp"
|
|
#include "game/game.hpp"
|
|
|
|
#include "console.hpp"
|
|
#include "network.hpp"
|
|
#include "party.hpp"
|
|
#include "scheduler.hpp"
|
|
|
|
#include <utils/cryptography.hpp>
|
|
#include <utils/string.hpp>
|
|
|
|
#include <discord_rpc.h>
|
|
|
|
namespace discord
|
|
{
|
|
namespace
|
|
{
|
|
DiscordRichPresence discord_presence;
|
|
|
|
void join_game(const char* join_secret)
|
|
{
|
|
game::Cbuf_AddText(0, utils::string::va("connect %s\n", join_secret));
|
|
}
|
|
|
|
void join_request(const DiscordUser* request)
|
|
{
|
|
#ifdef _DEBUG
|
|
console::info("Discord: Join request from %s (%s)\n", request->username, request->userId);
|
|
#endif
|
|
Discord_Respond(request->userId, DISCORD_REPLY_IGNORE);
|
|
}
|
|
|
|
void update_discord()
|
|
{
|
|
Discord_RunCallbacks();
|
|
|
|
if (!game::CL_IsCgameInitialized())
|
|
{
|
|
discord_presence.details = game::environment::is_sp() ? "Singleplayer" : "Multiplayer";
|
|
discord_presence.state = "Main Menu";
|
|
|
|
discord_presence.partySize = 0;
|
|
discord_presence.partyMax = 0;
|
|
|
|
discord_presence.startTimestamp = 0;
|
|
}
|
|
else
|
|
{
|
|
if (game::environment::is_sp()) return;
|
|
|
|
const auto* gametype = game::UI_LocalizeGametype(game::Dvar_FindVar("ui_gametype")->current.string);
|
|
const auto* map = game::UI_LocalizeMapname(game::Dvar_FindVar("ui_mapname")->current.string);
|
|
|
|
discord_presence.details = utils::string::va("%s on %s", gametype, map);
|
|
|
|
discord_presence.partySize = game::mp::cgArray->snap != nullptr
|
|
? game::mp::cgArray->snap->numClients
|
|
: 1;
|
|
|
|
if (game::Dvar_GetBool("xblive_privatematch"))
|
|
{
|
|
discord_presence.state = "Private Match";
|
|
discord_presence.partyMax = game::Dvar_GetInt("sv_maxclients");
|
|
}
|
|
else
|
|
{
|
|
auto* host_name = reinterpret_cast<char*>(0x14187EBC4);
|
|
utils::string::strip(host_name, host_name, std::strlen(host_name) + 1);
|
|
|
|
discord_presence.state = host_name;
|
|
discord_presence.partyMax = party::server_client_count();
|
|
|
|
std::hash<game::netadr_s> hash_fn;
|
|
static const auto nonce = utils::cryptography::random::get_integer();
|
|
|
|
const auto& address = party::get_target();
|
|
discord_presence.partyId = utils::string::va("%zu", hash_fn(address) ^ nonce);
|
|
discord_presence.joinSecret = network::net_adr_to_string(address);
|
|
}
|
|
|
|
if (!discord_presence.startTimestamp)
|
|
{
|
|
discord_presence.startTimestamp = std::chrono::duration_cast<std::chrono::seconds>(
|
|
std::chrono::system_clock::now().time_since_epoch()).count();
|
|
}
|
|
}
|
|
|
|
discord_presence.largeImageKey = "main_logo";
|
|
Discord_UpdatePresence(&discord_presence);
|
|
}
|
|
}
|
|
|
|
class component final : public component_interface
|
|
{
|
|
public:
|
|
void post_load() override
|
|
{
|
|
if (game::environment::is_dedi())
|
|
{
|
|
return;
|
|
}
|
|
|
|
DiscordEventHandlers handlers;
|
|
ZeroMemory(&handlers, sizeof(handlers));
|
|
handlers.ready = ready;
|
|
handlers.errored = errored;
|
|
handlers.disconnected = errored;
|
|
handlers.joinGame = join_game;
|
|
handlers.spectateGame = nullptr;
|
|
handlers.joinRequest = join_request;
|
|
|
|
Discord_Initialize("1116670204148195428", &handlers, 1, nullptr);
|
|
|
|
scheduler::loop(update_discord, scheduler::pipeline::main, 20s);
|
|
|
|
initialized_ = true;
|
|
}
|
|
|
|
void pre_destroy() override
|
|
{
|
|
if (!initialized_)
|
|
{
|
|
return;
|
|
}
|
|
|
|
Discord_Shutdown();
|
|
}
|
|
|
|
private:
|
|
bool initialized_ = false;
|
|
|
|
static void ready(const DiscordUser* /*request*/)
|
|
{
|
|
ZeroMemory(&discord_presence, sizeof(discord_presence));
|
|
|
|
discord_presence.instance = 1;
|
|
|
|
Discord_UpdatePresence(&discord_presence);
|
|
}
|
|
|
|
static void errored(const int error_code, const char* message)
|
|
{
|
|
console::error("Discord: (%i) %s\n", error_code, message);
|
|
}
|
|
};
|
|
}
|
|
|
|
#ifndef DEV_BUILD
|
|
REGISTER_COMPONENT(discord::component)
|
|
#endif
|