✨ feat: createContext support for custom defaultEqualityFn (#3)

* 🐛 fix: custom eq method without inheritance error

* chore: update comment

* chore: update peerdep

* Revert "chore: update peerdep"

This reverts commit dfc5b3a.

* feat: plan 2

* Revert "feat: plan 2"

This reverts commit b19ab77.

* feat: createContext support for defaultEqualityFn custom

src/context.ts

@@ -1,6 +1,7 @@
 import type { ReactNode } from 'react';
 import { createElement, createContext as reactCreateContext, useContext, useRef } from 'react';
 import type { StoreApi } from 'zustand';
+import { shallow } from 'zustand/shallow';
 import { useStoreWithEqualityFn } from 'zustand/traditional';
 
 export type UseContextStore<S extends StoreApi<unknown>> = {
@@ -12,11 +13,16 @@ type ExtractState<S> = S extends { getState: () => infer T } ? T : never;
 
 type WithoutCallSignature<T> = { [K in keyof T]: T[K] };
 
+interface CreateContextParams {
+  equalityFn?: <U>(a: U, b: U) => boolean;
+}
+
 /**
  * create context for individual App
  * mostly use for component
  */
-export const createContext = <S extends StoreApi<unknown>>() => {
+export const createContext = <S extends StoreApi<unknown>>(params: CreateContextParams = {}) => {
+  const defaultEqualityFn = params.equalityFn || shallow;
   const ZustandContext = reactCreateContext<S | undefined>(undefined);
 
   const Provider = ({ createStore, children }: { createStore: () => S; children: ReactNode }) => {
@@ -29,29 +35,27 @@ export const createContext = <S extends StoreApi<unknown>>() => {
     return createElement(ZustandContext.Provider, { value: storeRef.current }, children);
   };
 
-  const useContextStore: UseContextStore<S> = <StateSlice = ExtractState<S>>(
-    selector?: (state: ExtractState<S>) => StateSlice,
-    equalityFn?: (a: StateSlice, b: StateSlice) => boolean,
-  ) => {
+  const useStoreApi = () => {
     const store = useContext(ZustandContext);
     if (!store) {
       throw new Error('Seems like you have not used zustand provider as an ancestor.');
     }
+    return store as WithoutCallSignature<S>;
+  };
+
+  const useContextStore: UseContextStore<S> = <StateSlice = ExtractState<S>>(
+    selector?: (state: ExtractState<S>) => StateSlice,
+    equalityFn = defaultEqualityFn,
+  ) => {
+    const store = useStoreApi();
+
     return useStoreWithEqualityFn(
       store,
       selector as (state: ExtractState<S>) => StateSlice,
       equalityFn,
     );
   };
 
-  const useStoreApi = () => {
-    const store = useContext(ZustandContext);
-    if (!store) {
-      throw new Error('Seems like you have not used zustand provider as an ancestor.');
-    }
-    return store as WithoutCallSignature<S>;
-  };
-
   return {
     Provider,
     useStore: useContextStore,