26 lines
785 B
JavaScript
26 lines
785 B
JavaScript
'use strict';
|
|
|
|
var React = require('react');
|
|
var vanilla = require('zustand/vanilla');
|
|
|
|
const identity = (arg) => arg;
|
|
function useStore(api, selector = identity) {
|
|
const slice = React.useSyncExternalStore(
|
|
api.subscribe,
|
|
React.useCallback(() => selector(api.getState()), [api, selector]),
|
|
React.useCallback(() => selector(api.getInitialState()), [api, selector])
|
|
);
|
|
React.useDebugValue(slice);
|
|
return slice;
|
|
}
|
|
const createImpl = (createState) => {
|
|
const api = vanilla.createStore(createState);
|
|
const useBoundStore = (selector) => useStore(api, selector);
|
|
Object.assign(useBoundStore, api);
|
|
return useBoundStore;
|
|
};
|
|
const create = ((createState) => createState ? createImpl(createState) : createImpl);
|
|
|
|
exports.create = create;
|
|
exports.useStore = useStore;
|