c565b44965
* The easy ones * For certain values of easy * Easy test * Hair * Fix sandbox violations * Sort usings
49 lines
1.6 KiB
C#
49 lines
1.6 KiB
C#
using Content.Shared.Speech.EntitySystems;
|
|
using Content.Shared.StatusEffect;
|
|
using Content.Shared.Traits.Assorted;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Shared.Drunk;
|
|
|
|
public abstract class SharedDrunkSystem : EntitySystem
|
|
{
|
|
public static readonly ProtoId<StatusEffectPrototype> DrunkKey = "Drunk";
|
|
|
|
[Dependency] private readonly StatusEffectsSystem _statusEffectsSystem = default!;
|
|
[Dependency] private readonly SharedSlurredSystem _slurredSystem = default!;
|
|
|
|
public void TryApplyDrunkenness(EntityUid uid, float boozePower, bool applySlur = true,
|
|
StatusEffectsComponent? status = null)
|
|
{
|
|
if (!Resolve(uid, ref status, false))
|
|
return;
|
|
|
|
if (TryComp<LightweightDrunkComponent>(uid, out var trait))
|
|
boozePower *= trait.BoozeStrengthMultiplier;
|
|
|
|
if (applySlur)
|
|
{
|
|
_slurredSystem.DoSlur(uid, TimeSpan.FromSeconds(boozePower), status);
|
|
}
|
|
|
|
if (!_statusEffectsSystem.HasStatusEffect(uid, DrunkKey, status))
|
|
{
|
|
_statusEffectsSystem.TryAddStatusEffect<DrunkComponent>(uid, DrunkKey, TimeSpan.FromSeconds(boozePower), true, status);
|
|
}
|
|
else
|
|
{
|
|
_statusEffectsSystem.TryAddTime(uid, DrunkKey, TimeSpan.FromSeconds(boozePower), status);
|
|
}
|
|
}
|
|
|
|
public void TryRemoveDrunkenness(EntityUid uid)
|
|
{
|
|
_statusEffectsSystem.TryRemoveStatusEffect(uid, DrunkKey);
|
|
}
|
|
public void TryRemoveDrunkenessTime(EntityUid uid, double timeRemoved)
|
|
{
|
|
_statusEffectsSystem.TryRemoveTime(uid, DrunkKey, TimeSpan.FromSeconds(timeRemoved));
|
|
}
|
|
|
|
}
|