Remove FoodContainer, add whitelists, replace with SpawnItemsOnUse (#4358)

2021-07-25 08:41:50 -07:00
parent ae84ba07e6
commit a91f919018
18 changed files with 253 additions and 260 deletions
--- a/Content.Server/Storage/EntitySystems/SpawnItemsOnUseSystem.cs
+++ b/Content.Server/Storage/EntitySystems/SpawnItemsOnUseSystem.cs
@@ -0,0 +1,68 @@
+using System.Collections.Generic;
+using Content.Server.Storage.Components;
+using Content.Shared.Hands.Components;
+using Content.Shared.Interaction;
+using Robust.Shared.Audio;
+using Robust.Shared.GameObjects;
+using Robust.Shared.IoC;
+using Robust.Shared.Player;
+using Robust.Shared.Random;
+
+namespace Content.Server.Storage.EntitySystems
+{
+    public class SpawnItemsOnUseSystem : EntitySystem
+    {
+        [Dependency] private readonly IRobustRandom _random = default!;
+
+        public override void Initialize()
+        {
+            base.Initialize();
+
+            SubscribeLocalEvent<SpawnItemsOnUseComponent, UseInHandEvent>(OnUseInHand);
+        }
+
+        private void OnUseInHand(EntityUid uid, SpawnItemsOnUseComponent component, UseInHandEvent args)
+        {
+            if (args.Handled)
+                return;
+
+            var owner = EntityManager.GetEntity(uid);
+            var alreadySpawnedGroups = new List<string>();
+            IEntity? entityToPlaceInHands = null;
+            foreach (var storageItem in component.Items)
+            {
+                if (!string.IsNullOrEmpty(storageItem.GroupId) &&
+                    alreadySpawnedGroups.Contains(storageItem.GroupId)) continue;
+
+                if (storageItem.SpawnProbability != 1f &&
+                    !_random.Prob(storageItem.SpawnProbability))
+                {
+                    continue;
+                }
+
+                for (var i = 0; i < storageItem.Amount; i++)
+                {
+                    entityToPlaceInHands = EntityManager.SpawnEntity(storageItem.PrototypeId, args.User.Transform.Coordinates);
+                }
+
+                if (!string.IsNullOrEmpty(storageItem.GroupId)) alreadySpawnedGroups.Add(storageItem.GroupId);
+            }
+
+            if (component.Sound != null)
+                SoundSystem.Play(Filter.Pvs(owner), component.Sound.GetSound());
+
+            component.Uses--;
+            if (component.Uses == 0)
+            {
+                args.Handled = true;
+                owner.Delete();
+            }
+
+            if (entityToPlaceInHands != null
+                && args.User.TryGetComponent<SharedHandsComponent>(out var hands))
+            {
+                hands.TryPutInAnyHand(entityToPlaceInHands);
+            }
+        }
+    }
+}