✨ feat(core): let inspectors able to return a callback function

Author: so1ve

packages/core/src/types/index.ts

@@ -56,7 +56,7 @@ export type InspectorContext<C extends Commands = Commands> = HandlerContext<C>
   flags: FallbackType<TypeFlag<NonNullable<C[keyof C]["flags"]>>["flags"], Dict<any>>
 };
 export type Inspector<C extends Commands = Commands> = InspectorFn<C> | InspectorObject<C>;
-export type InspectorFn<C extends Commands = Commands> = (ctx: InspectorContext<C>, next: () => void) => void;
+export type InspectorFn<C extends Commands = Commands> = (ctx: InspectorContext<C>, next: () => void) => (() => void) | void;
 export interface InspectorObject<C extends Commands = Commands> {
   enforce?: "pre" | "post"
   fn: InspectorFn<C>

packages/core/src/utils.ts

@@ -126,11 +126,19 @@ export function compose(inspectors: Inspector[]) {
     ...inspectorMap.normal,
     ...inspectorMap.post,
   ];
+
   return (getCtx: () => InspectorContext) => {
-    return dispatch(0);
+    const callbacks: (() => void)[] = [];
+    dispatch(0);
+    for (let i = callbacks.length - 1; i >= 0; i--) {
+      callbacks[i]();
+    }
     function dispatch(i: number): void {
       const inspector = mergedInspectorFns[i];
-      return inspector(getCtx(), dispatch.bind(null, i + 1));
+      const cb = inspector(getCtx(), dispatch.bind(null, i + 1));
+      if (cb) {
+        callbacks.push(cb);
+      }
     }
   };
 }