94 lines
3.4 KiB
C#
94 lines
3.4 KiB
C#
using Content.Shared.Actions;
|
|
using Content.Shared.Devour.Components;
|
|
using Content.Shared.DoAfter;
|
|
using Content.Shared.Mobs;
|
|
using Content.Shared.Mobs.Components;
|
|
using Content.Shared.Popups;
|
|
using Robust.Shared.Containers;
|
|
using Robust.Shared.Serialization;
|
|
|
|
namespace Content.Shared.Devour;
|
|
|
|
public abstract class SharedDevourSystem : EntitySystem
|
|
{
|
|
[Dependency] protected readonly SharedAudioSystem _audioSystem = default!;
|
|
[Dependency] private readonly SharedDoAfterSystem _doAfterSystem = default!;
|
|
[Dependency] private readonly SharedPopupSystem _popupSystem = default!;
|
|
[Dependency] private readonly SharedActionsSystem _actionsSystem = default!;
|
|
[Dependency] private readonly SharedContainerSystem _containerSystem = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<DevourerComponent, ComponentStartup>(OnStartup);
|
|
SubscribeLocalEvent<DevourerComponent, DevourActionEvent>(OnDevourAction);
|
|
}
|
|
|
|
protected void OnStartup(EntityUid uid, DevourerComponent component, ComponentStartup args)
|
|
{
|
|
//Devourer doesn't actually chew, since he sends targets right into his stomach.
|
|
//I did it mom, I added ERP content into upstream. Legally!
|
|
component.Stomach = _containerSystem.EnsureContainer<Container>(uid, "stomach");
|
|
|
|
_actionsSystem.AddAction(uid, ref component.DevourActionEntity, component.DevourAction);
|
|
}
|
|
|
|
/// <summary>
|
|
/// The devour action
|
|
/// </summary>
|
|
protected void OnDevourAction(EntityUid uid, DevourerComponent component, DevourActionEvent args)
|
|
{
|
|
if (args.Handled || component.Whitelist?.IsValid(args.Target, EntityManager) != true)
|
|
return;
|
|
|
|
args.Handled = true;
|
|
var target = args.Target;
|
|
|
|
// Structure and mob devours handled differently.
|
|
if (TryComp(target, out MobStateComponent? targetState))
|
|
{
|
|
switch (targetState.CurrentState)
|
|
{
|
|
case MobState.Critical:
|
|
case MobState.Dead:
|
|
|
|
_doAfterSystem.TryStartDoAfter(new DoAfterArgs(EntityManager, uid, component.DevourTime, new DevourDoAfterEvent(), uid, target: target, used: uid)
|
|
{
|
|
BreakOnTargetMove = true,
|
|
BreakOnUserMove = true,
|
|
});
|
|
break;
|
|
default:
|
|
_popupSystem.PopupClient(Loc.GetString("devour-action-popup-message-fail-target-alive"), uid,uid);
|
|
break;
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
_popupSystem.PopupClient(Loc.GetString("devour-action-popup-message-structure"), uid, uid);
|
|
|
|
if (component.SoundStructureDevour != null)
|
|
_audioSystem.PlayPredicted(component.SoundStructureDevour, uid, uid, component.SoundStructureDevour.Params);
|
|
|
|
_doAfterSystem.TryStartDoAfter(new DoAfterArgs(EntityManager, uid, component.StructureDevourTime, new DevourDoAfterEvent(), uid, target: target, used: uid)
|
|
{
|
|
BreakOnTargetMove = true,
|
|
BreakOnUserMove = true,
|
|
});
|
|
}
|
|
}
|
|
|
|
public sealed partial class DevourActionEvent : EntityTargetActionEvent { }
|
|
|
|
[Serializable, NetSerializable]
|
|
public sealed partial class DevourDoAfterEvent : SimpleDoAfterEvent { }
|
|
|
|
[Serializable, NetSerializable]
|
|
public enum FoodPreference : byte
|
|
{
|
|
Humanoid = 0,
|
|
All = 1
|
|
}
|