d75e743dd7
* Creates Content.Shared.Chemistry.Solutions Copies Solution class to new namespace Obsoletes old Solution class * Switches over to the Solutions.Solution Solution * Creates Content.Shared.Chemistry.Containers Copies relevant components/systems to the new namespace Obsoletes old versions * Switches over to the Containers.XYZ namespace * Creates SolutionSystem and obsoletes old SolutionContainerSystem methods * Start using SolutionSystem for Solution manipulation * EnumerateSolutions * Move TryGetMixableSolution * Move EnsureSolution to Server * Create Solution Entities * Stop using obsolete solution system methods * Fix prototype component tests * Add using ..Audio.Systems; back * Wrap solution container slots in ContainerSlots * Actually add the slot to the solution container map * Dirty SolutionContainerComponent when ensuring solutions * Revert namespace changes * Remerge SolutionSystem and SolutionContainerSystem * SolutionContainerManagerComponent refactor * Avoid wrapping necessary code in DebugTools.Assert as it is removed when compiling for release * Readd examine reagent sorting * Fix errors * Poke tests * Fix solution names not being applied * Fix WoolyComponent including statement * Fix merge skew * Fix compile errors * Make reactions use solntities * Reindent solution class namespace * Field attribute changes * AutoGenerateComponentState for SolutionContainerComponent * SolutionContainerComponent -> ContainedSolutionComponent * ref ReactionAttemptEvent * Denetwork preinit solutions * Misc 1 * Nullable TryGetSolution out vars * Cache associated solutions * Fix merge skew * Use explicit regions in SharedSolutionContainerSystem.Capabilities * Add debug assert * Use explicit regions in SharedSolutionContainerSystem.Relay + ref SolutionContainerChangedEvent * ContainedSolutionComponent.Name -> ContainedSolutionComponent.ContainerName * SolutionComponent doc comments * Implicit DataField names and property purge * ReagentEffect DataField names * Local variables for readability * Sort using statements + Entity<T> event handlers * Fix compile erros * Fix compile errors --------- Co-authored-by: ElectroJr <leonsfriedrich@gmail.com>
89 lines
3.6 KiB
C#
89 lines
3.6 KiB
C#
using Content.Server.Chat.Systems;
|
|
using Content.Server.Chemistry.Containers.EntitySystems;
|
|
using Content.Server.NPC.Components;
|
|
using Content.Shared.Damage;
|
|
using Content.Shared.Emag.Components;
|
|
using Content.Shared.Interaction;
|
|
using Content.Shared.Mobs.Components;
|
|
using Content.Shared.Popups;
|
|
using Content.Shared.Silicons.Bots;
|
|
using Robust.Shared.Audio.Systems;
|
|
|
|
namespace Content.Server.NPC.HTN.PrimitiveTasks.Operators.Specific;
|
|
|
|
public sealed partial class MedibotInjectOperator : HTNOperator
|
|
{
|
|
[Dependency] private readonly IEntityManager _entMan = default!;
|
|
private ChatSystem _chat = default!;
|
|
private MedibotSystem _medibot = default!;
|
|
private SharedAudioSystem _audio = default!;
|
|
private SharedInteractionSystem _interaction = default!;
|
|
private SharedPopupSystem _popup = default!;
|
|
private SolutionContainerSystem _solutionContainer = default!;
|
|
|
|
/// <summary>
|
|
/// Target entity to inject.
|
|
/// </summary>
|
|
[DataField("targetKey", required: true)]
|
|
public string TargetKey = string.Empty;
|
|
|
|
public override void Initialize(IEntitySystemManager sysManager)
|
|
{
|
|
base.Initialize(sysManager);
|
|
_chat = sysManager.GetEntitySystem<ChatSystem>();
|
|
_medibot = sysManager.GetEntitySystem<MedibotSystem>();
|
|
_audio = sysManager.GetEntitySystem<SharedAudioSystem>();
|
|
_interaction = sysManager.GetEntitySystem<SharedInteractionSystem>();
|
|
_popup = sysManager.GetEntitySystem<SharedPopupSystem>();
|
|
_solutionContainer = sysManager.GetEntitySystem<SolutionContainerSystem>();
|
|
}
|
|
|
|
public override void TaskShutdown(NPCBlackboard blackboard, HTNOperatorStatus status)
|
|
{
|
|
base.TaskShutdown(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, _entMan) || _entMan.Deleted(target))
|
|
return HTNOperatorStatus.Failed;
|
|
|
|
if (!_entMan.TryGetComponent<MedibotComponent>(owner, out var botComp))
|
|
return HTNOperatorStatus.Failed;
|
|
|
|
|
|
if (!_entMan.TryGetComponent<DamageableComponent>(target, out var damage))
|
|
return HTNOperatorStatus.Failed;
|
|
|
|
if (!_solutionContainer.TryGetInjectableSolution(target, out var injectable, out _))
|
|
return HTNOperatorStatus.Failed;
|
|
|
|
if (!_interaction.InRangeUnobstructed(owner, target))
|
|
return HTNOperatorStatus.Failed;
|
|
|
|
var total = damage.TotalDamage;
|
|
|
|
// always inject healthy patients when emagged
|
|
if (total == 0 && !_entMan.HasComponent<EmaggedComponent>(owner))
|
|
return HTNOperatorStatus.Failed;
|
|
|
|
if (!_entMan.TryGetComponent<MobStateComponent>(target, out var mobState))
|
|
return HTNOperatorStatus.Failed;
|
|
|
|
var state = mobState.CurrentState;
|
|
if (!_medibot.TryGetTreatment(botComp, mobState.CurrentState, out var treatment) || !treatment.IsValid(total))
|
|
return HTNOperatorStatus.Failed;
|
|
|
|
_entMan.EnsureComponent<NPCRecentlyInjectedComponent>(target);
|
|
_solutionContainer.TryAddReagent(injectable.Value, treatment.Reagent, treatment.Quantity, out _);
|
|
_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: true, hideLog: true);
|
|
return HTNOperatorStatus.Finished;
|
|
}
|
|
}
|