t8-mod/source/proxy-dll/loader/component_loader.cpp

#include <std_include.hpp>
#include "component_loader.hpp"

#include <utils/nt.hpp>

void component_loader::register_component(std::unique_ptr<component_interface>&& component_)
{
	auto& components = get_components();
	components.push_back(std::move(component_));

	std::ranges::stable_sort(components, [](const std::unique_ptr<component_interface>& a,
	                                        const std::unique_ptr<component_interface>& b)
	{
		return a->priority() > b->priority();
	});
}

bool component_loader::pre_start()
{
	static auto res = []
	{
		clean();

		try
		{
			for (const auto& component_ : get_components())
			{
				component_->pre_start();
			}
		}
		catch (premature_shutdown_trigger&)
		{
			return false;
		}
		catch (const std::exception& e)
		{
			MessageBoxA(nullptr, e.what(), "Error", MB_ICONERROR);
			return false;
		}

		return true;
	}();

	return res;
}

void component_loader::post_unpack()
{
	static auto res = []
	{
		clean();

		try
		{
			for (const auto& component_ : get_components())
			{
				component_->post_unpack();
			}
		}
		catch (const std::exception& e)
		{
			MessageBoxA(nullptr, e.what(), "Error", MB_ICONERROR);
			return false;
		}

		return true;
	}();

	if (!res)
	{
		TerminateProcess(GetCurrentProcess(), 1);
	}
}

void component_loader::pre_destroy()
{
	static auto res = []
	{
		clean();

		try
		{
			for (const auto& component_ : get_components())
			{
				component_->pre_destroy();
			}
		}
		catch (const std::exception& e)
		{
			MessageBoxA(nullptr, e.what(), "Error", MB_ICONERROR);
			return false;
		}

		return true;
	}();

	if (!res)
	{
		TerminateProcess(GetCurrentProcess(), 1);
	}
}

void component_loader::clean()
{
	auto& components = get_components();
	for (auto i = components.begin(); i != components.end();)
	{
		if (!(*i)->is_supported())
		{
			(*i)->pre_destroy();
			i = components.erase(i);
		}
		else
		{
			++i;
		}
	}
}

void component_loader::trigger_premature_shutdown()
{
	throw premature_shutdown_trigger();
}

std::vector<std::unique_ptr<component_interface>>& component_loader::get_components()
{
	using component_vector = std::vector<std::unique_ptr<component_interface>>;
	using component_vector_container = std::unique_ptr<component_vector, std::function<void(component_vector*)>>;

	static component_vector_container components(new component_vector, [](const component_vector* component_vector)
	{
		pre_destroy();
		delete component_vector;
	});

	return *components;
}