import { PrismaClient } from '@prisma/client';

import { findManyAndCountExtension } from './extensions/findManyAndCount.extension';

function extendClient(base: PrismaClient) {
	// Add as many as you'd like - no ugly types required!
	return base.$extends(findManyAndCountExtension);
}

class UntypedExtendedClient extends PrismaClient {
	constructor(options?: ConstructorParameters<typeof PrismaClient>[0]) {
		super(options);

		return extendClient(this) as this;
	}
}

const ExtendedPrismaClient = UntypedExtendedClient as unknown as new (options?: ConstructorParameters<typeof PrismaClient>[0]) => ReturnType<typeof extendClient>;

export { ExtendedPrismaClient };