FilesExpand file tree

main

/

react-commit-cascade-write-probe.ts

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

53 lines (51 loc) · 1.98 KB

main

/

react-commit-cascade-write-probe.ts

Copy path

Top

File metadata and controls

• Code
• Blame

53 lines (51 loc) · 1.98 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

/**

* zustand middleware that samples store writes during a suspected React commit

* cascade, so the crash report names the code driving it.

*

* Why middleware and not a setState patch: create() hands slices the `set`

* closure built before `api` exists (zustand/esm/vanilla.mjs), so patching

* useAppStore.setState afterwards misses every slice-internal set() — and the

* slices are where the writes are. Wrapping the creator catches both, the same

* way store-listener-census has to reach the inner api to see hook subscribers.

*

* Cost when no cascade is suspected: one boolean field load per write.

*/

import {

noteReactCommitCascadeStoreWrite,

reactCommitCascadeWriteProbe

} from '@/lib/react-commit-cascade-store-write-samples'

import type { StateCreator } from 'zustand'

/** zustand's two `setState` call shapes, as a union so a wrapper can forward them uncast. */

export type StoreSetArgs<TState> =

| [

partial: TState | Partial<TState> | ((state: TState) => TState | Partial<TState>),

replace?: false

]

| [state: TState | ((state: TState) => TState), replace: true]

export function withReactCommitCascadeWriteProbe<TState>(

createState: StateCreator<TState, [], []>

): StateCreator<TState, [], []> {

return (set, get, api) => {

const probe = (write: typeof set): typeof set => {

const wrapped = (...args: StoreSetArgs<TState>): void => {

if (reactCommitCascadeWriteProbe.armed) {

try {

noteReactCommitCascadeStoreWrite(wrapped, args[0])

} catch {

// This is the app's universal write path: a diagnostic that throws here

// would drop the write itself.

}

}

if (args[1] === true) {

write(...args)

} else {

write(...args)

}

}

return wrapped

}

// Why separately: an outer middleware may hand slices a different `set` than api.setState.

api.setState = probe(api.setState)

return createState(probe(set), get, api)

}

}