create-selectors.ts 512 B

123456789101112131415
  1. import { StoreApi, UseBoundStore } from 'zustand';
  2. type WithSelectors<S> = S extends { getState: () => infer T }
  3. ? S & { use: { [K in keyof T]: () => T[K] } }
  4. : never;
  5. export const createSelectors = <S extends UseBoundStore<StoreApi<object>>>(_store: S) => {
  6. const store = _store as WithSelectors<typeof _store>;
  7. store.use = {};
  8. for (const k of Object.keys(store.getState())) {
  9. (store.use as Record<string, () => unknown>)[k] = () => store((s) => s[k as keyof typeof s]);
  10. }
  11. return store;
  12. };