quantum-space-buddies/QSB/SectorSync/WorldObjects/QSBSector.cs

using Cysharp.Threading.Tasks;
using OWML.Common;
using QSB.Utility;
using QSB.WorldSync;
using System;
using System.Linq;
using System.Threading;
using UnityEngine;

namespace QSB.SectorSync.WorldObjects
{
	public class QSBSector : WorldObject<Sector>
	{
		public Sector.Name Type => AttachedObject.GetName();
		public Transform Transform
		{
			get
			{
				if (AttachedObject == null)
				{
					DebugLog.ToConsole($"Error - Tried to get Transform from QSBSector {ObjectId} with null AttachedObject!\r\n{Environment.StackTrace}", MessageType.Error);
					return null;
				}

				return AttachedObject.transform;
			}
		}
		public Vector3 Position => Transform.position;

		public bool IsFakeSector => AttachedObject is FakeSector;
		public FakeSector FakeSector => (FakeSector)AttachedObject;

		public override async UniTask Init(CancellationToken ct)
		{
			if (IsFakeSector)
			{
				QSBSectorManager.Instance.FakeSectors.Add(this);
			}
		}

		public override void OnRemoval()
		{
			if (IsFakeSector)
			{
				QSBSectorManager.Instance.FakeSectors.Remove(this);
			}
		}

		public override void SendInitialState(uint to) { }

		public bool ShouldSyncTo(DynamicOccupant occupantType)
		{
			if (AttachedObject == null)
			{
				DebugLog.ToConsole($"Warning - AttachedObject for sector id:{ObjectId} is null!", MessageType.Warning);
				return false;
			}

			if (!AttachedObject.gameObject.activeInHierarchy)
			{
				return false;
			}

			if (occupantType == DynamicOccupant.Ship && Type == Sector.Name.Ship)
			{
				return false;
			}

			if (AttachedObject.name is "Sector_Shuttle" or "Sector_NomaiShuttleInterior")
			{
				if (QSBSceneManager.CurrentScene == OWScene.SolarSystem)
				{
					var shuttleController = AttachedObject.gameObject.GetComponentInParent<NomaiShuttleController>();
					if (shuttleController == null)
					{
						DebugLog.ToConsole($"Warning - Expected to find a NomaiShuttleController for {AttachedObject.name}!", MessageType.Warning);
						return false;
					}

					if (!shuttleController.IsPlayerInside())
					{
						return false;
					}
				}
				else if (QSBSceneManager.CurrentScene == OWScene.EyeOfTheUniverse)
				{
					var shuttleController = QSBWorldSync.GetUnityObjects<EyeShuttleController>().First();
					if (shuttleController == null)
					{
						DebugLog.ToConsole($"Warning - Expected to find a EyeShuttleController for {AttachedObject.name}!", MessageType.Warning);
						return false;
					}

					if (!shuttleController._isPlayerInside)
					{
						return false;
					}
				}
			}

			return true;
		}
	}
}