pocket-state 0.1.12 → 0.1.13

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pocket-state",
-  "version": "0.1.12",
+  "version": "0.1.13",
   "description": "tiny global store",
   "main": "src/index",
   "codegenConfig": {

package/src/globalState/hooks.ts

@@ -1,30 +1,29 @@
-import {useCallback, useRef, useSyncExternalStore} from 'react';
+import {useCallback, useSyncExternalStore} from 'react';
 import type {Store} from './type';
 
 export function useStore<T, S = T>(
   store: Store<T>,
   selector?: (state: T) => S,
 ): S {
-  const sliceRef = useRef<S>(
-    selector ? selector(store.getValue()) : (store.getValue() as unknown as S),
+  const getSnapshot = useCallback(
+    () =>
+      selector
+        ? selector(store.getValue())
+        : (store.getValue() as unknown as S),
+    [store, selector],
   );
 
   const subscribe = useCallback(
     (onChange: () => void) => {
-      if (selector) {
-        return store.subscribe(selector, (nextSlice: S) => {
-          sliceRef.current = nextSlice;
-          onChange();
-        });
+      if (selector && store.subscribe) {
+        return store.subscribe(selector, onChange);
       }
-      return store.subscribe((next: T) => {
-        sliceRef.current = next as unknown as S;
+      return store.subscribe(() => {
         onChange();
       });
     },
     [store, selector],
   );
-  const getSnapshot = useCallback(() => sliceRef.current, []);
 
   return useSyncExternalStore(subscribe, getSnapshot, getSnapshot);
 }