CatStation/Content.Server/Resist/ResistLockerSystem.cs

using Content.Server.Popups;
using Content.Server.Storage.EntitySystems;
using Content.Shared.DoAfter;
using Content.Shared.Lock;
using Content.Shared.Movement.Events;
using Content.Shared.Popups;
using Content.Shared.Resist;
using Content.Shared.Storage.Components;
using Content.Shared.Tools.Components;
using Content.Shared.Tools.Systems;
using Content.Shared.ActionBlocker;

namespace Content.Server.Resist;

public sealed partial class ResistLockerSystem : EntitySystem
{
    [Dependency] private EntityStorageSystem _entityStorage = default!;
    [Dependency] private LockSystem _lockSystem = default!;
    [Dependency] private PopupSystem _popupSystem = default!;
    [Dependency] private SharedDoAfterSystem _doAfterSystem = default!;
    [Dependency] private WeldableSystem _weldable = default!;
    [Dependency] private ActionBlockerSystem _actionBlocker = default!;

    public override void Initialize()
    {
        base.Initialize();
        SubscribeLocalEvent<ResistLockerComponent, ContainerRelayMovementEntityEvent>(OnRelayMovement);
        SubscribeLocalEvent<ResistLockerComponent, ResistLockerDoAfterEvent>(OnDoAfter);
    }

    private void OnRelayMovement(EntityUid uid, ResistLockerComponent component, ref ContainerRelayMovementEntityEvent args)
    {
        if (component.IsResisting)
            return;

        if (!TryComp(uid, out EntityStorageComponent? storageComponent) || !storageComponent.OpenOnMove)
            return;

        if (!_actionBlocker.CanMove(args.Entity))
            return;

        if (TryComp<LockComponent>(uid, out var lockComponent) && lockComponent.Locked || _weldable.IsWelded(uid))
        {
            AttemptResist(args.Entity, uid, storageComponent, component);
        }
    }

    private void AttemptResist(EntityUid user, EntityUid target, EntityStorageComponent? storageComponent = null, ResistLockerComponent? resistLockerComponent = null)
    {
        if (!Resolve(target, ref storageComponent, ref resistLockerComponent))
            return;

        var doAfterEventArgs = new DoAfterArgs(EntityManager, user, resistLockerComponent.ResistTime, new ResistLockerDoAfterEvent(), target, target: target)
        {
            BreakOnMove = true,
            BreakOnDamage = true,
            NeedHand = false, //No hands 'cause we be kickin'
        };

        // Make sure the do after is able to start
        if (!_doAfterSystem.TryStartDoAfter(doAfterEventArgs))
            return;

        resistLockerComponent.IsResisting = true;
        _popupSystem.PopupEntity(Loc.GetString("resist-locker-component-start-resisting"), user, user, PopupType.Large);
    }

    // TODO: Convert to DoAfterAttemptEvent
    private void OnDoAfter(EntityUid uid, ResistLockerComponent component, DoAfterEvent args)
    {
        if (args.Handled)
            return;

        component.IsResisting = false;

        if (args.Target != uid)
            return;

        if (args.Cancelled)
        {
            _popupSystem.PopupEntity(Loc.GetString("resist-locker-component-resist-interrupted"), args.User, args.User, PopupType.Medium);
            return;
        }

        if (TryComp(uid, out EntityStorageComponent? storageComponent))
        {
            WeldableComponent? weldable = null;
            if (_weldable.IsWelded(uid, weldable))
                _weldable.SetWeldedState(uid, false, weldable);

            if (TryComp<LockComponent>(uid, out var lockComponent))
                _lockSystem.Unlock(uid, args.User, lockComponent);

            if (storageComponent.OpenOnMove)
                _entityStorage.TryOpenStorage(args.User, uid);
        }

        args.Handled = true;
    }
}