quantum-space-buddies/QSB/Menus/MenuManager.cs
2022-04-30 15:33:55 +01:00

415 lines
12 KiB
C#

using EpicTransport;
using Mirror;
using QSB.Messaging;
using QSB.Player.TransformSync;
using QSB.SaveSync.Messages;
using QSB.Utility;
using System;
using System.Text;
using UnityEngine;
using UnityEngine.UI;
namespace QSB.Menus;
internal class MenuManager : MonoBehaviour, IAddComponentOnStart
{
public static MenuManager Instance;
private PopupMenu OneButtonInfoPopup;
private PopupMenu TwoButtonInfoPopup;
private bool _addedPauseLock;
// Pause menu only
private GameObject QuitButton;
private GameObject DisconnectButton;
private PopupMenu DisconnectPopup;
// title screen only
private GameObject ResumeGameButton;
private GameObject NewGameButton;
private Button HostButton;
private GameObject ConnectButton;
private PopupInputMenu ConnectPopup;
private Text _loadingText;
private StringBuilder _nowLoadingSB;
private const int _titleButtonIndex = 2;
private float _connectPopupOpenTime;
private const string HostString = "OPEN TO MULTIPLAYER";
private const string ConnectString = "CONNECT TO MULTIPLAYER";
private const string DisconnectString = "DISCONNECT";
private const string StopHostingString = "STOP HOSTING";
private const string UpdateChangelog = "QSB v0.18.0\r\nit's good";
private Action<bool> PopupClose;
private bool _intentionalDisconnect;
public void Start()
{
Instance = this;
MakeTitleMenus();
QSBSceneManager.OnSceneLoaded += OnSceneLoaded;
QSBNetworkManager.singleton.OnClientConnected += OnConnected;
QSBNetworkManager.singleton.OnClientDisconnected += OnDisconnected;
if (QSBCore.Storage.LastUsedVersion != QSBCore.QSBVersion)
{
// recently updated!
QSBCore.Storage.LastUsedVersion = QSBCore.QSBVersion;
QSBCore.Helper.Storage.Save(QSBCore.Storage, "storage.json");
QSBCore.MenuApi.RegisterStartupPopup(UpdateChangelog);
}
}
private void OnSceneLoaded(OWScene oldScene, OWScene newScene, bool isUniverse)
{
if (isUniverse)
{
// wait a frame or else the changes won't actually happen
Delay.RunNextFrame(InitPauseMenus);
return;
}
if (newScene == OWScene.TitleScreen)
{
// wait a frame or else the changes won't actually happen
Delay.RunNextFrame(MakeTitleMenus);
}
}
private void ResetStringBuilder()
{
if (_nowLoadingSB == null)
{
_nowLoadingSB = new StringBuilder();
return;
}
_nowLoadingSB.Length = 0;
}
private void Update()
{
if ((LoadManager.GetLoadingScene() == OWScene.SolarSystem || LoadManager.GetLoadingScene() == OWScene.EyeOfTheUniverse)
&& _loadingText != null)
{
var num = LoadManager.GetAsyncLoadProgress();
num = num < 0.1f
? Mathf.InverseLerp(0f, 0.1f, num) * 0.9f
: 0.9f + Mathf.InverseLerp(0.1f, 1f, num) * 0.1f;
ResetStringBuilder();
_nowLoadingSB.Append(UITextLibrary.GetString(UITextType.LoadingMessage));
_nowLoadingSB.Append(num.ToString("P0"));
_loadingText.text = _nowLoadingSB.ToString();
}
}
public void LoadGame(bool inEye)
{
var sceneToLoad = inEye ? OWScene.EyeOfTheUniverse : OWScene.SolarSystem;
LoadManager.LoadSceneAsync(sceneToLoad, true, LoadManager.FadeType.ToBlack, 1f, false);
Locator.GetMenuInputModule().DisableInputs();
}
private void OpenInfoPopup(string message, string okButtonText)
{
OneButtonInfoPopup.SetUpPopup(message, InputLibrary.menuConfirm, InputLibrary.cancel, new ScreenPrompt(okButtonText), null, true, false);
OWTime.Pause(OWTime.PauseType.Menu);
OWInput.ChangeInputMode(InputMode.Menu);
var pauseCommandListener = Locator.GetPauseCommandListener();
if (pauseCommandListener != null)
{
pauseCommandListener.AddPauseCommandLock();
_addedPauseLock = true;
}
OneButtonInfoPopup.EnableMenu(true);
}
private void OpenInfoPopup(string message, string okButtonText, string cancelButtonText)
{
TwoButtonInfoPopup.SetUpPopup(message, InputLibrary.menuConfirm, InputLibrary.cancel, new ScreenPrompt(okButtonText), new ScreenPrompt(cancelButtonText));
OWTime.Pause(OWTime.PauseType.Menu);
OWInput.ChangeInputMode(InputMode.Menu);
var pauseCommandListener = Locator.GetPauseCommandListener();
if (pauseCommandListener != null)
{
pauseCommandListener.AddPauseCommandLock();
_addedPauseLock = true;
}
TwoButtonInfoPopup.EnableMenu(true);
}
private void OnCloseInfoPopup(bool confirm)
{
var pauseCommandListener = Locator.GetPauseCommandListener();
if (pauseCommandListener != null && _addedPauseLock)
{
pauseCommandListener.RemovePauseCommandLock();
_addedPauseLock = false;
}
OWTime.Unpause(OWTime.PauseType.Menu);
OWInput.RestorePreviousInputs();
PopupClose?.SafeInvoke(confirm);
PopupClose = null;
}
private void CreateCommonPopups()
{
var text = QSBCore.DebugSettings.UseKcpTransport ? "Public IP Address" : "Product User ID";
ConnectPopup = QSBCore.MenuApi.MakeInputFieldPopup(text, text, "Connect", "Cancel");
ConnectPopup.CloseMenuOnOk(false);
ConnectPopup.OnPopupConfirm += () =>
{
// fixes dumb thing with using keyboard to open popup
if (OWMath.ApproxEquals(Time.time, _connectPopupOpenTime))
{
return;
}
ConnectPopup.EnableMenu(false);
Connect();
};
ConnectPopup.OnActivateMenu += () => _connectPopupOpenTime = Time.time;
OneButtonInfoPopup = QSBCore.MenuApi.MakeInfoPopup("", "");
OneButtonInfoPopup.OnPopupConfirm += () => OnCloseInfoPopup(true);
TwoButtonInfoPopup = QSBCore.MenuApi.MakeTwoChoicePopup("", "", "");
TwoButtonInfoPopup.OnPopupConfirm += () => OnCloseInfoPopup(true);
TwoButtonInfoPopup.OnPopupCancel += () => OnCloseInfoPopup(false);
}
private static void SetButtonActive(Button button, bool active)
=> SetButtonActive(button ? button.gameObject : null, active);
private static void SetButtonActive(GameObject button, bool active)
{
if (button == null)
{
DebugLog.DebugWrite($"Warning - Tried to set button to {active}, but it was null.", OWML.Common.MessageType.Warning);
return;
}
button.SetActive(active);
button.GetComponent<CanvasGroup>().alpha = active ? 1 : 0;
}
private void InitPauseMenus()
{
CreateCommonPopups();
DisconnectPopup = QSBCore.MenuApi.MakeTwoChoicePopup("Are you sure you want to disconnect?\r\nThis will send you back to the main menu.", "YES", "NO");
DisconnectPopup.OnPopupConfirm += Disconnect;
DisconnectButton = QSBCore.MenuApi.PauseMenu_MakeMenuOpenButton(DisconnectString, DisconnectPopup);
QuitButton = FindObjectOfType<PauseMenuManager>()._exitToMainMenuAction.gameObject;
if (QSBCore.IsInMultiplayer)
{
SetButtonActive(DisconnectButton, true);
SetButtonActive(QuitButton, false);
}
else
{
SetButtonActive(DisconnectButton, false);
SetButtonActive(QuitButton, true);
}
var text = QSBCore.IsHost
? StopHostingString
: DisconnectString;
DisconnectButton.transform.GetChild(0).GetChild(1).GetComponent<Text>().text = text;
var popupText = QSBCore.IsHost
? "Are you sure you want to stop hosting?\r\nThis will disconnect all clients and send everyone back to the main menu."
: "Are you sure you want to disconnect?\r\nThis will send you back to the main menu.";
DisconnectPopup._labelText.text = popupText;
}
private void MakeTitleMenus()
{
CreateCommonPopups();
HostButton = QSBCore.MenuApi.TitleScreen_MakeSimpleButton(HostString, _titleButtonIndex);
HostButton.onClick.AddListener(Host);
ConnectButton = QSBCore.MenuApi.TitleScreen_MakeMenuOpenButton(ConnectString, _titleButtonIndex + 1, ConnectPopup);
ResumeGameButton = GameObject.Find("MainMenuLayoutGroup/Button-ResumeGame");
NewGameButton = GameObject.Find("MainMenuLayoutGroup/Button-NewGame");
SetButtonActive(ConnectButton, true);
Delay.RunWhen(PlayerData.IsLoaded, () => SetButtonActive(ResumeGameButton, PlayerData.LoadLoopCount() > 1));
SetButtonActive(NewGameButton, true);
if (QSBCore.DebugSettings.SkipTitleScreen)
{
Application.runInBackground = true;
var titleScreenManager = FindObjectOfType<TitleScreenManager>();
var titleScreenAnimation = titleScreenManager._cameraController;
const float small = 1 / 1000f;
titleScreenAnimation._gamepadSplash = false;
titleScreenAnimation._introPan = false;
titleScreenAnimation._fadeDuration = small;
titleScreenAnimation.Start();
var titleAnimationController = titleScreenManager._gfxController;
titleAnimationController._logoFadeDelay = small;
titleAnimationController._logoFadeDuration = small;
titleAnimationController._echoesFadeDelay = small;
titleAnimationController._optionsFadeDelay = small;
titleAnimationController._optionsFadeDuration = small;
titleAnimationController._optionsFadeSpacing = small;
}
}
private void Disconnect()
{
_intentionalDisconnect = true;
QSBNetworkManager.singleton.StopHost();
SetButtonActive(DisconnectButton, false);
Locator.GetSceneMenuManager().pauseMenu._pauseMenu.EnableMenu(false);
Locator.GetSceneMenuManager().pauseMenu._isPaused = false;
OWInput.RestorePreviousInputs();
LoadManager.LoadScene(OWScene.TitleScreen, LoadManager.FadeType.ToBlack, 2f);
}
private void Host()
{
_intentionalDisconnect = false;
SetButtonActive(ConnectButton, false);
SetButtonActive(ResumeGameButton, false);
SetButtonActive(NewGameButton, false);
_loadingText = HostButton.transform.GetChild(0).GetChild(1).GetComponent<Text>();
if (!QSBCore.DebugSettings.UseKcpTransport)
{
var productUserId = EOSSDKComponent.LocalUserProductIdString;
PopupClose += confirm =>
{
if (confirm)
{
GUIUtility.systemCopyBuffer = productUserId;
}
LoadGame(PlayerData.GetWarpedToTheEye());
Delay.RunWhen(() => TimeLoop._initialized, QSBNetworkManager.singleton.StartHost);
};
OpenInfoPopup("Hosting server.\r\nClients will connect using your product user id, which is :\r\n" +
$"{productUserId}\r\n" +
"Do you want to copy this to the clipboard?"
, "YES"
, "NO");
}
else
{
LoadGame(PlayerData.GetWarpedToTheEye());
Delay.RunWhen(() => TimeLoop._initialized, QSBNetworkManager.singleton.StartHost);
}
}
private void Connect()
{
_intentionalDisconnect = false;
var address = ConnectPopup.GetInputText();
if (address == string.Empty)
{
address = QSBCore.DefaultServerIP;
}
SetButtonActive(HostButton, false);
SetButtonActive(ResumeGameButton, false);
SetButtonActive(NewGameButton, false);
_loadingText = ConnectButton.transform.GetChild(0).GetChild(1).GetComponent<Text>();
_loadingText.text = "CONNECTING...";
Locator.GetMenuInputModule().DisableInputs();
QSBNetworkManager.singleton.networkAddress = address;
// hack to get disconnect call if start client fails immediately
typeof(NetworkClient).GetProperty(nameof(NetworkClient.connection))!.SetValue(null, new NetworkConnectionToServer());
QSBNetworkManager.singleton.StartClient();
}
private static void OnConnected()
{
if (!QSBCore.IsHost)
{
Delay.RunWhen(() => PlayerTransformSync.LocalInstance,
() => new RequestGameStateMessage().Send());
}
}
public void OnKicked(string reason)
{
_intentionalDisconnect = true;
PopupClose += _ =>
{
if (QSBSceneManager.IsInUniverse)
{
LoadManager.LoadScene(OWScene.TitleScreen, LoadManager.FadeType.ToBlack, 2f);
}
};
OpenInfoPopup($"Server refused connection.\r\n{reason}", "OK");
}
private void OnDisconnected(string error)
{
if (_intentionalDisconnect)
{
DebugLog.DebugWrite("intentional disconnect. dont show popup");
_intentionalDisconnect = false;
}
else
{
PopupClose += _ =>
{
if (QSBSceneManager.IsInUniverse)
{
LoadManager.LoadScene(OWScene.TitleScreen, LoadManager.FadeType.ToBlack, 2f);
}
};
OpenInfoPopup($"Client disconnected with error!\r\n{error}", "OK");
}
SetButtonActive(DisconnectButton, false);
SetButtonActive(ConnectButton, true);
SetButtonActive(QuitButton, true);
SetButtonActive(HostButton, true);
SetButtonActive(ResumeGameButton, PlayerData.LoadLoopCount() > 1);
SetButtonActive(NewGameButton, true);
if (ConnectButton)
{
ConnectButton.transform.GetChild(0).GetChild(1).GetComponent<Text>().text = ConnectString;
}
if (HostButton)
{
HostButton.transform.GetChild(0).GetChild(1).GetComponent<Text>().text = HostString;
}
_loadingText = null;
Locator.GetMenuInputModule().EnableInputs();
}
}