YimMenuV2/src/core/scripting/LuaManager.cpp

#include "LuaManager.hpp"
#include "core/filemgr/FileMgr.hpp"
#include "core/backend/ScriptMgr.hpp"
#include "core/frontend/Notifications.hpp"

namespace YimMenu
{
	void LuaManager::AddUnloadedScript(std::string_view name, std::string_view path)
	{
		m_UnloadedScripts.push_back({std::string(name), std::string(path)});
	}

	void LuaManager::RegisterLibraryImpl(LuaLibrary* library)
	{
		m_Libraries.push_back(library);
	}

	void LuaManager::LoadLibrariesImpl(lua_State* state)
	{
		for (auto library : m_Libraries)
			library->Register(state);
	}

	void LuaManager::LoadScriptImpl(std::string path)
	{
		std::lock_guard lock(m_LoadMutex);
		m_ScriptsToLoad.push(path);
	}

	void LuaManager::RunScriptImpl()
	{
		m_MainThreadId = GetCurrentThreadId();

		auto scripts_dir = FileMgr::GetProjectFolder("./scripts");
		for (const auto& entry : std::filesystem::directory_iterator(scripts_dir.Path()))
		{
			if (entry.path().extension() != ".lua")
				continue;

			m_ScriptsToLoad.push(entry.path().string());
		}

		while (g_Running)
		{
			{
				// 1) process new load requests
				std::lock_guard lock(m_LoadMutex);
				while (!m_ScriptsToLoad.empty())
				{
					std::erase_if(m_UnloadedScripts, [this](auto& script) {
						return std::filesystem::equivalent(m_ScriptsToLoad.front(), script.m_Path);
					});
					m_LoadedScripts.push_back(std::make_shared<LuaScript>(m_ScriptsToLoad.front()));
					m_ScriptsToLoad.pop();
				}

				// 2) remove scripts if needed
				std::erase_if(m_LoadedScripts, [this](auto& script) {
					if (!script->SafeToUnload())
						return false;

					bool unload = false;

					if (script->IsMalfunctioning())
					{
						Notifications::Show("Lua Scripting", std::format("Script {} has been unloaded due to a malfunction", script->GetName()), NotificationType::Warning);
						unload = true;
					}
					else if (script->GetLoadState() == LuaScript::LoadState::WANT_UNLOAD)
					{
						unload = true;
					}
					else if (script->GetLoadState() == LuaScript::LoadState::WANT_RELOAD)
					{
						m_ScriptsToLoad.push(std::string(script->GetPath())); // will be reloaded next tick
						unload = true;
					}

					if (unload)
					{
						script->MarkUnloaded();
						AddUnloadedScript(script->GetName(), script->GetPath());
					}

					return unload;
				});

				// 3) refresh unloaded scripts (every 10 seconds)
				if (m_LastRefreshedUnloadedScripts + 10s < std::chrono::system_clock::now())
				{
					m_UnloadedScripts.clear();
					for (const auto& entry : std::filesystem::directory_iterator(scripts_dir.Path()))
					{
						if (entry.path().extension() != ".lua")
							continue;

						for (auto& script : m_LoadedScripts)
							if (std::filesystem::equivalent(script->GetPath(), entry.path().string()))
								continue;

						AddUnloadedScript(entry.path().filename().string(), entry.path().string());
					}
					m_LastRefreshedUnloadedScripts = std::chrono::system_clock::now();
				}
			}

			// 4) run tick coroutines
			for (auto& script : m_LoadedScripts)
				if (script->IsValid())
					script->Tick();

			ScriptMgr::Yield();
		}
	}

	bool LuaManager::IsRunningInMainThreadImpl()
	{
		return GetCurrentThreadId() == m_MainThreadId;
	}

	void LuaManager::SetRunningCoroutineImpl(lua_State* script)
	{
		if (!m_RunningCoroutine || !script)
		{
			m_RunningCoroutine = script;
		}
		else
		{
			LOGF(FATAL, "LuaManager::SetRunningCoroutineImpl: {} attempted to enter a coroutine when a coroutine from {} is already running", LuaScript::GetScript(script).GetName(), LuaScript::GetScript(m_RunningCoroutine).GetName());
			LuaScript::GetScript(script).SetMalfunctioning();
		}
	}

	void LuaManager::ForAllLoadedScriptsImpl(ForAllLoadedScriptsCallback callback)
	{
		std::lock_guard lock(m_LoadMutex);
		for (auto& script : m_LoadedScripts)
			callback(script);
	}

	void LuaManager::ForAllUnloadedScriptsImpl(ForAllUnloadedScriptsCallback callback)
	{
		std::lock_guard lock(m_LoadMutex);
		for (auto& script : m_UnloadedScripts)
			callback(script);
	}
}