54 lines
1.7 KiB
C#
54 lines
1.7 KiB
C#
using System.Linq;
|
|
using Content.Server.Research.Components;
|
|
using Content.Shared.Lathe;
|
|
using Content.Shared.Research.Prototypes;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Server.Lathe.Components
|
|
{
|
|
[RegisterComponent]
|
|
[ComponentReference(typeof(SharedLatheDatabaseComponent))]
|
|
public sealed class ProtolatheDatabaseComponent : SharedProtolatheDatabaseComponent
|
|
{
|
|
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
|
|
|
|
public override ComponentState GetComponentState()
|
|
{
|
|
return new ProtolatheDatabaseState(GetRecipeIdList());
|
|
}
|
|
|
|
/// <summary>
|
|
/// Adds unlocked recipes from technologies to the database.
|
|
/// </summary>
|
|
public void Sync()
|
|
{
|
|
if (!IoCManager.Resolve<IEntityManager>().TryGetComponent(Owner, out TechnologyDatabaseComponent? database)) return;
|
|
|
|
foreach (var technology in database.Technologies)
|
|
{
|
|
foreach (var id in technology.UnlockedRecipes)
|
|
{
|
|
var recipe = (LatheRecipePrototype) _prototypeManager.Index(typeof(LatheRecipePrototype), id);
|
|
UnlockRecipe(recipe);
|
|
}
|
|
}
|
|
|
|
Dirty();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Unlocks a recipe but only if it's one of the allowed recipes on this protolathe.
|
|
/// </summary>
|
|
/// <param name="recipe">The recipe</param>
|
|
/// <returns>Whether it could add it or not.</returns>
|
|
public bool UnlockRecipe(LatheRecipePrototype recipe)
|
|
{
|
|
if (!ProtolatheRecipes.Contains(recipe)) return false;
|
|
|
|
AddRecipe(recipe);
|
|
|
|
return true;
|
|
}
|
|
}
|
|
}
|