createMemoContainer

createMemoContainer.ts

/**
 *
 * @param initialValue {T} initial value
 * @param callback {function} used to update value
 */
export const createMemoContainer = <T, ExtraArgs extends unknown[] = []>(
  initialValue: T,
  callback: (currentValue: T, ...args: ExtraArgs) => T | null,
): ((...args: ExtraArgs) => T) => {
  let currentValue: T = initialValue;

  return (...args: ExtraArgs): T => {
    const newValue = callback(currentValue, ...args);

    if (newValue === null) return currentValue;

    currentValue = newValue;
    return newValue;
  };
};