94 lines
3.3 KiB
C#
94 lines
3.3 KiB
C#
using Content.Server.Power.Components;
|
|
using Content.Server.UserInterface;
|
|
using Content.Shared.Research.Components;
|
|
using Content.Shared.Research.Prototypes;
|
|
using Content.Shared.Sound;
|
|
using Robust.Server.GameObjects;
|
|
using Robust.Server.Player;
|
|
using Robust.Shared.Audio;
|
|
using Robust.Shared.Player;
|
|
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);
|
|
}
|
|
}
|
|
}
|