62 lines
1.5 KiB
JavaScript
62 lines
1.5 KiB
JavaScript
import ReactExports from 'react';
|
|
import { useStoreWithEqualityFn } from 'zustand/traditional';
|
|
|
|
const {
|
|
createElement,
|
|
createContext: reactCreateContext,
|
|
useContext,
|
|
useMemo,
|
|
useRef
|
|
} = ReactExports;
|
|
function createContext() {
|
|
if (process.env.NODE_ENV !== "production") {
|
|
console.warn(
|
|
"[DEPRECATED] `context` will be removed in a future version. Instead use `import { createStore, useStore } from 'zustand'`. See: https://github.com/pmndrs/zustand/discussions/1180."
|
|
);
|
|
}
|
|
const ZustandContext = reactCreateContext(void 0);
|
|
const Provider = ({
|
|
createStore,
|
|
children
|
|
}) => {
|
|
const storeRef = useRef();
|
|
if (!storeRef.current) {
|
|
storeRef.current = createStore();
|
|
}
|
|
return createElement(
|
|
ZustandContext.Provider,
|
|
{ value: storeRef.current },
|
|
children
|
|
);
|
|
};
|
|
const useContextStore = (selector, equalityFn) => {
|
|
const store = useContext(ZustandContext);
|
|
if (!store) {
|
|
throw new Error(
|
|
"Seems like you have not used zustand provider as an ancestor."
|
|
);
|
|
}
|
|
return useStoreWithEqualityFn(
|
|
store,
|
|
selector,
|
|
equalityFn
|
|
);
|
|
};
|
|
const useStoreApi = () => {
|
|
const store = useContext(ZustandContext);
|
|
if (!store) {
|
|
throw new Error(
|
|
"Seems like you have not used zustand provider as an ancestor."
|
|
);
|
|
}
|
|
return useMemo(() => ({ ...store }), [store]);
|
|
};
|
|
return {
|
|
Provider,
|
|
useStore: useContextStore,
|
|
useStoreApi
|
|
};
|
|
}
|
|
|
|
export { createContext as default };
|