Add TM Type

xsn34kzx · Jul 9, 2024 · c278061 · c278061
1 parent 2bd2152
commit c278061
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/modifier/modifier-type.ts b/src/modifier/modifier-type.ts
@@ -968,7 +968,10 @@ class SpeciesStatBoosterModifierTypeGenerator extends ModifierTypeGenerator {
 
 class TmModifierTypeGenerator extends ModifierTypeGenerator {
   constructor(tier: ModifierTier) {
-    super((party: Pokemon[]) => {
+    super((party: Pokemon[], pregenArgs?: any[]) => {
+      if (pregenArgs) {
+        return new TmModifierType(pregenArgs[0] as Moves);
+      }
       const partyMemberCompatibleTms = party.map(p => (p as PlayerPokemon).compatibleTms.filter(tm => !p.moveset.find(m => m.moveId === tm)));
       const tierUniqueCompatibleTms = partyMemberCompatibleTms.flat().filter(tm => tmPoolTiers[tm] === tier).filter(tm => !allMoves[tm].name.endsWith(" (N)")).filter((tm, i, array) => array.indexOf(tm) === i);
       if (!tierUniqueCompatibleTms.length) {
@@ -2192,6 +2195,7 @@ type GeneratorModifierOverride = {
     }
   | {
       name: "TM_COMMON" | "TM_GREAT" | "TM_ULTRA";
+      type?: Moves;
     }
 );