#nullable enable using Content.Server.GameObjects.Components.Strap; using Content.Server.GameObjects.EntitySystems; using Content.Server.Interfaces; using Content.Server.Interfaces.GameObjects.Components.Interaction; using Content.Server.Mobs; using Content.Server.Utility; using Content.Shared.GameObjects; using Content.Shared.GameObjects.Components.Mobs; using Content.Shared.GameObjects.Components.Strap; using Content.Shared.GameObjects.EntitySystems; using Robust.Server.GameObjects; using Robust.Server.GameObjects.EntitySystems; using Robust.Shared.GameObjects; using Robust.Shared.Interfaces.GameObjects; using Robust.Shared.IoC; using Robust.Shared.Localization; using Robust.Shared.Maths; using Robust.Shared.Serialization; using Robust.Shared.ViewVariables; namespace Content.Server.GameObjects.Components.Mobs { [RegisterComponent] public class BuckleComponent : SharedBuckleComponent, IInteractHand, IDragDrop { #pragma warning disable 649 [Dependency] private readonly IEntitySystemManager _entitySystem = default!; [Dependency] private readonly IServerNotifyManager _notifyManager = default!; #pragma warning restore 649 private StrapComponent? _buckledTo; private int _size; [ViewVariables] public StrapComponent? BuckledTo { get => _buckledTo; private set { _buckledTo = value; Dirty(); } } [ViewVariables] protected override bool Buckled => BuckledTo != null; [ViewVariables] public int Size => _size; private void BuckleStatus() { if (Owner.TryGetComponent(out ServerStatusEffectsComponent status)) { status.ChangeStatusEffectIcon(StatusEffect.Buckled, Buckled ? "/Textures/Interface/StatusEffects/Buckle/buckled.png" : "/Textures/Interface/StatusEffects/Buckle/unbuckled.png"); } } private bool TryBuckle(IEntity user, IEntity to) { if (user == null || user == to) { return false; } if (!ActionBlockerSystem.CanInteract(user)) { _notifyManager.PopupMessage(user, user, Loc.GetString("You can't do that!")); return false; } var strapPosition = Owner.Transform.MapPosition; var range = SharedInteractionSystem.InteractionRange / 2; if (!InteractionChecks.InRangeUnobstructed(user, strapPosition, range)) { _notifyManager.PopupMessage(user, user, Loc.GetString("You can't reach there!")); return false; } if (!user.TryGetComponent(out HandsComponent hands)) { _notifyManager.PopupMessage(user, user, Loc.GetString("You don't have hands!")); return false; } if (hands.GetActiveHand != null) { _notifyManager.PopupMessage(user, user, Loc.GetString("Your hand isn't free!")); return false; } if (Buckled) { _notifyManager.PopupMessage(Owner, user, Loc.GetString(Owner == user ? "You are already buckled in!" : "{0:They} are already buckled in!", Owner)); return false; } if (!to.TryGetComponent(out StrapComponent strap)) { _notifyManager.PopupMessage(Owner, user, Loc.GetString(Owner == user ? "You can't buckle yourself there!" : "You can't buckle {0:them} there!", Owner)); return false; } var parent = to.Transform.Parent; while (parent != null) { if (parent == user.Transform) { _notifyManager.PopupMessage(Owner, user, Loc.GetString(Owner == user ? "You can't buckle yourself there!" : "You can't buckle {0:them} there!", Owner)); return false; } parent = parent.Parent; } if (!strap.HasSpace(this)) { _notifyManager.PopupMessage(Owner, user, Loc.GetString(Owner == user ? "You can't fit there!" : "{0:They} can't fit there!", Owner)); return false; } _entitySystem.GetEntitySystem() .PlayFromEntity(strap.BuckleSound, Owner); if (!strap.TryAdd(this)) { _notifyManager.PopupMessage(Owner, user, Loc.GetString(Owner == user ? "You can't buckle yourself there!" : "You can't buckle {0:them} there!", Owner)); return false; } BuckledTo = strap; if (Owner.TryGetComponent(out AppearanceComponent appearance)) { appearance.SetData(BuckleVisuals.Buckled, true); } var ownTransform = Owner.Transform; var strapTransform = strap.Owner.Transform; ownTransform.GridPosition = strapTransform.GridPosition; ownTransform.AttachParent(strapTransform); switch (strap.Position) { case StrapPosition.None: ownTransform.WorldRotation = strapTransform.WorldRotation; break; case StrapPosition.Stand: StandingStateHelper.Standing(Owner); ownTransform.WorldRotation = strapTransform.WorldRotation; break; case StrapPosition.Down: StandingStateHelper.Down(Owner); ownTransform.WorldRotation = Angle.South; break; } BuckleStatus(); return true; } public bool TryUnbuckle(IEntity user, bool force = false) { if (BuckledTo == null) { return false; } if (!force) { if (!ActionBlockerSystem.CanInteract(user)) { _notifyManager.PopupMessage(user, user, Loc.GetString("You can't do that!")); return false; } var strapPosition = Owner.Transform.MapPosition; var range = SharedInteractionSystem.InteractionRange / 2; if (!InteractionChecks.InRangeUnobstructed(user, strapPosition, range)) { _notifyManager.PopupMessage(user, user, Loc.GetString("You can't reach there!")); return false; } } if (BuckledTo.Owner.TryGetComponent(out StrapComponent strap)) { strap.Remove(this); _entitySystem.GetEntitySystem() .PlayFromEntity(strap.UnbuckleSound, Owner); } Owner.Transform.DetachParent(); Owner.Transform.WorldRotation = BuckledTo.Owner.Transform.WorldRotation; BuckledTo = null; if (Owner.TryGetComponent(out AppearanceComponent appearance)) { appearance.SetData(BuckleVisuals.Buckled, false); } if (Owner.TryGetComponent(out StunnableComponent stunnable) && stunnable.KnockedDown) { StandingStateHelper.Down(Owner); } else { StandingStateHelper.Standing(Owner); } if (Owner.TryGetComponent(out SpeciesComponent species)) { species.CurrentDamageState.EnterState(Owner); } BuckleStatus(); return true; } public bool ToggleBuckle(IEntity user, IEntity to) { if (BuckledTo == null) { return TryBuckle(user, to); } else if (BuckledTo.Owner == to) { return TryUnbuckle(user); } else { return false; } } public override void ExposeData(ObjectSerializer serializer) { base.ExposeData(serializer); serializer.DataField(ref _size, "size", 100); } protected override void Startup() { base.Startup(); BuckleStatus(); } public override void OnRemove() { base.OnRemove(); if (BuckledTo != null && BuckledTo.Owner.TryGetComponent(out StrapComponent strap)) { strap.Remove(this); } BuckledTo = null; BuckleStatus(); } public override ComponentState GetComponentState() { return new BuckleComponentState(Buckled); } bool IInteractHand.InteractHand(InteractHandEventArgs eventArgs) { return TryUnbuckle(eventArgs.User); } bool IDragDrop.DragDrop(DragDropEventArgs eventArgs) { return TryBuckle(eventArgs.User, eventArgs.Target); } [Verb] private sealed class BuckleVerb : Verb { protected override void GetData(IEntity user, BuckleComponent component, VerbData data) { if (!ActionBlockerSystem.CanInteract(user) || !component.Buckled) { data.Visibility = VerbVisibility.Invisible; return; } data.Text = Loc.GetString("Unbuckle"); } protected override void Activate(IEntity user, BuckleComponent component) { component.TryUnbuckle(user); } } } }