Small instrument system fix (#7432)

2022-04-05 00:08:05 -07:00
parent b7ff72ba6e
commit 7f2ceeb8e2
1 changed files with 6 additions and 0 deletions
--- a/Content.Server/Instruments/InstrumentSystem.cs
+++ b/Content.Server/Instruments/InstrumentSystem.cs
@@ -67,6 +67,9 @@ public sealed partial class InstrumentSystem : SharedInstrumentSystem

    private void OnBoundUIClosed(EntityUid uid, InstrumentComponent component, BoundUIClosedEvent args)
    {
+        if (args.UiKey is not InstrumentUiKey)
+            return;
+
        if (HasComp<ActiveInstrumentComponent>(uid)
            && _userInterfaceSystem.TryGetUi(uid, args.UiKey, out var bui)
            && bui.SubscribedSessions.Count == 0)
@@ -79,6 +82,9 @@ public sealed partial class InstrumentSystem : SharedInstrumentSystem

    private void OnBoundUIOpened(EntityUid uid, InstrumentComponent component, BoundUIOpenedEvent args)
    {
+        if (args.UiKey is not InstrumentUiKey)
+            return;
+        
        EnsureComp<ActiveInstrumentComponent>(uid);
        Clean(uid, component);
    }