✨ feat(component-asset): createUseAssetStore 方法支持导出AssetProvider

ant-design · Jun 14, 2023 · d33a8b3 · d33a8b3
1 parent af0b2e5
commit d33a8b3
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/ComponentAsset/store/index.ts b/src/ComponentAsset/store/index.ts
@@ -3,6 +3,7 @@ import { UseContextStore, createContext, optionalDevtools } from 'zustand-utils'
 import { DevtoolsOptions } from 'zustand/middleware';
 
 import { PublicProEditorStore } from '@/ProEditor/store';
+import { ReactNode } from 'react';
 
 export interface AssetStoreOptions<T = any> {
   devtools?: boolean | DevtoolsOptions;
@@ -46,4 +47,11 @@ export type WithoutCallSignature<T> = {
 export const createUseAssetStore = <T>(): {
   useStore: UseContextStore<StoreApi<T & PublicProEditorStore>>;
   useStoreApi: () => WithoutCallSignature<StoreApi<T & PublicProEditorStore>>;
-} => ({ useStore, useStoreApi });
+  AssetProvider: ({
+    createStore,
+    children,
+  }: {
+    createStore: () => StoreApi<T>;
+    children: any;
+  }) => ReactNode;
+} => ({ useStore, useStoreApi, AssetProvider: Provider });