tbd-station-14/Content.Server/AI/Utility/Considerations/Inventory/CanPutTargetInHandsCon.cs

using Content.Server.AI.Utility.Curves;
using Content.Server.AI.WorldState;
using Content.Server.AI.WorldState.States;
using Content.Server.AI.WorldState.States.Hands;
using Content.Server.AI.WorldState.States.Inventory;
using Content.Server.GameObjects;

namespace Content.Server.AI.Utility.Considerations.Inventory
{
    public class CanPutTargetInHandsCon : Consideration
    {
        public CanPutTargetInHandsCon(IResponseCurve curve) : base(curve) {}

        public override float GetScore(Blackboard context)
        {
            // First check if target in inventory already
            // If not then check if we have a free hand
            var target = context.GetState<TargetEntityState>().GetValue();

            if (target == null || !target.HasComponent<ItemComponent>())
            {
                return 0.0f;
            }

            var inventory = context.GetState<InventoryState>().GetValue();

            foreach (var item in inventory)
            {
                if (item == target)
                {
                    return 1.0f;
                }
            }

            return context.GetState<AnyFreeHandState>().GetValue() ? 1.0f : 0.0f;
        }
    }
}