14b0fc5b4a
* solutionResource * fuck research system + QoL workbench update * colorful quartz returns * Refactor workbench skill requirements handling Replaces the SkillRequired requirement with a direct RequiredSkills field on CP14WorkbenchRecipePrototype. Updates client and server logic to check skills directly, simplifying recipe skill checks and related UI logic. * Initial commit Add initial project files and setup.
88 lines
2.5 KiB
C#
88 lines
2.5 KiB
C#
/*
|
|
* This file is sublicensed under MIT License
|
|
* https://github.com/space-wizards/space-station-14/blob/master/LICENSE.TXT
|
|
*/
|
|
|
|
using Content.Shared._CP14.Workbench.Prototypes;
|
|
using Content.Shared.Stacks;
|
|
using Robust.Shared.Prototypes;
|
|
using Robust.Shared.Utility;
|
|
|
|
namespace Content.Shared._CP14.Workbench.Requirements;
|
|
|
|
public sealed partial class StackGroupResource : CP14WorkbenchCraftRequirement
|
|
{
|
|
[DataField(required: true)]
|
|
public ProtoId<CP14StackGroupPrototype> Group;
|
|
|
|
[DataField]
|
|
public int Count = 1;
|
|
|
|
public override bool CheckRequirement(EntityManager entManager,
|
|
IPrototypeManager protoManager,
|
|
HashSet<EntityUid> placedEntities)
|
|
{
|
|
if (!protoManager.TryIndex(Group, out var indexedGroup))
|
|
return false;
|
|
|
|
var count = 0;
|
|
foreach (var ent in placedEntities)
|
|
{
|
|
if (!entManager.TryGetComponent<StackComponent>(ent, out var stack))
|
|
continue;
|
|
|
|
if (!indexedGroup.Stacks.Contains(stack.StackTypeId))
|
|
continue;
|
|
|
|
count += stack.Count;
|
|
}
|
|
|
|
if (count < Count)
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
public override void PostCraft(EntityManager entManager, IPrototypeManager protoManager,
|
|
HashSet<EntityUid> placedEntities)
|
|
{
|
|
var stackSystem = entManager.System<SharedStackSystem>();
|
|
|
|
if (!protoManager.TryIndex(Group, out var indexedGroup))
|
|
return;
|
|
|
|
var requiredCount = Count;
|
|
foreach (var placedEntity in placedEntities)
|
|
{
|
|
if (!entManager.TryGetComponent<StackComponent>(placedEntity, out var stack))
|
|
continue;
|
|
|
|
if (!indexedGroup.Stacks.Contains(stack.StackTypeId))
|
|
continue;
|
|
|
|
var count = (int)MathF.Min(requiredCount, stack.Count);
|
|
|
|
if (stack.Count - count <= 0)
|
|
entManager.DeleteEntity(placedEntity);
|
|
else
|
|
stackSystem.SetCount(placedEntity, stack.Count - count, stack);
|
|
|
|
requiredCount -= count;
|
|
}
|
|
}
|
|
|
|
public override string GetRequirementTitle(IPrototypeManager protoManager)
|
|
{
|
|
var indexedGroup = protoManager.Index(Group);
|
|
|
|
return $"{Loc.GetString(indexedGroup.Name)} x{Count}";
|
|
}
|
|
|
|
public override SpriteSpecifier? GetRequirementTexture(IPrototypeManager protoManager)
|
|
{
|
|
var indexedGroup = protoManager.Index(Group);
|
|
|
|
return !protoManager.TryIndex(indexedGroup.Stacks.FirstOrNull(), out var indexedStack) ? null : indexedStack.Icon;
|
|
}
|
|
}
|