38 lines
1.4 KiB
C#
38 lines
1.4 KiB
C#
|
using Content.Shared.Weapons.Ranged.Components;
|
||
|
|
using Robust.Shared.Containers;
|
||
|
|
using Robust.Shared.Utility;
|
||
|
|
|
||
|
|
namespace Content.Client.Weapons.Ranged.Systems;
|
||
|
|
|
||
|
|
public sealed partial class GunSystem
|
||
|
|
{
|
||
|
|
protected override void InitializeRevolver()
|
||
|
|
{
|
||
|
|
base.InitializeRevolver();
|
||
|
|
SubscribeLocalEvent<RevolverAmmoProviderComponent, AmmoCounterControlEvent>(OnRevolverCounter);
|
||
|
|
SubscribeLocalEvent<RevolverAmmoProviderComponent, UpdateAmmoCounterEvent>(OnRevolverAmmoUpdate);
|
||
|
|
SubscribeLocalEvent<RevolverAmmoProviderComponent, EntRemovedFromContainerMessage>(OnRevolverEntRemove);
|
||
|
|
}
|
||
|
|
|
||
|
|
private void OnRevolverEntRemove(EntityUid uid, RevolverAmmoProviderComponent component, EntRemovedFromContainerMessage args)
|
||
|
|
{
|
||
|
|
if (args.Container.ID != RevolverContainer) return;
|
||
|
|
|
||
|
|
// See ChamberMagazineAmmoProvider
|
||
|
|
if (!args.Entity.IsClientSide()) return;
|
||
|
|
|
||
|
|
QueueDel(args.Entity);
|
||
|
|
}
|
||
|
|
|
||
|
|
private void OnRevolverAmmoUpdate(EntityUid uid, RevolverAmmoProviderComponent component, UpdateAmmoCounterEvent args)
|
||
|
|
{
|
||
|
|
if (args.Control is not RevolverStatusControl control) return;
|
||
|
|
control.Update(component.CurrentIndex, component.Chambers);
|
||
|
|
}
|
||
|
|
|
||
|
|
private void OnRevolverCounter(EntityUid uid, RevolverAmmoProviderComponent component, AmmoCounterControlEvent args)
|
||
|
|
{
|
||
|
|
args.Control = new RevolverStatusControl();
|
||
|
|
}
|
||
|
|
}
|