import {
	type AutocompleteInteraction,
	type User,
	type Attachment,
	type Role,
	type GuildBasedChannel,
	type ApplicationCommandOptionChoiceData,
} from "discord.js";

export const OPTION_TYPES = {
	subcommand: 1,
	subcommandGroup: 2,
	string: 3,
	integer: 4,
	boolean: 5,
	user: 6,
	channel: 7,
	role: 8,
	mentionable: 9,
	number: 10,
	attachment: 11,
} as const;
type OptionType = keyof typeof OPTION_TYPES;

export interface Option<T, Req extends boolean> {
	type: OptionType;
	description?: string;
	required: Req;
	autocomplete?: (
		interaction: AutocompleteInteraction,
		value: T,
	) => Promise<ApplicationCommandOptionChoiceData[]> | ApplicationCommandOptionChoiceData[];
	__type?: T;
}

export interface CommandOption<T, Req extends boolean> extends Option<T, Req> {
	description: string;
}

export type InferOptions<O> = {
	[K in keyof O]: O[K] extends Option<infer Type, infer Required>
		? Required extends true
			? Type
			: Type | undefined
		: never;
};

export type AutocompleteFn<T> = (
	interaction: AutocompleteInteraction,
	value: T,
) => Promise<ApplicationCommandOptionChoiceData[]> | ApplicationCommandOptionChoiceData[];

const createOption = <T>(type: OptionType) => {
	return ((
		descriptionOrConfig?: string | { required?: boolean; autocomplete?: AutocompleteFn<T> },
		config?: { required?: boolean; autocomplete?: AutocompleteFn<T> },
	): Option<T, any> => {
		const hasDescription = typeof descriptionOrConfig === "string";
		const description = hasDescription ? descriptionOrConfig : undefined;
		const actualConfig = hasDescription ? config : descriptionOrConfig;

		return {
			type,
			description,
			required: actualConfig?.required ?? false,
			autocomplete: actualConfig?.autocomplete,
		};
	}) as {
		// w/ description for commands
		<Req extends boolean>(
			description: string,
			config: { required: Req; autocomplete?: AutocompleteFn<T> },
		): CommandOption<T, Req>;
		(
			description: string,
			config?: { required?: false; autocomplete?: AutocompleteFn<T> },
		): CommandOption<T, false>;

		// w/out description for components
		<Req extends boolean>(config: {
			required: Req;
			autocomplete?: AutocompleteFn<T>;
		}): Option<T, Req>;
		(config?: { required?: false; autocomplete?: AutocompleteFn<T> }): Option<T, false>;
	};
};

export const option = {
	string: createOption<string>("string"),
	integer: createOption<number>("integer"),
	number: createOption<number>("number"),
	boolean: createOption<boolean>("boolean"),
	user: createOption<User>("user"),
	channel: createOption<GuildBasedChannel>("channel"),
	role: createOption<Role>("role"),
	mentionable: createOption<User | Role | any>("mentionable"),
	attachment: createOption<Attachment>("attachment"),
};