TmpMenu/src/lua/lua_module.hpp

#pragma once
#include "bindings/gui/gui_element.hpp"
#include "core/data/menu_event.hpp"
#include "lua_patch.hpp"
#include "sol.hpp"
#include "../script.hpp"
#include <services/gui/gui_service.hpp>

namespace big
{
	class lua_module
	{
		sol::state m_state;

		std::filesystem::path m_module_path;

		std::string m_module_name;
		rage::joaat_t m_module_id;

		std::chrono::time_point<std::chrono::file_clock> m_last_write_time;

		std::mutex m_registered_scripts_mutex;

	public:
		std::vector<std::unique_ptr<script>> m_registered_scripts;
		std::vector<std::unique_ptr<lua_patch>> m_registered_patches;

		std::vector<big::tabs> m_owned_tabs;

		std::unordered_map<big::tabs, std::vector<big::tabs>> m_tab_to_sub_tabs;

		std::vector<std::unique_ptr<lua::gui::gui_element>> m_independent_gui;
		std::unordered_map<rage::joaat_t, std::vector<std::unique_ptr<lua::gui::gui_element>>> m_gui;
		std::unordered_map<menu_event, std::vector<sol::protected_function>> m_event_callbacks;
		std::vector<void*> m_allocated_memory;

		lua_module(const std::filesystem::path& module_path, folder& scripts_folder);
		~lua_module();

		const std::filesystem::path& module_path() const;

		rage::joaat_t module_id() const;
		const std::string& module_name() const;
		const std::chrono::time_point<std::chrono::file_clock> last_write_time() const;

		// used for sandboxing and limiting to only our custom search path for the lua require function
		void set_folder_for_lua_require(folder& scripts_folder);

		void sandbox_lua_os_library();
		void sandbox_lua_loads(folder& scripts_folder);

		void init_lua_api(folder& scripts_folder);

		void load_and_call_script();

		inline void for_each_script(auto func)
		{
			std::lock_guard guard(m_registered_scripts_mutex);

			for (auto& script : m_registered_scripts)
			{
				func(script.get());
			}
		}

		void tick_scripts();
		void cleanup_done_scripts();
	};
}