quantum-space-buddies/QSB/OwnershipSync/OwnedWorldObject.cs

using Cysharp.Threading.Tasks;
using QSB.Messaging;
using QSB.Player;
using QSB.WorldSync;
using System.Threading;
using UnityEngine;

namespace QSB.OwnershipSync;

/// <summary>
/// helper implementation of the interface
/// </summary>
public abstract class OwnedWorldObject<T> : WorldObject<T>, IOwnedWorldObject
	where T : MonoBehaviour
{
	public uint Owner { get; set; }
	public abstract bool CanOwn { get; }

	public override void SendInitialState(uint to) =>
		((IOwnedWorldObject)this).SendMessage(new OwnedWorldObjectMessage(Owner) { To = to });

	public override async UniTask Init(CancellationToken ct) =>
		QSBPlayerManager.OnRemovePlayer += OnPlayerLeave;

	public override void OnRemoval() =>
		QSBPlayerManager.OnRemovePlayer -= OnPlayerLeave;

	private void OnPlayerLeave(PlayerInfo player)
	{
		if (!QSBCore.IsHost)
		{
			return;
		}
		if (Owner == player.PlayerId)
		{
			// put CanOwn check here cuz it only does the thingy in OnReceiveRemote and we want to be able to own this
			((IOwnedWorldObject)this).SendMessage(new OwnedWorldObjectMessage(CanOwn ? QSBPlayerManager.LocalPlayerId : 0));
		}
	}
}