1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import { atom } from 'jotai'import type { Atom } from 'jotai'import { createMemoizeAtom } from './weakCache'
const memoizeAtom = createMemoizeAtom()const emptyArrayAtom = atom(() => [])
type ResolveAtom<T> = T extends Atom<infer V> ? V : Ttype AwaitedAtom<T> = Awaited<ResolveAtom<T>>
export function waitForAll<Atoms extends Atom<unknown>[]>(  atoms: readonly [...Atoms]): Atom<{  [K in keyof Atoms]: AwaitedAtom<Atoms[K]>}>
export function waitForAll<Atoms extends Record<string, Atom<unknown>>>(  atoms: Atoms): Atom<{  [K in keyof Atoms]: AwaitedAtom<Atoms[K]>}>
export function waitForAll<  Atoms extends Record<string, Atom<unknown>> | readonly Atom<unknown>[]>(atoms: Atoms) {  const createAtom = () => {    const unwrappedAtoms = unwrapAtoms(atoms)    const derivedAtom = atom((get) => {      const promises: Promise<unknown>[] = []      const values = unwrappedAtoms.map((anAtom, index) => {        try {          return get(anAtom)        } catch (e) {          if (e instanceof Promise) {            promises[index] = e          } else {            throw e          }        }      })      if (promises.length) {        throw Promise.all(promises)      }      return wrapResults(atoms, values) as {        [K in keyof Atoms]: AwaitedAtom<Atoms[K]>      }    })    return derivedAtom  }
  if (Array.isArray(atoms)) {    if (atoms.length) {      return memoizeAtom(createAtom, atoms)    }    return emptyArrayAtom  }  return createAtom()}
const unwrapAtoms = <  Atoms extends Record<string, Atom<unknown>> | readonly Atom<unknown>[]>(  atoms: Atoms): Atom<unknown>[] =>  Array.isArray(atoms)    ? atoms    : Object.getOwnPropertyNames(atoms).map((key) => atoms[key as keyof Atoms])
const wrapResults = <  Atoms extends Record<string, Atom<unknown>> | readonly Atom<unknown>[]>(  atoms: Atoms,  results: unknown[]): unknown[] | Record<string, unknown> =>  Array.isArray(atoms)    ? results    : Object.getOwnPropertyNames(atoms).reduce(        (out, key, idx) => ({ ...out, [key]: results[idx] }),        {}      )