45 lines
1.6 KiB
C#
45 lines
1.6 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using Content.Server.AI.Utility.Actions;
|
|
using Content.Server.AI.Utility.Actions.Nutrition.Drink;
|
|
using Content.Server.AI.Utility.Considerations;
|
|
using Content.Server.AI.Utility.Considerations.Nutrition.Drink;
|
|
using Content.Server.AI.WorldState;
|
|
using Content.Server.AI.WorldState.States;
|
|
using Content.Server.AI.WorldState.States.Inventory;
|
|
using Content.Server.Nutrition.Components;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
|
|
namespace Content.Server.AI.Utility.ExpandableActions.Nutrition
|
|
{
|
|
public sealed class UseDrinkInInventoryExp : ExpandableUtilityAction
|
|
{
|
|
public override float Bonus => UtilityAction.NeedsBonus;
|
|
|
|
protected override IEnumerable<Func<float>> GetCommonConsiderations(Blackboard context)
|
|
{
|
|
var considerationsManager = IoCManager.Resolve<ConsiderationsManager>();
|
|
return new[]
|
|
{
|
|
considerationsManager.Get<ThirstCon>().PresetCurve(context, PresetCurve.Nutrition)
|
|
};
|
|
}
|
|
|
|
public override IEnumerable<UtilityAction> GetActions(Blackboard context)
|
|
{
|
|
var owner = context.GetState<SelfState>().GetValue();
|
|
|
|
foreach (var entity in context.GetState<EnumerableInventoryState>().GetValue())
|
|
{
|
|
if (!IoCManager.Resolve<IEntityManager>().HasComponent<DrinkComponent>(entity))
|
|
{
|
|
continue;
|
|
}
|
|
|
|
yield return new UseDrinkInInventory {Owner = owner, Target = entity, Bonus = Bonus};
|
|
}
|
|
}
|
|
}
|
|
}
|