tommy/tbd-station-14
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

emag medibot to make it poison patients (#15377)

Browse Source 
Co-authored-by: deltanedas <@deltanedas:kde.org>
This commit is contained in:
deltanedas
2023-05-02 08:00:57 +00:00
committed by GitHub
parent 136edfc0b1
commit aceb354945
6 changed files with 154 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
Content.Server/NPC/HTN/PrimitiveTasks/Operators/Specific/MedibotInjectOperator.cs
View File

@@ -1,10 +1,11 @@
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.Emag.Components;
using Content.Shared.Interaction;
using Content.Shared.Popups;
using Content.Shared.Silicons.Bots;
using Robust.Shared.Audio;
using Robust.Shared.Player;
@@ -12,11 +13,12 @@ namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators.Specific;
public sealed class MedibotInjectOperator : HTNOperator
{
[Dependency] private readonly IEntityManager _entManager = default!;
[Dependency] private readonly IEntityManager _entMan = default!;
private SharedAudioSystem _audio = default!;
private ChatSystem _chat = default!;
private SharedInteractionSystem _interactionSystem = default!;
private SharedPopupSystem _popupSystem = default!;
private SolutionContainerSystem _solutionSystem = default!;
private SharedInteractionSystem _interaction = default!;
private SharedPopupSystem _popup = default!;
private SolutionContainerSystem _solution = default!;
/// <summary>
/// Target entity to inject.
@@ -27,10 +29,11 @@ public sealed class MedibotInjectOperator : HTNOperator
public override void Initialize(IEntitySystemManager sysManager)
{
base.Initialize(sysManager);
_audio = sysManager.GetEntitySystem<SharedAudioSystem>();
_chat = sysManager.GetEntitySystem<ChatSystem>();
_interactionSystem = sysManager.GetEntitySystem<SharedInteractionSystem>();
_popupSystem = sysManager.GetEntitySystem<SharedPopupSystem>();
_solutionSystem = sysManager.GetEntitySystem<SolutionContainerSystem>();
_interaction = sysManager.GetEntitySystem<SharedInteractionSystem>();
_popup = sysManager.GetEntitySystem<SharedPopupSystem>();
_solution = sysManager.GetEntitySystem<SolutionContainerSystem>();
}
public override void Shutdown(NPCBlackboard blackboard, HTNOperatorStatus status)
@@ -44,41 +47,48 @@ public sealed class MedibotInjectOperator : HTNOperator
// TODO: Wat
var owner = blackboard.GetValue<EntityUid>(NPCBlackboard.Owner);
if (!blackboard.TryGetValue<EntityUid>(TargetKey, out var target, _entManager) || _entManager.Deleted(target))
if (!blackboard.TryGetValue<EntityUid>(TargetKey, out var target, _entMan) || _entMan.Deleted(target))
return HTNOperatorStatus.Failed;
if (!_entManager.TryGetComponent<MedibotComponent>(owner, out var botComp))
if (!_entMan.TryGetComponent<MedibotComponent>(owner, out var botComp))
return HTNOperatorStatus.Failed;
// To avoid spam, the rest of this needs fixing.
_entManager.EnsureComponent<NPCRecentlyInjectedComponent>(target);
_entMan.EnsureComponent<NPCRecentlyInjectedComponent>(target);
if (!_entManager.TryGetComponent<DamageableComponent>(target, out var damage))
if (!_entMan.TryGetComponent<DamageableComponent>(target, out var damage))
return HTNOperatorStatus.Failed;
if (!_solutionSystem.TryGetInjectableSolution(target, out var injectable))
if (!_solution.TryGetInjectableSolution(target, out var injectable))
return HTNOperatorStatus.Failed;
if (!_interactionSystem.InRangeUnobstructed(owner, target))
if (!_interaction.InRangeUnobstructed(owner, target))
return HTNOperatorStatus.Failed;
if (damage.TotalDamage == 0)
return HTNOperatorStatus.Failed;
if (damage.TotalDamage >= MedibotComponent.EmergencyMedDamageThreshold)
// if emagged, always treat below-crit as injured (give funny juice to healthy people)
var total = damage.TotalDamage;
if (_entMan.HasComponent<EmaggedComponent>(owner) && total < MedibotComponent.EmergencyMedDamageThreshold)
{
_solutionSystem.TryAddReagent(target, injectable, botComp.EmergencyMed, botComp.EmergencyMedInjectAmount, out var accepted);
_popupSystem.PopupEntity(Loc.GetString("hypospray-component-feel-prick-message"), target, target);
SoundSystem.Play("/Audio/Items/hypospray.ogg", Filter.Pvs(target), target);
total = MedibotComponent.EmergencyMedDamageThreshold;
}
if (total == 0)
return HTNOperatorStatus.Failed;
if (total >= MedibotComponent.EmergencyMedDamageThreshold)
{
_solution.TryAddReagent(target, injectable, botComp.EmergencyMed, botComp.EmergencyMedAmount, out var accepted);
_popup.PopupEntity(Loc.GetString("hypospray-component-feel-prick-message"), target, target);
_audio.PlayPvs(botComp.InjectSound, target);
_chat.TrySendInGameICMessage(owner, Loc.GetString("medibot-finish-inject"), InGameICChatType.Speak, hideChat: false, hideGlobalGhostChat: true);
return HTNOperatorStatus.Finished;
}
if (damage.TotalDamage >= MedibotComponent.StandardMedDamageThreshold)
if (total >= MedibotComponent.StandardMedDamageThreshold)
{
_solutionSystem.TryAddReagent(target, injectable, botComp.StandardMed, botComp.StandardMedInjectAmount, out var accepted);
_popupSystem.PopupEntity(Loc.GetString("hypospray-component-feel-prick-message"), target, target);
SoundSystem.Play("/Audio/Items/hypospray.ogg", Filter.Pvs(target), target);
_solution.TryAddReagent(target, injectable, botComp.StandardMed, botComp.StandardMedAmount, out var accepted);
_popup.PopupEntity(Loc.GetString("hypospray-component-feel-prick-message"), target, target);
_audio.PlayPvs(botComp.InjectSound, target);
_chat.TrySendInGameICMessage(owner, Loc.GetString("medibot-finish-inject"), InGameICChatType.Speak, hideChat: false, hideGlobalGhostChat: true);
return HTNOperatorStatus.Finished;
}