From 56b5d346d4bcf9669d528bbf95a9b0fdda97f38a Mon Sep 17 00:00:00 2001
From: WantedDV <122710241+WantedDV@users.noreply.github.com>
Date: Tue, 11 Apr 2023 17:27:59 -0230
Subject: [PATCH] Move Datasource.StartMenuGameOptions

---
 .../DataSources_StartMenuGameOptions.lua      | 74 ++++++++++++++++++
 data/ui_scripts/party/__init__.lua            | 76 +------------------
 2 files changed, 75 insertions(+), 75 deletions(-)
 create mode 100644 data/ui_scripts/party/DataSources_StartMenuGameOptions.lua

diff --git a/data/ui_scripts/party/DataSources_StartMenuGameOptions.lua b/data/ui_scripts/party/DataSources_StartMenuGameOptions.lua
new file mode 100644
index 00000000..522b21f3
--- /dev/null
+++ b/data/ui_scripts/party/DataSources_StartMenuGameOptions.lua
@@ -0,0 +1,74 @@
+DataSources.StartMenuGameOptions = ListHelper_SetupDataSource("StartMenuGameOptions", function (controller)
+	local options = {}
+	if Engine.IsDemoPlaying() then
+		if not IsDemoRestrictedBasicMode() then
+			table.insert(options, {models = {displayText = Engine.ToUpper(Engine.Localize("MENU_UPLOAD_CLIP", Engine.GetDemoSegmentCount())), action = StartMenuUploadClip, disabledFunction = IsUploadClipButtonDisabled}, properties = {hideHelpItemLabel = true}})
+		end
+		if Engine.IsDemoHighlightReelMode() then
+			table.insert(options, {models = {displayText = Engine.ToUpper(Engine.Localize("MENU_DEMO_CUSTOMIZE_HIGHLIGHT_REEL")), action = StartMenuOpenCustomizeHighlightReel, disabledFunction = IsCustomizeHighlightReelButtonDisabled}})
+		end
+		table.insert(options, {models = {displayText = Engine.ToUpper(Engine.ToUpper(Engine.Localize("MENU_JUMP_TO_START"))), action = StartMenuJumpToStart, disabledFunction = IsJumpToStartButtonDisabled}, properties = {hideHelpItemLabel = true}})
+		local endDemoButtonText = nil
+		if Engine.IsDemoClipPlaying() then
+			endDemoButtonText = Engine.ToUpper(Engine.Localize("MENU_END_CLIP"))
+		else
+			endDemoButtonText = Engine.ToUpper(Engine.Localize("MENU_END_FILM"))
+		end
+		table.insert(options, {models = {displayText = Engine.ToUpper(endDemoButtonText), action = StartMenuEndDemo}})
+	elseif CoD.isCampaign then
+		table.insert(options, {models = {displayText = "MENU_RESUMEGAME_CAPS", action = StartMenuGoBack_ListElement}})
+		local inTrainingSim = CoD.SafeGetModelValue(Engine.GetModelForController(controller), "safehouse.inTrainingSim")
+		if not inTrainingSim then
+			inTrainingSim = 0
+		end
+		if Engine.IsLobbyHost(Enum.LobbyType.LOBBY_TYPE_GAME) then
+			if not CoD.isSafehouse and controller == Engine.GetPrimaryController() then
+				table.insert(options, {models = {displayText = "MENU_RESTART_MISSION_CAPS", action = RestartMission}})
+				if LUI.DEV ~= nil then
+					table.insert(options, {models = {displayText = "MENU_RESTART_CHECKPOINT_CAPS", action = RestartFromCheckpoint}})
+				end
+			end
+			if controller == Engine.GetPrimaryController() then
+				table.insert(options, {models = {displayText = "MENU_CHANGE_DIFFICULTY_CAPS", action = OpenDifficultySelect}})
+			end
+			if CoD.isSafehouse and inTrainingSim == 1 then
+				table.insert(options, {models = {displayText = "MENU_END_TRAINING_SIM", action = EndTrainingSim}})
+			elseif controller == Engine.GetPrimaryController() then
+				if Engine.DvarBool(0, "ui_blocksaves") then
+					table.insert(options, {models = {displayText = "MENU_EXIT_CAPS", action = SaveAndQuitGame}})
+				else
+					table.insert(options, {models = {displayText = "MENU_SAVE_AND_QUIT_CAPS", action = SaveAndQuitGame}})
+				end
+			end
+		elseif CoD.isSafehouse and inTrainingSim == 1 then
+			table.insert(options, {models = {displayText = "MENU_END_TRAINING_SIM", action = EndTrainingSim}})
+		else
+			table.insert(options, {models = {displayText = "MENU_LEAVE_PARTY_AND_EXIT_CAPS", action = QuitGame}})
+		end
+	elseif CoD.isMultiplayer then
+		if Engine.Team(controller, "name") ~= "TEAM_SPECTATOR" and Engine.GetGametypeSetting("disableClassSelection") ~= 1 then
+			table.insert(options, {models = {displayText = "MPUI_CHOOSE_CLASS_BUTTON_CAPS", action = ChooseClass}})
+		end
+		if not Engine.IsVisibilityBitSet(controller, Enum.UIVisibilityBit.BIT_ROUND_END_KILLCAM) and not Engine.IsVisibilityBitSet(controller, Enum.UIVisibilityBit.BIT_FINAL_KILLCAM) and CoD.IsTeamChangeAllowed() then
+			table.insert(options, {models = {displayText = "MPUI_CHANGE_TEAM_BUTTON_CAPS", action = ChooseTeam}})
+		end
+		if controller == 0 then
+			local endGameText = "MENU_QUIT_GAME_CAPS"
+			if Engine.IsLobbyHost(Enum.LobbyType.LOBBY_TYPE_GAME) and not CoD.isOnlineGame() then
+				endGameText = "MENU_END_GAME_CAPS"
+			end
+			table.insert(options, {models = {displayText = endGameText, action = QuitGame_MP}})
+		end
+	elseif CoD.isZombie then
+		table.insert(options, {models = {displayText = "MENU_RESUMEGAME_CAPS", action = StartMenuGoBack_ListElement}})
+		if Engine.IsLobbyHost(Enum.LobbyType.LOBBY_TYPE_GAME) and (not Engine.SessionModeIsMode(CoD.SESSIONMODE_SYSTEMLINK) or Engine.SessionModeIsMode(CoD.SESSIONMODE_OFFLINE)) then
+			table.insert(options, {models = {displayText = "MENU_RESTART_LEVEL_CAPS", action = RestartGame}})
+		end
+		if Engine.IsLobbyHost(Enum.LobbyType.LOBBY_TYPE_GAME) == true then
+			table.insert(options, {models = {displayText = "MENU_END_GAME_CAPS", action = QuitGame_MP}})
+		else
+			table.insert(options, {models = {displayText = "MENU_QUIT_GAME_CAPS", action = QuitGame_MP}})
+		end
+	end
+	return options
+end, true)
diff --git a/data/ui_scripts/party/__init__.lua b/data/ui_scripts/party/__init__.lua
index 7702e6df..2030b7bc 100644
--- a/data/ui_scripts/party/__init__.lua
+++ b/data/ui_scripts/party/__init__.lua
@@ -18,78 +18,4 @@ end
 
 Engine.SetDvar( "com_maxclients", maxclients() )
 
-
-DataSources.StartMenuGameOptions = ListHelper_SetupDataSource("StartMenuGameOptions", function (controller)
-	local options = {}
-	if Engine.IsDemoPlaying() then
-		if not IsDemoRestrictedBasicMode() then
-			table.insert(options, {models = {displayText = Engine.ToUpper(Engine.Localize("MENU_UPLOAD_CLIP", Engine.GetDemoSegmentCount())), action = StartMenuUploadClip, disabledFunction = IsUploadClipButtonDisabled}, properties = {hideHelpItemLabel = true}})
-		end
-		if Engine.IsDemoHighlightReelMode() then
-			table.insert(options, {models = {displayText = Engine.ToUpper(Engine.Localize("MENU_DEMO_CUSTOMIZE_HIGHLIGHT_REEL")), action = StartMenuOpenCustomizeHighlightReel, disabledFunction = IsCustomizeHighlightReelButtonDisabled}})
-		end
-		table.insert(options, {models = {displayText = Engine.ToUpper(Engine.ToUpper(Engine.Localize("MENU_JUMP_TO_START"))), action = StartMenuJumpToStart, disabledFunction = IsJumpToStartButtonDisabled}, properties = {hideHelpItemLabel = true}})
-		local endDemoButtonText = nil
-		if Engine.IsDemoClipPlaying() then
-			endDemoButtonText = Engine.ToUpper(Engine.Localize("MENU_END_CLIP"))
-		else
-			endDemoButtonText = Engine.ToUpper(Engine.Localize("MENU_END_FILM"))
-		end
-		table.insert(options, {models = {displayText = Engine.ToUpper(endDemoButtonText), action = StartMenuEndDemo}})
-	elseif CoD.isCampaign then
-		table.insert(options, {models = {displayText = "MENU_RESUMEGAME_CAPS", action = StartMenuGoBack_ListElement}})
-		local inTrainingSim = CoD.SafeGetModelValue(Engine.GetModelForController(controller), "safehouse.inTrainingSim")
-		if not inTrainingSim then
-			inTrainingSim = 0
-		end
-		if Engine.IsLobbyHost(Enum.LobbyType.LOBBY_TYPE_GAME) then
-			if not CoD.isSafehouse and controller == Engine.GetPrimaryController() then
-				table.insert(options, {models = {displayText = "MENU_RESTART_MISSION_CAPS", action = RestartMission}})
-				if LUI.DEV ~= nil then
-					table.insert(options, {models = {displayText = "MENU_RESTART_CHECKPOINT_CAPS", action = RestartFromCheckpoint}})
-				end
-			end
-			if controller == Engine.GetPrimaryController() then
-				table.insert(options, {models = {displayText = "MENU_CHANGE_DIFFICULTY_CAPS", action = OpenDifficultySelect}})
-			end
-			if CoD.isSafehouse and inTrainingSim == 1 then
-				table.insert(options, {models = {displayText = "MENU_END_TRAINING_SIM", action = EndTrainingSim}})
-			elseif controller == Engine.GetPrimaryController() then
-				if Engine.DvarBool(0, "ui_blocksaves") then
-					table.insert(options, {models = {displayText = "MENU_EXIT_CAPS", action = SaveAndQuitGame}})
-				else
-					table.insert(options, {models = {displayText = "MENU_SAVE_AND_QUIT_CAPS", action = SaveAndQuitGame}})
-				end
-			end
-		elseif CoD.isSafehouse and inTrainingSim == 1 then
-			table.insert(options, {models = {displayText = "MENU_END_TRAINING_SIM", action = EndTrainingSim}})
-		else
-			table.insert(options, {models = {displayText = "MENU_LEAVE_PARTY_AND_EXIT_CAPS", action = QuitGame}})
-		end
-	elseif CoD.isMultiplayer then
-		if Engine.Team(controller, "name") ~= "TEAM_SPECTATOR" and Engine.GetGametypeSetting("disableClassSelection") ~= 1 then
-			table.insert(options, {models = {displayText = "MPUI_CHOOSE_CLASS_BUTTON_CAPS", action = ChooseClass}})
-		end
-		if not Engine.IsVisibilityBitSet(controller, Enum.UIVisibilityBit.BIT_ROUND_END_KILLCAM) and not Engine.IsVisibilityBitSet(controller, Enum.UIVisibilityBit.BIT_FINAL_KILLCAM) and CoD.IsTeamChangeAllowed() then
-			table.insert(options, {models = {displayText = "MPUI_CHANGE_TEAM_BUTTON_CAPS", action = ChooseTeam}})
-		end
-		if controller == 0 then
-			local endGameText = "MENU_QUIT_GAME_CAPS"
-			if Engine.IsLobbyHost(Enum.LobbyType.LOBBY_TYPE_GAME) and not CoD.isOnlineGame() then
-				endGameText = "MENU_END_GAME_CAPS"
-			end
-			table.insert(options, {models = {displayText = endGameText, action = QuitGame_MP}})
-		end
-	elseif CoD.isZombie then
-		table.insert(options, {models = {displayText = "MENU_RESUMEGAME_CAPS", action = StartMenuGoBack_ListElement}})
-		if Engine.IsLobbyHost(Enum.LobbyType.LOBBY_TYPE_GAME) and (not Engine.SessionModeIsMode(CoD.SESSIONMODE_SYSTEMLINK) or Engine.SessionModeIsMode(CoD.SESSIONMODE_OFFLINE)) then
-			table.insert(options, {models = {displayText = "MENU_RESTART_LEVEL_CAPS", action = RestartGame}})
-		end
-		if Engine.IsLobbyHost(Enum.LobbyType.LOBBY_TYPE_GAME) == true then
-			table.insert(options, {models = {displayText = "MENU_END_GAME_CAPS", action = QuitGame_MP}})
-		else
-			table.insert(options, {models = {displayText = "MENU_QUIT_GAME_CAPS", action = QuitGame_MP}})
-		end
-	end
-	return options
-end, true)
+require("DataSources_StartMenuGameOptions")