refactors barsigncomponent to be ecs (#4332)

Co-authored-by: Paul <ritter.paul1+git@googlemail.com>
This commit is contained in:
Paul Ritter
2021-07-23 04:42:39 +02:00
committed by GitHub
parent 975ec19981
commit 4503a6f0ee
2 changed files with 99 additions and 95 deletions

View File

@@ -13,103 +13,11 @@ using Robust.Shared.ViewVariables;
namespace Content.Server.BarSign namespace Content.Server.BarSign
{ {
[RegisterComponent] [RegisterComponent]
public class BarSignComponent : Component, IMapInit public class BarSignComponent : Component
{ {
public override string Name => "BarSign"; public override string Name => "BarSign";
[Dependency] private readonly IPrototypeManager _prototypeManager = default!; [DataField("current")] [ViewVariables(VVAccess.ReadOnly)]
[Dependency] private readonly IRobustRandom _robustRandom = default!; public string? CurrentSign;
[DataField("current")]
private string? _currentSign;
[ViewVariables(VVAccess.ReadWrite)]
public string? CurrentSign
{
get => _currentSign;
set
{
_currentSign = value;
UpdateSignInfo();
}
}
private bool Powered => !Owner.TryGetComponent(out ApcPowerReceiverComponent? receiver) || receiver.Powered;
private void UpdateSignInfo()
{
if (_currentSign == null)
{
return;
}
if (!_prototypeManager.TryIndex(_currentSign, out BarSignPrototype? prototype))
{
Logger.ErrorS("barSign", $"Invalid bar sign prototype: \"{_currentSign}\"");
return;
}
if (Owner.TryGetComponent(out SpriteComponent? sprite))
{
if (!Powered)
{
sprite.LayerSetState(0, "empty");
sprite.LayerSetShader(0, "shaded");
}
else
{
sprite.LayerSetState(0, prototype.Icon);
sprite.LayerSetShader(0, "unshaded");
}
}
if (!string.IsNullOrEmpty(prototype.Name))
{
Owner.Name = prototype.Name;
}
else
{
Owner.Name = Loc.GetString("barsign-component-name");
}
Owner.Description = prototype.Description;
}
protected override void Initialize()
{
base.Initialize();
UpdateSignInfo();
}
public override void HandleMessage(ComponentMessage message, IComponent? component)
{
base.HandleMessage(message, component);
switch (message)
{
case PowerChangedMessage powerChanged:
PowerOnOnPowerStateChanged(powerChanged);
break;
}
}
private void PowerOnOnPowerStateChanged(PowerChangedMessage e)
{
UpdateSignInfo();
}
public void MapInit()
{
if (_currentSign != null)
{
return;
}
var prototypes = _prototypeManager.EnumeratePrototypes<BarSignPrototype>().Where(p => !p.Hidden)
.ToList();
var prototype = _robustRandom.Pick(prototypes);
CurrentSign = prototype.ID;
}
} }
} }

View File

@@ -0,0 +1,96 @@
using System.Linq;
using Content.Server.Power.Components;
using Robust.Server.GameObjects;
using Robust.Shared.GameObjects;
using Robust.Shared.IoC;
using Robust.Shared.Localization;
using Robust.Shared.Log;
using Robust.Shared.Prototypes;
using Robust.Shared.Random;
namespace Content.Server.BarSign.Systems
{
public class BarSignSystem : EntitySystem
{
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
[Dependency] private readonly IRobustRandom _random = default!;
public override void Initialize()
{
base.Initialize();
SubscribeLocalEvent<BarSignComponent, ComponentInit>(OnComponentInit);
SubscribeLocalEvent<BarSignComponent, MapInitEvent>(OnMapInit);
SubscribeLocalEvent<BarSignComponent, PowerChangedEvent>(OnPowerChanged);
}
public void SetBarSign(BarSignComponent component, string barSign)
{
component.CurrentSign = barSign;
UpdateBarSignVisuals(component);
}
private void OnComponentInit(EntityUid uid, BarSignComponent component, ComponentInit args)
{
UpdateBarSignVisuals(component);
}
private void OnMapInit(EntityUid uid, BarSignComponent component, MapInitEvent args)
{
if (component.CurrentSign != null)
{
return;
}
var prototypes = _prototypeManager.EnumeratePrototypes<BarSignPrototype>().Where(p => !p.Hidden)
.ToList();
var prototype = _random.Pick(prototypes);
component.CurrentSign = prototype.ID;
}
private void OnPowerChanged(EntityUid uid, BarSignComponent component, PowerChangedEvent args)
{
UpdateBarSignVisuals(component);
}
private void UpdateBarSignVisuals(BarSignComponent component)
{
if (component.CurrentSign == null)
{
return;
}
if (!_prototypeManager.TryIndex(component.CurrentSign, out BarSignPrototype? prototype))
{
Logger.ErrorS("barSign", $"Invalid bar sign prototype: \"{component.CurrentSign}\"");
return;
}
if (component.Owner.TryGetComponent(out SpriteComponent? sprite))
{
if (!component.Owner.TryGetComponent(out ApcPowerReceiverComponent? receiver) || receiver.Powered)
{
sprite.LayerSetState(0, "empty");
sprite.LayerSetShader(0, "shaded");
}
else
{
sprite.LayerSetState(0, prototype.Icon);
sprite.LayerSetShader(0, "unshaded");
}
}
if (!string.IsNullOrEmpty(prototype.Name))
{
component.Owner.Name = prototype.Name;
}
else
{
component.Owner.Name = Loc.GetString("barsign-component-name");
}
component.Owner.Description = prototype.Description;
}
}
}