NPC refactor (#10122)

Co-authored-by: metalgearsloth <metalgearsloth@gmail.com>

Content.Server/NPC/HTN/PrimitiveTasks/Operators/Specific/MedibotInjectOperator.cs

@@ -0,0 +1,87 @@
+using Content.Server.Chat.Systems;
+using Content.Server.Chemistry.EntitySystems;
+using Content.Server.NPC.Components;
+using Content.Server.Silicons.Bots;
+using Content.Shared.Damage;
+using Content.Shared.Interaction;
+using Content.Shared.Popups;
+using Robust.Shared.Audio;
+using Robust.Shared.Player;
+
+namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators.Specific;
+
+public sealed class MedibotInjectOperator : HTNOperator
+{
+    [Dependency] private readonly IEntityManager _entManager = default!;
+    private ChatSystem _chat = default!;
+    private SharedInteractionSystem _interactionSystem = default!;
+    private SharedPopupSystem _popupSystem = default!;
+    private SolutionContainerSystem _solutionSystem = default!;
+
+    /// <summary>
+    /// Target entity to inject.
+    /// </summary>
+    [ViewVariables, DataField("targetKey", required: true)]
+    public string TargetKey = string.Empty;
+
+    public override void Initialize(IEntitySystemManager sysManager)
+    {
+        base.Initialize(sysManager);
+        _chat = sysManager.GetEntitySystem<ChatSystem>();
+        _interactionSystem = sysManager.GetEntitySystem<SharedInteractionSystem>();
+        _popupSystem = sysManager.GetEntitySystem<SharedPopupSystem>();
+        _solutionSystem = sysManager.GetEntitySystem<SolutionContainerSystem>();
+    }
+
+    public override void Shutdown(NPCBlackboard blackboard, HTNOperatorStatus status)
+    {
+        base.Shutdown(blackboard, status);
+        blackboard.Remove<EntityUid>(TargetKey);
+    }
+
+    public override HTNOperatorStatus Update(NPCBlackboard blackboard, float frameTime)
+    {
+        // TODO: Wat
+        var owner = blackboard.GetValue<EntityUid>(NPCBlackboard.Owner);
+
+        if (!blackboard.TryGetValue<EntityUid>(TargetKey, out var target))
+            return HTNOperatorStatus.Failed;
+
+        if (!_entManager.TryGetComponent<MedibotComponent>(owner, out var botComp))
+            return HTNOperatorStatus.Failed;
+
+        if (!_entManager.TryGetComponent<DamageableComponent>(target, out var damage))
+            return HTNOperatorStatus.Failed;
+
+        if (!_solutionSystem.TryGetInjectableSolution(target, out var injectable))
+            return HTNOperatorStatus.Failed;
+
+        if (!_interactionSystem.InRangeUnobstructed(owner, target))
+            return HTNOperatorStatus.Failed;
+
+        if (damage.TotalDamage == 0)
+            return HTNOperatorStatus.Failed;
+
+        if (damage.TotalDamage <= MedibotComponent.StandardMedDamageThreshold)
+        {
+            _solutionSystem.TryAddReagent(target, injectable, botComp.StandardMed, botComp.StandardMedInjectAmount, out var accepted);
+            _entManager.EnsureComponent<NPCRecentlyInjectedComponent>(target);
+            _popupSystem.PopupEntity(Loc.GetString("hypospray-component-feel-prick-message"), target, Filter.Entities(target));
+            SoundSystem.Play("/Audio/Items/hypospray.ogg", Filter.Pvs(target), target);
+            _chat.TrySendInGameICMessage(owner, Loc.GetString("medibot-finish-inject"), InGameICChatType.Speak, false);
+            return HTNOperatorStatus.Finished;
+        }
+
+        if (damage.TotalDamage >= MedibotComponent.EmergencyMedDamageThreshold)
+        {
+            _solutionSystem.TryAddReagent(target, injectable, botComp.EmergencyMed, botComp.EmergencyMedInjectAmount, out var accepted);
+            _entManager.EnsureComponent<NPCRecentlyInjectedComponent>(target);
+            _popupSystem.PopupEntity(Loc.GetString("hypospray-component-feel-prick-message"), target, Filter.Entities(target));
+            SoundSystem.Play("/Audio/Items/hypospray.ogg", Filter.Pvs(target), target);
+            _chat.TrySendInGameICMessage(owner, Loc.GetString("medibot-finish-inject"), InGameICChatType.Speak, false);
+            return HTNOperatorStatus.Finished;
+        }
+
+        return HTNOperatorStatus.Failed;
+    }
+}

Content.Server/NPC/HTN/PrimitiveTasks/Operators/Specific/PickNearbyInjectableOperator.cs

@@ -0,0 +1,64 @@
+using System.Threading.Tasks;
+using Content.Server.Chemistry.Components.SolutionManager;
+using Content.Server.NPC.Components;
+using Content.Shared.Damage;
+using Content.Shared.MobState.Components;
+
+namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators.Specific;
+
+public sealed class PickNearbyInjectableOperator : HTNOperator
+{
+    [Dependency] private readonly IEntityManager _entManager = default!;
+    private EntityLookupSystem _lookup = default!;
+
+    [ViewVariables, DataField("rangeKey")] public string RangeKey = NPCBlackboard.MedibotInjectRange;
+
+    /// <summary>
+    /// Target entity to inject
+    /// </summary>
+    [ViewVariables, DataField("targetKey", required: true)]
+    public string TargetKey = string.Empty;
+
+    /// <summary>
+    /// Target entitycoordinates to move to.
+    /// </summary>
+    [ViewVariables, DataField("targetMoveKey", required: true)]
+    public string TargetMoveKey = string.Empty;
+
+    public override void Initialize(IEntitySystemManager sysManager)
+    {
+        base.Initialize(sysManager);
+        _lookup = sysManager.GetEntitySystem<EntityLookupSystem>();
+    }
+
+    public override async Task<(bool Valid, Dictionary<string, object>? Effects)> Plan(NPCBlackboard blackboard)
+    {
+        var owner = blackboard.GetValue<EntityUid>(NPCBlackboard.Owner);
+
+        if (!blackboard.TryGetValue<float>(RangeKey, out var range))
+            return (false, null);
+
+        var damageQuery = _entManager.GetEntityQuery<DamageableComponent>();
+        var injectQuery = _entManager.GetEntityQuery<InjectableSolutionComponent>();
+        var recentlyInjected = _entManager.GetEntityQuery<NPCRecentlyInjectedComponent>();
+        var mobState = _entManager.GetEntityQuery<MobStateComponent>();
+
+        foreach (var entity in _lookup.GetEntitiesInRange(owner, range))
+        {
+            if (mobState.HasComponent(entity) &&
+                injectQuery.HasComponent(entity) &&
+                damageQuery.TryGetComponent(entity, out var damage) &&
+                damage.TotalDamage > 0 &&
+                !recentlyInjected.HasComponent(entity))
+            {
+                return (true, new Dictionary<string, object>()
+                {
+                    {TargetKey, entity},
+                    {TargetMoveKey, _entManager.GetComponent<TransformComponent>(entity).Coordinates}
+                });
+            }
+        }
+
+        return (false, null);
+    }
+}