feat(debug): Add Scripts Tab (#243)

2026-10-10 20:03:17 +08:00 · 2025-04-28 23:27:13 +03:00
parent ff4164ed26
commit a7b861e4d3
17 changed files with 1689 additions and 274 deletions
@@ -0,0 +1,174 @@
+#include "Scripts.hpp"
+#include "core/backend/FiberPool.hpp"
+#include "core/backend/ScriptMgr.hpp"
+#include "core/frontend/Notifications.hpp"
+#include "game/gta/data/ScriptNames.hpp"
+#include "game/gta/data/StackSizes.hpp"
+#include "game/gta/Scripts.hpp"
+#include "game/gta/Natives.hpp"
+#include "game/pointers/Pointers.hpp"
+#include "types/rage/atArray.hpp"
+#include "types/script/GtaThread.hpp"
+#include "types/script/CGameScriptHandlerNetComponent.hpp"
+#include "types/network/CNetGamePlayer.hpp"
+#include "misc/cpp/imgui_stdlib.h"
+
+namespace YimMenu::Submenus
+{
+	std::shared_ptr<Category> BuildScriptsMenu()
+	{
+		auto scripts = std::make_unique<Category>("Scripts");
+
+		auto threads = std::make_unique<Group>("Threads");
+		auto script  = std::make_unique<Group>("Start Script");
+
+		static rage::scrThread* curThread      = nullptr;
+		static std::string scriptSearch        = "";
+		static std::string scriptName          = "";
+		static std::string stackSizeName       = stackSizes[0].first;
+		static int stackSize                   = 0;
+
+		threads->AddItem(std::make_unique<ImGuiItem>([] {
+			if (!Pointers.ScriptThreads || Pointers.ScriptThreads->size() == 0)
+			{
+				curThread = nullptr;
+				return ImGui::TextDisabled("None");
+			}
+
+			if (ImGui::BeginCombo("Thread", curThread ? curThread->m_ScriptName : "(Select)"))
+			{
+				for (auto thread : *Pointers.ScriptThreads)
+				{
+					if (thread)
+					{
+						if (thread->m_Context.m_State == rage::scrThread::State::KILLED || thread->m_Context.m_StackSize == 0)
+							continue;
+
+						ImGui::PushID(thread->m_Context.m_ThreadId);
+
+						if (ImGui::Selectable(thread->m_ScriptName, curThread == thread))
+						{
+							curThread = thread;
+						}
+
+						if (curThread == thread)
+							ImGui::SetItemDefaultFocus();
+
+						ImGui::PopID();
+					}
+				}
+
+				ImGui::EndCombo();
+			}
+
+			if (!curThread)
+				return;
+
+			ImGui::Combo("State", (int*)&curThread->m_Context.m_State, "Idle\0Running\0Killed\0Paused\0Unk4\0");
+
+			if (auto netComponent = reinterpret_cast<GtaThread*>(curThread)->m_NetComponent)
+			{
+				if (auto host = netComponent->GetHost())
+				{
+					ImGui::Text("Host: %s", host->GetName());
+				}
+				ImGui::SameLine();
+				ImGui::BeginDisabled(netComponent->IsLocalPlayerHost());
+				if (ImGui::SmallButton("Take Control"))
+				{
+					FiberPool::Push([] {
+						Scripts::ForceScriptHost(curThread);
+					});
+				}
+				ImGui::EndDisabled();
+			}
+
+			ImGui::Text("Thread ID: %d", curThread->m_Context.m_ThreadId);
+			ImGui::Text("Stack Size: %d", curThread->m_Context.m_StackSize);
+			ImGui::Text("Stack Pointer: 0x%X", &curThread->m_Context.m_StackPointer);
+			ImGui::Text("Program Counter: 0x%X", &curThread->m_Context.m_ProgramCounter); // This is not really accurate (always points to the WAIT)
+			if (curThread->m_Context.m_State == rage::scrThread::State::KILLED)
+				ImGui::Text("Exit Reason: %s", curThread->m_ErrorMessage);
+		}));
+
+		script->AddItem(std::make_unique<ImGuiItem>([] {
+			ImGui::InputTextWithHint("Script Name", "Search", &scriptSearch);
+
+			if (ImGui::BeginCombo("Stack Size", stackSizeName.c_str()))
+			{
+				for (auto& p : stackSizes)
+				{
+					if (ImGui::Selectable(std::format("{} ({})", p.first, (int)p.second).data(), stackSize == p.second))
+					{
+						stackSizeName = p.first;
+						stackSize     = p.second;
+					}
+
+					if (p.second == stackSize)
+						ImGui::SetItemDefaultFocus();
+				}
+
+				ImGui::EndCombo();
+			}
+
+			if (ImGui::BeginListBox("##scripts"))
+			{
+				std::string lowerSearch = scriptSearch;
+				std::transform(lowerSearch.begin(), lowerSearch.end(), lowerSearch.begin(), ::tolower);
+				for (auto& script : scriptNames)
+				{
+					std::string lowerScript = script;
+					std::transform(lowerScript.begin(), lowerScript.end(), lowerScript.begin(), ::tolower);
+					if (lowerScript.find(lowerSearch) != std::string::npos)
+					{
+						if (ImGui::Selectable(script, scriptName == script))
+						{
+							scriptSearch = script;
+							scriptName   = script;
+						}
+					}
+				}
+
+				ImGui::EndListBox();
+			}
+
+			if (ImGui::Button("Start Script"))
+			{
+				FiberPool::Push([] {
+					auto hash = Joaat(scriptName);
+
+					if (!SCRIPT::DOES_SCRIPT_WITH_NAME_HASH_EXIST(hash))
+					{
+						Notifications::Show("Start Script", "Script does not exist.", NotificationType::Error);
+						return;
+					}
+
+					if (SCRIPT::GET_NUMBER_OF_THREADS_RUNNING_THE_SCRIPT_WITH_THIS_HASH(hash) > 0)
+					{
+						Notifications::Show("Start Script", "Script is already running.", NotificationType::Error);
+						return;
+					}
+
+					if (MISC::GET_NUMBER_OF_FREE_STACKS_OF_THIS_SIZE(stackSize) == 0)
+					{
+						Notifications::Show("Start Script", "No free stack.", NotificationType::Error);
+						return;
+					}
+
+					while (!SCRIPT::HAS_SCRIPT_WITH_NAME_HASH_LOADED(hash))
+					{
+						SCRIPT::REQUEST_SCRIPT_WITH_NAME_HASH(hash);
+						ScriptMgr::Yield();
+					}
+
+					BUILTIN::START_NEW_SCRIPT_WITH_NAME_HASH(hash, stackSize);
+					SCRIPT::SET_SCRIPT_WITH_NAME_HASH_AS_NO_LONGER_NEEDED(hash);
+				});
+			}
+		}));
+
+		scripts->AddItem(std::move(threads));
+		scripts->AddItem(std::move(script));
+		return scripts;
+	}
+}