import type { ModalSubmitInteraction } from "discord.js";
import type { ModalDefinition } from "./ModalDefinition";

export interface RegisteredModal<Args extends string[]> {
	definition: ModalDefinition<any, Args>;
	run(interaction: ModalSubmitInteraction, args: Args): unknown | Promise<unknown>;
}

class ModalRegistry {
	private modals = new Map<string, RegisteredModal<any>>();

	public register<Args extends string[]>(
		definition: ModalDefinition<any, Args>,
		run: (interaction: ModalSubmitInteraction, args: Args) => unknown | Promise<unknown>,
	) {
		this.modals.set(definition.id, { definition, run });
	}

	public get(id: string) {
		return this.modals.get(id);
	}
}

export const modalRegistry = new ModalRegistry();