import { useSearchParams } from "wouter";

function isValidRedirect(value: string | null | undefined): value is string {
  return !!(value && /^[~/]/.test(value));
}

/**
 * Gets redirect path from query params and validates it.
 *
 * Returns `null` if no path is found, or it is invalid.
 */
export function useRedirectPath() {
  const [params] = useSearchParams();
  const redirectTo = params.get("redirectTo");

  if (isValidRedirect(redirectTo)) {
    return redirectTo;
  }

  return null;
}