Add Living/Twisted Garden stager

m-h-c-t · Nov 30, 2023 · bc10bb3 · bc10bb3
1 parent 7bca78d
commit bc10bb3
Show file tree

Hide file tree

Showing 9 changed files with 113 additions and 25 deletions.
diff --git a/src/scripts/main.js b/src/scripts/main.js
@@ -1185,7 +1185,6 @@ import * as detailingFuncs from './modules/details/legacy';
         "Furoma Rift": stagingFuncs.addFuromaRiftStage,
         "Gnawnian Express Station": stagingFuncs.addTrainStage,
         "Iceberg": stagingFuncs.addIcebergStage,
-        "Living Garden": stagingFuncs.addGardenStage,
         "Lost City": stagingFuncs.addLostCityStage,
         "Muridae Market": stagingFuncs.addMuridaeMarketStage,
         "Queso Geyser": stagingFuncs.addQuesoGeyserStage,
@@ -1194,7 +1193,6 @@ import * as detailingFuncs from './modules/details/legacy';
         "Sunken City": stagingFuncs.addSunkenCityStage,
         "Table of Contents": stagingFuncs.addTableOfContentsStage,
         "Toxic Spill": stagingFuncs.addToxicSpillStage,
-        "Twisted Garden": stagingFuncs.addGardenStage,
         "Valour Rift": stagingFuncs.addValourRiftStage,
         "Whisker Woods Rift": stagingFuncs.addWhiskerWoodsRiftStage,
         "Zokor": stagingFuncs.addZokorStage,

diff --git a/src/scripts/modules/stages/environments/livingGarden.ts b/src/scripts/modules/stages/environments/livingGarden.ts
@@ -0,0 +1,20 @@
+import {type User} from '@scripts/types/hg';
+import {type IntakeMessage} from '@scripts/types/mhct';
+import {type IStager} from '../stages.types';
+
+export class LivingGardenStager implements IStager {
+    readonly environment: string = 'Living Garden';
+
+    /**
+     * Read the bucket / vial state to determine the stage for Living & Twisted garden.
+     */
+    addStage(message: IntakeMessage, userPre: User, userPost: User, journal: unknown): void {
+        const quest = userPre.quests.QuestLivingGarden;
+        if (!quest) {
+            throw new Error('QuestLivingGarden is undefined');
+        }
+
+        const container_status = (quest.is_normal) ? quest.minigame.bucket_state : quest.minigame.vials_state;
+        message.stage = (container_status === "dumped") ? "Pouring" : "Not Pouring";
+    }
+}
diff --git a/src/scripts/modules/stages/environments/twistedGarden.ts b/src/scripts/modules/stages/environments/twistedGarden.ts
@@ -0,0 +1,20 @@
+import {type User} from '@scripts/types/hg';
+import {type IntakeMessage} from '@scripts/types/mhct';
+import {type IStager} from '../stages.types';
+
+export class TwistedGardenStager implements IStager {
+    readonly environment: string = 'Twisted Garden';
+
+    /**
+     * Read the bucket / vial state to determine the stage for Living & Twisted garden.
+     */
+    addStage(message: IntakeMessage, userPre: User, userPost: User, journal: unknown): void {
+        const quest = userPre.quests.QuestLivingGarden;
+        if (!quest) {
+            throw new Error('QuestLivingGarden is undefined');
+        }
+
+        const container_status = (quest.is_normal) ? quest.minigame.bucket_state : quest.minigame.vials_state;
+        message.stage = (container_status === "dumped") ? "Pouring" : "Not Pouring";
+    }
+}
diff --git a/src/scripts/modules/stages/index.ts b/src/scripts/modules/stages/index.ts
@@ -8,10 +8,12 @@ import {FungalCavernStager} from './environments/fungalCavern';
 import {HarbourStager} from './environments/harbour';
 import {IceFortressStager} from './environments/iceFortress';
 import {LabyrinthStager} from './environments/labyrinth';
+import {LivingGardenStager} from './environments/livingGarden';
 import {MoussuPicchuStager} from './environments/moussuPicchu';
 import {MousoleumStager} from './environments/mousoleum';
 import {SlushyShorelineStager} from './environments/slushyShoreline';
 import {SuperBrieFactoryStager} from './environments/superBrieFactory';
+import {TwistedGardenStager} from './environments/twistedGarden';
 
 const stageModules: IStager[]  = [
     new BalacksCoveStager(),
@@ -23,10 +25,12 @@ const stageModules: IStager[]  = [
     new HarbourStager(),
     new IceFortressStager(),
     new LabyrinthStager(),
+    new LivingGardenStager(),
     new MoussuPicchuStager(),
     new MousoleumStager(),
     new SlushyShorelineStager(),
     new SuperBrieFactoryStager(),
+    new TwistedGardenStager(),
 ];
 
 export {stageModules};
diff --git a/src/scripts/modules/stages/legacy.js b/src/scripts/modules/stages/legacy.js
@@ -114,19 +114,6 @@ export function addSeasonalGardenStage(message, user, user_post, hunt) {
     }
 }
 
-/**
- * Read the bucket / vial state to determine the stage for Living & Twisted garden.
- * @param {Object <string, any>} message The message to be sent.
- * @param {Object <string, any>} user The user state object, when the hunt was invoked (pre-hunt).
- * @param {Object <string, any>} user_post The user state object, after the hunt.
- * @param {Object <string, any>} hunt The journal entry corresponding to the active hunt.
- */
-export function addGardenStage(message, user, user_post, hunt) {
-    const quest = user.quests.QuestLivingGarden;
-    const container_status = (quest.is_normal) ? quest.minigame.bucket_state : quest.minigame.vials_state;
-    message.stage = (container_status === "dumped") ? "Pouring" : "Not Pouring";
-}
-
 /**
  * Determine if there is a stampede active
  * @param {Object <string, any>} message The message to be sent.

diff --git a/src/scripts/types/hg.ts b/src/scripts/types/hg.ts
@@ -51,7 +51,7 @@ export interface Quests {
     QuestIceberg?: quests.QuestIceberg;
     QuestIceFortress?: quests.QuestIceFortress;
     QuestLabyrinth?: quests.QuestLabyrinth
-    QuestLivingGarden?: unknown
+    QuestLivingGarden?: quests.QuestLivingGarden
     QuestLostCity?: unknown
     QuestMousoleum?: quests.QuestMousoleum
     QuestMoussuPicchu?: quests.QuestMoussuPicchu

diff --git a/src/scripts/types/quests/index.ts b/src/scripts/types/quests/index.ts
@@ -8,6 +8,7 @@ export * from '@scripts/types/quests/harbour';
 export * from '@scripts/types/quests/iceberg';
 export * from '@scripts/types/quests/iceFortress';
 export * from '@scripts/types/quests/labyrinth';
+export * from '@scripts/types/quests/livingGarden';
 export * from '@scripts/types/quests/mousoleum';
 export * from '@scripts/types/quests/moussuPicchu';
 export * from '@scripts/types/quests/superBrieFactory';

diff --git a/src/scripts/types/quests/livingGarden.ts b/src/scripts/types/quests/livingGarden.ts
@@ -0,0 +1,17 @@
+export type QuestLivingGarden = LivingGardenState | TwistedGardenState
+
+interface LivingGardenState {
+    is_normal: true
+    minigame: {
+        bucket_state: PourStatus
+    }
+}
+
+interface TwistedGardenState {
+    is_normal: false
+    minigame: {
+         vials_state: PourStatus
+    }
+}
+
+type PourStatus = 'filling' | 'dumped';
diff --git a/tests/scripts/modules/stages/environments/livingGarden.spec.ts b/tests/scripts/modules/stages/environments/livingGarden.spec.ts
@@ -1,10 +1,18 @@
-import {addGardenStage} from "@scripts/modules/stages/legacy";
+import {LivingGardenStager} from "@scripts/modules/stages/environments/livingGarden";
+import {TwistedGardenStager} from "@scripts/modules/stages/environments/twistedGarden";
 import {User} from "@scripts/types/hg";
 import {IntakeMessage} from "@scripts/types/mhct";
 
 describe('Living/Twisted Garden stages', () => {
     describe('Living', () => {
+        it('should be for the "Living Garden" environment', () => {
+            const stager = new LivingGardenStager();
+            expect(stager.environment).toBe('Living Garden');
+        });
+
         it('should set stage to "Pouring" if bucket dumped', () => {
+            const stager = new LivingGardenStager();
+
             const message = {} as IntakeMessage;
             const preUser = {quests: {QuestLivingGarden: {
                 is_normal: true,
@@ -15,29 +23,49 @@ describe('Living/Twisted Garden stages', () => {
             const postUser = {} as User;
             const journal = {};
 
-            addGardenStage(message, preUser, postUser, journal);
+            stager.addStage(message, preUser, postUser, journal);
 
             expect(message.stage).toBe('Pouring');
         });
 
         it('should set stage to "Not Pouring" if bucket not dumped', () => {
+            const stager = new LivingGardenStager();
+
             const message = {} as IntakeMessage;
             const preUser = {quests: {QuestLivingGarden: {
                 is_normal: true,
                 minigame: {
                     bucket_state: 'filling',
                 },
-            }}} as User;            const postUser = {} as User;
+            }}} as User;
+            const postUser = {} as User;
             const journal = {};
 
-            addGardenStage(message, preUser, postUser, journal);
+            stager.addStage(message, preUser, postUser, journal);
+        });
+
+        it.each([undefined, null])('should throw if quest is %p', (quest) => {
+            const stager = new LivingGardenStager();
+
+            const message = {} as IntakeMessage;
+            const preUser = {quests: {QuestLivingGarden: quest}} as User;
+            const postUser = {} as User;
+            const journal = {};
 
-            expect(message.stage).toBe('Not Pouring');
+            expect(() => stager.addStage(message, preUser, postUser, journal))
+                .toThrow('QuestLivingGarden is undefined');
         });
     });
 
     describe('Twisted', () => {
+        it('should be for the "Twisted Garden" environment', () => {
+            const stager = new TwistedGardenStager();
+            expect(stager.environment).toBe('Twisted Garden');
+        });
+
         it('should set stage to "Pouring" if bucket dumped', () => {
+            const stager = new TwistedGardenStager();
+
             const message = {} as IntakeMessage;
             const preUser = {quests: {QuestLivingGarden: {
                 is_normal: false,
@@ -48,24 +76,37 @@ describe('Living/Twisted Garden stages', () => {
             const postUser = {} as User;
             const journal = {};
 
-            addGardenStage(message, preUser, postUser, journal);
+            stager.addStage(message, preUser, postUser, journal);
 
             expect(message.stage).toBe('Pouring');
         });
 
         it('should set stage to "Not Pouring" if bucket not dumped', () => {
+            const stager = new TwistedGardenStager();
+
             const message = {} as IntakeMessage;
             const preUser = {quests: {QuestLivingGarden: {
                 is_normal: false,
                 minigame: {
                     vials_state: 'filling',
                 },
-            }}} as User;            const postUser = {} as User;
+            }}} as User;
+            const postUser = {} as User;
             const journal = {};
 
-            addGardenStage(message, preUser, postUser, journal);
+            stager.addStage(message, preUser, postUser, journal);
+        });
+
+        it.each([undefined, null])('should throw if quest is %p', (quest) => {
+            const stager = new TwistedGardenStager();
+
+            const message = {} as IntakeMessage;
+            const preUser = {quests: {QuestLivingGarden: quest}} as User;
+            const postUser = {} as User;
+            const journal = {};
 
-            expect(message.stage).toBe('Not Pouring');
+            expect(() => stager.addStage(message, preUser, postUser, journal))
+                .toThrow('QuestLivingGarden is undefined');
         });
     });
 });