Files
Nitrox/NitroxPatcher/Patches/Dynamic/PropulsionCannon_GrabObject_Patch.cs
2025-07-06 00:23:46 +02:00

78 lines
2.4 KiB
C#

using System.Reflection;
using NitroxClient.GameLogic;
using NitroxClient.GameLogic.PlayerLogic;
using NitroxClient.GameLogic.Simulation;
using NitroxClient.MonoBehaviours;
using NitroxClient.MonoBehaviours.Gui.HUD;
using NitroxModel.DataStructures;
using NitroxModel.Helper;
using UnityEngine;
namespace NitroxPatcher.Patches.Dynamic;
public sealed partial class PropulsionCannon_GrabObject_Patch : NitroxPatch, IDynamicPatch
{
private static readonly MethodInfo TARGET_METHOD = Reflect.Method((PropulsionCannon t) => t.GrabObject(default(GameObject)));
private static bool skipPrefixPatch;
public static bool Prefix(PropulsionCannon __instance, GameObject target)
{
if (skipPrefixPatch)
{
return true;
}
if (!target.TryGetIdOrWarn(out NitroxId id))
{
return true;
}
if (Resolve<SimulationOwnership>().HasExclusiveLock(id))
{
Log.Debug($"Already have an exclusive lock on the grabbed propulsion cannon object: {id}");
return true;
}
if (IsInvalidGrabTarget(target))
{
return false;
}
PropulsionGrab context = new(__instance, target);
LockRequest<PropulsionGrab> lockRequest = new(id, SimulationLockType.EXCLUSIVE, ReceivedSimulationLockResponse, context);
Resolve<SimulationOwnership>().RequestSimulationLock(lockRequest);
return false;
}
private static void ReceivedSimulationLockResponse(NitroxId id, bool lockAquired, PropulsionGrab context)
{
if (lockAquired)
{
// In case what we grabbed wasn't a vehicle, we'll be watching it with the regular entity position broadcast system
if (!Resolve<SimulationOwnership>().TreatVehicleEntity(id, true, SimulationLockType.EXCLUSIVE))
{
EntityPositionBroadcaster.WatchEntity(id);
}
skipPrefixPatch = true;
context.Cannon.GrabObject(context.GrabbedObject);
skipPrefixPatch = false;
}
else
{
context.GrabbedObject.AddComponent<DenyOwnershipHand>();
}
}
/// <summary>
/// Prevents certain entities like players from being grabbed
/// </summary>
private static bool IsInvalidGrabTarget(GameObject target)
{
return target.GetComponent<RemotePlayerIdentifier>();
}
}