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
import { useCallback, useContext } from 'react'import type { SetAtom, WritableAtom } from './atom'import { getScopeContext } from './contexts'import { WRITE_ATOM } from './store'import type { VersionObject } from './store'import type { ExtractAtomResult, ExtractAtomUpdate } from './typeUtils'
type Scope = NonNullable<Parameters<typeof getScopeContext>[0]>
export function useSetAtom<Value, Update, Result extends void | Promise<void>>(  atom: WritableAtom<Value, Update, Result>,  scope?: Scope): SetAtom<Update, Result>
export function useSetAtom<  AtomType extends WritableAtom<any, any, void | Promise<void>>>(  atom: AtomType,  scope?: Scope): SetAtom<ExtractAtomUpdate<AtomType>, ExtractAtomResult<AtomType>>
export function useSetAtom<Value, Update, Result extends void | Promise<void>>(  atom: WritableAtom<Value, Update, Result>,  scope?: Scope) {  const ScopeContext = getScopeContext(scope)  const { s: store, w: versionedWrite } = useContext(ScopeContext)  const setAtom = useCallback(    (update: Update) => {      if (__DEV__ && !('write' in atom)) {        // useAtom can pass non writable atom with wrong type assertion,        // so we should check here.        throw new Error('not writable atom')      }      const write = (version?: VersionObject) =>        store[WRITE_ATOM](atom, update, version)      return versionedWrite ? versionedWrite(write) : write()    },    [store, versionedWrite, atom]  )  return setAtom as SetAtom<Update, Result>}