tbd-station-14/Content.Server/Lathe/Components/MaterialStorageComponent.cs

using Content.Shared.Lathe;
using Content.Shared.Whitelist;
using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype.List;
using Content.Shared.Materials;
using Robust.Shared.Audio;

namespace Content.Server.Lathe.Components
{
    [RegisterComponent]
    [ComponentReference(typeof(SharedMaterialStorageComponent))]
    public sealed class MaterialStorageComponent : SharedMaterialStorageComponent
    {
        [ViewVariables]
        protected override Dictionary<string, int> Storage { get; set; } = new();

        /// <summary>
        ///     How much material the storage can store in total.
        /// </summary>
        [ViewVariables]
        public int StorageLimit => _storageLimit;
        [DataField("StorageLimit")]
        private int _storageLimit = -1;

        /// <summary>
        /// Whitelist for specifying the kind of items that can be insert into this entity.
        /// </summary>
        [ViewVariables]
        [DataField("whitelist")]
        public EntityWhitelist? EntityWhitelist;

        /// <summary>
        /// Whitelist generated on runtime for what specific materials can be inserted into this entity.
        /// </summary>
        [ViewVariables]
        [DataField("materialWhiteList", customTypeSerializer: typeof(PrototypeIdListSerializer<MaterialPrototype>))]
        public List<string> MaterialWhiteList = new();

        /// <summary>
        /// The sound that plays when inserting an item into the storage
        /// </summary>
        [DataField("insertingSound")]
        public SoundSpecifier? InsertingSound;

        public override ComponentState GetComponentState()
        {
            return new MaterialStorageState(Storage);
        }

        /// <summary>
        ///     Checks if the storage can take a volume of material without surpassing its own limits.
        /// </summary>
        /// <param name="amount">The volume of material</param>
        /// <returns></returns>
        public bool CanTakeAmount(int amount)
        {
            return CurrentAmount + amount <= StorageLimit;
        }

        /// <summary>
        ///     Checks if it can insert a material.
        /// </summary>
        /// <param name="id">Material ID</param>
        /// <param name="amount">How much to insert</param>
        /// <returns>Whether it can insert the material or not.</returns>
        public bool CanInsertMaterial(string id, int amount)
        {
            return (CanTakeAmount(amount) || StorageLimit < 0) && (!Storage.ContainsKey(id) || Storage[id] + amount >= 0);
        }

        /// <summary>
        ///     Inserts material into the storage.
        /// </summary>
        /// <param name="id">Material ID</param>
        /// <param name="amount">How much to insert</param>
        /// <returns>Whether it inserted it or not.</returns>
        public bool InsertMaterial(string id, int amount)
        {
            if (!CanInsertMaterial(id, amount)) return false;

            if (!Storage.ContainsKey(id))
                Storage.Add(id, 0);

            Storage[id] += amount;

            Dirty();

            return true;
        }

        /// <summary>
        ///     Removes material from the storage.
        /// </summary>
        /// <param name="id">Material ID</param>
        /// <param name="amount">How much to remove</param>
        /// <returns>Whether it removed it or not.</returns>
        public bool RemoveMaterial(string id, int amount)
        {
            return InsertMaterial(id, -amount);
        }

        // forgive me I needed to write a crumb of e/c code to not go fucking insane i swear i will ecs this entire shitty fucking system one day
        public int GetMaterialAmount(string id)
        {
            if (!Storage.TryGetValue(id, out var amount))
                return 0;

            return amount;
        }
    }
}