crystall-punk-14/Content.Server/Research/Systems/ResearchSystem.Client.cs

using System.Linq;
using Content.Server.Research.Components;
using Content.Shared.Research.Components;

namespace Content.Server.Research.Systems;

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)
    {
        var allServers = EntityQuery<ResearchServerComponent>(true).ToArray();
        if (allServers.Length > 0)
            RegisterClientServer(component, allServers[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 names = GetServerNames();
        var state = new ResearchClientBoundInterfaceState(names.Length, names,
            GetServerIds(), component.ConnectedToServer ? component.Server!.Id : -1);

        _uiSystem.TrySetUiState(component.Owner, ResearchClientUiKey.Key, state);
    }

    private bool RegisterClientServer(ResearchClientComponent component, ResearchServerComponent? server = null)
    {
        return server != null && RegisterServerClient(server, component.Owner, component);
    }

    private void UnregisterClientServer(ResearchClientComponent component)
    {
        if (component.Server == null)
            return;

        UnregisterServerClient(component.Server, component.Owner, component);
    }
}