Delta-v/Content.Shared/EntityEffects/Effects/ArtifactEntityEffectsSystem.cs

73 lines
2.8 KiB
C#

using Content.Shared.Popups;
using Content.Shared.Xenoarchaeology.Artifact;
using Content.Shared.Xenoarchaeology.Artifact.Components;
using Robust.Shared.Prototypes;
namespace Content.Shared.EntityEffects.Effects;
/// <summary>
/// Restores durability on this artifact
/// </summary>
/// <inheritdoc cref="EntityEffectSystem{T,TEffect}"/>
public sealed partial class ArtifactDurabilityRestoreEntityEffectsSystem : EntityEffectSystem<XenoArtifactComponent, ArtifactDurabilityRestore>
{
[Dependency] private readonly SharedXenoArtifactSystem _xenoArtifact = default!;
protected override void Effect(Entity<XenoArtifactComponent> entity, ref EntityEffectEvent<ArtifactDurabilityRestore> args)
{
var durability = args.Effect.RestoredDurability;
foreach (var node in _xenoArtifact.GetActiveNodes(entity))
{
_xenoArtifact.AdjustNodeDurability(node.Owner, durability);
}
}
}
/// <summary>
/// Unlocks a node on this artifact. Only works this effect hasn't been applied before.
/// </summary>
/// <inheritdoc cref="EntityEffectSystem{T,TEffect}"/>
public sealed partial class ArtifactUnlockEntityEffectSystem : EntityEffectSystem<XenoArtifactComponent, ArtifactUnlock>
{
[Dependency] private readonly SharedPopupSystem _popup = default!;
[Dependency] private readonly SharedXenoArtifactSystem _xenoArtifact = default!;
protected override void Effect(Entity<XenoArtifactComponent> entity, ref EntityEffectEvent<ArtifactUnlock> args)
{
if (EnsureComp<XenoArtifactUnlockingComponent>(entity, out var unlocking))
{
if (unlocking.ArtifexiumApplied)
return;
_popup.PopupEntity(Loc.GetString("artifact-activation-artifexium"), entity, PopupType.Medium);
}
else
{
_xenoArtifact.TriggerXenoArtifact(entity, null, force: true);
}
_xenoArtifact.SetArtifexiumApplied((entity, unlocking), true);
}
}
/// <inheritdoc cref="EntityEffect"/>
public sealed partial class ArtifactDurabilityRestore : EntityEffectBase<ArtifactDurabilityRestore>
{
/// <summary>
/// Amount of durability that will be restored per effect interaction.
/// </summary>
[DataField]
public int RestoredDurability = 1;
public override string EntityEffectGuidebookText(IPrototypeManager prototype, IEntitySystemManager entSys) =>
Loc.GetString("entity-effect-guidebook-artifact-durability-restore", ("restored", RestoredDurability));
}
/// <inheritdoc cref="EntityEffect"/>
public sealed partial class ArtifactUnlock : EntityEffectBase<ArtifactUnlock>
{
public override string EntityEffectGuidebookText(IPrototypeManager prototype, IEntitySystemManager entSys) =>
Loc.GetString("entity-effect-guidebook-artifact-unlock", ("chance", Probability));
}