b9196d0a10
* Add test that puts all components on an entity and checks for no exceptions Also fix all the exceptions that happened because of this * Add comments to the test * Fix nullable errors * Fix more nullable errors * More nullable error fixes * Unignore basic actor component * Fix more nullable errors * NULLABLE ERROR * Add string interpolation * Merge if checks * Remove redundant pragma warning disable 649 * Address reviews * Remove null wrappers around TryGetComponent * Merge conflict fixes * APC battery component error fix * Fix power test * Fix atmos mapgrid usages
116 lines
3.7 KiB
C#
116 lines
3.7 KiB
C#
#nullable enable
|
|
using Content.Server.GameObjects.EntitySystems;
|
|
using Content.Shared.GameObjects.Components.Research;
|
|
using Content.Shared.Interfaces.GameObjects.Components;
|
|
using Robust.Server.GameObjects.Components.UserInterface;
|
|
using Robust.Server.Interfaces.GameObjects;
|
|
using Robust.Server.Interfaces.Player;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.Interfaces.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.ViewVariables;
|
|
|
|
namespace Content.Server.GameObjects.Components.Research
|
|
{
|
|
[RegisterComponent]
|
|
public class ResearchClientComponent : SharedResearchClientComponent, IActivate
|
|
{
|
|
[Dependency] private readonly IEntitySystemManager _entitySystemManager = default!;
|
|
|
|
// TODO: Create GUI for changing RD server.
|
|
private BoundUserInterface? UserInterface =>
|
|
Owner.TryGetComponent(out ServerUserInterfaceComponent? ui) &&
|
|
ui.TryGetBoundUserInterface(ResearchClientUiKey.Key, out var boundUi)
|
|
? boundUi
|
|
: null;
|
|
|
|
public bool ConnectedToServer => Server != null;
|
|
|
|
[ViewVariables(VVAccess.ReadOnly)]
|
|
public ResearchServerComponent? Server { get; set; }
|
|
|
|
public bool RegisterServer(ResearchServerComponent? server)
|
|
{
|
|
var result = server != null && server.RegisterClient(this);
|
|
return result;
|
|
}
|
|
|
|
public void UnregisterFromServer()
|
|
{
|
|
Server?.UnregisterClient(this);
|
|
}
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
// For now it just registers on the first server it can find.
|
|
var servers = _entitySystemManager.GetEntitySystem<ResearchSystem>().Servers;
|
|
|
|
if (servers.Count > 0)
|
|
RegisterServer(servers[0]);
|
|
|
|
if (UserInterface != null)
|
|
{
|
|
UserInterface.OnReceiveMessage += UserInterfaceOnOnReceiveMessage;
|
|
}
|
|
}
|
|
|
|
public void OpenUserInterface(IPlayerSession session)
|
|
{
|
|
UpdateUserInterface();
|
|
UserInterface?.Open(session);
|
|
}
|
|
|
|
void IActivate.Activate(ActivateEventArgs eventArgs)
|
|
{
|
|
if (!eventArgs.User.TryGetComponent(out IActorComponent? actor))
|
|
return;
|
|
|
|
OpenUserInterface(actor.playerSession);
|
|
}
|
|
|
|
public void UpdateUserInterface()
|
|
{
|
|
UserInterface?.SetState(GetNewUiState());
|
|
}
|
|
|
|
private ResearchClientBoundInterfaceState GetNewUiState()
|
|
{
|
|
var rd = _entitySystemManager.GetEntitySystem<ResearchSystem>();
|
|
|
|
return new ResearchClientBoundInterfaceState(rd.Servers.Count, rd.GetServerNames(),
|
|
rd.GetServerIds(), ConnectedToServer ? Server!.Id : -1);
|
|
}
|
|
|
|
private void UserInterfaceOnOnReceiveMessage(ServerBoundUserInterfaceMessage msg)
|
|
{
|
|
switch (msg.Message)
|
|
{
|
|
case ResearchClientSyncMessage _:
|
|
UpdateUserInterface();
|
|
break;
|
|
|
|
case ResearchClientServerSelectedMessage selectedMessage:
|
|
UnregisterFromServer();
|
|
RegisterServer(_entitySystemManager.GetEntitySystem<ResearchSystem>().GetServerById(selectedMessage.ServerId));
|
|
UpdateUserInterface();
|
|
break;
|
|
|
|
case ResearchClientServerDeselectedMessage _:
|
|
UnregisterFromServer();
|
|
UpdateUserInterface();
|
|
break;
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
protected override void Shutdown()
|
|
{
|
|
base.Shutdown();
|
|
UnregisterFromServer();
|
|
|
|
}
|
|
}
|
|
}
|