fix: types

This commit is contained in:
abearxiong 2025-03-26 14:06:10 +08:00
parent 152fda350e
commit 4d7af80d15

View File

@ -2,7 +2,7 @@ import { createContext, useContext, useEffect, useMemo, useState } from 'react';
import { StateCreator } from '../../src/store'; import { StateCreator } from '../../src/store';
import { shallow, useShallow } from 'zustand/shallow'; import { shallow, useShallow } from 'zustand/shallow';
import { useContextKey } from '../../src/web-context'; import { useContextKey } from '../../src/web-context';
import { UseBoundStore, StoreApi } from 'zustand';
export const StoreContext = createContext<any>(null); export const StoreContext = createContext<any>(null);
export const initStoreFn: StateCreator<any, [], [], any> = (set, get, store) => { export const initStoreFn: StateCreator<any, [], [], any> = (set, get, store) => {
@ -38,7 +38,9 @@ export const StoreContextProvider = ({
return <StoreContext.Provider value={state}>{children}</StoreContext.Provider>; return <StoreContext.Provider value={state}>{children}</StoreContext.Provider>;
}; };
export const useStore = (selector?: any) => { type SimpleObject = Record<string, any>;
export const useStore = (selector: any): any => {
const store = useContext(StoreContext); const store = useContext(StoreContext);
const allState = store.getState(); const allState = store.getState();
const selectedState = selector ? selector(allState) : allState; const selectedState = selector ? selector(allState) : allState;
@ -56,3 +58,5 @@ export const useStore = (selector?: any) => {
return state; return state;
}; };
export type BoundStore<T> = UseBoundStore<StoreApi<T>>;