53 lines
2.0 KiB
C#
53 lines
2.0 KiB
C#
using Content.Server.Construction.Components;
|
|
using Content.Server.Stack;
|
|
using Content.Shared.Construction;
|
|
using Content.Shared.DoAfter;
|
|
using Content.Shared.Interaction;
|
|
using Content.Shared.Stacks;
|
|
using Content.Shared.Tools;
|
|
using Robust.Shared.Serialization;
|
|
|
|
namespace Content.Server.Construction
|
|
{
|
|
public sealed class RefiningSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly SharedToolSystem _toolSystem = default!;
|
|
[Dependency] private readonly StackSystem _stackSystem = default!;
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
SubscribeLocalEvent<WelderRefinableComponent, InteractUsingEvent>(OnInteractUsing);
|
|
SubscribeLocalEvent<WelderRefinableComponent, WelderRefineDoAfterEvent>(OnDoAfter);
|
|
}
|
|
|
|
private void OnInteractUsing(EntityUid uid, WelderRefinableComponent component, InteractUsingEvent args)
|
|
{
|
|
if (args.Handled)
|
|
return;
|
|
|
|
args.Handled = _toolSystem.UseTool(args.Used, args.User, uid, component.RefineTime, component.QualityNeeded, new WelderRefineDoAfterEvent());
|
|
}
|
|
|
|
private void OnDoAfter(EntityUid uid, WelderRefinableComponent component, WelderRefineDoAfterEvent args)
|
|
{
|
|
if (args.Cancelled)
|
|
return;
|
|
|
|
// get last owner coordinates and delete it
|
|
var resultPosition = Transform(uid).Coordinates;
|
|
EntityManager.DeleteEntity(uid);
|
|
|
|
// spawn each result after refine
|
|
foreach (var result in component.RefineResult!)
|
|
{
|
|
var droppedEnt = EntityManager.SpawnEntity(result, resultPosition);
|
|
|
|
// TODO: If something has a stack... Just use a prototype with a single thing in the stack.
|
|
// This is not a good way to do it.
|
|
if (TryComp<StackComponent?>(droppedEnt, out var stack))
|
|
_stackSystem.SetCount(droppedEnt, 1, stack);
|
|
}
|
|
}
|
|
}
|
|
}
|