Make CuffableComponent and CuffableSystem not Crash (Hopefully) (#39123)

* This system is ancient

* Destroy that API

* Address reviews

* Destroy merge conflicts from orbit

* seems to work fine

---------

Co-authored-by: Princess Cheeseballs <66055347+Pronana@users.noreply.github.com>
This commit is contained in:
Princess Cheeseballs
2025-11-01 06:22:13 -07:00
committed by GitHub
parent 3d095c8eed
commit 04b71d8203
6 changed files with 89 additions and 58 deletions

View File

@@ -1,3 +1,4 @@
using System.Diagnostics.CodeAnalysis;
using System.Linq;
using Content.Shared.ActionBlocker;
using Content.Shared.Administration.Components;
@@ -260,7 +261,7 @@ namespace Content.Shared.Cuffs
{
if (args.Handled)
return;
TryUncuff(ent, ent, cuffable: ent.Comp);
TryUncuff((ent, ent.Comp), ent);
args.Handled = true;
}
@@ -278,7 +279,7 @@ namespace Content.Shared.Cuffs
Verb verb = new()
{
Act = () => TryUncuff(uid, args.User, cuffable: component),
Act = () => TryUncuff((uid, component), args.User),
DoContactInteraction = true,
Text = Loc.GetString("uncuff-verb-get-data-text")
};
@@ -585,41 +586,31 @@ namespace Content.Shared.Cuffs
return true;
}
/// <inheritdoc cref="TryUncuff(Entity{CuffableComponent?},EntityUid,Entity{HandcuffComponent?})"/>
public void TryUncuff(Entity<CuffableComponent?> target, EntityUid user)
{
if (!TryGetLastCuff(target, out var cuff))
return;
TryUncuff(target, user, cuff.Value);
}
/// <summary>
/// Attempt to uncuff a cuffed entity. Can be called by the cuffed entity, or another entity trying to help uncuff them.
/// If the uncuffing succeeds, the cuffs will drop on the floor.
/// </summary>
/// <param name="target"></param>
/// <param name="user">The cuffed entity</param>
/// <param name="cuffsToRemove">Optional param for the handcuff entity to remove from the cuffed entity. If null, uses the most recently added handcuff entity.</param>
/// <param name="cuffable"></param>
/// <param name="cuff"></param>
public void TryUncuff(EntityUid target, EntityUid user, EntityUid? cuffsToRemove = null, CuffableComponent? cuffable = null, HandcuffComponent? cuff = null)
/// <param name="target">The entity we're trying to remove cuffs from.</param>
/// <param name="user">The entity doing the cuffing.</param>
/// <param name="cuff">The handcuff entity we're attempting to remove.</param>
public void TryUncuff(Entity<CuffableComponent?> target, EntityUid user, Entity<HandcuffComponent?> cuff)
{
if (!Resolve(target, ref cuffable))
if (!Resolve(target, ref target.Comp) || !Resolve(cuff, ref cuff.Comp))
return;
var isOwner = user == target;
var isOwner = user == target.Owner;
if (cuffsToRemove == null)
{
if (cuffable.Container.ContainedEntities.Count == 0)
{
return;
}
cuffsToRemove = cuffable.LastAddedCuffs;
}
else
{
if (!cuffable.Container.ContainedEntities.Contains(cuffsToRemove.Value))
{
Log.Warning("A user is trying to remove handcuffs that aren't in the owner's container. This should never happen!");
}
}
if (!Resolve(cuffsToRemove.Value, ref cuff))
return;
if (!target.Comp.Container.ContainedEntities.Contains(cuff))
Log.Warning("A user is trying to remove handcuffs that aren't in the owner's container. This should never happen!");
var attempt = new UncuffAttemptEvent(user, target);
RaiseLocalEvent(user, ref attempt, true);
@@ -629,29 +620,28 @@ namespace Content.Shared.Cuffs
return;
}
if (!isOwner && !_interaction.InRangeUnobstructed(user, target))
if (!isOwner && !_interaction.InRangeUnobstructed(user, target.Owner))
{
_popup.PopupClient(Loc.GetString("cuffable-component-cannot-remove-cuffs-too-far-message"), user, user);
return;
}
var ev = new ModifyUncuffDurationEvent(user, target, isOwner ? cuff.BreakoutTime : cuff.UncuffTime);
var ev = new ModifyUncuffDurationEvent(user, target, isOwner ? cuff.Comp.BreakoutTime : cuff.Comp.UncuffTime);
RaiseLocalEvent(user, ref ev);
var uncuffTime = ev.Duration;
if (isOwner)
{
if (!TryComp(cuffsToRemove.Value, out UseDelayComponent? useDelay))
if (!TryComp(cuff, out UseDelayComponent? useDelay))
return;
if (!_delay.TryResetDelay((cuffsToRemove.Value, useDelay), true))
if (!_delay.TryResetDelay((cuff, useDelay), true))
{
return;
}
}
var doAfterEventArgs = new DoAfterArgs(EntityManager, user, uncuffTime, new UnCuffDoAfterEvent(), target, target, cuffsToRemove)
var doAfterEventArgs = new DoAfterArgs(EntityManager, user, uncuffTime, new UnCuffDoAfterEvent(), target, target, cuff)
{
BreakOnMove = true,
BreakOnWeightlessMove = false,
@@ -666,7 +656,7 @@ namespace Content.Shared.Cuffs
_adminLog.Add(LogType.Action, LogImpact.High, $"{ToPrettyString(user):player} is trying to uncuff {ToPrettyString(target):subject}");
var popupText = user == target
var popupText = user == target.Owner
? "cuffable-component-start-uncuffing-self-observer"
: "cuffable-component-start-uncuffing-observer";
_popup.PopupEntity(
@@ -678,7 +668,7 @@ namespace Content.Shared.Cuffs
.RemoveWhere(e => e.AttachedEntity == target || e.AttachedEntity == user),
true);
if (target == user)
if (isOwner)
{
_popup.PopupClient(Loc.GetString("cuffable-component-start-uncuffing-self"), user, user);
}
@@ -694,7 +684,7 @@ namespace Content.Shared.Cuffs
target);
}
_audio.PlayPredicted(isOwner ? cuff.StartBreakoutSound : cuff.StartUncuffSound, target, user);
_audio.PlayPredicted(isOwner ? cuff.Comp.StartBreakoutSound : cuff.Comp.StartUncuffSound, target, user);
}
public void Uncuff(EntityUid target, EntityUid? user, EntityUid cuffsToRemove, CuffableComponent? cuffable = null, HandcuffComponent? cuff = null)
@@ -818,9 +808,56 @@ namespace Content.Shared.Cuffs
#endregion
public IReadOnlyList<EntityUid> GetAllCuffs(CuffableComponent component)
/// <summary>
/// Tries to get a list of all the handcuffs stored in an entity's <see cref="CuffableComponent"/>.
/// </summary>
/// <param name="entity">The cuffable entity in question.</param>
/// <param name="cuffs">A list of cuffs if it exists.</param>
/// <returns>True if a list of cuffs with cuffs exists. False if no list exists or if it is empty.</returns>
public bool TryGetAllCuffs(Entity<CuffableComponent?> entity, out IReadOnlyList<EntityUid> cuffs)
{
return component.Container.ContainedEntities;
cuffs = GetAllCuffs(entity);
return cuffs.Count > 0;
}
/// <summary>
/// Tries to get a list of all the handcuffs stored in a entity's <see cref="CuffableComponent"/>.
/// </summary>
/// <param name="entity">The cuffable entity in question.</param>
/// <returns>A list of cuffs if it exists, or null if there are no cuffs.</returns>
public IReadOnlyList<EntityUid> GetAllCuffs(Entity<CuffableComponent?> entity)
{
if (!Resolve(entity, ref entity.Comp))
return [];
return entity.Comp.Container.ContainedEntities;
}
/// <summary>
/// Tries to get the most recently added pair of handcuffs added to an entity with <see cref="CuffableComponent"/>.
/// </summary>
/// <param name="entity">The cuffable entity in question.</param>
/// <param name="cuff">The most recently added cuff.</param>
/// <returns>Returns true if a cuff exists and false if one doesn't.</returns>
public bool TryGetLastCuff(Entity<CuffableComponent?> entity, [NotNullWhen(true)] out EntityUid? cuff)
{
cuff = GetLastCuffOrNull(entity);
return cuff != null;
}
/// <summary>
/// Tries to get the most recently added pair of handcuffs added to an entity with <see cref="CuffableComponent"/>.
/// </summary>
/// <param name="entity">The cuffable entity in question.</param>
/// <returns>The most recently added cuff or null if none exists.</returns>
public EntityUid? GetLastCuffOrNull(Entity<CuffableComponent?> entity)
{
if (!Resolve(entity, ref entity.Comp))
return null;
return entity.Comp.Container.ContainedEntities.Count == 0 ? null : entity.Comp.Container.ContainedEntities.Last();
}
}