change type definition to allow passing synchronous functions

johnsonjo4531 · Oct 8, 2019 · f2841c2 · f2841c2
1 parent 0abc04d
commit f2841c2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/index.d.ts b/index.d.ts
@@ -1,3 +1,5 @@
-type AsyncFunction<T> = (...args: any[]) => Promise<T>;
+type AsyncFunction<S extends any[], T> = (...args: S) => Promise<T>;
 
-export default function greenlet<T extends AsyncFunction<U>, U>(fn: T): T;
+type MaybeAsyncFunction<S extends any[], T> = (...args: S) => (T | Promise<T>);
+
+export default function greenlet<S extends any[], T>(fn: MaybeAsyncFunction<S, T>): AsyncFunction<S, T>;