🔀Merge pull request #408 from TF2Autobot/option-exclude-partial-price

✅ opt to exclude items for partial price update
TF2Autobot · Mar 4, 2021 · e97de9d · e97de9d
2 parents ca2bc1b + 5a6b0ee
commit e97de9d
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/src/classes/Options.ts b/src/classes/Options.ts
@@ -72,7 +72,8 @@ export const DEFAULTS = {
     pricelist: {
         partialPriceUpdate: {
             enable: false,
-            thresholdInSeconds: 604800 // 7 days
+            thresholdInSeconds: 604800, // 7 days
+            excludeSKU: []
         },
         filterCantAfford: {
             enable: false
@@ -1058,6 +1059,7 @@ interface Pricelist {
 
 interface PartialPriceUpdate extends OnlyEnable {
     thresholdInSeconds?: number;
+    excludeSKU?: string[];
 }
 
 interface PriceAge {

diff --git a/src/classes/Pricelist.ts b/src/classes/Pricelist.ts
@@ -697,6 +697,7 @@ export default class Pricelist extends EventEmitter {
             // Go through our pricelist
             const oldCount = old.length;
             const opt = this.options.pricelist.partialPriceUpdate;
+            const excludedSKU = ['5021;6'].concat(opt.excludeSKU);
             const keyPrice = this.getKeyPrice.metal;
 
             for (let i = 0; i < oldCount; i++) {
@@ -733,8 +734,9 @@ export default class Pricelist extends EventEmitter {
                             const currSellingValue = currPrice.sell.toValue(keyPrice);
 
                             const isNotExceedThreshold = newestPrice.time - currPrice.time < opt.thresholdInSeconds;
+                            const isNotExcluded = !excludedSKU.includes(currPrice.sku);
 
-                            if (opt.enable && isInStock && isNotExceedThreshold && currPrice.sku !== '5021;6') {
+                            if (opt.enable && isInStock && isNotExceedThreshold && isNotExcluded) {
                                 // if optPartialUpdate.enable is true and the item is currently in stock
                                 // and difference between latest time and time recorded in pricelist is less than threshold
 
@@ -859,13 +861,14 @@ export default class Pricelist extends EventEmitter {
             const optPartialUpdate = opt.pricelist.partialPriceUpdate;
             const isInStock = this.bot.inventoryManager.getInventory.getAmount(match.sku, true) > 0;
             const isNotExceedThreshold = data.time - match.time < optPartialUpdate.thresholdInSeconds;
+            const isNotExcluded = !['5021;6'].concat(optPartialUpdate.excludeSKU).includes(match.sku);
 
             if (
                 optPartialUpdate.enable &&
                 isInStock &&
                 isNotExceedThreshold &&
                 this.globalKeyPrices !== undefined &&
-                match.sku !== '5021;6'
+                isNotExcluded
             ) {
                 // if optPartialUpdate.enable is true and the item is currently in stock
                 // and difference between latest time and time recorded in pricelist is less than threshold

diff --git a/src/schemas/options-json/options.ts b/src/schemas/options-json/options.ts
@@ -509,9 +509,12 @@ export const optionsSchema: jsonschema.Schema = {
                         thresholdInSeconds: {
                             type: 'integer',
                             minimum: 86400 // 1 day
+                        },
+                        excludeSKU: {
+                            type: '#/definitions/string-array'
                         }
                     },
-                    required: ['enable', 'thresholdInSeconds'],
+                    required: ['enable', 'thresholdInSeconds', 'excludeSKU'],
                     additionalProperties: false
                 },
                 filterCantAfford: {