12345678910111213141516171819202122232425262728293031323334353637383940
import { atom } from 'jotai'import type { Atom, Getter } from 'jotai'import { createMemoizeAtom } from './weakCache'
const memoizeAtom = createMemoizeAtom()
const deepFreeze = (obj: any) => {  if (typeof obj !== 'object' || obj === null) return  Object.freeze(obj)  const propNames = Object.getOwnPropertyNames(obj)  for (const name of propNames) {    const value = obj[name]    deepFreeze(value)  }  return obj}
export function freezeAtom<AtomType extends Atom<any>>(  anAtom: AtomType): AtomType {  return memoizeAtom(() => {    const frozenAtom: any = atom(      (get) => deepFreeze(get(anAtom)),      (_get, set, arg) => set(anAtom as any, arg)    )    return frozenAtom  }, [anAtom])}
export function freezeAtomCreator<  CreateAtom extends (...params: any[]) => Atom<any>>(createAtom: CreateAtom) {  return ((...params: any[]) => {    const anAtom = createAtom(...params)    const origRead = anAtom.read    anAtom.read = (get: Getter) => deepFreeze(origRead(get))    return anAtom  }) as CreateAtom}