NAK_CVR_Mods/MenuScalePatch/HarmonyPatches.cs

using ABI_RC.Core;
using ABI_RC.Core.InteractionSystem;
using ABI_RC.Core.Player;
using ABI_RC.Core.Savior;
using ABI_RC.Systems.InputManagement;
using ABI_RC.Systems.InputManagement.InputModules;
using HarmonyLib;
using NAK.MenuScalePatch.Helpers;
using UnityEngine;

namespace NAK.MenuScalePatch.HarmonyPatches;

/**
    ViewManager.SetScale runs once a second when it should only run when aspect ratio changes- CVR bug
    assuming its caused by cast from int to float getting the screen size, something floating point bleh
**/

[HarmonyPatch]
internal class HarmonyPatches
{
    //stuff needed on start
    [HarmonyPostfix]
    [HarmonyPatch(typeof(PlayerSetup), nameof(PlayerSetup.Start))]
    private static void Postfix_PlayerSetup_Start()
    {
        MSP_MenuInfo.CameraTransform = PlayerSetup.Instance.GetActiveCamera().transform;
        MenuScalePatch.UpdateSettings();
        QuickMenuHelper.Instance.CreateWorldAnchors();
        MainMenuHelper.Instance.CreateWorldAnchors();
    }

    [HarmonyPrefix]
    [HarmonyPatch(typeof(CVR_MenuManager), nameof(CVR_MenuManager.SetScale))]
    private static bool Prefix_CVR_MenuManager_SetScale(float avatarHeight, ref float ____scaleFactor)
    {
        ____scaleFactor = avatarHeight / 1.8f;
        if (MetaPort.Instance.isUsingVr) ____scaleFactor *= 0.5f;
        MSP_MenuInfo.ScaleFactor = ____scaleFactor;
        if (!MSP_MenuInfo.PlayerAnchorMenus)
        {
            QuickMenuHelper.Instance.NeedsPositionUpdate = true;
            MainMenuHelper.Instance.NeedsPositionUpdate = true;
        }
        return false;
    }

    [HarmonyPrefix]
    [HarmonyPatch(typeof(ViewManager), nameof(ViewManager.SetScale))]
    private static bool Prefix_ViewManager_SetScale()
    {
        return false;
    }

    //nuke UpdateMenuPosition methods
    //there are 2 Jobs calling this each second, which is fucking my shit
    [HarmonyPrefix]
    [HarmonyPatch(typeof(CVR_MenuManager), nameof(CVR_MenuManager.UpdateMenuPosition))]
    private static bool Prefix_CVR_MenuManager_UpdateMenuPosition()
    {
        if (QuickMenuHelper.Instance == null) return true;
        return !QuickMenuHelper.Instance.MenuIsOpen;
    }

    [HarmonyPrefix]
    [HarmonyPatch(typeof(ViewManager), nameof(ViewManager.UpdateMenuPosition))]
    private static bool Prefix_ViewManager_UpdateMenuPosition(ref float ___cachedScreenAspectRatio)
    {
        if (MainMenuHelper.Instance == null) return true;
        //this is called once a second, so ill fix their dumb aspect ratio shit
        float ratio = (float)Screen.width / (float)Screen.height;
        float clamp = Mathf.Clamp(ratio, 0f, 1.8f);
        MSP_MenuInfo.AspectRatio = 1.7777779f / clamp;
        return !MainMenuHelper.Instance.MenuIsOpen;
    }

    //Set QM stuff
    [HarmonyPostfix]
    [HarmonyPatch(typeof(CVR_MenuManager), nameof(CVR_MenuManager.Start))]
    private static void Postfix_CVR_MenuManager_Start(ref CVR_MenuManager __instance, ref GameObject ____leftVrAnchor)
    {
        QuickMenuHelper helper = __instance.quickMenu.gameObject.AddComponent<QuickMenuHelper>();
        helper.handAnchor = ____leftVrAnchor.transform;
    }

    //Set MM stuff
    [HarmonyPostfix]
    [HarmonyPatch(typeof(ViewManager), nameof(ViewManager.Start))]
    private static void Postfix_ViewManager_Start(ref ViewManager __instance)
    {
        __instance.gameObject.AddComponent<MainMenuHelper>();
    }

    //hook quickmenu open/close
    [HarmonyPrefix]
    [HarmonyPatch(typeof(CVR_MenuManager), nameof(CVR_MenuManager.ToggleQuickMenu))]
    private static bool Prefix_CVR_MenuManager_ToggleQuickMenu(bool show, ref CVR_MenuManager __instance, ref bool ____quickMenuOpen)
    {
        if (QuickMenuHelper.Instance == null) return true;
        if (show != ____quickMenuOpen)
        {
            ____quickMenuOpen = show;
            __instance.quickMenu.ShouldRender = show;
            __instance.quickMenuAnimator.SetBool("Open", show);
            QuickMenuHelper.Instance.MenuIsOpen = show;
            QuickMenuHelper.Instance.UpdateWorldAnchors(show);
            //shouldnt run if switching menus on desktop
            if (!MetaPort.Instance.isUsingVr)
            {
                if (!show && MainMenuHelper.Instance.MenuIsOpen)
                {
                    return false;
                }
                ViewManager.Instance.UiStateToggle(false);
            }
            MSP_MenuInfo.ToggleDesktopInputMethod(show);
            CVRPlayerManager.Instance.ReloadAllNameplates();
        }
        return false;
    }

    // Hook menu open/close
    [HarmonyPrefix]
    [HarmonyPatch(typeof(ViewManager), nameof(ViewManager.UiStateToggle), new Type[] { typeof(bool) })]
    private static bool Prefix_ViewManager_UiStateToggle(bool show, ref ViewManager __instance)
    {
        if (MainMenuHelper.Instance == null) return true;
        if (show != __instance._gameMenuOpen)
        {
            __instance._gameMenuOpen = show;
            __instance.gameMenuView.ShouldRender = show;
            __instance.uiMenuAnimator.SetBool("Open", show);
            MainMenuHelper.Instance.MenuIsOpen = show;
            MainMenuHelper.Instance.UpdateWorldAnchors(show);

            // Shouldn't run if switching menus on desktop
            if (!MetaPort.Instance.isUsingVr)
            {
                if (!show && QuickMenuHelper.Instance.MenuIsOpen)
                {
                    return false;
                }
                CVR_MenuManager.Instance.ToggleQuickMenu(false);
            }

            MSP_MenuInfo.ToggleDesktopInputMethod(show);
            CVRPlayerManager.Instance.ReloadAllNameplates();
        }
        return false;
    }

    //add independent head movement to important input
    [HarmonyPostfix]
    [HarmonyPatch(typeof(CVRInputModule_Keyboard), nameof(CVRInputModule_Keyboard.Update_Always))]
    private static void Postfix_CVRInputModule_Keyboard_Update_Always(ref CVRInputModule_Keyboard __instance)
    {
        if (!__instance._inputManager.inputEnabled) //only run while in menu, so input isnt handled twice
        {
            __instance._inputManager.independentHeadTurn |= CVRInputManager.Controls.MouseAndKeyboard.Character.IndependantHeadTurn.IsPressed();
            if (CVRInputManager.Controls.MouseAndKeyboard.Character.IndependantHeadTurnToggle.WasPerformedThisFrame())
            {
                __instance._inputManager.independentHeadToggle = !__instance._inputManager.independentHeadToggle;
            }
        }
    }

    //Support for changing VRMode during runtime.
    [HarmonyPostfix]
    [HarmonyPatch(typeof(CVRTools), nameof(CVRTools.ConfigureHudAffinity))]
    private static void Postfix_CVRTools_ConfigureHudAffinity()
    {
        MSP_MenuInfo.CameraTransform = PlayerSetup.Instance.GetActiveCamera().transform;
    }
}