ECS and bandaid research (#9251)
This commit is contained in:
@@ -1,89 +1,5 @@
|
||||
using Content.Server.Power.Components;
|
||||
using Content.Server.UserInterface;
|
||||
using Content.Shared.Research.Components;
|
||||
using Content.Shared.Research.Prototypes;
|
||||
using Robust.Server.GameObjects;
|
||||
using Robust.Shared.Prototypes;
|
||||
|
||||
namespace Content.Server.Research.Components
|
||||
{
|
||||
[RegisterComponent]
|
||||
public sealed class ResearchConsoleComponent : SharedResearchConsoleComponent
|
||||
{
|
||||
[Dependency] private readonly IEntityManager _entMan = default!;
|
||||
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
|
||||
|
||||
[ViewVariables] private bool Powered => !_entMan.TryGetComponent(Owner, out ApcPowerReceiverComponent? receiver) || receiver.Powered;
|
||||
|
||||
[ViewVariables] private BoundUserInterface? UserInterface => Owner.GetUIOrNull(ResearchConsoleUiKey.Key);
|
||||
|
||||
protected override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
|
||||
Owner.EnsureComponentWarn<ServerUserInterfaceComponent>();
|
||||
|
||||
if (UserInterface != null)
|
||||
{
|
||||
UserInterface.OnReceiveMessage += UserInterfaceOnOnReceiveMessage;
|
||||
}
|
||||
|
||||
Owner.EnsureComponent<ResearchClientComponent>();
|
||||
}
|
||||
|
||||
private void UserInterfaceOnOnReceiveMessage(ServerBoundUserInterfaceMessage message)
|
||||
{
|
||||
if (!_entMan.TryGetComponent(Owner, out TechnologyDatabaseComponent? database))
|
||||
return;
|
||||
if (!_entMan.TryGetComponent(Owner, out ResearchClientComponent? client))
|
||||
return;
|
||||
if (!Powered)
|
||||
return;
|
||||
|
||||
switch (message.Message)
|
||||
{
|
||||
case ConsoleUnlockTechnologyMessage msg:
|
||||
if (!_prototypeManager.TryIndex(msg.Id, out TechnologyPrototype? tech)) break;
|
||||
if (client.Server == null) break;
|
||||
if (!client.Server.CanUnlockTechnology(tech)) break;
|
||||
if (client.Server.UnlockTechnology(tech))
|
||||
{
|
||||
database.SyncWithServer();
|
||||
database.Dirty();
|
||||
UpdateUserInterface();
|
||||
}
|
||||
|
||||
break;
|
||||
|
||||
case ConsoleServerSyncMessage _:
|
||||
database.SyncWithServer();
|
||||
UpdateUserInterface();
|
||||
break;
|
||||
|
||||
case ConsoleServerSelectionMessage _:
|
||||
client.OpenUserInterface(message.Session);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Method to update the user interface on the clients.
|
||||
/// </summary>
|
||||
public void UpdateUserInterface()
|
||||
{
|
||||
UserInterface?.SetState(GetNewUiState());
|
||||
}
|
||||
|
||||
private ResearchConsoleBoundInterfaceState GetNewUiState()
|
||||
{
|
||||
if (!_entMan.TryGetComponent(Owner, out ResearchClientComponent? client) ||
|
||||
client.Server == null)
|
||||
return new ResearchConsoleBoundInterfaceState(default, default);
|
||||
|
||||
var points = client.ConnectedToServer ? client.Server.Point : 0;
|
||||
var pointsPerSecond = client.ConnectedToServer ? client.Server.PointsPerSecond : 0;
|
||||
|
||||
return new ResearchConsoleBoundInterfaceState(points, pointsPerSecond);
|
||||
}
|
||||
}
|
||||
public sealed class ResearchConsoleComponent : Component {}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user