ECS and bandaid research (#9251)

Content.Server/Research/Systems/ResearchSystem.Client.cs

@@ -0,0 +1,87 @@
+using Content.Server.Research.Components;
+using Content.Shared.Research.Components;
+
+namespace Content.Server.Research;
+
+public sealed partial class ResearchSystem
+{
+    private void InitializeClient()
+    {
+        SubscribeLocalEvent<ResearchClientComponent, ComponentStartup>(OnClientStartup);
+        SubscribeLocalEvent<ResearchClientComponent, ComponentShutdown>(OnClientShutdown);
+        SubscribeLocalEvent<ResearchClientComponent, BoundUIOpenedEvent>(OnClientUIOpen);
+
+        SubscribeLocalEvent<ResearchClientComponent, ResearchClientSyncMessage>(OnClientSyncMessage);
+        SubscribeLocalEvent<ResearchClientComponent, ResearchClientServerSelectedMessage>(OnClientSelected);
+        SubscribeLocalEvent<ResearchClientComponent, ResearchClientServerDeselectedMessage>(OnClientDeselected);
+    }
+
+    #region UI
+
+    private void OnClientDeselected(EntityUid uid, ResearchClientComponent component, ResearchClientServerDeselectedMessage args)
+    {
+        UnregisterClientServer(component);
+        UpdateClientInterface(component);
+
+        if (TryComp<ResearchConsoleComponent>(uid, out var console))
+        {
+            UpdateConsoleInterface(console, component);
+        }
+    }
+
+    private void OnClientSelected(EntityUid uid, ResearchClientComponent component, ResearchClientServerSelectedMessage args)
+    {
+        UnregisterClientServer(component);
+        RegisterClientServer(component, GetServerById(args.ServerId));
+        UpdateClientInterface(component);
+
+        if (TryComp<ResearchConsoleComponent>(uid, out var console))
+        {
+            UpdateConsoleInterface(console, component);
+        }
+    }
+
+    private void OnClientSyncMessage(EntityUid uid, ResearchClientComponent component, ResearchClientSyncMessage args)
+    {
+        UpdateClientInterface(component);
+    }
+
+    #endregion
+
+    private void OnClientStartup(EntityUid uid, ResearchClientComponent component, ComponentStartup args)
+    {
+        if (Servers.Count > 0)
+            RegisterClientServer(component, Servers[0]);
+    }
+
+    private void OnClientShutdown(EntityUid uid, ResearchClientComponent component, ComponentShutdown args)
+    {
+        UnregisterClientServer(component);
+    }
+
+    private void OnClientUIOpen(EntityUid uid, ResearchClientComponent component, BoundUIOpenedEvent args)
+    {
+        UpdateClientInterface(component);
+    }
+
+    private void UpdateClientInterface(ResearchClientComponent component)
+    {
+        var state = new ResearchClientBoundInterfaceState(Servers.Count, GetServerNames(),
+            GetServerIds(), component.ConnectedToServer ? component.Server!.Id : -1);
+
+        _uiSystem.GetUiOrNull(component.Owner, ResearchClientUiKey.Key)?.SetState(state);
+    }
+
+    private bool RegisterClientServer(ResearchClientComponent component, ResearchServerComponent? server = null)
+    {
+        if (server == null) return false;
+        return RegisterServerClient(server, component);
+    }
+
+    private void UnregisterClientServer(ResearchClientComponent component)
+    {
+        if (component.Server == null) return;
+
+        UnregisterServerClient(component.Server, component);
+    }
+}