Add a test that puts all components on an entity and checks for no exceptions (#1815)
* 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
This commit is contained in:
@@ -1,4 +1,5 @@
|
||||
using Content.Server.GameObjects.Components.Power.ApcNetComponents;
|
||||
#nullable enable
|
||||
using Content.Server.GameObjects.Components.Power.ApcNetComponents;
|
||||
using Content.Shared.Audio;
|
||||
using Content.Shared.GameObjects.Components.Research;
|
||||
using Content.Shared.Interfaces.GameObjects.Components;
|
||||
@@ -14,6 +15,7 @@ using Robust.Shared.Interfaces.Random;
|
||||
using Robust.Shared.IoC;
|
||||
using Robust.Shared.Prototypes;
|
||||
using Robust.Shared.Random;
|
||||
using Robust.Shared.ViewVariables;
|
||||
|
||||
namespace Content.Server.GameObjects.Components.Research
|
||||
{
|
||||
@@ -21,31 +23,38 @@ namespace Content.Server.GameObjects.Components.Research
|
||||
[ComponentReference(typeof(IActivate))]
|
||||
public class ResearchConsoleComponent : SharedResearchConsoleComponent, IActivate
|
||||
{
|
||||
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
|
||||
[Dependency] private readonly IRobustRandom _random = default!;
|
||||
|
||||
#pragma warning disable 649
|
||||
[Dependency] private readonly IPrototypeManager _prototypeManager;
|
||||
[Dependency] private readonly IRobustRandom _random;
|
||||
#pragma warning restore 649
|
||||
private const string SoundCollectionName = "keyboard";
|
||||
|
||||
private BoundUserInterface _userInterface;
|
||||
private ResearchClientComponent _client;
|
||||
private PowerReceiverComponent _powerReceiver;
|
||||
private const string _soundCollectionName = "keyboard";
|
||||
private bool Powered => !Owner.TryGetComponent(out PowerReceiverComponent? receiver) || receiver.Powered;
|
||||
|
||||
private bool Powered => _powerReceiver.Powered;
|
||||
[ViewVariables]
|
||||
private BoundUserInterface? UserInterface =>
|
||||
Owner.TryGetComponent(out ServerUserInterfaceComponent? ui) &&
|
||||
ui.TryGetBoundUserInterface(ResearchConsoleUiKey.Key, out var boundUi)
|
||||
? boundUi
|
||||
: null;
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
_userInterface = Owner.GetComponent<ServerUserInterfaceComponent>().GetBoundUserInterface(ResearchConsoleUiKey.Key);
|
||||
_userInterface.OnReceiveMessage += UserInterfaceOnOnReceiveMessage;
|
||||
_client = Owner.GetComponent<ResearchClientComponent>();
|
||||
_powerReceiver = Owner.GetComponent<PowerReceiverComponent>();
|
||||
|
||||
if (UserInterface != null)
|
||||
{
|
||||
UserInterface.OnReceiveMessage += UserInterfaceOnOnReceiveMessage;
|
||||
}
|
||||
|
||||
Owner.EnsureComponent<ResearchClientComponent>();
|
||||
}
|
||||
|
||||
private void UserInterfaceOnOnReceiveMessage(ServerBoundUserInterfaceMessage message)
|
||||
{
|
||||
if (!Owner.TryGetComponent(out TechnologyDatabaseComponent database)) return;
|
||||
if (!Owner.TryGetComponent(out TechnologyDatabaseComponent? database))
|
||||
return;
|
||||
if (!Owner.TryGetComponent(out ResearchClientComponent? client))
|
||||
return;
|
||||
if (!Powered)
|
||||
return;
|
||||
|
||||
@@ -54,8 +63,9 @@ namespace Content.Server.GameObjects.Components.Research
|
||||
case ConsoleUnlockTechnologyMessage msg:
|
||||
var protoMan = IoCManager.Resolve<IPrototypeManager>();
|
||||
if (!protoMan.TryIndex(msg.Id, out TechnologyPrototype tech)) break;
|
||||
if(!_client.Server.CanUnlockTechnology(tech)) break;
|
||||
if (_client.Server.UnlockTechnology(tech))
|
||||
if (client.Server == null) break;
|
||||
if (!client.Server.CanUnlockTechnology(tech)) break;
|
||||
if (client.Server.UnlockTechnology(tech))
|
||||
{
|
||||
database.SyncWithServer();
|
||||
database.Dirty();
|
||||
@@ -64,13 +74,12 @@ namespace Content.Server.GameObjects.Components.Research
|
||||
|
||||
break;
|
||||
|
||||
case ConsoleServerSyncMessage msg:
|
||||
case ConsoleServerSyncMessage _:
|
||||
database.SyncWithServer();
|
||||
UpdateUserInterface();
|
||||
break;
|
||||
|
||||
case ConsoleServerSelectionMessage msg:
|
||||
if (!Owner.TryGetComponent(out ResearchClientComponent client)) break;
|
||||
case ConsoleServerSelectionMessage _:
|
||||
client.OpenUserInterface(message.Session);
|
||||
break;
|
||||
}
|
||||
@@ -81,13 +90,17 @@ namespace Content.Server.GameObjects.Components.Research
|
||||
/// </summary>
|
||||
public void UpdateUserInterface()
|
||||
{
|
||||
_userInterface.SetState(GetNewUiState());
|
||||
UserInterface?.SetState(GetNewUiState());
|
||||
}
|
||||
|
||||
private ResearchConsoleBoundInterfaceState GetNewUiState()
|
||||
{
|
||||
var points = _client.ConnectedToServer ? _client.Server.Point : 0;
|
||||
var pointsPerSecond = _client.ConnectedToServer ? _client.Server.PointsPerSecond : 0;
|
||||
if (!Owner.TryGetComponent(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);
|
||||
}
|
||||
@@ -98,12 +111,12 @@ namespace Content.Server.GameObjects.Components.Research
|
||||
/// <param name="session">Session where the UI will be shown</param>
|
||||
public void OpenUserInterface(IPlayerSession session)
|
||||
{
|
||||
_userInterface.Open(session);
|
||||
UserInterface?.Open(session);
|
||||
}
|
||||
|
||||
void IActivate.Activate(ActivateEventArgs eventArgs)
|
||||
{
|
||||
if (!eventArgs.User.TryGetComponent(out IActorComponent actor))
|
||||
if (!eventArgs.User.TryGetComponent(out IActorComponent? actor))
|
||||
return;
|
||||
if (!Powered)
|
||||
{
|
||||
@@ -112,17 +125,14 @@ namespace Content.Server.GameObjects.Components.Research
|
||||
|
||||
OpenUserInterface(actor.playerSession);
|
||||
PlayKeyboardSound();
|
||||
return;
|
||||
}
|
||||
|
||||
private void PlayKeyboardSound()
|
||||
{
|
||||
var soundCollection = _prototypeManager.Index<SoundCollectionPrototype>(_soundCollectionName);
|
||||
var soundCollection = _prototypeManager.Index<SoundCollectionPrototype>(SoundCollectionName);
|
||||
var file = _random.Pick(soundCollection.PickFiles);
|
||||
var audioSystem = EntitySystem.Get<AudioSystem>();
|
||||
audioSystem.PlayFromEntity(file,Owner,AudioParams.Default);
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user