safe-react-cache.ts

import { safeParse, safeStringify } from '@shared/utils-json';
import { cache } from 'react';

/**
 * By default react cache only supports primitive arguments.
 *
 * This fn wraps react cache to support a single object argument.
 */
export function safeCache<R>(fn: () => R): () => R;
export function safeCache<T, R>(fn: (p: T) => R): (p: T) => R;
export function safeCache<T, R>(fn: (p?: T) => R): (p?: T) => R {
  const cachedFn = cache((x?: string) => fn(safeParse(x) as T));

  return p => cachedFn(safeStringify(p));
}