72 lines
2.8 KiB
C#
72 lines
2.8 KiB
C#
using Content.Server.Hands.Components;
|
|
using Content.Server.Items;
|
|
using Content.Server.Nutrition.Components;
|
|
using Content.Shared.Nutrition.Components;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Random;
|
|
|
|
namespace Content.Server.AI.Operators.Nutrition
|
|
{
|
|
public class UseFoodInInventoryOperator : AiOperator
|
|
{
|
|
private readonly IEntity _owner;
|
|
private readonly IEntity _target;
|
|
private float _interactionCooldown;
|
|
|
|
public UseFoodInInventoryOperator(IEntity owner, IEntity target)
|
|
{
|
|
_owner = owner;
|
|
_target = target;
|
|
}
|
|
|
|
public override Outcome Execute(float frameTime)
|
|
{
|
|
if (_interactionCooldown >= 0)
|
|
{
|
|
_interactionCooldown -= frameTime;
|
|
return Outcome.Continuing;
|
|
}
|
|
|
|
// TODO: Also have this check storage a la backpack etc.
|
|
if ((!IoCManager.Resolve<IEntityManager>().EntityExists(_target.Uid) ? EntityLifeStage.Deleted : IoCManager.Resolve<IEntityManager>().GetComponent<MetaDataComponent>(_target.Uid).EntityLifeStage) >= EntityLifeStage.Deleted ||
|
|
!IoCManager.Resolve<IEntityManager>().TryGetComponent(_owner.Uid, out HandsComponent? handsComponent) ||
|
|
!IoCManager.Resolve<IEntityManager>().TryGetComponent(_target.Uid, out ItemComponent? itemComponent))
|
|
{
|
|
return Outcome.Failed;
|
|
}
|
|
|
|
FoodComponent? foodComponent = null;
|
|
|
|
foreach (var slot in handsComponent.ActivePriorityEnumerable())
|
|
{
|
|
if (handsComponent.GetItem(slot) != itemComponent) continue;
|
|
handsComponent.ActiveHand = slot;
|
|
if (!IoCManager.Resolve<IEntityManager>().TryGetComponent(_target.Uid, out foodComponent))
|
|
{
|
|
return Outcome.Failed;
|
|
}
|
|
|
|
// This should also implicitly open it.
|
|
handsComponent.ActivateItem();
|
|
_interactionCooldown = IoCManager.Resolve<IRobustRandom>().NextFloat() + 0.5f;
|
|
}
|
|
|
|
if (foodComponent == null)
|
|
{
|
|
return Outcome.Failed;
|
|
}
|
|
|
|
if ((!IoCManager.Resolve<IEntityManager>().EntityExists(_target.Uid) ? EntityLifeStage.Deleted : IoCManager.Resolve<IEntityManager>().GetComponent<MetaDataComponent>(_target.Uid).EntityLifeStage) >= EntityLifeStage.Deleted ||
|
|
foodComponent.UsesRemaining == 0 ||
|
|
IoCManager.Resolve<IEntityManager>().TryGetComponent(_owner.Uid, out HungerComponent? hungerComponent) &&
|
|
hungerComponent.CurrentHunger >= hungerComponent.HungerThresholds[HungerThreshold.Okay])
|
|
{
|
|
return Outcome.Success;
|
|
}
|
|
|
|
return Outcome.Continuing;
|
|
}
|
|
}
|
|
}
|