bitburner-src/utils/YesNoBox.ts

/**
 * Generic Yes-No Pop-up box
 * Used to create pop-up boxes that require a yes/no response from player
 *
 * There are two types of pop ups:
 *  1. Just a Yes/No response from player
 *  2. Popup also includes a text input field in addition to the Yes/No response
 */
import { clearEventListeners } from "./uiHelpers/clearEventListeners";
import { KEY } from "./helpers/keyCodes";
import * as ReactDOM from "react-dom";

export let yesNoBoxOpen = false;

const yesNoBoxContainer: HTMLElement | null = document.getElementById("yes-no-box-container");
const yesNoBoxTextElement: HTMLElement | null = document.getElementById("yes-no-box-text");

function yesNoBoxHotkeyHandler(e: KeyboardEvent): void {
    if (e.keyCode === KEY.ESC) {
        yesNoBoxClose();
    } else if (e.keyCode === KEY.ENTER) {
        const yesBtn: HTMLElement | null = document.getElementById("yes-no-box-yes");
        if (yesBtn) {
            yesBtn.click();
        } else {
            console.error(`Could not find YesNoBox Yes button DOM element`)
        }
    }
}

export function yesNoBoxClose(): boolean {
    if (yesNoBoxContainer) {
        yesNoBoxContainer.style.display = "none";
    } else {
        console.error("Container not found for YesNoBox");
    }
    yesNoBoxOpen = false;

    // Remove hotkey handler
    document.removeEventListener("keydown", yesNoBoxHotkeyHandler);

    return false; //So that 'return yesNoBoxClose()' is return false in event listeners
}

export function yesNoBoxGetYesButton(): HTMLElement | null {
    return clearEventListeners("yes-no-box-yes");
}

export function yesNoBoxGetNoButton(): HTMLElement | null {
    return clearEventListeners("yes-no-box-no");
}

export function yesNoBoxCreate(txt: string | JSX.Element): boolean {
    if (yesNoBoxOpen) { return false; }   //Already open
    yesNoBoxOpen = true;

    if (yesNoBoxTextElement) {
        ReactDOM.unmountComponentAtNode(yesNoBoxTextElement);
        yesNoBoxTextElement.innerHTML = '';
        if(typeof txt === 'string') {
            yesNoBoxTextElement.innerHTML = txt as string;
        } else {
            ReactDOM.render(txt, yesNoBoxTextElement);
        }
    } else {
        console.error(`Text element not found for YesNoBox`);
    }

    if (yesNoBoxContainer) {
        yesNoBoxContainer.style.display = "flex";
    } else {
        console.error("Container not found for YesNoBox");
    }

    // Add event listener for Esc and Enter hotkeys
    document.addEventListener("keydown", yesNoBoxHotkeyHandler);

    return true;
}

/**
 * Yes-No pop up box with text input field
 */
const yesNoTextInputBoxContainer: HTMLElement | null = document.getElementById("yes-no-text-input-box-container");
const yesNoTextInputBoxInput: HTMLInputElement | null = document.getElementById("yes-no-text-input-box-input") as HTMLInputElement;
const yesNoTextInputBoxTextElement: HTMLElement | null = document.getElementById("yes-no-text-input-box-text");

export function yesNoTxtInpBoxHotkeyHandler(e: KeyboardEvent): void {
    if (e.keyCode === KEY.ESC) {
        yesNoTxtInpBoxClose();
    } else if (e.keyCode === KEY.ENTER) {
        const yesBtn: HTMLElement | null = document.getElementById("yes-no-text-input-box-yes");
        if (yesBtn) {
            yesBtn.click();
        } else {
            console.error(`Could not find YesNoTxtInputBox Yes button DOM element`)
        }
    }
}

export function yesNoTxtInpBoxClose(): boolean {
    if (yesNoTextInputBoxContainer != null) {
        yesNoTextInputBoxContainer.style.display = "none";
    } else {
        console.error("Container not found for YesNoTextInputBox");
        return false;
    }
    if(!yesNoTextInputBoxInput) throw new Error("yesNoTextInputBoxInput was not set");
    yesNoBoxOpen = false;
    yesNoTextInputBoxInput.value = "";

    // Remove hotkey handler
    document.removeEventListener("keydown", yesNoTxtInpBoxHotkeyHandler);

    return false;
}

export function yesNoTxtInpBoxGetYesButton(): HTMLElement {
    const elem = clearEventListeners("yes-no-text-input-box-yes");
    if(elem === null) throw new Error("Could not find element with id: 'yes-no-text-input-box-yes'");
    return elem;
}

export function yesNoTxtInpBoxGetNoButton(): HTMLElement {
    const elem = clearEventListeners("yes-no-text-input-box-no");
    if(elem === null) throw new Error("Could not find element with id: 'yes-no-text-input-box-no'");
    return elem;
}

export function yesNoTxtInpBoxGetInput(): string {
    if (!yesNoTextInputBoxInput) {
        console.error("Could not find YesNoTextInputBox input element");
        return "";
    }
    let val: string = yesNoTextInputBoxInput.value;
    val = val.replace(/\s+/g, '');
    return val;
}

export function yesNoTxtInpBoxCreate(txt: string | JSX.Element): void {
    yesNoBoxOpen = true;


    if (yesNoTextInputBoxTextElement) {
        ReactDOM.unmountComponentAtNode(yesNoTextInputBoxTextElement);
        yesNoTextInputBoxTextElement.innerHTML = '';
        if(typeof txt === 'string') {
            yesNoTextInputBoxTextElement.innerHTML = txt;
        } else {
            ReactDOM.render(txt, yesNoTextInputBoxTextElement);
        }
    }

    if (yesNoTextInputBoxContainer) {
        yesNoTextInputBoxContainer.style.display = "flex";
    } else {
        console.error("Container not found for YesNoTextInputBox");
    }

    // Add event listener for Esc and Enter hotkeys
    document.addEventListener("keydown", yesNoTxtInpBoxHotkeyHandler);

    if(!yesNoTextInputBoxInput) throw new Error("yesNoTextInputBoxInput was not set");
    yesNoTextInputBoxInput.focus();
}