mirror of
https://github.com/DocNR/POWR.git
synced 2025-04-23 01:01:27 +00:00
17 lines
513 B
TypeScript
17 lines
513 B
TypeScript
|
// utils/createSelectors.ts
|
||
|
|
import { StoreApi, UseBoundStore } from 'zustand';
|
||
|
|
|
||
|
|
type WithSelectors<S> = S extends { getState: () => infer T }
|
||
|
|
? S & { use: { [K in keyof T]: () => T[K] } }
|
||
|
|
: never;
|
||
|
|
|
||
|
|
export const createSelectors = <S extends UseBoundStore<StoreApi<object>>>(
|
||
|
|
_store: S,
|
||
|
|
) => {
|
||
|
|
let store = _store as WithSelectors<typeof _store>;
|
||
|
|
store.use = {};
|
||
|
|
for (let k of Object.keys(store.getState())) {
|
||
|
|
(store.use as any)[k] = () => store((s) => s[k as keyof typeof s]);
|
||
|
|
}
|
||
|
|
return store;
|
||
|
|
};
|