import type { Atom, WritableAtom } from 'jotai'
import { atom } from 'jotai/vanilla'
import { RESET } from './actions'

export type History<Value> = Value[]

/**
 * @param targetAtom an atom or derived atom
 * @param limit the maximum number of history states to keep
 * @returns an atom with an array of history states
 */
export function withHistory<Value>(
  targetAtom: Atom<Value>,
  limit: number
): WritableAtom<History<Value>, [RESET], void> {
  const refreshAtom = atom(0)
  refreshAtom.debugPrivate = true
  const historyAtom = {
    read: (get) => (get(refreshAtom), { history: [] }),
    write: (get, set) => {
      get(historyAtom).history.length = 0
      set(refreshAtom, (v) => v + 1)
    },
    onMount: (reset) => reset,
    debugPrivate: true,
    init: 1, // dirty hack to bypass hasInitialValue
  } as WritableAtom<{ history: Value[] }, [], void>
  return atom(
    (get) => {
      const ref = get(historyAtom)
      return (ref.history = [get(targetAtom), ...ref.history].slice(0, limit))
    },
    (_, set, action: RESET) => {
      if (action === RESET) {
        set(historyAtom)
      }
    }
  )
}