using NAK.Melons.DesktopVRSwitch.Patches;
using System.Collections;
using UnityEngine;
using UnityEngine.XR;
using Valve.VR;

namespace NAK.Melons.DesktopVRSwitch;

public class DesktopVRSwitch : MonoBehaviour
{
    //Settings
    public bool _reloadLocalAvatar = true;

    //Internal Stuff
    private bool _switchInProgress = false;

    void Start()
    {
        //do not pause game, this breaks dynbones & trackers
        SteamVR_Settings.instance.pauseGameWhenDashboardVisible = false;
    }

    void Update()
    {
        if (Input.GetKeyDown(KeyCode.F6) && Input.GetKey(KeyCode.LeftControl))
        {
            SwitchVRMode();
        }
    }

    public void SwitchVRMode()
    {
        if (_switchInProgress) return;
        if (!IsInVR())
        {
            StartCoroutine(StartVRSystem());
        }
        else
        {
            StartCoroutine(StopVR());
        }
    }

    public bool IsInVR() => XRSettings.enabled;

    private IEnumerator StartVRSystem()
    {
        BeforeVRModeSwitch(true);
        XRSettings.LoadDeviceByName("OpenVR");
        yield return null;
        if (string.IsNullOrEmpty(XRSettings.loadedDeviceName))
        {
            DesktopVRSwitchMod.Logger.Error("Initializing VR Failed. Is there no VR device connected?");
        }
        else
        {
            DesktopVRSwitchMod.Logger.Msg("Starting SteamVR...");
            XRSettings.enabled = true;
            //force steamvr to reinitialize input
            //this does SteamVR_Input.actionSets[0].Activate() for us (we deactivate in StopVR())
            //but only if SteamVR_Settings.instance.activateFirstActionSetOnStart is enabled
            //which in ChilloutVR, it is, because all those settings are default
            SteamVR_Input.Initialize(true);
            yield return null;
            AfterVRModeSwitch(true);
        }
        yield break;
    }

    private IEnumerator StopVR()
    {
        BeforeVRModeSwitch(false);
        yield return null;
        if (!string.IsNullOrEmpty(XRSettings.loadedDeviceName))
        {
            //deactivate the action set so SteamVR_Input.Initialize can reactivate
            SteamVR_Input.actionSets[0].Deactivate(SteamVR_Input_Sources.Any);
            SteamVR.SafeDispose(); //idk
            XRSettings.LoadDeviceByName("");
            XRSettings.enabled = false;
            yield return null;
            //reset physics time to Desktop default
            Time.fixedDeltaTime = 0.02f;
            AfterVRModeSwitch(false);
        }
        yield break;
    }

    //one frame before switch attempt
    public void BeforeVRModeSwitch(bool enterVR)
    {
        //let tracked objects know we are attempting to switch
        VRModeSwitchTracker.PreVRModeSwitch(enterVR);
    }

    //one frame after switch attempt
    public void AfterVRModeSwitch(bool enterVR)
    {
        //close the menus
        TryCatchHell.CloseCohtmlMenus();

        //these two must come first
        TryCatchHell.SetCheckVR(enterVR);
        TryCatchHell.SetMetaPort(enterVR);

        //the bulk of funni changes
        TryCatchHell.RepositionCohtmlHud(enterVR);
        TryCatchHell.UpdateHudOperations(enterVR);
        TryCatchHell.DisableMirrorCanvas();
        TryCatchHell.SwitchActiveCameraRigs(enterVR);
        TryCatchHell.ResetCVRInputManager();
        TryCatchHell.UpdateRichPresence();
        TryCatchHell.UpdateGestureReconizerCam();

        //let tracked objects know we switched
        VRModeSwitchTracker.PostVRModeSwitch(enterVR);

        //reload avatar by default, optional for debugging
        if (_reloadLocalAvatar)
        {
            TryCatchHell.ReloadLocalAvatar();
        }

        _switchInProgress = false;
    }
}