Add lua getter for Game from string

etternagame · Apr 3, 2021 · 83572df · 83572df
1 parent 6d90f71
commit 83572df
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/Etterna/Singletons/GameManager.cpp b/src/Etterna/Singletons/GameManager.cpp
@@ -1824,6 +1824,16 @@ class LunaGameManager : public Luna<GameManager>
 		GameLoop::ChangeGame(game_name, theme);
 		return 0;
 	}
+	static int GetGameFromString(T* p, lua_State* L)
+	{
+		std::string game = SArg(1);
+		const auto* g = p->StringToGame(game);
+		if (g == nullptr)
+			luaL_error(L, "GetGameFromString: Invalid Game: %s", game.c_str());
+		else
+			const_cast<Game*>(g)->PushSelf(L);
+		return 1;
+	}
 
 	LunaGameManager()
 	{
@@ -1833,6 +1843,7 @@ class LunaGameManager : public Luna<GameManager>
 		ADD_METHOD(GetStylesForGame);
 		ADD_METHOD(GetEnabledGames);
 		ADD_METHOD(SetGame);
+		ADD_METHOD(GetGameFromString);
 	};
 };