import type { ReactNode } from 'react';
import type { LazyRouteFunction, RouteObject, UIMatch } from 'react-router';

export { useRouteTitles } from './useRouteTitles';
export { usePrompt } from './usePrompt';

declare module 'react-router' {
	export interface IndexRouteObject {
		meta?: KnownRouteObjectMeta;
	}

	export interface NonIndexRouteObject {
		meta?: KnownRouteObjectMeta;
	}
}

export interface KnownRouteObjectHandle {
	title?: RouteHandleTitle;

	[key: string]: any;
}

type RouteHandleTitle = string | ((data: any, match: UIMatch<unknown, KnownRouteObjectHandle>) => string);

export interface KnownRouteObjectMeta {
	id?: string;
	title?: string;
	icon?: ReactNode;
	menu?: Record<string, any>;

	[key: string]: any;
}

export type RouteObjects = RouteObject[];

export type LazyRouteObject<R extends RouteObject = RouteObject> = Awaited<ReturnType<LazyRouteFunction<R>>>;

export function lazyRoute(
	loader: () => Promise<
		| Awaited<object>
		| {
				default: any;
		  }
		| { route: object }
	>,
): LazyRouteFunction<RouteObject> {
	return () =>
		loader().then((v) => {
			if ('route' in v) {
				return v.route;
			}
			if ('default' in v) {
				return v.default;
			}
			return v;
		});
}