about summary refs log tree commit diff
path: root/src/lib/async/revertible.ts
blob: 3c8e3e8f9e05ae7340d27ab853ac68379d66bbbc (plain) (blame)
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
import {runInAction} from 'mobx'
import {deepObserve} from 'mobx-utils'
import set from 'lodash.set'

const ongoingActions = new Set<any>()

export const updateDataOptimistically = async <
  T extends Record<string, any>,
  U,
>(
  model: T,
  preUpdate: () => void,
  serverUpdate: () => Promise<U>,
  postUpdate?: (res: U) => void,
): Promise<void> => {
  if (ongoingActions.has(model)) {
    return
  }
  ongoingActions.add(model)

  const prevState: Map<string, any> = new Map<string, any>()
  const dispose = deepObserve(model, (change, path) => {
    if (change.observableKind === 'object') {
      if (change.type === 'update') {
        prevState.set(
          [path, change.name].filter(Boolean).join('.'),
          change.oldValue,
        )
      } else if (change.type === 'add') {
        prevState.set([path, change.name].filter(Boolean).join('.'), undefined)
      }
    }
  })
  preUpdate()
  dispose()

  try {
    const res = await serverUpdate()
    runInAction(() => {
      postUpdate?.(res)
    })
  } catch (error) {
    runInAction(() => {
      prevState.forEach((value, path) => {
        set(model, path, value)
      })
    })
    throw error
  } finally {
    ongoingActions.delete(model)
  }
}