From 4d7af80d15827fc61e3ba3838e6136b941586ff4 Mon Sep 17 00:00:00 2001
From: abearxiong <xiongxiao1012@outlook.com>
Date: Wed, 26 Mar 2025 14:06:10 +0800
Subject: [PATCH] fix: types

---
 react/src/Store.tsx | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/react/src/Store.tsx b/react/src/Store.tsx
index 3a62911..59d1f19 100644
--- a/react/src/Store.tsx
+++ b/react/src/Store.tsx
@@ -2,7 +2,7 @@ import { createContext, useContext, useEffect, useMemo, useState } from 'react';
 import { StateCreator } from '../../src/store';
 import { shallow, useShallow } from 'zustand/shallow';
 import { useContextKey } from '../../src/web-context';
-
+import { UseBoundStore, StoreApi } from 'zustand';
 export const StoreContext = createContext<any>(null);
 
 export const initStoreFn: StateCreator<any, [], [], any> = (set, get, store) => {
@@ -38,7 +38,9 @@ export const StoreContextProvider = ({
   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 allState = store.getState();
   const selectedState = selector ? selector(allState) : allState;
@@ -56,3 +58,5 @@ export const useStore = (selector?: any) => {
 
   return state;
 };
+
+export type BoundStore<T> = UseBoundStore<StoreApi<T>>;