From b23664cb9e221ac7ed41b74914aff5ebd128ecde Mon Sep 17 00:00:00 2001
From: Felix Staud <felix.staud@headwire.com>
Date: Fri, 12 Jul 2024 14:40:47 -0700
Subject: [PATCH] extend MysteryEncounterOptionBuilder

---
 .../mystery-encounter-option.ts               | 42 ++++++++++++++++++-
 1 file changed, 41 insertions(+), 1 deletion(-)

diff --git a/src/data/mystery-encounters/mystery-encounter-option.ts b/src/data/mystery-encounters/mystery-encounter-option.ts
index 7bbde3f3830..43ccd258ce3 100644
--- a/src/data/mystery-encounters/mystery-encounter-option.ts
+++ b/src/data/mystery-encounters/mystery-encounter-option.ts
@@ -1,4 +1,4 @@
-import { OptionTextDisplay } from "#app/data/mystery-encounters/mystery-encounter-dialogue";
+import { OptionTextDisplay, TextDisplay } from "#app/data/mystery-encounters/mystery-encounter-dialogue";
 import { Moves } from "#app/enums/moves.js";
 import { PlayerPokemon } from "#app/field/pokemon";
 import BattleScene from "../../battle-scene";
@@ -216,8 +216,48 @@ export class MysteryEncounterOptionBuilder implements Partial<MysteryEncounterOp
     return Object.assign(this, { secondaryPokemonRequirements: this.secondaryPokemonRequirements });
   }
 
+  /**
+   * Se the full dialogue object to the option. Will override anything already set
+   *
+   * @param dialogue see {@linkcode OptionTextDisplay}
+   * @returns
+   */
   withDialogue(dialogue: OptionTextDisplay) {
     this.dialogue = dialogue;
     return this;
   }
+
+  /**
+   * Set the button label and disabled label of the option
+   *
+   * @param label the button label
+   * @returns
+   */
+  withButtonLabel(label: string, disabledLabel?: string) {
+    this.dialogue.buttonLabel = label;
+    this.dialogue.disabledButtonLabel = disabledLabel ?? label;
+    return this;
+  }
+  /**
+   * Set the button tooltip and disabled tooltip of the option
+   *
+   * @param tooltip the button tooltip
+   * @returns
+   */
+  withButtonTooltip(tooltip: string, disabledTooltip?: string) {
+    this.dialogue.buttonTooltip = tooltip;
+    this.dialogue.disabledButtonTooltip = disabledTooltip;
+    return this;
+  }
+
+  /**
+   * Set the dialogue to display when the player selects the option
+   *
+   * @param dialogue the text to display on selecting the option
+   * @returns
+   */
+  withSelectedDialogue(dialogue: TextDisplay[]) {
+    this.dialogue.selected = dialogue;
+    return this;
+  }
 }