OldThink/Content.Shared/_White/Crossbow/PenetratedSystem.cs

using Content.Shared.Interaction;
using Content.Shared.Movement.Events;
using Content.Shared.Projectiles;
using Robust.Shared.Physics;
using Robust.Shared.Physics.Components;
using Robust.Shared.Physics.Systems;

namespace Content.Shared._White.Crossbow;

public sealed class PenetratedSystem : EntitySystem
{
    [Dependency] private readonly SharedPhysicsSystem _physics = default!;
    [Dependency] private readonly SharedTransformSystem _transform = default!;
    [Dependency] private readonly SharedProjectileSystem _projectile = default!;

    public override void Initialize()
    {
        base.Initialize();

        SubscribeLocalEvent<PenetratedComponent, MoveInputEvent>(OnMoveInput);
        SubscribeLocalEvent<PenetratedComponent, InteractHandEvent>(OnInteract);
    }

    private bool AttemptEmbedRemove(EntityUid uid, EntityUid user, PenetratedComponent component)
    {
        if (component is {ProjectileUid: not null, IsPinned: true})
        {
            if (!_projectile.AttemptEmbedRemove(component.ProjectileUid.Value, user))
                FreePenetrated(uid, component);
            else
                return true;
        }
        else if (component.ProjectileUid == null && TryComp(uid, out PhysicsComponent? physics) &&
                 physics.BodyType == BodyType.Static)
        {
            FreePenetrated(uid, component, physics);
        }
        return false;
    }

    private void OnInteract(EntityUid uid, PenetratedComponent component, InteractHandEvent args)
    {
        if (AttemptEmbedRemove(uid, args.User, component))
            args.Handled = true;
    }

    private void OnMoveInput(EntityUid uid, PenetratedComponent component, ref MoveInputEvent args)
    {
        AttemptEmbedRemove(uid, uid, component);
    }

    public void FreePenetrated(EntityUid uid, PenetratedComponent? penetrated = null, PhysicsComponent? physics = null)
    {
        var xform = Transform(uid);
        _transform.AttachToGridOrMap(uid, xform);

        if (Resolve(uid, ref physics, false))
        {
            _physics.SetBodyType(uid, BodyType.KinematicController, body: physics, xform: xform);
            _physics.WakeBody(uid, body: physics);
        }

        if (!Resolve(uid, ref penetrated, false))
            return;

        penetrated.ProjectileUid = null;
        penetrated.IsPinned = false;
        Dirty(uid, penetrated);
    }
}