✨ feat: 添加 styled api，支持 styled 方法响应自定义 token

ant-design · Feb 17, 2023 · a152c33 · a152c33
1 parent 01cbbe3
commit a152c33
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/factories/createThemeProvider/index.tsx b/src/factories/createThemeProvider/index.tsx
@@ -1,5 +1,6 @@
 import { ReactElement } from 'react';
 
+import { createUseTheme } from '@/factories/createUseTheme';
 import { UseTheme } from '@/types';
 import AntdProvider from './AntdProvider';
 import ThemeSwitcher from './ThemeSwitcher';
@@ -28,9 +29,12 @@ export const createThemeProvider =
     defaultAppearance,
     onAppearanceChange,
     themeMode,
-    StyledThemeProvider = defaultStyledThemeProvider,
-    useTheme = defaultUseTheme,
+    styled,
   }) => {
+    const StyledThemeProvider = styled?.ThemeProvider || defaultStyledThemeProvider;
+
+    const useTheme = createUseTheme(styled?.useTheme || defaultUseTheme);
+
     return (
       <ThemeSwitcher
         themeMode={themeMode}

diff --git a/src/factories/createThemeProvider/type.ts b/src/factories/createThemeProvider/type.ts
@@ -25,6 +25,10 @@ export interface ThemeProviderProps<T, S = Record<string, string>> {
    * 自定义 Stylish
    */
   customStylish?: GetCustomStylish<S>;
+  styled?: {
+    ThemeProvider?: StyledThemeProvider;
+    useTheme?: () => any;
+  };
   StyledThemeProvider?: StyledThemeProvider;
   useTheme?: () => Theme;