tbd-station-14/Content.Server/Construction/Completions/SetStackCount.cs

#nullable enable
using System.Threading.Tasks;
using Content.Server.GameObjects.Components.Stack;
using Content.Shared.Construction;
using JetBrains.Annotations;
using Robust.Shared.Interfaces.GameObjects;
using System;
using Robust.Shared.Log;
using Robust.Shared.Serialization;

namespace Content.Server.Construction.Completions
{
    [UsedImplicitly]
    public class SetStackCount : IGraphAction
    {
        public int Amount { get; private set; }

        public void ExposeData(ObjectSerializer serializer)
        {
            serializer.DataField(this, x => x.Amount, "amount", 1);
        }

        public async Task PerformAction(IEntity entity, IEntity? user)
        {
            if (entity.Deleted) return;
            if(!entity.TryGetComponent(out StackComponent? stackComponent)) return;

            stackComponent.Count = Math.Min(stackComponent.MaxCount, Amount);

            if (Amount > stackComponent.MaxCount)
            {
                Logger.Warning("StackCount is bigger than maximum stack capacity, for entity " + entity.Name);
            }
        }
    }
}