Predict stripping (#32478)
* Predict stripping Stops mob verbs from getting moved around again. * Bola * Fix ftl
This commit is contained in:
metalgearsloth
@@ -1,7 +1,21 @@
|
||||
using System.Linq;
|
||||
using Content.Shared.Alert;
|
||||
using Content.Shared.Body.Part;
|
||||
using Content.Shared.Body.Systems;
|
||||
using Content.Shared.CombatMode.Pacification;
|
||||
using Content.Shared.Damage.Components;
|
||||
using Content.Shared.Damage.Systems;
|
||||
using Content.Shared.DoAfter;
|
||||
using Content.Shared.Ensnaring.Components;
|
||||
using Content.Shared.Hands.EntitySystems;
|
||||
using Content.Shared.IdentityManagement;
|
||||
using Content.Shared.Movement.Systems;
|
||||
using Robust.Shared.GameStates;
|
||||
using Content.Shared.Popups;
|
||||
using Content.Shared.StepTrigger.Systems;
|
||||
using Content.Shared.Strip.Components;
|
||||
using Content.Shared.Throwing;
|
||||
using Robust.Shared.Audio.Systems;
|
||||
using Robust.Shared.Containers;
|
||||
using Robust.Shared.Serialization;
|
||||
|
||||
namespace Content.Shared.Ensnaring;
|
||||
@@ -13,36 +27,82 @@ public sealed partial class EnsnareableDoAfterEvent : SimpleDoAfterEvent
|
||||
|
||||
public abstract class SharedEnsnareableSystem : EntitySystem
|
||||
{
|
||||
[Dependency] private readonly MovementSpeedModifierSystem _speedModifier = default!;
|
||||
[Dependency] private readonly AlertsSystem _alerts = default!;
|
||||
[Dependency] private readonly MovementSpeedModifierSystem _speedModifier = default!;
|
||||
[Dependency] protected readonly SharedAppearanceSystem Appearance = default!;
|
||||
[Dependency] private readonly SharedAudioSystem _audio = default!;
|
||||
[Dependency] private readonly SharedBodySystem _body = default!;
|
||||
[Dependency] protected readonly SharedContainerSystem Container = default!;
|
||||
[Dependency] private readonly SharedDoAfterSystem _doAfter = default!;
|
||||
[Dependency] private readonly SharedHandsSystem _hands = default!;
|
||||
[Dependency] protected readonly SharedPopupSystem Popup = default!;
|
||||
[Dependency] private readonly StaminaSystem _stamina = default!;
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
|
||||
SubscribeLocalEvent<EnsnareableComponent, ComponentInit>(OnEnsnareInit);
|
||||
SubscribeLocalEvent<EnsnareableComponent, RefreshMovementSpeedModifiersEvent>(MovementSpeedModify);
|
||||
SubscribeLocalEvent<EnsnareableComponent, EnsnareEvent>(OnEnsnare);
|
||||
SubscribeLocalEvent<EnsnareableComponent, EnsnareRemoveEvent>(OnEnsnareRemove);
|
||||
SubscribeLocalEvent<EnsnareableComponent, EnsnaredChangedEvent>(OnEnsnareChange);
|
||||
SubscribeLocalEvent<EnsnareableComponent, ComponentGetState>(OnGetState);
|
||||
SubscribeLocalEvent<EnsnareableComponent, ComponentHandleState>(OnHandleState);
|
||||
SubscribeLocalEvent<EnsnareableComponent, AfterAutoHandleStateEvent>(OnHandleState);
|
||||
SubscribeLocalEvent<EnsnareableComponent, StrippingEnsnareButtonPressed>(OnStripEnsnareMessage);
|
||||
SubscribeLocalEvent<EnsnareableComponent, RemoveEnsnareAlertEvent>(OnRemoveEnsnareAlert);
|
||||
SubscribeLocalEvent<EnsnareableComponent, EnsnareableDoAfterEvent>(OnDoAfter);
|
||||
|
||||
SubscribeLocalEvent<EnsnaringComponent, ComponentRemove>(OnComponentRemove);
|
||||
SubscribeLocalEvent<EnsnaringComponent, StepTriggerAttemptEvent>(AttemptStepTrigger);
|
||||
SubscribeLocalEvent<EnsnaringComponent, StepTriggeredOffEvent>(OnStepTrigger);
|
||||
SubscribeLocalEvent<EnsnaringComponent, ThrowDoHitEvent>(OnThrowHit);
|
||||
SubscribeLocalEvent<EnsnaringComponent, AttemptPacifiedThrowEvent>(OnAttemptPacifiedThrow);
|
||||
}
|
||||
|
||||
private void OnHandleState(EntityUid uid, EnsnareableComponent component, ref ComponentHandleState args)
|
||||
protected virtual void OnEnsnareInit(Entity<EnsnareableComponent> ent, ref ComponentInit args)
|
||||
{
|
||||
if (args.Current is not EnsnareableComponentState state)
|
||||
return;
|
||||
ent.Comp.Container = Container.EnsureContainer<Container>(ent.Owner, "ensnare");
|
||||
}
|
||||
|
||||
if (state.IsEnsnared == component.IsEnsnared)
|
||||
return;
|
||||
|
||||
component.IsEnsnared = state.IsEnsnared;
|
||||
private void OnHandleState(EntityUid uid, EnsnareableComponent component, ref AfterAutoHandleStateEvent args)
|
||||
{
|
||||
RaiseLocalEvent(uid, new EnsnaredChangedEvent(component.IsEnsnared));
|
||||
}
|
||||
|
||||
private void OnGetState(EntityUid uid, EnsnareableComponent component, ref ComponentGetState args)
|
||||
private void OnDoAfter(EntityUid uid, EnsnareableComponent component, DoAfterEvent args)
|
||||
{
|
||||
args.State = new EnsnareableComponentState(component.IsEnsnared);
|
||||
if (args.Args.Target == null)
|
||||
return;
|
||||
|
||||
if (args.Handled || !TryComp<EnsnaringComponent>(args.Args.Used, out var ensnaring))
|
||||
return;
|
||||
|
||||
if (args.Cancelled || !Container.Remove(args.Args.Used.Value, component.Container))
|
||||
{
|
||||
if (args.User == args.Target)
|
||||
Popup.PopupPredicted(Loc.GetString("ensnare-component-try-free-fail", ("ensnare", args.Args.Used)), uid, args.User, PopupType.MediumCaution);
|
||||
else if (args.Target != null)
|
||||
Popup.PopupPredicted(Loc.GetString("ensnare-component-try-free-fail-other", ("ensnare", args.Args.Used), ("user", args.Target)), uid, args.User, PopupType.MediumCaution);
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
component.IsEnsnared = component.Container.ContainedEntities.Count > 0;
|
||||
Dirty(uid, component);
|
||||
ensnaring.Ensnared = null;
|
||||
|
||||
_hands.PickupOrDrop(args.Args.User, args.Args.Used.Value);
|
||||
|
||||
if (args.User == args.Target)
|
||||
Popup.PopupPredicted(Loc.GetString("ensnare-component-try-free-complete", ("ensnare", args.Args.Used)), uid, args.User, PopupType.Medium);
|
||||
else if (args.Target != null)
|
||||
Popup.PopupPredicted(Loc.GetString("ensnare-component-try-free-complete-other", ("ensnare", args.Args.Used), ("user", args.Target)), uid, args.User, PopupType.Medium);
|
||||
|
||||
UpdateAlert(args.Args.Target.Value, component);
|
||||
var ev = new EnsnareRemoveEvent(ensnaring.WalkSpeed, ensnaring.SprintSpeed);
|
||||
RaiseLocalEvent(uid, ev);
|
||||
|
||||
args.Handled = true;
|
||||
}
|
||||
|
||||
private void OnEnsnare(EntityUid uid, EnsnareableComponent component, EnsnareEvent args)
|
||||
@@ -85,4 +145,178 @@ public abstract class SharedEnsnareableSystem : EntitySystem
|
||||
|
||||
args.ModifySpeed(component.WalkSpeed, component.SprintSpeed);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Used where you want to try to free an entity with the <see cref="EnsnareableComponent"/>
|
||||
/// </summary>
|
||||
/// <param name="target">The entity that will be freed</param>
|
||||
/// <param name="user">The entity that is freeing the target</param>
|
||||
/// <param name="ensnare">The entity used to ensnare</param>
|
||||
/// <param name="component">The ensnaring component</param>
|
||||
public void TryFree(EntityUid target, EntityUid user, EntityUid ensnare, EnsnaringComponent component)
|
||||
{
|
||||
// Don't do anything if they don't have the ensnareable component.
|
||||
if (!HasComp<EnsnareableComponent>(target))
|
||||
return;
|
||||
|
||||
var freeTime = user == target ? component.BreakoutTime : component.FreeTime;
|
||||
var breakOnMove = !component.CanMoveBreakout;
|
||||
|
||||
var doAfterEventArgs = new DoAfterArgs(EntityManager, user, freeTime, new EnsnareableDoAfterEvent(), target, target: target, used: ensnare)
|
||||
{
|
||||
BreakOnMove = breakOnMove,
|
||||
BreakOnDamage = false,
|
||||
NeedHand = true,
|
||||
BreakOnDropItem = false,
|
||||
};
|
||||
|
||||
if (!_doAfter.TryStartDoAfter(doAfterEventArgs))
|
||||
return;
|
||||
|
||||
if (user == target)
|
||||
Popup.PopupPredicted(Loc.GetString("ensnare-component-try-free", ("ensnare", ensnare)), target, target);
|
||||
else
|
||||
Popup.PopupPredicted(Loc.GetString("ensnare-component-try-free-other", ("ensnare", ensnare), ("user", Identity.Entity(target, EntityManager))), user, user);
|
||||
}
|
||||
|
||||
private void OnStripEnsnareMessage(EntityUid uid, EnsnareableComponent component, StrippingEnsnareButtonPressed args)
|
||||
{
|
||||
foreach (var entity in component.Container.ContainedEntities)
|
||||
{
|
||||
if (!TryComp<EnsnaringComponent>(entity, out var ensnaring))
|
||||
continue;
|
||||
|
||||
TryFree(uid, args.Actor, entity, ensnaring);
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
private void OnAttemptPacifiedThrow(Entity<EnsnaringComponent> ent, ref AttemptPacifiedThrowEvent args)
|
||||
{
|
||||
args.Cancel("pacified-cannot-throw-snare");
|
||||
}
|
||||
|
||||
private void OnRemoveEnsnareAlert(Entity<EnsnareableComponent> ent, ref RemoveEnsnareAlertEvent args)
|
||||
{
|
||||
if (args.Handled)
|
||||
return;
|
||||
|
||||
foreach (var ensnare in ent.Comp.Container.ContainedEntities)
|
||||
{
|
||||
if (!TryComp<EnsnaringComponent>(ensnare, out var ensnaringComponent))
|
||||
continue;
|
||||
|
||||
TryFree(ent, ent, ensnare, ensnaringComponent);
|
||||
|
||||
args.Handled = true;
|
||||
// Only one snare at a time.
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
private void OnComponentRemove(EntityUid uid, EnsnaringComponent component, ComponentRemove args)
|
||||
{
|
||||
if (!TryComp<EnsnareableComponent>(component.Ensnared, out var ensnared))
|
||||
return;
|
||||
|
||||
if (ensnared.IsEnsnared)
|
||||
ForceFree(uid, component);
|
||||
}
|
||||
|
||||
private void AttemptStepTrigger(EntityUid uid, EnsnaringComponent component, ref StepTriggerAttemptEvent args)
|
||||
{
|
||||
args.Continue = true;
|
||||
}
|
||||
|
||||
private void OnStepTrigger(EntityUid uid, EnsnaringComponent component, ref StepTriggeredOffEvent args)
|
||||
{
|
||||
TryEnsnare(args.Tripper, uid, component);
|
||||
}
|
||||
|
||||
private void OnThrowHit(EntityUid uid, EnsnaringComponent component, ThrowDoHitEvent args)
|
||||
{
|
||||
if (!component.CanThrowTrigger)
|
||||
return;
|
||||
|
||||
if (TryEnsnare(args.Target, uid, component))
|
||||
{
|
||||
_audio.PlayPvs(component.EnsnareSound, uid);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Used where you want to try to ensnare an entity with the <see cref="EnsnareableComponent"/>
|
||||
/// </summary>
|
||||
/// <param name="target">The entity that will be ensnared</param>
|
||||
/// <paramref name="ensnare"> The entity that is used to ensnare</param>
|
||||
/// <param name="component">The ensnaring component</param>
|
||||
public bool TryEnsnare(EntityUid target, EntityUid ensnare, EnsnaringComponent component)
|
||||
{
|
||||
//Don't do anything if they don't have the ensnareable component.
|
||||
if (!TryComp<EnsnareableComponent>(target, out var ensnareable))
|
||||
return false;
|
||||
|
||||
// Need to insert before free legs check.
|
||||
Container.Insert(ensnare, ensnareable.Container);
|
||||
|
||||
var legs = _body.GetBodyChildrenOfType(target, BodyPartType.Leg).Count();
|
||||
var ensnaredLegs = (2 * ensnareable.Container.ContainedEntities.Count);
|
||||
var freeLegs = legs - ensnaredLegs;
|
||||
|
||||
if (freeLegs > 0)
|
||||
return false;
|
||||
|
||||
// Apply stamina damage to target if they weren't ensnared before.
|
||||
if (ensnareable.IsEnsnared != true)
|
||||
{
|
||||
if (TryComp<StaminaComponent>(target, out var stamina))
|
||||
{
|
||||
_stamina.TakeStaminaDamage(target, component.StaminaDamage, with: ensnare, component: stamina);
|
||||
}
|
||||
}
|
||||
|
||||
component.Ensnared = target;
|
||||
ensnareable.IsEnsnared = true;
|
||||
Dirty(target, ensnareable);
|
||||
|
||||
UpdateAlert(target, ensnareable);
|
||||
var ev = new EnsnareEvent(component.WalkSpeed, component.SprintSpeed);
|
||||
RaiseLocalEvent(target, ev);
|
||||
return true;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Used to force free someone for things like if the <see cref="EnsnaringComponent"/> is removed
|
||||
/// </summary>
|
||||
public void ForceFree(EntityUid ensnare, EnsnaringComponent component)
|
||||
{
|
||||
if (component.Ensnared == null)
|
||||
return;
|
||||
|
||||
if (!TryComp<EnsnareableComponent>(component.Ensnared, out var ensnareable))
|
||||
return;
|
||||
|
||||
var target = component.Ensnared.Value;
|
||||
|
||||
Container.Remove(ensnare, ensnareable.Container, force: true);
|
||||
ensnareable.IsEnsnared = ensnareable.Container.ContainedEntities.Count > 0;
|
||||
Dirty(component.Ensnared.Value, ensnareable);
|
||||
component.Ensnared = null;
|
||||
|
||||
UpdateAlert(target, ensnareable);
|
||||
var ev = new EnsnareRemoveEvent(component.WalkSpeed, component.SprintSpeed);
|
||||
RaiseLocalEvent(ensnare, ev);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Update the Ensnared alert for an entity.
|
||||
/// </summary>
|
||||
/// <param name="target">The entity that has been affected by a snare</param>
|
||||
public void UpdateAlert(EntityUid target, EnsnareableComponent component)
|
||||
{
|
||||
if (!component.IsEnsnared)
|
||||
_alerts.ClearAlert(target, component.EnsnaredAlert);
|
||||
else
|
||||
_alerts.ShowAlert(target, component.EnsnaredAlert);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user