From 459ec0d101d708c1a5e0e4944b0a001dd0f2257e Mon Sep 17 00:00:00 2001
From: Platinum <platinumgamer@hotmail.com>
Date: Sun, 11 Jun 2023 16:13:19 +1000
Subject: [PATCH] Restrict to one slot wide

---
 config.json |  5 ++---
 src/mod.ts  | 10 ++++++----
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/config.json b/config.json
index 5760617..2dbe420 100644
--- a/config.json
+++ b/config.json
@@ -1,5 +1,4 @@
 {
-  "minMagazineSizeToBeIncluded": 30,
-  "maxMagazineSizeToBeIncluded": 60,
-  "newMagazineInventorySlotSize": 2
+  "minMagazineCapacityToBeIncluded": 40,
+  "maxMagazineCapacityToBeIncluded": 50
 }
\ No newline at end of file
diff --git a/src/mod.ts b/src/mod.ts
index cceec60..f1d5bae 100644
--- a/src/mod.ts
+++ b/src/mod.ts
@@ -35,20 +35,22 @@ class TwoSlotExtendedMags implements IPostDBLoadMod {
   }
 
   private updateInventorySlotSize(itemProp: Props): void {
-    itemProp.Height = config.newMagazineInventorySlotSize;
+    itemProp.Height = 2;
     // itemProp.ExtraSizeDown = config.newMagazineInventorySlotSize--;
   }
 
   private isExtendedMag(item: ITemplateItem): boolean {
     const magazineCategoryId = "5448bc234bdc2d3c308b4569";
     
-    return item._parent == magazineCategoryId && this.isMagazineSizeWithinCapacity(item._props);
+    return item._parent == magazineCategoryId &&
+      item._props.Width == 1 && // We don't want to make horizontal mags like P90's to be 4 squares wide or change drum mags (for now)
+      this.isWithinMagazineSizeCapacity(item._props);
   }
 
-  private isMagazineSizeWithinCapacity(itemProp: Props): boolean {
+  private isWithinMagazineSizeCapacity(itemProp: Props): boolean {
     const capacity = this.getMagazineCapacity(itemProp);
 
-    return capacity >= config.minMagazineSizeToBeIncluded && capacity <= config.maxMagazineSizeToBeIncluded;
+    return capacity >= config.minMagazineCapacityToBeIncluded && capacity <= config.maxMagazineCapacityToBeIncluded;
   }
 
   private getMagazineCapacity(itemProp: Props): number {