From 71c207b165151012a0d9fc986aa7113ed695edde Mon Sep 17 00:00:00 2001
From: Platinum <platinumgamer@hotmail.com>
Date: Sat, 18 Feb 2023 11:06:16 +1100
Subject: [PATCH] feature: add limit attachment flea prices

---
 config.json |  2 ++
 src/mod.ts  | 40 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 42 insertions(+)

diff --git a/config.json b/config.json
index 113be60..c1190cf 100644
--- a/config.json
+++ b/config.json
@@ -1,5 +1,7 @@
 {
   "disableBsgBlacklist": true,
+  "limitMaxPriceOfAttachments": true,
+  "maxFleaPriceOfAttachmentsToHandbookPrice": 5,
   "blacklistedItemPriceMultiplier": 1,
   "blacklistedAmmoAdditionalPriceMultiplier": 1,
   "blacklistedArmourAdditionalPriceMultiplier": 1,
diff --git a/src/mod.ts b/src/mod.ts
index a5654c5..532c361 100644
--- a/src/mod.ts
+++ b/src/mod.ts
@@ -26,6 +26,7 @@ import { IRagfairConfig } from "@spt-aki/models/spt/config/IRagfairConfig";
 import { ConfigTypes } from "@spt-aki/models/enums/ConfigTypes";
 import { RagfairOfferGenerator } from "@spt-aki/generators/RagfairOfferGenerator";
 import { RagfairPriceService } from "@spt-aki/services/RagfairPriceService";
+import { Category } from "@spt-aki/models/eft/common/tables/IHandbookBase";
 
 import config from "../config.json";
 import advancedConfig from "../advancedConfig.json";
@@ -39,6 +40,8 @@ class TheBlacklistMod implements IPostDBLoadMod {
   private baselineBullet: ITemplateItem;
   private baselineArmour: ITemplateItem;
 
+  // Store the category IDs of all attachments in the handbook so we don't have to manually enter them in json
+  private attachmentCategoryIds: string[] = [];
 
   public postDBLoad(container: DependencyContainer): void {
     this.logger = container.resolve<ILogger>("WinstonLogger");
@@ -67,6 +70,11 @@ class TheBlacklistMod implements IPostDBLoadMod {
     this.baselineArmour = itemTable[advancedConfig.baselineArmourId];
 
     let blacklistedItemsCount = 0;
+    let attachmentPriceLimitedCount = 0;
+
+    if (config.limitMaxPriceOfAttachments) {
+      this.initialiseAttachmentCategoryIds(tables.templates.handbook.Categories);
+    }
 
     // Find all items to update by looping through handbook which is a better indicator of useable items.
     handbookItems.forEach(handbookItem => {
@@ -82,6 +90,18 @@ class TheBlacklistMod implements IPostDBLoadMod {
         return;
       }
 
+      if (config.limitMaxPriceOfAttachments && this.attachmentCategoryIds.includes(handbookItem.ParentId)) {
+        const handbookPrice = handbookItem.Price;
+        const existingFleaPrice = prices[item._id];
+        const maxFleaPrice = handbookPrice * config.maxFleaPriceOfAttachmentsToHandbookPrice;
+        
+        if (existingFleaPrice > maxFleaPrice) {
+          prices[item._id] = maxFleaPrice;
+          attachmentPriceLimitedCount++;
+          this.debug(`Attachment ${item._id} - ${item._name} was updated from ${existingFleaPrice} to ${maxFleaPrice}.`)
+        }
+      }
+
       const itemProps = item._props;
       if (!itemProps.CanSellOnRagfair) {
         // Some blacklisted items are hard to balance or just shouldn't be allowed so we will keep them blacklisted.
@@ -113,9 +133,29 @@ class TheBlacklistMod implements IPostDBLoadMod {
     (ragfairPriceService as any).generateDynamicPrices();
     ragfairOfferGenerator.generateDynamicOffers().then(() => {
       this.logger.success(`${this.modName}: Success! Found ${blacklistedItemsCount} blacklisted items to update.`);
+      this.logger.success(`${this.modName}: config.limitMaxPriceOfAttachments is enabled! Updated ${attachmentPriceLimitedCount} flea prices of attachments.`)
     });
   }
 
+  private initialiseAttachmentCategoryIds(handbookCategories: Category[]) {
+    const weaponPartsAndModsId = "5b5f71a686f77447ed5636ab";
+    const weaponPartsChildrenCategories = this.getChildCategoriesRecursively(handbookCategories, weaponPartsAndModsId);
+    const childrenIds = weaponPartsChildrenCategories.map(category => category.Id);
+
+    this.attachmentCategoryIds.push(weaponPartsAndModsId);
+    this.attachmentCategoryIds = this.attachmentCategoryIds.concat(childrenIds);
+  }
+
+  private getChildCategoriesRecursively(handbookCategories: Category[], parentId: string): Category[] {
+    const childCategories = handbookCategories.filter(category => category.ParentId === parentId);
+    const grandChildrenCategories = childCategories.reduce(
+      (memo, category) => memo.concat(this.getChildCategoriesRecursively(handbookCategories, category.Id)), 
+      []
+    );
+  
+    return childCategories.concat(grandChildrenCategories);
+  }
+
   private getUpdatedPrice(item: ITemplateItem, prices: Record<string, number>): number | undefined {
     // Note that this price can be affected by other mods like Lua's market updater.
     const currentFleaPrice = prices[item._id];