import yargs from "yargs";
import { hideBin } from "yargs/helpers";
import type { CliAction } from "../../core/core.models.js";
import type { CliArgumentsSetter, Command, CommandOption } from "../cli.models.js";

export function argumentsSetter(): CliArgumentsSetter<CliAction> {
	const argv = yargs(hideBin(process.argv));

	return {
		withCommand(command: Command<CliAction>): CliArgumentsSetter<CliAction> {
			argv.command(command.name, command.description, (yargs) => {
				for (const example of command.examples) {
					yargs.example(command.name, example);
				}

				command.options.forEach((option) => {
					yargs.positional(option.name, {
						alias: option.alias,
						describe: option.description,
						type: option.type,
						demandOption: option.isRequired,
					});
				});
			});

			return this;
		},
		withOption(option: CommandOption): CliArgumentsSetter<CliAction> {
			argv.option(option.name, {
				alias: option.alias,
				description: option.description,
				type: option.type,
				demandOption: option.isRequired,
			});

			return this;
		},
		registerCommands(): void {
			argv.parseSync();
		},
	};
}