Function wrapCommandAction

packages/cli-v3/src/cli/common.ts:50–98

(
  name: string,
  schema: T,
  options: unknown,
  action: (opts: z.output<T>) => Promise<TResult>
)

export const tracer = getTracer();

export async function wrapCommandAction<T extends z.AnyZodObject, TResult>(
  name: string,
  schema: T,
  options: unknown,
  action: (opts: z.output<T>) => Promise<TResult>
): Promise<TResult> {
  return await tracer.startActiveSpan(name, async (span) => {
    try {
      const parsedOptions = schema.safeParse(options);

      if (!parsedOptions.success) {
        throw new Error(fromZodError(parsedOptions.error).toString());
      }

      span.setAttributes({
        ...flattenAttributes(parsedOptions.data, "cli.options"),
      });

      logger.loggerLevel = parsedOptions.data.logLevel;

      logger.debug(`Running "${name}" with the following options`, {
        options: options,
        spanContext: span?.spanContext(),
      });

      const result = await action(parsedOptions.data);

      span.end();

      return result;
    } catch (e) {
      if (e instanceof SkipLoggingError) {
        recordSpanException(span, e);
      } else if (e instanceof OutroCommandError) {
        outro("Operation cancelled");
      } else if (e instanceof SkipCommandError) {
        // do nothing
      } else {
        recordSpanException(span, e);

        logger.log(`${chalkError("X Error:")} ${e instanceof Error ? e.message : String(e)}`);
      }

      span.end();

      throw e;
    }
  });
}