crystall-punk-14/Content.Shared/Hands/EntitySystems/SharedHandsSystem.Drop.cs

using System.Numerics;
using Content.Shared.Database;
using Content.Shared.Hands.Components;
using Content.Shared.Interaction;
using Content.Shared.Inventory.VirtualItem;
using Content.Shared.Storage.Components;
using Content.Shared.Tag;
using Robust.Shared.Containers;
using Robust.Shared.Map;
using Robust.Shared.Prototypes;
using Robust.Shared.Utility;

namespace Content.Shared.Hands.EntitySystems;

public abstract partial class SharedHandsSystem
{
    [Dependency] private readonly TagSystem _tagSystem = default!;

    private static readonly ProtoId<TagPrototype> BypassDropChecksTag = "BypassDropChecks";

    private void InitializeDrop()
    {
        SubscribeLocalEvent<HandsComponent, EntRemovedFromContainerMessage>(HandleEntityRemoved);
        SubscribeLocalEvent<HandsComponent, EntityStorageIntoContainerAttemptEvent>(OnEntityStorageDump);
    }

    protected virtual void HandleEntityRemoved(EntityUid uid, HandsComponent hands, EntRemovedFromContainerMessage args)
    {
        if (!TryGetHand(uid, args.Container.ID, out var hand))
        {
            return;
        }

        var gotUnequipped = new GotUnequippedHandEvent(uid, args.Entity, hand.Value);
        RaiseLocalEvent(args.Entity, gotUnequipped);

        var didUnequip = new DidUnequipHandEvent(uid, args.Entity, hand.Value);
        RaiseLocalEvent(uid, didUnequip);

        if (TryComp(args.Entity, out VirtualItemComponent? @virtual))
            _virtualSystem.DeleteVirtualItem((args.Entity, @virtual), uid);
    }


    private void OnEntityStorageDump(Entity<HandsComponent> ent, ref EntityStorageIntoContainerAttemptEvent args)
    {
        // If you're physically carrying an EntityStroage which tries to dump its contents out,
        // we want those contents to fall to the floor.
        args.Cancelled = true;
    }

    private bool ShouldIgnoreRestrictions(EntityUid user)
    {
        //Checks if the Entity is something that shouldn't care about drop distance or walls ie Aghost
        return !_tagSystem.HasTag(user, BypassDropChecksTag);
    }

    /// <summary>
    ///     Checks whether an entity can drop a given entity. Will return false if they are not holding the entity.
    /// </summary>
    public bool CanDrop(Entity<HandsComponent?> ent, EntityUid entity, bool checkActionBlocker = true)
    {
        if (!Resolve(ent, ref ent.Comp, false))
            return false;

        if (!IsHolding(ent, entity, out var hand))
            return false;

        return CanDropHeld(ent, hand, checkActionBlocker);
    }

    /// <summary>
    ///     Checks if the contents of a hand is able to be removed from its container.
    /// </summary>
    public bool CanDropHeld(EntityUid uid, string handId, bool checkActionBlocker = true)
    {
        if (!ContainerSystem.TryGetContainer(uid, handId, out var container))
            return false;

        if (container.ContainedEntities.FirstOrNull() is not {} held)
            return false;

        if (!ContainerSystem.CanRemove(held, container))
            return false;

        if (checkActionBlocker && !_actionBlocker.CanDrop(uid))
            return false;

        return true;
    }

    /// <summary>
    ///     Attempts to drop the item in the currently active hand.
    /// </summary>
    public bool TryDrop(Entity<HandsComponent?> ent, EntityCoordinates? targetDropLocation = null, bool checkActionBlocker = true, bool doDropInteraction = true)
    {
        if (!Resolve(ent, ref ent.Comp, false))
            return false;

        if (ent.Comp.ActiveHandId == null)
            return false;

        return TryDrop(ent, ent.Comp.ActiveHandId, targetDropLocation, checkActionBlocker, doDropInteraction);
    }

    /// <summary>
    ///     Drops an item at the target location.
    /// </summary>
    public bool TryDrop(Entity<HandsComponent?> ent, EntityUid entity, EntityCoordinates? targetDropLocation = null, bool checkActionBlocker = true, bool doDropInteraction = true)
    {
        if (!Resolve(ent, ref ent.Comp, false))
            return false;

        if (!IsHolding(ent, entity, out var hand))
            return false;

        return TryDrop(ent, hand, targetDropLocation, checkActionBlocker, doDropInteraction);
    }

    /// <summary>
    ///     Drops a hands contents at the target location.
    /// </summary>
    public bool TryDrop(Entity<HandsComponent?> ent, string handId, EntityCoordinates? targetDropLocation = null, bool checkActionBlocker = true, bool doDropInteraction = true)
    {
        if (!Resolve(ent, ref ent.Comp, false))
            return false;

        if (!CanDropHeld(ent, handId, checkActionBlocker))
            return false;

        if (!TryGetHeldItem(ent, handId, out var entity))
            return false;

        // if item is a fake item (like with pulling), just delete it rather than bothering with trying to drop it into the world
        if (TryComp(entity, out VirtualItemComponent? @virtual))
            _virtualSystem.DeleteVirtualItem((entity.Value, @virtual), ent);

        if (TerminatingOrDeleted(entity))
            return true;

        var itemXform = Transform(entity.Value);
        if (itemXform.MapUid == null)
            return true;

        var userXform = Transform(ent);
        var isInContainer = ContainerSystem.IsEntityOrParentInContainer(ent, xform: userXform);

        // if the user is in a container, drop the item inside the container
        if (isInContainer)
        {
            TransformSystem.DropNextTo((entity.Value, itemXform), (ent, userXform));
            return true;
        }

        // drop the item with heavy calculations from their hands and place it at the calculated interaction range position
        // The DoDrop is handle if there's no drop target
        DoDrop(ent, handId, doDropInteraction: doDropInteraction);

        // if there's no drop location stop here
        if (targetDropLocation == null)
            return true;

        // otherwise, also move dropped item and rotate it properly according to grid/map
        var (itemPos, itemRot) = TransformSystem.GetWorldPositionRotation(entity.Value);
        var origin = new MapCoordinates(itemPos, itemXform.MapID);
        var target = TransformSystem.ToMapCoordinates(targetDropLocation.Value);
        TransformSystem.SetWorldPositionRotation(entity.Value, GetFinalDropCoordinates(ent, origin, target, entity.Value), itemRot);
        return true;
    }

    /// <summary>
    ///     Attempts to move a held item from a hand into a container that is not another hand, without dropping it on the floor in-between.
    /// </summary>
    public bool TryDropIntoContainer(Entity<HandsComponent?> ent, EntityUid entity, BaseContainer targetContainer, bool checkActionBlocker = true)
    {
        if (!Resolve(ent, ref ent.Comp, false))
            return false;

        if (!IsHolding(ent, entity, out var hand))
            return false;

        if (!CanDropHeld(ent, hand, checkActionBlocker))
            return false;

        if (!ContainerSystem.CanInsert(entity, targetContainer))
            return false;

        DoDrop(ent, hand, false);
        ContainerSystem.Insert(entity, targetContainer);
        return true;
    }

    /// <summary>
    ///     Calculates the final location a dropped item will end up at, accounting for max drop range and collision along the targeted drop path, Does a check to see if a user should bypass those checks as well.
    /// </summary>
    private Vector2 GetFinalDropCoordinates(EntityUid user, MapCoordinates origin, MapCoordinates target, EntityUid held)
    {
        var dropVector = target.Position - origin.Position;
        var requestedDropDistance = dropVector.Length();
        var dropLength = dropVector.Length();

        if (ShouldIgnoreRestrictions(user))
        {
            if (dropVector.Length() > SharedInteractionSystem.InteractionRange)
            {
                dropVector = dropVector.Normalized() * SharedInteractionSystem.InteractionRange;
                target = new MapCoordinates(origin.Position + dropVector, target.MapId);
            }

            dropLength = _interactionSystem.UnobstructedDistance(origin, target, predicate: e => e == user || e == held);
        }

        if (dropLength < requestedDropDistance)
            return origin.Position + dropVector.Normalized() * dropLength;
        return target.Position;
    }

    /// <summary>
    ///     Removes the contents of a hand from its container. Assumes that the removal is allowed. In general, you should not be calling this directly.
    /// </summary>
    public virtual void DoDrop(Entity<HandsComponent?> ent,
        string handId,
        bool doDropInteraction = true,
        bool log = true)
    {
        if (!Resolve(ent, ref ent.Comp, false))
            return;

        if (!ContainerSystem.TryGetContainer(ent, handId, out var container))
            return;

        if (!TryGetHeldItem(ent, handId, out var entity))
            return;

        if (TerminatingOrDeleted(ent) || TerminatingOrDeleted(entity))
            return;

        if (!ContainerSystem.Remove(entity.Value, container))
        {
            Log.Error($"Failed to remove {ToPrettyString(entity)} from users hand container when dropping. User: {ToPrettyString(ent)}. Hand: {handId}.");
            return;
        }

        Dirty(ent);

        if (doDropInteraction)
            _interactionSystem.DroppedInteraction(ent, entity.Value);

        if (log)
            _adminLogger.Add(LogType.Drop, LogImpact.Low, $"{ToPrettyString(ent):user} dropped {ToPrettyString(entity):entity}");

        if (handId == ent.Comp.ActiveHandId)
            RaiseLocalEvent(entity.Value, new HandDeselectedEvent(ent));
    }
}