/// <reference types="jest" />

type AsyncFunction = (...args: any[]) => Promise<unknown>;

type PromiseValue<PromiseType, Otherwise = PromiseType> = PromiseType extends Promise<infer Value>
	? { 0: PromiseValue<Value>; 1: Value }[PromiseType extends Promise<unknown> ? 0 : 1]
	: Otherwise;

type AsyncReturnType<Target extends AsyncFunction> = PromiseValue<ReturnType<Target>>;

declare module '@async-fn/jest' {
  type MockedAsyncFn<T extends (...args: any[]) => any> =
    & Omit<
      jest.MockInstance<ReturnType<T>, Parameters<T>>,
      | 'mockImplementation'
      | 'mockImplementationOnce'
      | 'mockReturnValue'
      | 'mockReturnValueOnce'
    >
    & T & {
      resolve: (resolvedValue: AsyncReturnType<T>) => Promise<void>,
      reject: (rejectValue: any) => Promise<void>
    };
  export type { MockedAsyncFn };

  export default function asyncFn<T, Y extends any[]>(implementation?: (...args: Y) => T): jest.Mock<T, Y>;
  export default function asyncFn(): jest.Mock;
}