new-horizons/NewHorizons/Handlers/VesselWarpHandler.cs

using NewHorizons.Builder.General;
using NewHorizons.Builder.Props;
using NewHorizons.Components;
using NewHorizons.Components.EyeOfTheUniverse;
using NewHorizons.Utility;
using NewHorizons.Utility.OuterWilds;
using NewHorizons.Utility.OWML;
using UnityEngine;
using System.Collections;
using static NewHorizons.Main;
using static NewHorizons.Utility.Files.AssetBundleUtilities;

namespace NewHorizons.Handlers
{
    public static class VesselWarpHandler
    {
        public static GameObject VesselPrefab { get; private set; }
        public static GameObject VesselObject { get; private set; }
        public static VesselWarpController WarpController { get; private set; }

        private static SpawnPoint _vesselSpawnPoint;
        public static SpawnPoint VesselSpawnPoint => _vesselSpawnPoint;

        public static void Initialize()
        {
            VesselPrefab = NHPrivateAssetBundle.LoadAsset<GameObject>("Vessel_Body");
        }

        public static bool IsVesselPresentAndActive()
        {
            var vesselConfig = SystemDict[Instance.CurrentStarSystem].Config?.Vessel;
            var vesselIsPresent = vesselConfig?.alwaysPresent ?? false;
            return Instance.IsWarpingFromVessel || vesselIsPresent;
        }

        public static bool IsVesselPresent()
        {
            var isDefaultSolarSystem = Instance.CurrentStarSystem == "SolarSystem";
            var isEyeOfTheUniverse = Instance.CurrentStarSystem == "EyeOfTheUniverse";
            return IsVesselPresentAndActive() || isDefaultSolarSystem || isEyeOfTheUniverse;
        }

        public static bool ShouldSpawnAtVessel()
        {
            var vesselConfig = SystemDict[Instance.CurrentStarSystem].Config?.Vessel;
            var shouldSpawnOnVessel = IsVesselPresent() && (vesselConfig?.spawnOnVessel ?? false);
            return !Instance.IsWarpingFromShip && (Instance.IsWarpingFromVessel || Instance.DidWarpFromVessel || shouldSpawnOnVessel);
        }

        public static void LoadVessel()
        {
            var system = SystemDict[Instance.CurrentStarSystem];
            if (Instance.CurrentStarSystem == "EyeOfTheUniverse")
            {
                _vesselSpawnPoint = SearchUtilities.Find("Vessel_Body/SPAWN_Vessel").GetComponent<EyeSpawnPoint>();
                return;
            }

            if (IsVesselPresentAndActive())
            {
                _vesselSpawnPoint = Instance.CurrentStarSystem == "SolarSystem" ? UpdateVessel() : CreateVessel();
            }
            else
            {
                var vesselDimension = SearchUtilities.Find("DB_VesselDimension_Body/Sector_VesselDimension");
                var vesselDimensionSpawn = vesselDimension.GetComponentInChildren<SpawnPoint>(true);
                var vesselWarpController = vesselDimension.GetComponentInChildren<VesselWarpController>(true);

                var defaultPlayerWarpPoint = new GameObject("DefaultPlayerWarpPos");
                defaultPlayerWarpPoint.transform.SetParent(vesselWarpController.transform, false);
                defaultPlayerWarpPoint.transform.localPosition = new Vector3(0, -5.82f, -6.56f);
                vesselWarpController._defaultPlayerWarpPoint = defaultPlayerWarpPoint.transform;

                var vesselSpawnObj = new GameObject("SPAWN_Vessel");
                vesselSpawnObj.transform.SetParent(vesselWarpController.transform.parent.parent, false);
                vesselSpawnObj.transform.localPosition = new Vector3(-0.3f, -5.18f, -6.35f);
                var vesselSpawnPoint = vesselSpawnObj.AddComponent<VesselSpawnPoint>();
                vesselSpawnPoint.WarpController = vesselWarpController;
                vesselSpawnPoint._triggerVolumes = vesselDimensionSpawn._triggerVolumes;
                _vesselSpawnPoint = vesselSpawnPoint;
            }
        }

        public static void TeleportToVessel()
        {
            var playerSpawner = Object.FindObjectOfType<PlayerSpawner>();
            if (_vesselSpawnPoint is VesselSpawnPoint vesselSpawnPoint)
            {
                NHLogger.LogVerbose("Relative warping into vessel");
                vesselSpawnPoint.WarpPlayer();

                // #1034 Vessel warp sometimes has the player get flung away into space and die
                // We do what we do with regular spawns where we keep resetting their position to the right one while invincible until we're relatively certain
                // that the spawning sequence is done
                Delay.StartCoroutine(FixPlayerSpawning(25, vesselSpawnPoint));
            }
            else
            {
                NHLogger.LogVerbose("Debug warping into vessel");
                playerSpawner.DebugWarp(_vesselSpawnPoint);
            }
            Builder.General.SpawnPointBuilder.SuitUp();

            LoadDB();
        }

        private static IEnumerator FixPlayerSpawning(int frameInterval, VesselSpawnPoint vesselSpawn)
        {
            InvulnerabilityHandler.MakeInvulnerable(true);

            var frameCount = 0;
            while (frameCount <= frameInterval)
            {
                vesselSpawn.WarpPlayer();
                frameCount++;
                yield return null; // Wait for the next frame
            }

            InvulnerabilityHandler.MakeInvulnerable(false);
            var playerBody = SearchUtilities.Find("Player_Body").GetAttachedOWRigidbody();
            var resources = playerBody.GetComponent<PlayerResources>();
            resources._currentHealth = 100f;
        }

        public static void LoadDB()
        {
            if (Instance.CurrentStarSystem == "SolarSystem")
            {
                //Deactivate lock since we aren't in timber anymore
                GameObject.Destroy(SearchUtilities.Find("TimberHearth_Body/StreamingGroup_TH").GetComponent<StreamingLock>());

                // Deactivate village music because for some reason it still plays.
                SearchUtilities.Find("TimberHearth_Body/Sector_TH/Sector_Village/Volumes_Village/MusicVolume_Village").GetComponent<VillageMusicVolume>().Deactivate();

                // Loads it manually so the player doesn't start falling and then vessel loads in on them.
                SectorStreaming ss = SearchUtilities.Find("DB_VesselDimension_Body/Sector_VesselDimension/Sector_Streaming").GetComponent<SectorStreaming>();
                ss.enabled = true;
                ss._streamingGroup.LoadRequiredAssets();
                ss._streamingGroup.LoadRequiredColliders();
                ss._streamingGroup.LoadGeneralAssets();
                StreamingManager.loadingPriority = StreamingManager.LoadingPriority.High;
            }
        }

        public static VesselSpawnPoint CreateVessel()
        {
            var system = SystemDict[Instance.CurrentStarSystem];

            NHLogger.LogVerbose("Checking for Vessel Prefab");
            if (VesselPrefab == null) return null;

            NHLogger.LogVerbose("Creating Vessel");
            var vesselObject = GeneralPropBuilder.MakeFromPrefab(VesselPrefab, VesselPrefab.name, null, null, system.Config.Vessel?.vesselSpawn);
            VesselObject = vesselObject;

            var vesselAO = vesselObject.AddComponent<EyeAstroObject>();
            vesselAO._owRigidbody = vesselObject.GetComponent<OWRigidbody>();
            vesselAO._rootSector = vesselObject.GetComponentInChildren<Sector>(true);
            vesselAO._customName = "Vessel";
            vesselAO._name = AstroObject.Name.CustomString;
            vesselAO._type = AstroObject.Type.SpaceStation;
            vesselAO.Register();
            vesselObject.GetComponentInChildren<ReferenceFrameVolume>(true)._referenceFrame._attachedAstroObject = vesselAO;

            VesselSingularityRoot singularityRoot = vesselObject.GetComponentInChildren<VesselSingularityRoot>(true);

            VesselWarpController vesselWarpController = vesselObject.GetComponentInChildren<VesselWarpController>(true);
            WarpController = vesselWarpController;

            GameObject WarpPlatform = SearchUtilities.Find("DB_VesselDimension_Body/Sector_VesselDimension/Sector_VesselBridge/Interactibles_VesselBridge/WarpController/Prefab_NOM_WarpPlatform");
            GameObject warpBH = WarpPlatform.transform.Find("BlackHole").gameObject;
            GameObject warpWH = WarpPlatform.transform.Find("WhiteHole").gameObject;

            GameObject sourceBH = Object.Instantiate(warpBH, vesselWarpController._sourceWarpPlatform.transform, false);
            sourceBH.name = "BlackHole";
            vesselWarpController._sourceWarpPlatform._blackHole = sourceBH.GetComponentInChildren<SingularityController>();
            vesselWarpController._sourceWarpPlatform._blackHole.OnCollapse += vesselWarpController._sourceWarpPlatform.OnBlackHoleCollapse;

            GameObject sourceWH = Object.Instantiate(warpWH, vesselWarpController._sourceWarpPlatform.transform, false);
            sourceWH.name = "WhiteHole";
            vesselWarpController._sourceWarpPlatform._whiteHole = sourceWH.GetComponentInChildren<SingularityController>();
            vesselWarpController._sourceWarpPlatform._whiteHole.OnCollapse += vesselWarpController._sourceWarpPlatform.OnWhiteHoleCollapse;

            GameObject targetBH = Object.Instantiate(warpBH, vesselWarpController._targetWarpPlatform.transform, false);
            targetBH.name = "BlackHole";
            vesselWarpController._targetWarpPlatform._blackHole = targetBH.GetComponentInChildren<SingularityController>();
            vesselWarpController._targetWarpPlatform._blackHole.OnCollapse += vesselWarpController._targetWarpPlatform.OnBlackHoleCollapse;

            GameObject targetWH = Object.Instantiate(warpWH, vesselWarpController._targetWarpPlatform.transform, false);
            targetWH.name = "WhiteHole";
            vesselWarpController._targetWarpPlatform._whiteHole = targetWH.GetComponentInChildren<SingularityController>();
            vesselWarpController._targetWarpPlatform._whiteHole.OnCollapse += vesselWarpController._targetWarpPlatform.OnWhiteHoleCollapse;

            GameObject blackHole = SearchUtilities.Find("DB_VesselDimension_Body/Sector_VesselDimension/Sector_VesselBridge/Interactibles_VesselBridge/BlackHole");
            GameObject newBlackHole = Object.Instantiate(blackHole, singularityRoot.transform);
            newBlackHole.transform.localPosition = Vector3.zero;
            newBlackHole.transform.localRotation = Quaternion.identity;
            newBlackHole.transform.localScale = Vector3.one;
            newBlackHole.name = "BlackHole";
            vesselWarpController._blackHole = newBlackHole.GetComponentInChildren<SingularityController>();
            vesselWarpController._blackHoleOneShot = vesselWarpController._blackHole.transform.parent.Find("BlackHoleAudio_OneShot").GetComponent<OWAudioSource>();

            GameObject whiteHole = SearchUtilities.Find("DB_VesselDimension_Body/Sector_VesselDimension/Sector_VesselBridge/Interactibles_VesselBridge/WhiteHole");
            GameObject newWhiteHole = Object.Instantiate(whiteHole, singularityRoot.transform);
            newWhiteHole.transform.localPosition = Vector3.zero;
            newWhiteHole.transform.localRotation = Quaternion.identity;
            newWhiteHole.transform.localScale = Vector3.one;
            newWhiteHole.name = "WhiteHole";
            vesselWarpController._whiteHole = newWhiteHole.GetComponentInChildren<SingularityController>();
            vesselWarpController._whiteHoleOneShot = vesselWarpController._whiteHole.transform.parent.Find("WhiteHoleAudio_OneShot").GetComponent<OWAudioSource>();
            vesselWarpController._whiteHole._startActive = true;

            vesselObject.GetComponent<MapMarker>()._labelID = (UITextType)TranslationHandler.AddUI("VESSEL");

            var hasParentBody = !string.IsNullOrEmpty(system.Config.Vessel?.vesselSpawn?.parentBody);
            var hasPhysics = system.Config.Vessel?.hasPhysics ?? !hasParentBody;
            var planetGO = hasParentBody ? vesselObject.transform.parent.gameObject : null;

            if (hasPhysics)
            {
                vesselObject.transform.parent = null;
            }
            else
            {
                vesselAO._owRigidbody = null;
                Object.DestroyImmediate(vesselObject.GetComponent<KinematicRigidbody>());
                Object.DestroyImmediate(vesselObject.GetComponent<CenterOfTheUniverseOffsetApplier>());
                Object.DestroyImmediate(vesselObject.GetComponent<OWRigidbody>());
                Object.DestroyImmediate(vesselObject.GetComponent<Rigidbody>());
                var rfVolume = vesselObject.transform.Find("RFVolume");
                if (rfVolume != null)
                {
                    Object.Destroy(rfVolume.gameObject);
                }
            }
            
            if (hasParentBody)
            {
                foreach (OWRigidbody dynamicProp in vesselObject.GetComponentsInChildren<OWRigidbody>(true))
                {
                    if (dynamicProp.GetComponent<NomaiInterfaceOrb>() == null)
                    {
                        dynamicProp.gameObject.AddComponent<FixPhysics>();
                    }
                }
            }

            var attachWarpExitToVessel = system.Config.Vessel?.warpExit?.attachToVessel ?? false;
            var warpExitParent = vesselWarpController._targetWarpPlatform.transform.parent;

            var warpExit = GeneralPropBuilder.MakeFromExisting(vesselWarpController._targetWarpPlatform.gameObject, planetGO, null, system.Config.Vessel?.warpExit, defaultParent: attachWarpExitToVessel ? warpExitParent : null);
            if (attachWarpExitToVessel)
            {
                warpExit.transform.parent = warpExitParent;
            }
            vesselWarpController._targetWarpPlatform._owRigidbody = warpExit.GetAttachedOWRigidbody();

            var hasZeroGravityVolume = system.Config.Vessel?.hasZeroGravityVolume ?? !hasParentBody;
            if (!hasZeroGravityVolume)
            {
                var zeroGVolume = vesselObject.transform.Find("Sector_VesselBridge/Volumes_VesselBridge/ZeroGVolume");
                if (zeroGVolume != null)
                {
                    Object.Destroy(zeroGVolume.gameObject);
                }
            }

            VesselSpawnPoint spawnPoint = vesselObject.GetComponentInChildren<VesselSpawnPoint>(true);
            if (ShouldSpawnAtVessel())
            {
                SpawnPointBuilder.OverridePlayerSpawn(spawnPoint);
            }

            vesselObject.SetActive(true);

            var power = vesselWarpController.transform.Find("PowerSwitchInterface");
            var orb = power.GetComponentInChildren<NomaiInterfaceOrb>(true);
            // Run after Start (when pillar is lowered immediately) and then skip 100 frames so that the weird orb rising thing doesn't happen #958
            Delay.RunWhenAndInNUpdates(() => SetupWarpController(vesselWarpController, orb), () => !vesselWarpController._coordinateInterface._pillarRaised, 100);

            return spawnPoint;
        }

        public static VesselSpawnPoint UpdateVessel()
        {
            var system = SystemDict[Instance.CurrentStarSystem];

            NHLogger.LogVerbose("Updating DB Vessel");
            var vectorSector = SearchUtilities.Find("DB_VesselDimension_Body/Sector_VesselDimension");
            VesselObject = vectorSector;

            var spawnPoint = vectorSector.GetComponentInChildren<SpawnPoint>(true);

            VesselWarpController vesselWarpController = vectorSector.GetComponentInChildren<VesselWarpController>(true);
            WarpController = vesselWarpController;

            if (vesselWarpController._whiteHole == null)
            {
                GameObject whiteHole = SearchUtilities.Find("DB_VesselDimension_Body/Sector_VesselDimension/Sector_VesselBridge/Interactibles_VesselBridge/WhiteHole");
                vesselWarpController._whiteHole = whiteHole.GetComponentInChildren<SingularityController>();
                vesselWarpController._whiteHoleOneShot = vesselWarpController._whiteHole.transform.parent.Find("WhiteHoleAudio_OneShot").GetComponent<OWAudioSource>();
            }

            vesselWarpController._whiteHole._startActive = true;
            vesselWarpController._whiteHole.Stabilize();

            var defaultPlayerWarpPoint = new GameObject("DefaultPlayerWarpPos");
            defaultPlayerWarpPoint.transform.SetParent(vesselWarpController.transform, false);
            defaultPlayerWarpPoint.transform.localPosition = new Vector3(0, -5.82f, -6.56f);
            vesselWarpController._defaultPlayerWarpPoint = defaultPlayerWarpPoint.transform;

            var vesselSpawnObj = new GameObject("SPAWN_Vessel");
            vesselSpawnObj.transform.SetParent(vesselWarpController.transform.parent.parent, false);
            vesselSpawnObj.transform.localPosition = new Vector3(-0.3f, -5.18f, -6.35f);
            var vesselSpawnPoint = vesselSpawnObj.AddComponent<VesselSpawnPoint>();
            vesselSpawnPoint.WarpController = vesselWarpController;
            vesselSpawnPoint._triggerVolumes = spawnPoint._triggerVolumes;

            var power = vesselWarpController.transform.Find("PowerSwitchInterface");
            var orb = power.GetComponentInChildren<NomaiInterfaceOrb>(true);
            // Run after Start (when pillar is lowered immediately) and then skip 100 frames so that the weird orb rising thing doesn't happen #958
            Delay.RunWhenAndInNUpdates(() => SetupWarpController(vesselWarpController, orb, true), () => !vesselWarpController._coordinateInterface._pillarRaised, 100);

            return vesselSpawnPoint;
        }

        public static void SetupWarpController(VesselWarpController vesselWarpController, NomaiInterfaceOrb orb, bool db = false)
        {
            if (db)
            {
                //Make warp core
                foreach (WarpCoreItem core in Resources.FindObjectsOfTypeAll<WarpCoreItem>())
                {
                    if (core.GetWarpCoreType().Equals(WarpCoreType.Vessel))
                    {
                        var newCore = Object.Instantiate(core, AstroObjectLocator.GetAstroObject("Vessel Dimension")?.transform ?? Locator.GetPlayerBody()?.transform);
                        newCore._visible = true;
                        foreach (OWRenderer render in newCore._renderers)
                        {
                            if (render)
                            {
                                render.enabled = true;
                                render.SetActivation(true);
                                render.SetLODActivation(true);
                                if (render.GetRenderer()) render.GetRenderer().enabled = true;
                            }
                        }
                        foreach (ParticleSystem particleSystem in newCore._particleSystems)
                        {
                            if (particleSystem) particleSystem.Play(true);
                        }
                        foreach (OWLight2 light in newCore._lights)
                        {
                            if (light)
                            {
                                light.enabled = true;
                                light.SetActivation(true);
                                light.SetLODActivation(true);
                                if (light.GetLight()) light.GetLight().enabled = true;
                            }
                        }
                        vesselWarpController._coreSocket._socketedItem = newCore;
                        newCore.SocketItem(vesselWarpController._coreSocket._socketTransform, vesselWarpController._coreSocket._sector);
                        newCore.PlaySocketAnimation();
                        vesselWarpController._coreSocket.enabled = true;
                        vesselWarpController.SetPowered(true);
                        break;
                    }
                }
            }
            else
            {
                foreach (NomaiLamp lamp in vesselWarpController.transform.root.GetComponentsInChildren<NomaiLamp>(true))
                {
                    lamp._startOn = true;
                    lamp.Awake();
                }
            }

            vesselWarpController.OnSlotDeactivated(vesselWarpController._coordinatePowerSlot);
            vesselWarpController._gravityVolume.SetFieldMagnitude(vesselWarpController._origGravityMagnitude);
            vesselWarpController._coreCable.SetPowered(true);
            vesselWarpController._warpPlatformCable.SetPowered(false);
            orb.SetOrbPosition(vesselWarpController._coordinatePowerSlot.transform.position);
            orb._occupiedSlot = vesselWarpController._coordinatePowerSlot;
            orb._enterSlotTime = Time.time;
            vesselWarpController.OnSlotActivated(vesselWarpController._coordinatePowerSlot);
            vesselWarpController._coordinateCable.SetPowered(true);
            vesselWarpController._cageClosed = true;
            if (vesselWarpController._cageAnimator != null)
            {
                vesselWarpController._cageAnimator.TranslateToLocalPosition(new Vector3(0.0f, -8.1f, 0.0f), 0.1f);
                vesselWarpController._cageAnimator.RotateToLocalEulerAngles(new Vector3(0.0f, 180f, 0.0f), 0.1f);
                vesselWarpController._cageAnimator.OnTranslationComplete -= new TransformAnimator.AnimationEvent(vesselWarpController.OnCageAnimationComplete);
                vesselWarpController._cageAnimator.OnTranslationComplete += new TransformAnimator.AnimationEvent(vesselWarpController.OnCageAnimationComplete);
            }

            // Normally the power-on sound is 2D/global, we set it to 3D/local so it isn't audible if the player isn't nearby
            vesselWarpController._audioSource.spatialBlend = 1f;
            vesselWarpController._audioSource.rolloffMode = AudioRolloffMode.Linear;
        }
    }
}