Aaron
104 lines
2.9 KiB
TypeScript
104 lines
2.9 KiB
TypeScript
import { CAC, Command } from 'cac';
|
|
import consola, { LogLevels } from 'consola';
|
|
import { filterTruthy, toArray } from '~/core/utils/arrays';
|
|
import { printHeader } from '~/core/utils/log';
|
|
import { formatDuration } from '~/core/utils/time';
|
|
import { ValidationError } from '~/core/utils/validation';
|
|
import { registerWxt } from '~/core/wxt';
|
|
|
|
/**
|
|
* Wrap an action handler to add a timer, error handling, and maybe enable debug mode.
|
|
*/
|
|
export function wrapAction(
|
|
cb: (
|
|
...args: any[]
|
|
) => void | { isOngoing?: boolean } | Promise<void | { isOngoing?: boolean }>,
|
|
options?: {
|
|
disableFinishedLog?: boolean;
|
|
},
|
|
) {
|
|
return async (...args: any[]) => {
|
|
// Enable consola's debug mode globally at the start of all commands when the `--debug` flag is
|
|
// passed
|
|
const isDebug = !!args.find((arg) => arg?.debug);
|
|
if (isDebug) {
|
|
consola.level = LogLevels.debug;
|
|
}
|
|
|
|
const startTime = Date.now();
|
|
try {
|
|
printHeader();
|
|
|
|
const status = await cb(...args);
|
|
|
|
if (!status?.isOngoing && !options?.disableFinishedLog)
|
|
consola.success(
|
|
`Finished in ${formatDuration(Date.now() - startTime)}`,
|
|
);
|
|
} catch (err) {
|
|
consola.fail(
|
|
`Command failed after ${formatDuration(Date.now() - startTime)}`,
|
|
);
|
|
if (err instanceof ValidationError) {
|
|
// Don't log these errors, they've already been logged
|
|
} else {
|
|
consola.error(err);
|
|
}
|
|
process.exit(1);
|
|
}
|
|
};
|
|
}
|
|
|
|
/**
|
|
* Array flags, when not passed, are either `undefined` or `[undefined]`. This function filters out
|
|
* the
|
|
*/
|
|
export function getArrayFromFlags<T>(
|
|
flags: any,
|
|
name: string,
|
|
): T[] | undefined {
|
|
const array = toArray<T | undefined>(flags[name]);
|
|
const result = filterTruthy(array);
|
|
return result.length ? result : undefined;
|
|
}
|
|
|
|
const aliasCommandNames = new Set<string>();
|
|
/**
|
|
* @param base Command to add this one to
|
|
* @param name The command name to add
|
|
* @param alias The CLI tool being aliased
|
|
* @param bin The CLI tool binary name. Usually the same as the alias
|
|
* @param docsUrl URL to the docs for the aliased CLI tool
|
|
*/
|
|
export function createAliasedCommand(
|
|
base: CAC,
|
|
name: string,
|
|
alias: string,
|
|
bin: string,
|
|
docsUrl: string,
|
|
) {
|
|
const aliasedCommand = base
|
|
.command(name, `Alias for ${alias} (${docsUrl})`)
|
|
.allowUnknownOptions()
|
|
.action(async () => {
|
|
try {
|
|
await registerWxt('build');
|
|
|
|
const args = process.argv.slice(
|
|
process.argv.indexOf(aliasedCommand.name) + 1,
|
|
);
|
|
const { execa } = await import('execa');
|
|
await execa(bin, args, {
|
|
stdio: 'inherit',
|
|
});
|
|
} catch {
|
|
// Let the other aliased CLI log errors, just exit
|
|
process.exit(1);
|
|
}
|
|
});
|
|
aliasCommandNames.add(aliasedCommand.name);
|
|
}
|
|
export function isAliasedCommand(command: Command | undefined): boolean {
|
|
return !!command && aliasCommandNames.has(command.name);
|
|
}
|