import type { StructuralSharingOption, ValidateSelected } from './structuralSharing';
import type {
	AnyRouter,
	FromPathOption,
	MakeRouteMatch,
	MakeRouteMatchUnion,
	RegisteredRouter,
	ResolveUseLoaderData,
	ResolveUseLoaderDeps,
	ResolveUseParams,
	ResolveUseSearch,
	RouterState,
	StrictOrFrom,
	ThrowConstraint,
	ThrowOrOptional,
	UseLoaderDataResult,
	UseLoaderDepsResult,
	UseNavigateResult,
	UseParamsResult,
	UseRouteContextBaseOptions,
	UseRouteContextOptions,
	UseRouteContextResult,
	UseSearchResult,
} from '@tanstack/router-core';

export interface UseMatchBaseOptions<
	TRouter extends AnyRouter,
	TFrom,
	TStrict extends boolean,
	TThrow extends boolean,
	TSelected,
	TStructuralSharing extends boolean,
> {
	select?: (
		match: MakeRouteMatch<TRouter['routeTree'], TFrom, TStrict>,
	) => ValidateSelected<TRouter, TSelected, TStructuralSharing>;
	shouldThrow?: TThrow;
}

export type UseMatchRoute<out TFrom> = <
	TRouter extends AnyRouter = RegisteredRouter,
	TSelected = unknown,
	TStructuralSharing extends boolean = boolean,
>(
	opts?: UseMatchBaseOptions<TRouter, TFrom, true, true, TSelected, TStructuralSharing> &
		StructuralSharingOption<TRouter, TSelected, TStructuralSharing>,
) => UseMatchResult<TRouter, TFrom, true, TSelected>;

export type UseMatchOptions<
	TRouter extends AnyRouter,
	TFrom extends string | undefined,
	TStrict extends boolean,
	TThrow extends boolean,
	TSelected,
	TStructuralSharing extends boolean,
> = StrictOrFrom<TRouter, TFrom, TStrict> &
	UseMatchBaseOptions<TRouter, TFrom, TStrict, TThrow, TSelected, TStructuralSharing> &
	StructuralSharingOption<TRouter, TSelected, TStructuralSharing>;

export type UseMatchResult<
	TRouter extends AnyRouter,
	TFrom,
	TStrict extends boolean,
	TSelected,
> = unknown extends TSelected
	? TStrict extends true
		? MakeRouteMatch<TRouter['routeTree'], TFrom, TStrict>
		: MakeRouteMatchUnion<TRouter>
	: TSelected;

export interface UseParamsBaseOptions<
	TRouter extends AnyRouter,
	TFrom,
	TStrict extends boolean,
	TThrow extends boolean,
	TSelected,
	TStructuralSharing,
> {
	select?: (
		params: ResolveUseParams<TRouter, TFrom, TStrict>,
	) => ValidateSelected<TRouter, TSelected, TStructuralSharing>;
	shouldThrow?: TThrow;
}

export type UseParamsRoute<out TFrom> = <
	TRouter extends AnyRouter = RegisteredRouter,
	TSelected = unknown,
	TStructuralSharing extends boolean = boolean,
>(
	opts?: UseParamsBaseOptions<TRouter, TFrom, true, true, TSelected, TStructuralSharing> &
		StructuralSharingOption<TRouter, TSelected, TStructuralSharing>,
) => UseParamsResult<TRouter, TFrom, true, TSelected>;

export type UseParamsOptions<
	TRouter extends AnyRouter,
	TFrom extends string | undefined,
	TStrict extends boolean,
	TThrow extends boolean,
	TSelected,
	TStructuralSharing,
> = StrictOrFrom<TRouter, TFrom, TStrict> &
	UseParamsBaseOptions<TRouter, TFrom, TStrict, TThrow, TSelected, TStructuralSharing> &
	StructuralSharingOption<TRouter, TSelected, TStructuralSharing>;

export interface UseSearchBaseOptions<
	TRouter extends AnyRouter,
	TFrom,
	TStrict extends boolean,
	TThrow extends boolean,
	TSelected,
	TStructuralSharing,
> {
	select?: (
		search: ResolveUseSearch<TRouter, TFrom, TStrict>,
	) => ValidateSelected<TRouter, TSelected, TStructuralSharing>;
	shouldThrow?: TThrow;
}

export type UseSearchRoute<out TFrom> = <
	TRouter extends AnyRouter = RegisteredRouter,
	TSelected = unknown,
	TStructuralSharing extends boolean = boolean,
>(
	opts?: UseSearchBaseOptions<TRouter, TFrom, true, true, TSelected, TStructuralSharing> &
		StructuralSharingOption<TRouter, TSelected, TStructuralSharing>,
) => UseSearchResult<TRouter, TFrom, true, TSelected>;

export type UseSearchOptions<
	TRouter extends AnyRouter,
	TFrom,
	TStrict extends boolean,
	TThrow extends boolean,
	TSelected,
	TStructuralSharing,
> = StrictOrFrom<TRouter, TFrom, TStrict> &
	UseSearchBaseOptions<TRouter, TFrom, TStrict, TThrow, TSelected, TStructuralSharing> &
	StructuralSharingOption<TRouter, TSelected, TStructuralSharing>;

export interface UseLoaderDataBaseOptions<
	TRouter extends AnyRouter,
	TFrom,
	TStrict extends boolean,
	TSelected,
	TStructuralSharing,
> {
	select?: (
		data: ResolveUseLoaderData<TRouter, TFrom, TStrict>,
	) => ValidateSelected<TRouter, TSelected, TStructuralSharing>;
}

export type UseLoaderDataRoute<out TFrom> = <
	TRouter extends AnyRouter = RegisteredRouter,
	TSelected = unknown,
	TStructuralSharing extends boolean = boolean,
>(
	opts?: UseLoaderDataBaseOptions<TRouter, TFrom, true, TSelected, TStructuralSharing> &
		StructuralSharingOption<TRouter, TSelected, TStructuralSharing>,
) => UseLoaderDataResult<TRouter, TFrom, true, TSelected>;

export type UseLoaderDataOptions<
	TRouter extends AnyRouter,
	TFrom extends string | undefined,
	TStrict extends boolean,
	TSelected,
	TStructuralSharing,
> = StrictOrFrom<TRouter, TFrom, TStrict> &
	UseLoaderDataBaseOptions<TRouter, TFrom, TStrict, TSelected, TStructuralSharing> &
	StructuralSharingOption<TRouter, TSelected, TStructuralSharing>;

export interface UseLoaderDepsBaseOptions<
	TRouter extends AnyRouter,
	TFrom,
	TSelected,
	TStructuralSharing,
> {
	select?: (
		deps: ResolveUseLoaderDeps<TRouter, TFrom>,
	) => ValidateSelected<TRouter, TSelected, TStructuralSharing>;
}

export type UseLoaderDepsRoute<out TFrom> = <
	TRouter extends AnyRouter = RegisteredRouter,
	TSelected = unknown,
	TStructuralSharing extends boolean = boolean,
>(
	opts?: UseLoaderDepsBaseOptions<TRouter, TFrom, TSelected, TStructuralSharing> &
		StructuralSharingOption<TRouter, TSelected, TStructuralSharing>,
) => UseLoaderDepsResult<TRouter, TFrom, TSelected>;

export type UseLoaderDepsOptions<
	TRouter extends AnyRouter,
	TFrom extends string | undefined,
	TSelected,
	TStructuralSharing,
> = StrictOrFrom<TRouter, TFrom> &
	UseLoaderDepsBaseOptions<TRouter, TFrom, TSelected, TStructuralSharing> &
	StructuralSharingOption<TRouter, TSelected, TStructuralSharing>;

export type UseRouteContextRoute<out TFrom> = <
	TRouter extends AnyRouter = RegisteredRouter,
	TSelected = unknown,
>(
	opts?: UseRouteContextBaseOptions<TRouter, TFrom, true, TSelected>,
) => UseRouteContextResult<TRouter, TFrom, true, TSelected>;

export interface UseLocationBaseOptions<
	TRouter extends AnyRouter,
	TSelected,
	TStructuralSharing extends boolean = boolean,
> {
	select?: (
		location: RouterState<TRouter['routeTree']>['location'],
	) => ValidateSelected<TRouter, TSelected, TStructuralSharing>;
}

export type UseLocationResult<TRouter extends AnyRouter, TSelected> = unknown extends TSelected
	? RouterState<TRouter['routeTree']>['location']
	: TSelected;

export interface UseMatchesBaseOptions<TRouter extends AnyRouter, TSelected, TStructuralSharing> {
	select?: (
		matches: Array<MakeRouteMatchUnion<TRouter>>,
	) => ValidateSelected<TRouter, TSelected, TStructuralSharing>;
}

export type UseMatchesResult<TRouter extends AnyRouter, TSelected> = unknown extends TSelected
	? Array<MakeRouteMatchUnion<TRouter>>
	: TSelected;