UNPKG

zustand

Version:

🐻 Bear necessities for state management in React

30 lines (29 loc) 1.47 kB
import type { Draft } from 'immer'; import type { StateCreator, StoreMutatorIdentifier } from 'zustand/vanilla'; type Immer = <T, Mps extends [StoreMutatorIdentifier, unknown][] = [], Mcs extends [StoreMutatorIdentifier, unknown][] = []>(initializer: StateCreator<T, [...Mps, ['zustand/immer', never]], Mcs>) => StateCreator<T, Mps, [['zustand/immer', never], ...Mcs]>; declare module '../vanilla' { interface StoreMutators<S, A> { ['zustand/immer']: WithImmer<S>; } } type Write<T, U> = Omit<T, keyof U> & U; type SkipTwo<T> = T extends { length: 0; } ? [] : T extends { length: 1; } ? [] : T extends { length: 0 | 1; } ? [] : T extends [unknown, unknown, ...infer A] ? A : T extends [unknown, unknown?, ...infer A] ? A : T extends [unknown?, unknown?, ...infer A] ? A : never; type SetStateType<T extends unknown[]> = Exclude<T[0], (...args: any[]) => any>; type WithImmer<S> = Write<S, StoreImmer<S>>; type StoreImmer<S> = S extends { setState: infer SetState; } ? SetState extends { (...a: infer A1): infer Sr1; (...a: infer A2): infer Sr2; } ? { setState(nextStateOrUpdater: SetStateType<A2> | Partial<SetStateType<A2>> | ((state: Draft<SetStateType<A2>>) => void), shouldReplace?: false, ...a: SkipTwo<A1>): Sr1; setState(nextStateOrUpdater: SetStateType<A2> | ((state: Draft<SetStateType<A2>>) => void), shouldReplace: true, ...a: SkipTwo<A2>): Sr2; } : never : never; export declare const immer: Immer; export {};