import {
	type AutocompleteInteraction,
	type User,
	type Attachment,
	type Role,
	type GuildBasedChannel,
	type ApplicationCommandOptionChoiceData,
	type LocalizationMap,
} 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 OptionChoice<T> {
	name: string;
	nameLocalizations?: LocalizationMap;
	value: T;
}

export interface Option<T, Req extends boolean> {
	type: OptionType;
	description?: string;
	required: Req;
	autocomplete?: (
		interaction: AutocompleteInteraction,
		value: T,
	) => Promise<ApplicationCommandOptionChoiceData[]> | ApplicationCommandOptionChoiceData[];
	choices?: OptionChoice<T>[];
	nameLocalizations?: LocalizationMap;
	descriptionLocalizations?: LocalizationMap;
	__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[];

export interface OptionBuilder<T> {
	(
		description: string,
		config: {
			required: true;
			autocomplete?: AutocompleteFn<T>;
			choices?: OptionChoice<T>[];
			nameLocalizations?: LocalizationMap;
			descriptionLocalizations?: LocalizationMap;
		},
	): CommandOption<T, true>;
	(
		description: string,
		config?: {
			required?: false;
			autocomplete?: AutocompleteFn<T>;
			choices?: OptionChoice<T>[];
			nameLocalizations?: LocalizationMap;
			descriptionLocalizations?: LocalizationMap;
		},
	): CommandOption<T, false>;
	(config: {
		required: true;
		autocomplete?: AutocompleteFn<T>;
		choices?: OptionChoice<T>[];
		nameLocalizations?: LocalizationMap;
		descriptionLocalizations?: LocalizationMap;
	}): Option<T, true>;
	(config?: {
		required?: false;
		autocomplete?: AutocompleteFn<T>;
		choices?: OptionChoice<T>[];
		nameLocalizations?: LocalizationMap;
		descriptionLocalizations?: LocalizationMap;
	}): Option<T, false>;

	required(
		description: string,
		config?: {
			autocomplete?: AutocompleteFn<T>;
			choices?: OptionChoice<T>[];
			nameLocalizations?: LocalizationMap;
			descriptionLocalizations?: LocalizationMap;
		},
	): CommandOption<T, true>;
	optional(
		description: string,
		config?: {
			autocomplete?: AutocompleteFn<T>;
			choices?: OptionChoice<T>[];
			nameLocalizations?: LocalizationMap;
			descriptionLocalizations?: LocalizationMap;
		},
	): CommandOption<T, false>;
}

const createOption = <T>(type: OptionType): OptionBuilder<T> => {
	const fn = (
		descriptionOrConfig?:
			| string
			| {
					required?: boolean;
					autocomplete?: AutocompleteFn<T>;
					choices?: OptionChoice<T>[];
					nameLocalizations?: LocalizationMap;
					descriptionLocalizations?: LocalizationMap;
			  },
		config?: {
			required?: boolean;
			autocomplete?: AutocompleteFn<T>;
			choices?: OptionChoice<T>[];
			nameLocalizations?: LocalizationMap;
			descriptionLocalizations?: LocalizationMap;
		},
	): 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,
			choices: actualConfig?.choices,
			nameLocalizations: actualConfig?.nameLocalizations,
			descriptionLocalizations: actualConfig?.descriptionLocalizations,
		};
	};

	fn.required = (
		description: string,
		config?: {
			autocomplete?: AutocompleteFn<T>;
			choices?: OptionChoice<T>[];
			nameLocalizations?: LocalizationMap;
			descriptionLocalizations?: LocalizationMap;
		},
	) => {
		return fn(description, {
			required: true,
			autocomplete: config?.autocomplete,
			choices: config?.choices,
			nameLocalizations: config?.nameLocalizations,
			descriptionLocalizations: config?.descriptionLocalizations,
		});
	};

	fn.optional = (
		description: string,
		config?: {
			autocomplete?: AutocompleteFn<T>;
			choices?: OptionChoice<T>[];
			nameLocalizations?: LocalizationMap;
			descriptionLocalizations?: LocalizationMap;
		},
	) => {
		return fn(description, {
			required: false,
			autocomplete: config?.autocomplete,
			choices: config?.choices,
			nameLocalizations: config?.nameLocalizations,
			descriptionLocalizations: config?.descriptionLocalizations,
		});
	};

	return fn as OptionBuilder<T>;
};

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>("mentionable"),
	attachment: createOption<Attachment>("attachment"),
};