31 lines
943 B
C#
31 lines
943 B
C#
using Content.Shared.Lathe;
|
|
using Content.Shared.Research.Prototypes;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.IoC;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Client.Lathe.Components
|
|
{
|
|
[RegisterComponent]
|
|
[ComponentReference(typeof(SharedLatheDatabaseComponent))]
|
|
public class LatheDatabaseComponent : SharedLatheDatabaseComponent
|
|
{
|
|
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
|
|
|
|
public override void HandleComponentState(ComponentState? curState, ComponentState? nextState)
|
|
{
|
|
base.HandleComponentState(curState, nextState);
|
|
|
|
if (curState is not LatheDatabaseState state) return;
|
|
|
|
Clear();
|
|
|
|
foreach (var ID in state.Recipes)
|
|
{
|
|
if (!_prototypeManager.TryIndex(ID, out LatheRecipePrototype? recipe)) continue;
|
|
AddRecipe(recipe);
|
|
}
|
|
}
|
|
}
|
|
}
|