pocket-state 0.0.2 → 0.0.4

package/package.json

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

package/src/globalState/event.ts

@@ -78,4 +78,13 @@ export class EventEmitter implements IEventEmitter {
     this.events.clear();
     this.onceWrappers.clear();
   }
+
+  getNumberOfSubscriber(event?: string): number {
+    if (event) {
+      return this.events.get(event)?.size ?? 0;
+    }
+    let total = 0;
+    this.events.forEach(set => (total += set.size));
+    return total;
+  }
 }

package/src/globalState/hooks.ts

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

package/src/globalState/store.ts

@@ -1,18 +1,19 @@
 // store.ts
-// import deepEqual from 'fast-deep-equal';
 import {IEventEmitter, Listener, Middleware, Store, UseStoreGet} from './type';
 import {EventEmitter} from './event';
 import {Draft, produce} from 'immer';
-// import {shallow} from './shallowEqual';
-import deepEqual from 'fast-deep-equal';
+import {shallow} from './shallowEqual';
+
 export function createStore<T>(
   initialState: T,
   middlewares: Middleware<T>[] = [],
+  equalityFn?: (a: any, b: any) => boolean,
 ): Store<T> {
   const emitter: IEventEmitter = new EventEmitter();
   let state = initialState;
 
-  // Coalesce emits trong cùng 1 microtask
+  const areEqual = equalityFn ?? shallow;
+
   let emitScheduled = false;
   const emitState = () => {
     if (emitScheduled) return;
@@ -25,13 +26,12 @@ export function createStore<T>(
 
   function baseSet(delta: Partial<T>) {
     const nextState = Array.isArray(state)
-      ? (delta as unknown as T) // với array: replace
-      : {...state, ...delta}; // với object: shallow merge
+      ? (delta as unknown as T)
+      : {...state, ...delta};
 
-    if (!deepEqual(state, nextState)) {
+    if (!areEqual(state, nextState)) {
       state = nextState;
 
-      // Dev guard: phát hiện mutate ngoài store
       if (process.env.NODE_ENV !== 'production') {
         try {
           Object.freeze(state as any);
@@ -61,22 +61,20 @@ export function createStore<T>(
     let wrapped: Listener<any>;
 
     if (typeof maybeListener === 'function') {
-      // subscribe(selector, listener)
       const selector: (s: T) => any = selectorOrListener;
       let prevSlice = selector(state);
       wrapped = (next: T) => {
         const slice = selector(next);
-        if (!deepEqual(slice, prevSlice)) {
+        if (!areEqual(slice, prevSlice)) {
           prevSlice = slice;
           maybeListener(slice);
         }
       };
     } else {
-      // subscribe(listener)
       const listener: Listener<T> = selectorOrListener;
       let prev = state;
       wrapped = (next: T) => {
-        if (!deepEqual(next, prev)) {
+        if (!areEqual(next, prev)) {
           prev = next;
           listener(next);
         }
@@ -84,10 +82,6 @@ export function createStore<T>(
     }
 
     emitter.on('state', wrapped);
-
-    // Lưu ý: KHÔNG gọi listener ngay khi subscribe
-    // (để tương thích useSyncExternalStore: initial snapshot lấy qua getSnapshot)
-
     return () => emitter.off('state', wrapped);
   }
 
@@ -111,7 +105,7 @@ export function createStore<T>(
   function setImmer(updater: (draft: Draft<T>) => void): void {
     try {
       const nextState = produce(state, updater);
-      if (deepEqual(state, nextState)) return;
+      if (areEqual(state, nextState)) return;
 
       if (Array.isArray(state)) {
         setFn(nextState as unknown as Partial<T>);
@@ -168,7 +162,7 @@ export function createStore<T>(
       }
     }
     const current = getValue();
-    if (!deepEqual(current, next)) {
+    if (!areEqual(current, next)) {
       setFn(next as unknown as Partial<T>);
     }
   }
@@ -184,7 +178,11 @@ export function createStore<T>(
   }
 
   function isDirty() {
-    return !deepEqual(state, initialState);
+    return !areEqual(state, initialState);
+  }
+
+  function getNumberOfSubscriber() {
+    return emitter.getNumberOfSubscriber();
   }
   return {
     getValue,
@@ -194,5 +192,6 @@ export function createStore<T>(
     reset,
     subscribe,
     isDirty,
+    getNumberOfSubscriber,
   };
 }

package/src/globalState/type.d.ts

@@ -97,6 +97,9 @@ export interface IEventEmitter {
 
   /** Remove all listeners for all events. */
   clear(): void;
+
+  /** Get number of subscribe */
+  getNumberOfSubscriber(event?: string): number;
 }
 
 /**
@@ -194,6 +197,9 @@ export interface Store<T> {
    * ```
    */
   isDirty(): boolean;
+
+  /** Get number of subscriber for current store */
+  getNumberOfSubscriber(): number;
 }
 
 /**