671324bef8
Co-authored-by: metalgearsloth <31366439+metalgearsloth@users.noreply.github.com> Co-authored-by: metalgearsloth <comedian_vs_clown@hotmail.com>
34 lines
1.1 KiB
C#
34 lines
1.1 KiB
C#
using Content.Client.Implants.UI;
|
|
using Content.Client.Items;
|
|
using Content.Shared.Implants;
|
|
using Content.Shared.Implants.Components;
|
|
using Robust.Shared.GameStates;
|
|
|
|
namespace Content.Client.Implants;
|
|
|
|
public sealed class ImplanterSystem : SharedImplanterSystem
|
|
{
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
SubscribeLocalEvent<ImplanterComponent, ComponentHandleState>(OnHandleImplanterState);
|
|
SubscribeLocalEvent<ImplanterComponent, ItemStatusCollectMessage>(OnItemImplanterStatus);
|
|
}
|
|
|
|
private void OnHandleImplanterState(EntityUid uid, ImplanterComponent component, ref ComponentHandleState args)
|
|
{
|
|
if (args.Current is not ImplanterComponentState state)
|
|
return;
|
|
|
|
component.CurrentMode = state.CurrentMode;
|
|
component.ImplantOnly = state.ImplantOnly;
|
|
component.UiUpdateNeeded = true;
|
|
}
|
|
|
|
private void OnItemImplanterStatus(EntityUid uid, ImplanterComponent component, ItemStatusCollectMessage args)
|
|
{
|
|
args.Controls.Add(new ImplanterStatusControl(component));
|
|
}
|
|
}
|