index.d.ts

/* eslint-disable @typescript-eslint/ban-types */

type Last<T extends readonly unknown[]> = T extends [...any, infer L]
	? L
	: never;
type DropLast<T extends readonly unknown[]> = T extends [...(infer U), any]
	? U
	: [];

type StringEndsWith<S, X extends string> = S extends `${infer _}${X}` ? true : false;

interface Options<Includes extends readonly unknown[], Excludes extends readonly unknown[], MultiArgs extends boolean = false, ErrorFirst extends boolean = true, ExcludeMain extends boolean = false> {
	multiArgs?: MultiArgs;
	include?: Includes;
	exclude?: Excludes;
	errorFirst?: ErrorFirst;
	promiseModule?: PromiseConstructor;
	excludeMain?: ExcludeMain;
}

interface InternalOptions<Includes extends readonly unknown[], Excludes extends readonly unknown[], MultiArgs extends boolean = false, ErrorFirst extends boolean = true> {
	multiArgs: MultiArgs;
	include: Includes;
	exclude: Excludes;
	errorFirst: ErrorFirst;
}

type Promisify<Args extends readonly unknown[], GenericOptions extends InternalOptions<readonly unknown[], readonly unknown[], boolean, boolean>> = (
	...args: DropLast<Args>
) =>
Last<Args> extends (...args: any) => any
// For single-argument functions when errorFirst: true we just return Promise<unknown> as it will always reject.
	? Parameters<Last<Args>> extends [infer SingleCallbackArg] ? GenericOptions extends {errorFirst: true} ? Promise<unknown> : Promise<SingleCallbackArg>
		: Promise<
		GenericOptions extends {multiArgs: false}
			? Last<Parameters<Last<Args>>>
			: Parameters<Last<Args>>
		>
	// Functions without a callback will return a promise that never settles. We model this as Promise<unknown>
	: Promise<unknown>;

type PromisifyModule<
	Module extends Record<string, any>,
	MultiArgs extends boolean,
	ErrorFirst extends boolean,
	Includes extends ReadonlyArray<keyof Module>,
	Excludes extends ReadonlyArray<keyof Module>,
> = {
	[K in keyof Module]: Module[K] extends (...args: infer Args) => any
		? K extends Includes[number]
			? Promisify<Args, InternalOptions<Includes, Excludes, MultiArgs>>
			: K extends Excludes[number]
				? Module[K]
				: StringEndsWith<K, 'Sync' | 'Stream'> extends true
					? Module[K]
					: Promisify<Args, InternalOptions<Includes, Excludes, MultiArgs, ErrorFirst>>
		: Module[K];
};

declare function pify<
	FirstArg,
	Args extends readonly unknown[],
	MultiArgs extends boolean = false,
	ErrorFirst extends boolean = true,
>(
	input: (arg: FirstArg, ...args: Args) => any,
	options?: Options<[], [], MultiArgs, ErrorFirst>
): Promisify<[FirstArg, ...Args], InternalOptions<[], [], MultiArgs, ErrorFirst>>;
declare function pify<
	Module extends Record<string, any>,
	Includes extends ReadonlyArray<keyof Module> = [],
	Excludes extends ReadonlyArray<keyof Module> = [],
	MultiArgs extends boolean = false,
	ErrorFirst extends boolean = true,
>(
	// eslint-disable-next-line unicorn/prefer-module
	module: Module,
	options?: Options<Includes, Excludes, MultiArgs, ErrorFirst, true>
): PromisifyModule<Module, MultiArgs, ErrorFirst, Includes, Excludes>;

export = pify;