feat: support before/after hooks

src/_utils.ts

@@ -1,5 +1,13 @@
 import type { Resolvable } from "./types";
 
+export  function runIfFunction<T extends (...args: any[]) => Promise<void> | void>(val: T | undefined) {
+  return async function (...args: Parameters<T>) {
+    if (typeof val === 'function') {
+      await val(...args)
+    }
+  }
+}
+
 export function toArray(val: any) {
   if (Array.isArray(val)) {
     return val;

src/command.ts

@@ -1,5 +1,5 @@
 import type { CommandContext, CommandDef, ArgsDef } from "./types";
-import { CLIError, resolveValue } from "./_utils";
+import { CLIError, resolveValue, runIfFunction } from "./_utils";
 import { parseArgs } from "./args";
 
 export function defineCommand<T extends ArgsDef = ArgsDef>(
@@ -29,9 +29,7 @@ export async function runCommand<T extends ArgsDef = ArgsDef>(
   };
 
   // Setup hook
-  if (typeof cmd.setup === "function") {
-    await cmd.setup(context);
-  }
+  await runIfFunction(cmd.setup)(context);
 
   // Handle sub command
   try {
@@ -60,13 +58,11 @@ export async function runCommand<T extends ArgsDef = ArgsDef>(
     }
 
     // Handle main command
-    if (typeof cmd.run === "function") {
-      await cmd.run(context);
-    }
+    await runIfFunction(cmd.before)(context);
+    await runIfFunction(cmd.run)(context);
   } finally {
-    if (typeof cmd.cleanup === "function") {
-      await cmd.cleanup(context);
-    }
+    await runIfFunction(cmd.after)(context);
+    await runIfFunction(cmd.cleanup)(context);
   }
 }
 

src/types.ts

@@ -54,6 +54,8 @@ export type CommandDef<T extends ArgsDef = ArgsDef> = {
   meta?: Resolvable<CommandMeta>;
   args?: Resolvable<T>;
   subCommands?: Resolvable<SubCommandsDef>;
+  before?: (context: CommandContext<T>) => any | Promise<any>;
+  after?: (context: CommandContext<T>) => any | Promise<any>;
   setup?: (context: CommandContext<T>) => any | Promise<any>;
   cleanup?: (context: CommandContext<T>) => any | Promise<any>;
   run?: (context: CommandContext<T>) => any | Promise<any>;