using Content.Server.Botany.Components;
using Content.Server.Botany.Systems;
using Content.Server.Chat.Systems;
using Content.Shared.Chat;
using Content.Shared.Damage;
using Content.Shared.Emag.Components;
using Content.Shared.Interaction;
using Content.Shared.Popups;
using Content.Shared.Silicons.Bots;
using Content.Shared.Tag;
using Robust.Shared.Audio.Systems;
using Robust.Shared.Prototypes;
namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators.Specific;
public sealed partial class PlantbotServiceOperator : HTNOperator
{
[Dependency] private readonly IEntityManager _entMan = default!;
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
private ChatSystem _chat = default!;
private SharedAudioSystem _audio = default!;
private SharedInteractionSystem _interaction = default!;
private SharedPopupSystem _popup = default!;
private PlantHolderSystem _plantHolderSystem = default!;
private DamageableSystem _damageableSystem = default!;
private TagSystem _tagSystem = default!;
public const float RequiredWaterLevelToService = 80f;
public const float RequiredWeedsAmountToWeed = 1f;
public const float WaterTransferAmount = 10f;
public const float WeedsRemovedAmount = 1f;
public const string SiliconTag = "SiliconMob";
///
/// Target entity to inject.
///
[DataField(required: true)]
public string TargetKey = string.Empty;
public override void Initialize(IEntitySystemManager sysManager)
{
base.Initialize(sysManager);
_chat = sysManager.GetEntitySystem();
_audio = sysManager.GetEntitySystem();
_interaction = sysManager.GetEntitySystem();
_popup = sysManager.GetEntitySystem();
_plantHolderSystem = sysManager.GetEntitySystem();
}
public override void TaskShutdown(NPCBlackboard blackboard, HTNOperatorStatus status)
{
base.TaskShutdown(blackboard, status);
blackboard.Remove(TargetKey);
}
public override HTNOperatorStatus Update(NPCBlackboard blackboard, float frameTime)
{
var owner = blackboard.GetValue(NPCBlackboard.Owner);
if (!blackboard.TryGetValue(TargetKey, out var target, _entMan) || _entMan.Deleted(target))
return HTNOperatorStatus.Failed;
if (!_entMan.TryGetComponent(owner, out var botComp)
|| !_entMan.TryGetComponent(target, out var plantHolderComponent)
|| !_interaction.InRangeUnobstructed(owner, target)
|| (plantHolderComponent is { WaterLevel: >= RequiredWaterLevelToService, WeedLevel: <= RequiredWeedsAmountToWeed } && (!_entMan.HasComponent(owner) || plantHolderComponent.Dead || plantHolderComponent.WaterLevel <= 0f)))
return HTNOperatorStatus.Failed;
if (botComp.IsEmagged)
{
_plantHolderSystem.AdjustWater(target, -WaterTransferAmount);
_audio.PlayPvs(botComp.RemoveWaterSound, target);
}
else
{
if (plantHolderComponent.WaterLevel <= RequiredWaterLevelToService)
{
_plantHolderSystem.AdjustWater(target, 10);
_audio.PlayPvs(botComp.WaterSound, target);
_chat.TrySendInGameICMessage(owner, Loc.GetString("plantbot-add-water"), InGameICChatType.Speak, hideChat: true, hideLog: true);
}
else if (plantHolderComponent.WeedLevel >= RequiredWeedsAmountToWeed)
{
plantHolderComponent.WeedLevel -= WeedsRemovedAmount;
_audio.PlayPvs(botComp.WeedSound, target);
_chat.TrySendInGameICMessage(owner, Loc.GetString("plantbot-remove-weeds"), InGameICChatType.Speak, hideChat: true, hideLog: true);
}
else
return HTNOperatorStatus.Failed;
}
return HTNOperatorStatus.Finished;
}
}