OptiScaler/CyberXeSS/dllmain.cpp

#pragma once
#include "pch.h"
#include <iostream>

#include "spdlog/sinks/basic_file_sink.h"
#include "spdlog/sinks/stdout_color_sinks.h"
#include "spdlog/sinks/callback_sink.h"

#include "Config.h"

HMODULE dllModule;

static bool InitializeConsole()
{
	// Allocate a console for this app
	if (!AllocConsole()) {
		return false;
	}

	FILE* pFile;

	// Redirect STDIN if the console has an input handle
	if (GetStdHandle(STD_INPUT_HANDLE) != INVALID_HANDLE_VALUE) {
		if (freopen_s(&pFile, "CONIN$", "r", stdin) != 0) {
			return false;
		}
	}

	// Redirect STDOUT if the console has an output handle
	if (GetStdHandle(STD_OUTPUT_HANDLE) != INVALID_HANDLE_VALUE) {
		if (freopen_s(&pFile, "CONOUT$", "w", stdout) != 0) {
			return false;
		}
	}

	// Redirect STDERR if the console has an error handle
	if (GetStdHandle(STD_ERROR_HANDLE) != INVALID_HANDLE_VALUE) {
		if (freopen_s(&pFile, "CONOUT$", "w", stderr) != 0) {
			return false;
		}
	}

	// Clear the error state for each of the C++ standard streams
	std::cin.clear();
	std::cout.clear();
	std::cerr.clear();
	std::wcin.clear();
	std::wcout.clear();
	std::wcerr.clear();

	// Make C++ standard streams point to console as well.
	std::ios::sync_with_stdio();

	return true;
}

static void PrepareLogger()
{
	try
	{
		if (Config::Instance()->LoggingEnabled.value_or(true))
		{
			if (Config::Instance()->OpenConsole.value_or(false))
				InitializeConsole();

			std::shared_ptr<spdlog::logger> shared_logger = nullptr;

			std::vector<spdlog::sink_ptr> sinks;

			if (Config::Instance()->LogToConsole.value_or(false))
			{
				auto console_sink = std::make_shared<spdlog::sinks::stdout_color_sink_mt>();
				console_sink->set_level(spdlog::level::level_enum::info);
				console_sink->set_pattern("[%H:%M:%S.%f] [%L] %v");

				sinks.push_back(console_sink);
			}

			if (Config::Instance()->LogToFile.value_or(false))
			{
				auto file_sink = std::make_shared<spdlog::sinks::basic_file_sink_mt>(Config::Instance()->LogFileName.value(), true);
				file_sink->set_level((spdlog::level::level_enum)Config::Instance()->LogLevel.value_or(2));
				file_sink->set_pattern("[%H:%M:%S.%f] [%L] %v");

				sinks.push_back(file_sink);
			}

			auto callback_sink = std::make_shared<spdlog::sinks::callback_sink_mt>([](const spdlog::details::log_msg& msg)
				{
					if (Config::Instance()->LogToNGX.value_or(false) &&
						!Config::Instance()->NVSDK_Logger.DisableOtherLoggingSinks &&
						Config::Instance()->NVSDK_Logger.LoggingCallback != nullptr &&
						Config::Instance()->NVSDK_Logger.MinimumLoggingLevel != NVSDK_NGX_LOGGING_LEVEL_OFF &&
						(Config::Instance()->NVSDK_Logger.MinimumLoggingLevel == NVSDK_NGX_LOGGING_LEVEL_VERBOSE || msg.level >= spdlog::level::info))
					{
						Config::Instance()->NVSDK_Logger.LoggingCallback(msg.payload.data(), NVSDK_NGX_LOGGING_LEVEL_ON, NVSDK_NGX_Feature_SuperSampling);
					}
				});

			callback_sink->set_level(spdlog::level::level_enum::trace);
			callback_sink->set_pattern("[%H:%M:%S.%f] [%L] %v");

			sinks.push_back(callback_sink);

			spdlog::logger logger("multi_sink", sinks.begin(), sinks.end());
			shared_logger = std::make_shared<spdlog::logger>(logger);

			shared_logger->set_level(spdlog::level::level_enum::trace);

#if _DEBUG
			shared_logger->flush_on(spdlog::level::debug);
#else
			shared_logger->flush_on(spdlog::level::err);
#endif // _DEBUG

			spdlog::set_default_logger(shared_logger);
		}
	}
	catch (const spdlog::spdlog_ex& ex)
	{
		std::cerr << ex.what() << std::endl;

		auto logger = spdlog::stdout_color_mt("xess");
		logger->set_pattern("[%H:%M:%S.%f] [%L] %v");
		logger->set_level((spdlog::level::level_enum)2);
		spdlog::set_default_logger(logger);
	}
}

static void CloseLogger()
{
	spdlog::default_logger()->flush();
	spdlog::shutdown();
}

BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved)
{
	switch (ul_reason_for_call)
	{
	case DLL_PROCESS_ATTACH:
		DisableThreadLibraryCalls(hModule);
		dllModule = hModule;
		PrepareLogger();
		break;

	case DLL_THREAD_ATTACH:
		break;

	case DLL_THREAD_DETACH:
		break;

	case DLL_PROCESS_DETACH:
		CloseLogger();
		break;
	}

	return TRUE;
}