From 496537d7b6993f8a64d3c0eff9632d3b817e2436 Mon Sep 17 00:00:00 2001
From: quaK <Joel.rautavuori@gmail.com>
Date: Sun, 4 Aug 2024 00:39:49 +0300
Subject: [PATCH] small improvements to ui

---
 data/cdata/ui_scripts/MainMenu/CPMainMenu.lua | 13 +++++++-
 .../SystemLinkMenu/ServerButtons.lua          |  2 ++
 .../SystemLinkMenu/SysLinkBrowser.lua         | 32 +++++--------------
 3 files changed, 22 insertions(+), 25 deletions(-)

diff --git a/data/cdata/ui_scripts/MainMenu/CPMainMenu.lua b/data/cdata/ui_scripts/MainMenu/CPMainMenu.lua
index 9e98133..59de819 100644
--- a/data/cdata/ui_scripts/MainMenu/CPMainMenu.lua
+++ b/data/cdata/ui_scripts/MainMenu/CPMainMenu.lua
@@ -5,7 +5,6 @@ local f0_local0 = function ( f1_arg0, f1_arg1 )
 end
 
 local f0_local1 = function ( f2_arg0, f2_arg1 )
-	print ("Are we running?!?")
 	f0_local0( f2_arg0, f2_arg1 )
 end
 
@@ -183,4 +182,16 @@ function CPMainMenuButtons( menu, controller )
 	return VNavigator
 end
 
+CPMainMenu_original = MenuBuilder.m_types["CPMainMenu"]
+function CPMainMenuStub( menu, controller )
+	ret = CPMainMenu_original( menu, controller )
+	
+	-- play music immediately
+	Engine.PlayMusic( CoD.Music.MainCPMusic )
+
+	return ret
+end
+
+MenuBuilder.m_types["CPMainMenu"] = CPMainMenuStub
+
 MenuBuilder.m_types["CPMainMenuButtons"] = CPMainMenuButtons
\ No newline at end of file
diff --git a/data/cdata/ui_scripts/SystemLinkMenu/ServerButtons.lua b/data/cdata/ui_scripts/SystemLinkMenu/ServerButtons.lua
index 7a3f37a..905303b 100644
--- a/data/cdata/ui_scripts/SystemLinkMenu/ServerButtons.lua
+++ b/data/cdata/ui_scripts/SystemLinkMenu/ServerButtons.lua
@@ -53,6 +53,8 @@ function SystemLinkMenuButtons( menu, controller )
 	if CONDITIONS.IsCoreMultiplayer( self ) then
 		f1_local5:addEventHandler( "button_action", function ( f2_arg0, f2_arg1 )
 			local f2_local0 = f2_arg1.controller or f1_local1
+			ClientWeapon.SetWeaponVisible( 0, true )
+			ClientCharacter.SetCharacterVisible( 0, true )
 			ACTIONS.OpenCreateAClass( self, f2_arg1 )
 		end )
 	end
diff --git a/data/cdata/ui_scripts/SystemLinkMenu/SysLinkBrowser.lua b/data/cdata/ui_scripts/SystemLinkMenu/SysLinkBrowser.lua
index 84685b2..14b3b42 100644
--- a/data/cdata/ui_scripts/SystemLinkMenu/SysLinkBrowser.lua
+++ b/data/cdata/ui_scripts/SystemLinkMenu/SysLinkBrowser.lua
@@ -7,10 +7,7 @@ serverBrowserOffsets = {
 }
 
 local f0_local4 = function ( f5_arg0, f5_arg1 )
-	Engine.ExecNow( "xstopprivateparty", f5_arg1.controller )
-	Engine.SetDvarBool( "systemlink", false )
-	Engine.SetDvarBool( "systemlink_host", false )
-	Engine.StopGameMode()
+	--Engine.StopGameMode()
 	LUI.FlowManager.RequestLeaveMenu( f5_arg0, true )
 end
 
@@ -58,9 +55,6 @@ local f0_local7 = function ( f11_arg0, f11_arg1, f11_arg2 )
 	else
 		Engine.ExecNow( MPConfig.default_systemlink, f11_arg1 )
 	end
-	if not Lobby.IsInPrivateParty() then
-		Engine.ExecNow( "xstartprivateparty", f11_arg1 )
-	end
 
 	Engine.Exec( MPConfig.default_dvars )
 
@@ -99,26 +93,11 @@ local f0_local7 = function ( f11_arg0, f11_arg1, f11_arg2 )
 			end
 		end
 	end )
-
 	
 	Engine.PLMRefreshData()
 	if Engine.IsCoreMode() then
-		FrontEndScene.CurrentMissionTeam = MissionDirector.InvalidTeamID
-		FrontEndScene.SetScene( "private_lobby_menu" )
-		ClientCharacter.SetCharacterVisible( FrontEndScene.ClientCharacters.Self, true )
-		if Fences.OfflineDataFetched( f11_arg1 ) then
-			local f11_local1 = CoD.GetPrivateLoadoutDataSource()
-			local f11_local2 = f11_local1.squadMembers.head:GetValue( f11_arg1 )
-			local f11_local3 = f11_local1.squadMembers.body:GetValue( f11_arg1 )
-			local f11_local4 = f11_local1.squadMembers.archetype:GetValue( f11_arg1 )
-			local f11_local5 = f11_local1.squadMembers.archetypeSuper:GetValue( f11_arg1 )
-			FrontEndScene.SetCharacterModelsByIndex( FrontEndScene.ClientCharacters.Self, f11_local3, f11_local2 )
-			FrontEndScene.SetWeaponForSuper( FrontEndScene.ClientCharacters.Self, f11_local5, f11_local4 )
-			FrontEndScene.PlayIdleForSuper( FrontEndScene.ClientCharacters.Self, f11_local5, f11_local4 )
-			ClientWeapon.SetWeaponIsViewModel( 0, true )
-			ClientWeapon.SetWeaponVisible( 0, false )
-			ClientCharacter.SetCharacterVisible( 0, true )
-		end
+		ClientWeapon.SetWeaponVisible( 0, false )
+		ClientCharacter.SetCharacterVisible( 0, false )
 	end
 	Lobby.SetPartyUIRoot( PartyUIRoot.SYSTEM_LINK )
 	local f11_local1 = LUI.FlowManager.GetScopedData( f11_arg0 )
@@ -844,5 +823,10 @@ MenuBuilder.m_types["SystemLinkMenu"] = function ( menu, controller )
 	return self
 end
 
+local f0_local3 = function ()
+	clearMatchData()
+	WipeGlobalModelsAtPath( f0_local0 )
+end
+
 LUI.FlowManager.RegisterStackPushBehaviour( "SystemLinkMenu", PushFunc )
 LUI.FlowManager.RegisterStackPopBehaviour( "SystemLinkMenu", f0_local3 )
\ No newline at end of file