fish-station/Content.Shared/Hands/EntitySystems/SharedHandsSystem.Pickup.cs

using Content.Shared.Database;
using Content.Shared.Hands.Components;
using Content.Shared.Item;
using Robust.Shared.Containers;
using Robust.Shared.Physics;
using Robust.Shared.Physics.Components;
using Robust.Shared.Utility;
using Robust.Shared.Timing; // Sunrise-add
using Content.Shared._Sunrise.Item; // Sunrise-add

namespace Content.Shared.Hands.EntitySystems;

public abstract partial class SharedHandsSystem
{
    [Dependency] private readonly IGameTiming _timing = default!; // Sunrise-add

    private void InitializePickup()
    {
        SubscribeLocalEvent<HandsComponent, EntInsertedIntoContainerMessage>(HandleEntityInserted);
    }

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

        var didEquip = new DidEquipHandEvent(uid, args.Entity, hand.Value);
        RaiseLocalEvent(uid, didEquip);

        var gotEquipped = new GotEquippedHandEvent(uid, args.Entity, hand.Value);
        RaiseLocalEvent(args.Entity, gotEquipped);
    }

    /// <summary>
    ///     Maximum pickup distance for which the pickup animation plays.
    /// </summary>
    public const float MaxAnimationRange = 10;

    /// <summary>
    ///     Attempts to pick up an item into any empty hand. Prioritizes the currently active hand.
    /// </summary>
    /// <remarks>
    ///     If one empty hand fails to pick up the item, this will NOT check other hands. If ever hand-specific item
    ///     restrictions are added, there a might need to be a TryPickupAllHands or something like that.
    /// </remarks>
    public bool TryPickupAnyHand(
        EntityUid uid,
        EntityUid entity,
        bool checkActionBlocker = true,
        bool animateUser = false,
        bool animate = true,
        HandsComponent? handsComp = null,
        ItemComponent? item = null)
    {
        if (!Resolve(uid, ref handsComp, false))
            return false;

        if (!TryGetEmptyHand((uid, handsComp), out var hand))
            return false;

        return TryPickup(uid, entity, hand, checkActionBlocker, animateUser, animate, handsComp, item);
    }

    /// <summary>
    ///     Tries to pick up an entity to a specific hand. If no explicit hand is specified, defaults to using the currently active hand.
    /// </summary>
    public bool TryPickup(
        EntityUid uid,
        EntityUid entity,
        string? handId = null,
        bool checkActionBlocker = true,
        bool animateUser = false,
        bool animate = true,
        HandsComponent? handsComp = null,
        ItemComponent? item = null)
    {
        if (!Resolve(uid, ref handsComp, false))
            return false;

        handId ??= handsComp.ActiveHandId;

        if (handId == null)
            return false;

        if (!Resolve(entity, ref item, false))
            return false;

        if (!CanPickupToHand(uid, entity, handId, checkActionBlocker: checkActionBlocker, showPopup: true, handsComp: handsComp, item: item))
            return false;

        if (!BeforeDoPickup((uid, handsComp), entity))
            return false;

        // Sunrise-start
        if (!CheckRepickupDuration(entity))
            return false;
        // Sunrise-end

        if (animate)
        {
            var xform = Transform(uid);
            var coordinateEntity = xform.ParentUid.IsValid() ? xform.ParentUid : uid;
            var itemXform = Transform(entity);
            var itemPos = TransformSystem.GetMapCoordinates(entity, xform: itemXform);

            if (itemPos.MapId == xform.MapID
                && (itemPos.Position - TransformSystem.GetMapCoordinates(uid, xform: xform).Position).Length() <= MaxAnimationRange
                && MetaData(entity).VisibilityMask == MetaData(uid).VisibilityMask) // Don't animate aghost pickups.
            {
                var initialPosition = TransformSystem.ToCoordinates(coordinateEntity, itemPos);
                _storage.PlayPickupAnimation(entity, initialPosition, xform.Coordinates, itemXform.LocalRotation, uid);
            }
        }
        DoPickup(uid, handId, entity, handsComp);

        return true;
    }

    // Sunrise-start
    /// <summary>
    /// Проверка для <see cref="ItemRepickupCooldownComponent"/>. Остановка досрочного поднятия
    /// Всю логику нельзя запихнуть в <see cref="ItemRepickupCooldownSystem"/>, потому что
    /// <see cref="PickupAttemptEvent"/> вызывается на мобе, который пытается поднять, а не на цели. Создавать новый
    /// ивент... Это пока не надо, мне кажется только больше конфликтов будет
    /// </summary>
    /// <param name="entity">Для какой сущности проверять</param>
    /// <returns>Результат проверки</returns>
    public bool CheckRepickupDuration(Entity<ItemRepickupCooldownComponent?> entity)
    {
        // У сущности просто нет запрашиваемого компонента
        if (!Resolve(entity.Owner, ref entity.Comp, logMissing: false))
            return true;

        if (entity.Comp.PrevDrop is not null &&
            _timing.CurTime - entity.Comp.PrevDrop < entity.Comp.CooldownDuration)
            return false;

        return true;
    }
    // Sunrise-end

    /// <summary>
    /// Tries to pick up an entity into a hand, forcing to drop an item if its not free.
    /// By default it does check if it's possible to drop items.
    /// </summary>
    public bool TryForcePickup(
        Entity<HandsComponent?> ent,
        EntityUid entity,
        string hand,
        bool checkActionBlocker = true,
        bool animate = true,
        HandsComponent? handsComp = null,
        ItemComponent? item = null)
    {
        if (!Resolve(ent, ref ent.Comp, false))
            return false;

        TryDrop(ent, hand, checkActionBlocker: checkActionBlocker);

        return TryPickup(ent, entity, hand, checkActionBlocker, animate: animate, handsComp: handsComp, item: item);
    }

    /// <summary>
    ///     Tries to pick up an entity into any hand, forcing to drop an item if there are no free hands
    ///     By default it does check if it's possible to drop items
    /// </summary>
    public bool TryForcePickupAnyHand(EntityUid uid, EntityUid entity, bool checkActionBlocker = true, HandsComponent? handsComp = null, ItemComponent? item = null)
    {
        if (!Resolve(uid, ref handsComp, false))
            return false;

        if (TryPickupAnyHand(uid, entity, checkActionBlocker: checkActionBlocker, handsComp: handsComp))
            return true;

        foreach (var hand in handsComp.Hands.Keys)
        {
            if (TryDrop((uid, handsComp), hand, checkActionBlocker: checkActionBlocker) &&
                TryPickup(uid, entity, hand, checkActionBlocker: checkActionBlocker, handsComp: handsComp))
            {
                return true;
            }
        }
        return false;
    }

    /// <summary>
    /// Checks whether a given item will fit into the user's first free hand.
    /// Unless otherwise specified, this will also check the general CanPickup action blocker.
    /// </summary>
    public bool CanPickupAnyHand(EntityUid uid, EntityUid entity, bool checkActionBlocker = true, bool showPopup = false, HandsComponent? handsComp = null, ItemComponent? item = null)
    {
        if (!Resolve(uid, ref handsComp, false))
            return false;

        if (!TryGetEmptyHand((uid, handsComp), out var hand))
            return false;

        return CanPickupToHand(uid, entity, hand, checkActionBlocker, showPopup, handsComp, item);
    }

    /// <summary>
    /// Checks whether a given item will fit into a specific user's hand.
    /// Unless otherwise specified, this will also check the general CanPickup action blocker.
    /// </summary>
    public bool CanPickupToHand(EntityUid uid, EntityUid entity, string handId, bool checkActionBlocker = true, bool showPopup = false, HandsComponent? handsComp = null, ItemComponent? item = null)
    {
        if (!Resolve(uid, ref handsComp, false))
            return false;

        if (!ContainerSystem.TryGetContainer(uid, handId, out var handContainer))
            return false;

        if (handContainer.ContainedEntities.FirstOrNull() != null)
            return false;

        // Huh, seems kinda weird that this system passes item comp around
        // everywhere but it's never actually used besides being resolved.
        // I wouldn't be surprised if there's some API simplifications that
        // could be made with respect to that.
        if (!Resolve(entity, ref item, false))
            return false;

        if (TryComp(entity, out PhysicsComponent? physics) && physics.BodyType == BodyType.Static)
            return false;

        if (checkActionBlocker && !_actionBlocker.CanPickup(uid, entity, showPopup))
            return false;

        if (!CheckWhitelists((uid, handsComp), handId, entity))
            return false;

        if (ContainerSystem.TryGetContainingContainer((entity, null, null), out var container))
        {
            if (!ContainerSystem.CanRemove(entity, container))
                return false;

            if (_inventory.TryGetSlotEntity(uid, container.ID, out var slotEnt) &&
                slotEnt == entity &&
                !_inventory.CanUnequip(uid, entity, container.ID, out _))
                return false;
        }

        // check can insert (including raising attempt events).
        return ContainerSystem.CanInsert(entity, handContainer);
    }

    /// <summary>
    ///     Puts an item into any hand, preferring the active hand, or puts it on the floor.
    /// </summary>
    /// <param name="dropNear">If true, the item will be dropped near the owner of the hand if possible.</param>
    public void PickupOrDrop(
        EntityUid? uid,
        EntityUid entity,
        bool checkActionBlocker = true,
        bool animateUser = false,
        bool animate = true,
        bool dropNear = false,
        HandsComponent? handsComp = null,
        ItemComponent? item = null)
    {
        if (uid == null
            || !Resolve(uid.Value, ref handsComp, false)
            || !TryPickupAnyHand(uid.Value, entity, checkActionBlocker, animateUser, animate, handsComp, item))
        {
            // TODO make this check upwards for any container, and parent to that.
            // Currently this just checks the direct parent, so items can still teleport through containers.
            ContainerSystem.AttachParentToContainerOrGrid((entity, Transform(entity)));

            if (dropNear && uid.HasValue)
            {
                TransformSystem.PlaceNextTo(entity, uid.Value);
            }
        }
    }

    /// <summary>
    /// Small helper function meant as a last step before <see cref="DoPickup"/>
    /// is called. Used to run a cancelable before pickup event that can have
    /// side effects, unlike the side effect free <see cref="GettingPickedUpAttemptEvent"/>.
    /// </summary>
    private bool BeforeDoPickup(Entity<HandsComponent?> user, EntityUid item)
    {
        if (!Resolve(user, ref user.Comp))
            return false;

        var userEv = new BeforeEquippingHandEvent(item);
        RaiseLocalEvent(user, ref userEv);

        if (userEv.Cancelled)
            return false;

        var itemEv = new BeforeGettingEquippedHandEvent(user);
        RaiseLocalEvent(item, ref itemEv);

        return !itemEv.Cancelled;
    }

    /// <summary>
    ///     Puts an entity into the player's hand, assumes that the insertion is allowed. In general, you should not be calling this function directly.
    /// </summary>
    public virtual void DoPickup(EntityUid uid, string hand, EntityUid entity, HandsComponent? hands = null, bool log = true)
    {
        if (!Resolve(uid, ref hands))
            return;

        if (!ContainerSystem.TryGetContainer(uid, hand, out var handContainer))
            return;

        if (handContainer.ContainedEntities.FirstOrNull() != null)
            return;

        if (!ContainerSystem.Insert(entity, handContainer))
        {
            Log.Error($"Failed to insert {ToPrettyString(entity)} into users hand container when picking up. User: {ToPrettyString(uid)}. Hand: {hand}.");
            return;
        }

        _interactionSystem.DoContactInteraction(uid, entity); //Possibly fires twice if manually picked up via interacting with the object

        if (log)
            _adminLogger.Add(LogType.Pickup, LogImpact.Low, $"{ToPrettyString(uid):user} picked up {ToPrettyString(entity):entity}");

        Dirty(uid, hands);

        if (hand == hands.ActiveHandId)
            RaiseLocalEvent(entity, new HandSelectedEvent(uid));
    }
}