67 lines
2.0 KiB
C#
67 lines
2.0 KiB
C#
using Content.Server.GameObjects;
|
|
using Content.Server.GameObjects.Components.Items.Storage;
|
|
using Content.Server.GameObjects.EntitySystems.Click;
|
|
using Content.Server.Utility;
|
|
using Robust.Shared.Containers;
|
|
using Robust.Shared.Interfaces.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
|
|
namespace Content.Server.AI.Operators.Inventory
|
|
{
|
|
public class PickupEntityOperator : AiOperator
|
|
{
|
|
// Input variables
|
|
private readonly IEntity _owner;
|
|
private readonly IEntity _target;
|
|
|
|
public PickupEntityOperator(IEntity owner, IEntity target)
|
|
{
|
|
_owner = owner;
|
|
_target = target;
|
|
}
|
|
|
|
// TODO: When I spawn new entities they seem to duplicate clothing or something?
|
|
public override Outcome Execute(float frameTime)
|
|
{
|
|
if (_target == null ||
|
|
_target.Deleted ||
|
|
!_target.HasComponent<ItemComponent>() ||
|
|
ContainerHelpers.IsInContainer(_target) ||
|
|
!InteractionChecks.InRangeUnobstructed(_owner, _target.Transform.MapPosition))
|
|
{
|
|
return Outcome.Failed;
|
|
}
|
|
|
|
if (!_owner.TryGetComponent(out HandsComponent handsComponent))
|
|
{
|
|
return Outcome.Failed;
|
|
}
|
|
|
|
var emptyHands = false;
|
|
|
|
foreach (var hand in handsComponent.ActivePriorityEnumerable())
|
|
{
|
|
if (handsComponent.GetHand(hand) == null)
|
|
{
|
|
if (handsComponent.ActiveIndex != hand)
|
|
{
|
|
handsComponent.ActiveIndex = hand;
|
|
}
|
|
|
|
emptyHands = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (!emptyHands)
|
|
{
|
|
return Outcome.Failed;
|
|
}
|
|
|
|
var interactionSystem = IoCManager.Resolve<IEntitySystemManager>().GetEntitySystem<InteractionSystem>();
|
|
interactionSystem.Interaction(_owner, _target);
|
|
return Outcome.Success;
|
|
}
|
|
}
|
|
}
|