proxy-memoize/src/memoizeWithArgs.ts

Intuitive magical memoization library with Proxy and WeakMap

import { memoize } from './memoize.js';

type Options = Omit<NonNullable<Parameters<typeof memoize>[1]>, 'noWeakMap'>;

/**
 * Create a memoized function with args
 *
 * @example
 * import { memoizeWithArgs } from 'proxy-memoize';
 *
 * const fn = memoizeWithArgs((a, b) => ({ sum: a.v + b.v, diff: a.v - b.v }));
 *
 * @param options
 * @param options.size - (default: 1)
 */
export const memoizeWithArgs = <Args extends unknown[], Result>(
  fnWithArgs: (...args: Args) => Result,
  options?: Options,
) => {
  const fn = memoize((args: Args) => fnWithArgs(...args), {
    ...options,
    noWeakMap: true,
  });
  return (...args: Args) => fn(args);
};