NAK_CVR_Mods/DesktopVRSwitch/Main.cs

using MelonLoader;
using NAK.DesktopVRSwitch.VRModeTrackers;
using UnityEngine;

namespace NAK.DesktopVRSwitch;

public class DesktopVRSwitch : MelonMod
{
    internal static MelonLogger.Instance Logger;

    public static readonly MelonPreferences_Category Category =
        MelonPreferences.CreateCategory(nameof(DesktopVRSwitch));

    public static readonly MelonPreferences_Entry<bool> EntryEnterCalibrationOnSwitch =
        Category.CreateEntry("Enter Calibration on Switch", true, description: "Should you automatically be placed into calibration after switch if FBT is available? Overridden by Save Calibration IK setting.");

    public static readonly MelonPreferences_Entry<bool> EntryUseTransitionOnSwitch =
        Category.CreateEntry("Use Transition on Switch", true, description: "Should the world transition play on VRMode switch?");

    //public static readonly MelonPreferences_Entry<bool> EntryRenderVRGameView =
    //    Category.CreateEntry("Render VR Game View", true, description: "Should the VR view be displayed in the game window after VRMode switch?");

    public static readonly MelonPreferences_Entry<bool> EntrySwitchToDesktopOnExit =
        Category.CreateEntry("Switch to Desktop on SteamVR Exit", false, description: "Should the game switch to Desktop when SteamVR quits?");

    public override void OnInitializeMelon()
    {
        Logger = LoggerInstance;

        RegisterVRModeTrackers();

        // main manager
        ApplyPatches(typeof(HarmonyPatches.CheckVRPatches));
        // nameplate fixes
        ApplyPatches(typeof(HarmonyPatches.CameraFacingObjectPatches));
        // pickup fixes
        ApplyPatches(typeof(HarmonyPatches.CVRPickupObjectPatches));
        // lazy fix to reset iksystem
        ApplyPatches(typeof(HarmonyPatches.IKSystemPatches));
        // post processing fixes
        ApplyPatches(typeof(HarmonyPatches.CVRWorldPatches));

        // prevent steamvr behaviour from closing game
        ApplyPatches(typeof(HarmonyPatches.SteamVRBehaviourPatches));
    }

    public override void OnUpdate()
    {
        if (Input.GetKeyDown(KeyCode.F6) && Input.GetKey(KeyCode.LeftControl))
        {
            VRModeSwitchManager.Instance?.AttemptSwitch();
        }
    }

    private static void RegisterVRModeTrackers()
    {
        // Core trackers
        VRModeSwitchManager.RegisterVRModeTracker(new CheckVRTracker());
        VRModeSwitchManager.RegisterVRModeTracker(new MetaPortTracker());

        // HUD trackers
        VRModeSwitchManager.RegisterVRModeTracker(new CohtmlHudTracker());
        VRModeSwitchManager.RegisterVRModeTracker(new HudOperationsTracker());

        // Player trackers
        VRModeSwitchManager.RegisterVRModeTracker(new PlayerSetupTracker());
        VRModeSwitchManager.RegisterVRModeTracker(new MovementSystemTracker());
        VRModeSwitchManager.RegisterVRModeTracker(new IKSystemTracker());

        // Menu trackers
        VRModeSwitchManager.RegisterVRModeTracker(new CVR_MenuManagerTracker());
        VRModeSwitchManager.RegisterVRModeTracker(new ViewManagerTracker());

        // Interaction trackers
        VRModeSwitchManager.RegisterVRModeTracker(new CVRInputManagerTracker());
        VRModeSwitchManager.RegisterVRModeTracker(new CVR_InteractableManagerTracker());
        VRModeSwitchManager.RegisterVRModeTracker(new CVRGestureRecognizerTracker());

        // Portable camera tracker
        VRModeSwitchManager.RegisterVRModeTracker(new PortableCameraTracker());

        // CVRWorld tracker - Must come after PlayerSetupTracker
        VRModeSwitchManager.RegisterVRModeTracker(new CVRWorldTracker());
    }

    private void ApplyPatches(Type type)
    {
        try
        {
            HarmonyInstance.PatchAll(type);
        }
        catch (Exception e)
        {
            LoggerInstance.Msg($"Failed while patching {type.Name}!");
            LoggerInstance.Error(e);
        }
    }
}