feat: preserves overloads for non-async functions

src/common/als.ts

@@ -1,14 +1,13 @@
+import type { Any } from "src/common/types";
 import { AsyncLocalStorage } from "node:async_hooks";
 
 export function als<T, IArgs extends any[]>(
   storage: AsyncLocalStorage<T>,
   init: (...args: IArgs) => T
-) {
-  return function <FArgs extends IArgs, FReturn>(
-    fn: (...args: FArgs) => FReturn
-  ) {
+): <Fn extends Any.Function>(fn: Fn) => Fn {
+  return function (fn) {
     function newFn(...args: Parameters<typeof fn>) {
-      // @ts-ignore TS2683
+      // @ts-expect-error TS2683
       const result = storage.run(init(...args), fn.bind(this), ...args);
       return result;
     }
@@ -20,6 +19,6 @@ export function als<T, IArgs extends any[]>(
       }
     );
 
-    return newFn;
+    return newFn as never;
   };
 }

src/common/memoize.ts

@@ -1,13 +1,15 @@
-export function memoize<H extends (...args: any[]) => any>(
+import { Any } from "src/common/types";
+
+export function memoize<H extends Any.Function>(
   hash: H,
   cache = Object.create(null)
-) {
-  return <FArgs extends any[], FReturn>(fn: (...args: FArgs) => FReturn) => {
+): <Fn extends Any.Function>(fn: Fn) => Fn {
+  return (fn) => {
     function newFn(...args: Parameters<typeof fn>) {
-      // @ts-ignore TS2683
+      // @ts-expect-error TS2683
       const key = hash.apply(this, args);
       if (!(key in cache)) {
-        // @ts-ignore TS2683
+        // @ts-expect-error TS2683
         cache[key] = fn.apply(this, args);
       }
       return cache[key] as ReturnType<typeof fn>;
@@ -20,6 +22,6 @@ export function memoize<H extends (...args: any[]) => any>(
       }
     );
 
-    return newFn;
+    return newFn as never;
   };
 }

src/common/trace.ts

@@ -1,10 +1,10 @@
-export const trace = <FArgs extends any[], FReturn>(
-  fn: (...args: FArgs) => FReturn
-) => {
-  function newFn(...args: Parameters<typeof fn>) {
+import type { Any } from "src/common/types";
+
+export const trace = <Fn extends Any.Function>(fn: Fn): Fn => {
+  function newFn(...args: Parameters<Fn>) {
     const tag = `${fn.name}(${args}) | duration`;
     console.time(tag);
-    // @ts-ignore TS2683
+    // @ts-expect-error TS2683
     const result = fn.apply(this, args);
     console.timeEnd(tag);
     return result;
@@ -17,5 +17,5 @@ export const trace = <FArgs extends any[], FReturn>(
     }
   );
 
-  return newFn;
+  return newFn as never;
 };

src/common/types.ts

@@ -0,0 +1,8 @@
+export type {
+  Any,
+}
+
+declare namespace Any {
+  type Function<I extends Array<unknown> = Array<any>, O = any> = { (...args: I): O }
+  type Array<T = unknown> = ReadonlyArray<T>
+}

src/wrapper.ts

@@ -45,17 +45,17 @@ type Spreadable<T> = T extends any[] ? T : never;
 
 type Replace<Source, A, B, Sentinel = never> = Source extends Sentinel
   ? Omit<Source, keyof Sentinel> extends infer T
-    ? Replace<LeftIfNotEqual<T, {}>, A, B, Sentinel> extends infer U
-      ? U | B
-      : never // Dummy ternary just so we can get an alias T
-    : never // Dummy ternary just so we can get an alias U
+  ? Replace<LeftIfNotEqual<T, {}>, A, B, Sentinel> extends infer U
+  ? U | B
+  : never // Dummy ternary just so we can get an alias T
+  : never // Dummy ternary just so we can get an alias U
   : Source extends Promise<infer T>
   ? Promise<Replace<T, A, B, Sentinel>>
   : Source extends [infer AItem, ...infer ARest]
   ? [
-      Replace<AItem, A, B, Sentinel>,
-      ...Spreadable<Replace<ARest, A, B, Sentinel>>
-    ]
+    Replace<AItem, A, B, Sentinel>,
+    ...Spreadable<Replace<ARest, A, B, Sentinel>>
+  ]
   : Source extends Record<string | number | symbol, any>
   ? { [K in keyof Source]: Replace<Source[K], A, B, Sentinel> }
   : Source extends A

test/trace.test.ts

@@ -1,5 +1,10 @@
 import test from "ava";
 import { trace } from "../src";
+import type { Any } from "src/common/types";
+
+
+declare function myFn<Args extends Any.Array>(arg_0: Args): string | number
+declare function myFn<T, U>(arg_0: T, arg_1: U): T | U
 
 test("maintains reference to this", async (t) => {
   const data = {