seyfert/packages/helpers/src/builders/components/MessageSelectMenuBuilder.ts

import type { DiscordSelectOption, DiscordSelectMenuComponent, } from '@biscuitland/api-types';
import type { ComponentEmoji } from '@biscuitland/core';
import { MessageComponentTypes } from '@biscuitland/api-types';

export class SelectMenuOptionBuilder {
	constructor() {
		this.#data = {} as DiscordSelectOption;
	}

	#data: DiscordSelectOption;

	setLabel(label: string): SelectMenuOptionBuilder {
		this.#data.label = label;
		return this;
	}

	setValue(value: string): SelectMenuOptionBuilder {
		this.#data.value = value;
		return this;
	}

	setDescription(description: string): SelectMenuOptionBuilder {
		this.#data.description = description;
		return this;
	}

	setDefault(Default = true): SelectMenuOptionBuilder {
		this.#data.default = Default;
		return this;
	}

	setEmoji(emoji: ComponentEmoji): SelectMenuOptionBuilder {
		this.#data.emoji = emoji;
		return this;
	}

	toJSON(): DiscordSelectOption {
		return { ...this.#data };
	}
}

export class SelectMenuBuilder {
	constructor() {
		this.#data = {} as DiscordSelectMenuComponent;
		this.type = MessageComponentTypes.SelectMenu;
		this.options = [];
	}

	#data: DiscordSelectMenuComponent;
	type: MessageComponentTypes.SelectMenu |
		MessageComponentTypes.RoleSelect |
		MessageComponentTypes.UserSelect |
		MessageComponentTypes.MentionableSelect |
		MessageComponentTypes.ChannelSelect;

	options: SelectMenuOptionBuilder[];

	setType(type: this['type']) {
		this.type = type;
		return this;
	}

	setPlaceholder(placeholder: string): this {
		this.#data.placeholder = placeholder;
		return this;
	}

	setValues(max?: number, min?: number): this {
		this.#data.max_values = max;
		this.#data.min_values = min;
		return this;
	}

	setDisabled(disabled = true): this {
		this.#data.disabled = disabled;
		return this;
	}

	setCustomId(id: string): this {
		this.#data.custom_id = id;
		return this;
	}

	setOptions(...options: SelectMenuOptionBuilder[]): this {
		this.options.splice(
			0,
			this.options.length,
			...options,
		);
		return this;
	}

	addOptions(...options: SelectMenuOptionBuilder[]): this {
		this.options.push(
			...options,
		);
		return this;
	}

	toJSON(): DiscordSelectMenuComponent {
		return { ...this.#data, type: this.type, options: this.options.map(option => option.toJSON()) };
	}
}