p-memoize

pMemoize: <
    FunctionToMemoize extends AnyAsyncFunction,
    CacheKeyType = Parameters<FunctionToMemoize>[0],
    OptionsType = Options<FunctionToMemoize, CacheKeyType>
>(
    fn: FunctionToMemoize,
    options?: OptionsType & Options<FunctionToMemoize, CacheKeyType>
) => FunctionToMemoize;

pMemoizeClear: (fn: AnyAsyncFunction) => void;

pMemoizeDecorator: <
    FunctionToMemoize extends AnyAsyncFunction = AnyAsyncFunction,
    CacheKeyType = Parameters<FunctionToMemoize>[0]
>(
    options?: Options<NoInfer<FunctionToMemoize>, NoInfer<CacheKeyType>>
) => <This, Value extends FunctionToMemoize>(
    value: Value,
    context: ClassMethodDecoratorContext<This, Value>
) => void;

type AnyAsyncFunction = (...arguments_: readonly any[]) => Promise<unknown | void>;

type CacheStorage<KeyType, ValueType> = {
    has: (key: KeyType) => Promise<boolean> | boolean;
    get: (key: KeyType) => Promise<ValueType | undefined> | ValueType | undefined;
    set: (key: KeyType, value: ValueType) => Promise<unknown> | unknown;
    delete: (key: KeyType) => unknown;
    clear?: () => unknown;
};

type Options<FunctionToMemoize extends AnyAsyncFunction, CacheKeyType> = {
    /**
    Determines the cache key for storing the result based on the function arguments. By default, __only the first argument is considered__ and it only works with [primitives](https://developer.mozilla.org/docs/Glossary/Primitive).

    A `cacheKey` function can return any type supported by `Map` (or whatever structure you use in the `cache` option).

    You can have it cache **all** the arguments by value with `JSON.stringify`, if they are compatible:

    ```
    import pMemoize from 'p-memoize';

    pMemoize(function_, {cacheKey: JSON.stringify});
    ```

    Or you can use a more full-featured serializer like [serialize-javascript](https://github.com/yahoo/serialize-javascript) to add support for `RegExp`, `Date` and so on.

    ```
    import pMemoize from 'p-memoize';
    import serializeJavascript from 'serialize-javascript';

    pMemoize(function_, {cacheKey: serializeJavascript});
    ```

    @default arguments_ => arguments_[0]
    @example arguments_ => JSON.stringify(arguments_)
    */
    readonly cacheKey?: (arguments_: Parameters<FunctionToMemoize>) => CacheKeyType;
    /**
    Use a different cache storage. Must implement the following methods: `.has(key)`, `.get(key)`, `.set(key, value)`, `.delete(key)`, and optionally `.clear()`. You could for example use a `WeakMap` instead or [`quick-lru`](https://github.com/sindresorhus/quick-lru) for a LRU cache. To disable caching so that only concurrent executions resolve with the same value, pass `false`.

    @default new Map()
    @example new WeakMap()
    */
    readonly cache?:
        | CacheStorage<CacheKeyType, AsyncReturnType<FunctionToMemoize>>
        | false;
    /**
    Controls whether a fulfilled value should be written to the cache.

    It runs after the function fulfills and before `cache.set`.

    - Omit to keep current behavior (always write).
    - Return `false` to skip writing to the cache (in-flight de-duplication is still cleared).
    - Throw or reject to propagate the error and skip caching.

    @example
    ```
    import pMemoize from 'p-memoize';

    // Only cache defined values
    const getMaybe = pMemoize(async key => db.get(key), {
        shouldCache: value => value !== undefined,
    });

    // Only cache non-empty arrays
    const search = pMemoize(async query => fetchResults(query), {
        shouldCache: value => Array.isArray(value) && value.length > 0,
    });
    ```

    Note: This only affects writes; reads from the cache are unchanged.
    */
    readonly shouldCache?: ShouldCache<
        CacheKeyType,
        AsyncReturnType<FunctionToMemoize>,
        Parameters<FunctionToMemoize>
    >;
};

type ShouldCache<
    KeyType,
    ValueType,
    ArgumentsList extends readonly unknown[] = readonly unknown[]
> = (
    value: ValueType,
    context: {
        key: KeyType;
        argumentsList: ArgumentsList;
    }
) => boolean | Promise<boolean>;