67 lines
2.2 KiB
C#
67 lines
2.2 KiB
C#
using System.Collections.Generic;
|
|
using System.Threading.Tasks;
|
|
using Content.Server.Stack;
|
|
using Content.Shared.ActionBlocker;
|
|
using Content.Shared.Damage;
|
|
using Content.Shared.Damage.Components;
|
|
using Content.Shared.Interaction;
|
|
using Content.Shared.Interaction.Helpers;
|
|
using Content.Shared.Stacks;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.Serialization.Manager.Attributes;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.ViewVariables;
|
|
|
|
namespace Content.Server.Medical.Components
|
|
{
|
|
[RegisterComponent]
|
|
public class HealingComponent : Component, IAfterInteract
|
|
{
|
|
public override string Name => "Healing";
|
|
|
|
// TODO PROTOTYPE Replace this datafield variable with prototype references, once they are supported.
|
|
// This also requires changing the dictionary type, and removing a _prototypeManager.Index() call.
|
|
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
|
|
[DataField("heal", required: true )]
|
|
[ViewVariables(VVAccess.ReadWrite)]
|
|
public Dictionary<string, int> Heal = new();
|
|
|
|
async Task<bool> IAfterInteract.AfterInteract(AfterInteractEventArgs eventArgs)
|
|
{
|
|
if (eventArgs.Target == null)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (!eventArgs.Target.TryGetComponent(out IDamageableComponent? damageable))
|
|
{
|
|
return true;
|
|
}
|
|
|
|
if (!EntitySystem.Get<ActionBlockerSystem>().CanInteract(eventArgs.User))
|
|
{
|
|
return true;
|
|
}
|
|
|
|
if (eventArgs.User != eventArgs.Target &&
|
|
!eventArgs.InRangeUnobstructed(ignoreInsideBlocker: true, popup: true))
|
|
{
|
|
return true;
|
|
}
|
|
|
|
if (Owner.TryGetComponent<SharedStackComponent>(out var stack) && !EntitySystem.Get<StackSystem>().Use(Owner.Uid, stack, 1))
|
|
{
|
|
return true;
|
|
}
|
|
|
|
foreach (var (damageTypeID, amount) in Heal)
|
|
{
|
|
damageable.TryChangeDamage(_prototypeManager.Index<DamageTypePrototype>(damageTypeID), -amount, true);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|
|
}
|