add an option to disable accepting friend request

TF2Autobot · Jul 15, 2020 · c4b3e85 · c4b3e85
1 parent b0c7495
commit c4b3e85
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 1 deletion.
diff --git a/src/classes/Friends.ts b/src/classes/Friends.ts
@@ -6,6 +6,7 @@ import { UnknownDictionary } from '../types/common';
 import Bot from './Bot';
 
 import request from '@nicklason/request-retry';
+import MyHandler from './MyHandler';
 
 export = class Friends {
     private readonly bot: Bot;
@@ -85,10 +86,13 @@ export = class Friends {
                 const result = body.response;
                 const level = result.player_level;
 
+                const friendToKeep = (this.bot.handler as MyHandler).getFriendToKeep();
+                const disableAddFriends = process.env.DISABLE_ADD_FRIENDS === 'true';
+
                 const base = 250;
                 const multiplier = 5;
 
-                this.maxFriends = base + level * multiplier;
+                this.maxFriends = disableAddFriends ? friendToKeep : base + level * multiplier;
 
                 resolve(this.maxFriends);
             });

diff --git a/src/classes/MyHandler.ts b/src/classes/MyHandler.ts
@@ -183,6 +183,10 @@ export = class MyHandler extends Handler {
         }, 1000);
     }
 
+    getFriendToKeep(): number {
+        return this.friendsToKeep.length;
+    }
+
     hasDupeCheckEnabled(): boolean {
         return this.dupeCheckEnabled;
     }

diff --git a/template.ecosystem.json b/template.ecosystem.json
@@ -41,6 +41,7 @@
                 "DISABLE_MESSAGES": false,
                 "DISABLE_SOMETHING_WRONG_ALERT": false,
                 "DISABLE_CRAFTWEAPON_AS_CURRENCY": false,
+                "DISABLE_ADD_FRIENDS": false,
 
                 "TRADES_MADE_STARTER_VALUE": 0,
                 "LAST_TOTAL_TRADES": 0,

diff --git a/template.env b/template.env
@@ -28,6 +28,7 @@ DISABLE_CRAFTING_WEAPONS=false
 DISABLE_MESSAGES=false
 DISABLE_SOMETHING_WRONG_ALERT=false
 DISABLE_CRAFTWEAPON_AS_CURRENCY=false
+DISABLE_ADD_FRIENDS=false
 
 TRADES_MADE_STARTER_VALUE=0
 LAST_TOTAL_TRADES=0