import type {
	DefinitionType,
	EndpointDefinitions,
	MutationDefinition,
	QueryDefinition,
	InfiniteQueryDefinition,
} from '@reduxjs/toolkit/query';
import type { UseInfiniteQuery, UseLazyQuery, UseMutation, UseQuery } from './buildHooks';

type QueryHookNames<Definitions extends EndpointDefinitions> = {
	[
		K in keyof Definitions as Definitions[K] extends {
			type: DefinitionType.query;
		}
			? `use${Capitalize<K & string>}Query`
			: never
	]: UseQuery<Extract<Definitions[K], QueryDefinition<any, any, any, any>>>;
};

type LazyQueryHookNames<Definitions extends EndpointDefinitions> = {
	[
		K in keyof Definitions as Definitions[K] extends {
			type: DefinitionType.query;
		}
			? `useLazy${Capitalize<K & string>}Query`
			: never
	]: UseLazyQuery<Extract<Definitions[K], QueryDefinition<any, any, any, any>>>;
};

type InfiniteQueryHookNames<Definitions extends EndpointDefinitions> = {
	[
		K in keyof Definitions as Definitions[K] extends {
			type: DefinitionType.infinitequery;
		}
			? `use${Capitalize<K & string>}InfiniteQuery`
			: never
	]: UseInfiniteQuery<Extract<Definitions[K], InfiniteQueryDefinition<any, any, any, any, any>>>;
};

type MutationHookNames<Definitions extends EndpointDefinitions> = {
	[
		K in keyof Definitions as Definitions[K] extends {
			type: DefinitionType.mutation;
		}
			? `use${Capitalize<K & string>}Mutation`
			: never
	]: UseMutation<Extract<Definitions[K], MutationDefinition<any, any, any, any>>>;
};

export type HooksWithUniqueNames<Definitions extends EndpointDefinitions> =
	QueryHookNames<Definitions> &
		LazyQueryHookNames<Definitions> &
		InfiniteQueryHookNames<Definitions> &
		MutationHookNames<Definitions>;