Max output 3 items for free

new-frontiers-14 · Aug 7, 2023 · 13b69f8 · 13b69f8
1 parent ce6bf1b
commit 13b69f8
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 9 deletions.
diff --git a/Content.Server/VendingMachines/VendingMachineSystem.cs b/Content.Server/VendingMachines/VendingMachineSystem.cs
@@ -415,17 +415,21 @@ public void EjectRandom(EntityUid uid, bool throwItem, bool forceEject = false,
 
             var item = _random.Pick(availableItems);
 
-            if (forceEject)
+            if (vendComponent.EjectRandomMax > vendComponent.EjectRandomCounter)
             {
-                vendComponent.NextItemToEject = item.ID;
-                vendComponent.ThrowNextItem = throwItem;
-                var entry = GetEntry(uid, item.ID, item.Type, vendComponent);
-                if (entry != null)
-                    entry.Amount--;
-                //EjectItem(uid, vendComponent, forceEject); // Stop vending machine from giving free items
+                if (forceEject)
+                {
+                    vendComponent.NextItemToEject = item.ID;
+                    vendComponent.ThrowNextItem = throwItem;
+                    var entry = GetEntry(uid, item.ID, item.Type, vendComponent);
+                    if (entry != null)
+                        entry.Amount--;
+                    EjectItem(uid, vendComponent, forceEject);
+                }
+                else
+                TryEjectVendorItem(uid, item.Type, item.ID, throwItem, 0, vendComponent);
+                vendComponent.EjectRandomCounter += 1;
             }
-            //else
-            //TryEjectVendorItem(uid, item.Type, item.ID, throwItem, 0, vendComponent); // Stop vending machine from giving free items
         }
 
         private void EjectItem(EntityUid uid, VendingMachineComponent? vendComponent = null, bool forceEject = false)

diff --git a/Content.Shared/VendingMachines/VendingMachineComponent.cs b/Content.Shared/VendingMachines/VendingMachineComponent.cs
@@ -32,6 +32,18 @@ public sealed class VendingMachineComponent : Component
         [DataField("ejectDelay")]
         public float EjectDelay = 1.2f;
 
+        /// <summary>
+        /// Used by the server to determine how many items the machine allowed to eject from random triggers.
+        /// </summary>
+        [DataField("ejectRandomMax")]
+        public float EjectRandomMax = 3f;
+
+        /// <summary>
+        /// Used by the server to determine how many items the machine ejected from random triggers.
+        /// </summary>
+        [DataField("ejectRandomCounter")]
+        public float EjectRandomCounter = 0f;
+
         [ViewVariables]
         public Dictionary<string, VendingMachineInventoryEntry> Inventory = new();