72 lines
2.2 KiB
C#
72 lines
2.2 KiB
C#
using Content.Shared.Interaction.Events;
|
|
using Content.Shared.Weapons.Ranged.Components;
|
|
using Content.Shared.Weapons.Ranged.Systems;
|
|
using Robust.Shared.Audio.Systems;
|
|
using Robust.Shared.Containers;
|
|
using Robust.Shared.Serialization;
|
|
|
|
namespace Content.Shared._White.Crossbow;
|
|
|
|
public sealed class DrawableSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly SharedAudioSystem _audio = default!;
|
|
[Dependency] private readonly SharedAppearanceSystem _appearance = default!;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<DrawableComponent, ComponentStartup>(OnStartup);
|
|
SubscribeLocalEvent<DrawableComponent, EntRemovedFromContainerMessage>(OnItemRemove);
|
|
SubscribeLocalEvent<DrawableComponent, AttemptShootEvent>(OnAttemptShoot);
|
|
SubscribeLocalEvent<DrawableComponent, UseInHandEvent>(OnUse);
|
|
}
|
|
|
|
private void OnUse(EntityUid uid, DrawableComponent component, UseInHandEvent args)
|
|
{
|
|
if (component.Drawn || component.Provider.Count == 0)
|
|
return;
|
|
|
|
args.Handled = true;
|
|
|
|
_audio.PlayPredicted(component.SoundDraw, uid, args.User);
|
|
component.Drawn = true;
|
|
|
|
UpdateDrawableAppearance(uid, component);
|
|
}
|
|
|
|
private void OnItemRemove(EntityUid uid, DrawableComponent component, EntRemovedFromContainerMessage args)
|
|
{
|
|
if (!component.Drawn || args.Container.ID != component.Provider.Container.ID)
|
|
return;
|
|
|
|
component.Drawn = false;
|
|
UpdateDrawableAppearance(uid, component);
|
|
}
|
|
|
|
private void OnStartup(EntityUid uid, DrawableComponent component, ComponentStartup args)
|
|
{
|
|
component.Provider = EnsureComp<BallisticAmmoProviderComponent>(uid);
|
|
}
|
|
|
|
private void OnAttemptShoot(EntityUid uid, DrawableComponent component, ref AttemptShootEvent args)
|
|
{
|
|
if (!component.Drawn)
|
|
args.Cancelled = true;
|
|
}
|
|
|
|
private void UpdateDrawableAppearance(EntityUid uid, DrawableComponent component)
|
|
{
|
|
if (!TryComp<AppearanceComponent>(uid, out var appearance))
|
|
return;
|
|
|
|
_appearance.SetData(uid, DrawableVisuals.Drawn, component.Drawn, appearance);
|
|
}
|
|
}
|
|
|
|
[Serializable, NetSerializable]
|
|
public enum DrawableVisuals : byte
|
|
{
|
|
Drawn
|
|
}
|