// CatchNotFound — a CatchBoundary specialized to `notFound()` errors (port of
// react-router's not-found.tsx). Anything that isn't a NotFoundError is rethrown
// (from onCatch during the catch render) so it reaches the next error boundary
// up; the reset key is `not-found-${pathname}-${status}` so navigating away (or a
// new load settling) clears the not-found UI.
import { isNotFound } from '@tanstack/router-core';
import type { ErrorComponentProps, NotFoundError } from '@tanstack/router-core';
import type { OctaneNode } from 'octane';
import type { ErrorInfo } from './route.ts';
import { useRouter } from './context.ts';
import { useStore } from './useStore.ts';
import { CatchBoundary } from './CatchBoundary.tsrx';

function NotFoundFallback(props: {
	error: unknown;
	render?: (error: NotFoundError) => OctaneNode;
}) {
	if (isNotFound(props.error)) {
		return props.render ? props.render(props.error) : null;
	}
	throw props.error;
}

export function CatchNotFound(props: {
	fallback?: (error: NotFoundError) => OctaneNode;
	onCatch?: (error: Error, errorInfo: ErrorInfo) => void;
	children: OctaneNode;
}) @{
	const router = useRouter();
	const pathname = useStore(router.stores.location, (l: any) => l.pathname as string);
	const status = useStore(router.stores.status, (s: any) => s as string);
	const resetKey = `not-found-${pathname}-${status}`;

	<CatchBoundary
		getResetKey={() => resetKey}
		onCatch={(error: Error, errorInfo: ErrorInfo) => {
			if (isNotFound(error)) {
				if (props.onCatch) props.onCatch(error, errorInfo);
			} else {
				throw error;
			}
		}}
		errorComponent={(p: ErrorComponentProps) => NotFoundFallback({
			error: p.error,
			render: props.fallback,
		})}
	>{props.children}</CatchBoundary>
}

export function DefaultGlobalNotFound() @{
	<p>Not Found</p>
}