AmongUsRevamped/Patches/GameStartManagerPatch.cs

using System;
using TMPro;
using UnityEngine;

namespace AmongUsRevamped;

[HarmonyPatch(typeof(GameStartManager), nameof(GameStartManager.Update))]
public static class GameStartManagerUpdatePatch
{
    public static bool CustomTimerApplied;
    public static bool Autostarting;
    public static bool CountingDown;

    public static void Prefix(GameStartManager __instance)
    {
        if (__instance == null || MeetingHud.Instance != null) return;
        if (AmongUsClient.Instance == null || !AmongUsClient.Instance.AmHost) return;

        if (__instance.StartButton == null || __instance.GameStartText == null) return;

        __instance.MinPlayers = 1;

        if (!Utils.CheckValidData())
        {
            __instance.StartButton.gameObject.SetActive(false);
        }
        else if (__instance.startState != GameStartManager.StartingStates.Countdown)
        {
           __instance.StartButton.gameObject.SetActive(true); 
        }

        if (Main.AutoStart.Value && OnGameJoinedPatch.AutoStartCheck && GameStartManager.InstanceExists && __instance.startState != GameStartManager.StartingStates.Countdown && GameData.Instance?.PlayerCount >= Options.PlayerAutoStart.GetInt())
        {
            __instance.startState = GameStartManager.StartingStates.Countdown;
            __instance.countDownTimer = Options.AutoStartTimer.GetFloat();
            __instance.StartButton.gameObject.SetActive(false);
            Autostarting = true;
            return;
        }

        if (__instance.startState == GameStartManager.StartingStates.Countdown)
        {
            CountingDown = true;
            if (!CustomTimerApplied && !Autostarting)
            {
                __instance.countDownTimer = Options.StartCountdown.GetInt();
                CustomTimerApplied = true;
            }
        }
        else CountingDown = false;

        if (__instance.startState != GameStartManager.StartingStates.Countdown)
        {
            Autostarting = false;
            CustomTimerApplied = false;
        }
    }

    public static void Postfix(GameStartManager __instance)
    {
        string warningMessage = "";

        if (__instance == null || MeetingHud.Instance != null) return;
        if (AmongUsClient.Instance == null || !AmongUsClient.Instance.AmHost) return;

        if (__instance.StartButton == null || __instance.GameStartText == null) return;

        if (GameStartManagerStartPatch.warningText != null)
        {
            if (warningMessage == "")
            {
                GameStartManagerStartPatch.warningText.gameObject.SetActive(false);
            }
            else
            {
                GameStartManagerStartPatch.warningText.text = warningMessage;
                GameStartManagerStartPatch.warningText.gameObject.SetActive(true);
            }
        }

        if (__instance.GameStartText != null)
        {
            __instance.GameStartText.transform.localPosition = new Vector3(__instance.GameStartText.transform.localPosition.x, 2f, __instance.GameStartText.transform.localPosition.z);
        }

        if (GameStartManagerStartPatch.cancelButton != null)
        {
            GameStartManagerStartPatch.cancelButton.gameObject.SetActive(__instance.startState == GameStartManager.StartingStates.Countdown);
        }
    }
}

[HarmonyPatch(typeof(GameStartManager), nameof(GameStartManager.Start))]
public class GameStartManagerStartPatch
{
    public static TextMeshPro warningText;
    public static PassiveButton cancelButton;

    public static void Postfix(GameStartManager __instance)
    {
        if (__instance == null || !Utils.IsLobby) return;

        warningText = UnityEngine.Object.Instantiate(__instance.GameStartText, __instance.transform.parent);
        warningText.name = "WarningText";
        warningText.transform.localPosition = new Vector3(0f, __instance.transform.localPosition.y + 3f, -1f);
        warningText.gameObject.SetActive(false);

        cancelButton = UnityEngine.Object.Instantiate(__instance.StartButton, __instance.transform);
        cancelButton.name = "CancelButton";

        var cancelLabel = cancelButton.buttonText;
        cancelLabel.DestroyTranslator();
        cancelLabel.text = "Cancel";

        cancelButton.activeTextColor = cancelButton.inactiveTextColor = Color.white;
        cancelButton.gameObject.SetActive(false);

        cancelButton.OnClick = new();
        cancelButton.OnClick.AddListener((UnityEngine.Events.UnityAction)(() =>
        {
            __instance.ResetStartState();
        }));
    }
}

[HarmonyPatch(typeof(GameStartManager), nameof(GameStartManager.BeginGame))]
public class GameStartManagerBeginGamePatch
{
    public static void Prefix(GameStartManager __instance)
    {
        if (Options.NoGameEnd.GetBool()) Logger.SendInGame(Translator.Get("noGameEndWarning"));
        if (Main.GM.Value) Logger.SendInGame(Translator.Get("gameMasterWarningTwo"));
    }
}