diff --git a/project/src/generators/BotGenerator.ts b/project/src/generators/BotGenerator.ts index 01900dcc..18cdcadb 100644 --- a/project/src/generators/BotGenerator.ts +++ b/project/src/generators/BotGenerator.ts @@ -592,21 +592,24 @@ export class BotGenerator return botInfo.GameVersion; } - // More color = more op + // Choose random weighted game version for bot botInfo.GameVersion = this.weightedRandomHelper.getWeightedValue(this.pmcConfig.gameVersionWeight); - // EOD and UhD must have a category of 2 for their unique icons to show - if ([GameEditions.EDGE_OF_DARKNESS, GameEditions.UNHEARD].includes(botInfo.GameVersion)) + // Choose appropriate member category value + switch (botInfo.GameVersion) { - botInfo.MemberCategory = 2; - } - else - { - // Everyone else gets a weighted randomised category - botInfo.MemberCategory = Number.parseInt( - this.weightedRandomHelper.getWeightedValue(this.pmcConfig.accountTypeWeight), - 10, - ); + case GameEditions.EDGE_OF_DARKNESS: + botInfo.MemberCategory = MemberCategory.UNIQUE_ID; + break; + case GameEditions.UNHEARD: + botInfo.MemberCategory = MemberCategory.UNHEARD; + break; + default: + // Everyone else gets a weighted randomised category + botInfo.MemberCategory = Number.parseInt( + this.weightedRandomHelper.getWeightedValue(this.pmcConfig.accountTypeWeight), + 10, + ); } return botInfo.GameVersion; diff --git a/project/src/models/enums/MemberCategory.ts b/project/src/models/enums/MemberCategory.ts index be692e2a..ac98d8c8 100644 --- a/project/src/models/enums/MemberCategory.ts +++ b/project/src/models/enums/MemberCategory.ts @@ -11,4 +11,5 @@ export enum MemberCategory UNIT_TEST = 128, SHERPA = 256, EMISSARY = 512, + UNHEARD = 1024, }