diff --git a/src/index.tsx b/src/index.tsx
index 81f0895..edcbfe0 100644
--- a/src/index.tsx
+++ b/src/index.tsx
@@ -6,13 +6,7 @@ import type { CustomRender } from './interface';
const defaultBlackList: string[] = [
'ConfigProvider',
- 'Drawer',
'Grid',
- 'Modal',
- 'Popconfirm',
- 'Popover',
- 'Tooltip',
- 'Tour',
];
const ComponentCustomizeRender: Record<
@@ -33,6 +27,40 @@ const ComponentCustomizeRender: Record<
Menu: (Menu) =>
,
QRCode: (QRCode) => ,
Tree: (Tree) => ,
+ Tag: (Tag) => (
+ <>
+ Tag
+ Tag
+ >
+ ),
+ Badge: (Badge: any) => (
+ <>
+
+
+ >
+ ),
+ Space: (Space: any) => (
+ <>
+
+
+
+
+ >
+ ),
+ Modal: (Modal: any) => (
+ <>
+
+
+ >
+ ),
+ message: (message: any) => {
+ const { _InternalPanelDoNotUseOrYouWillBeFired: PurePanel } = message;
+ return ;
+ },
+ notification: (notification: any) => {
+ const { _InternalPanelDoNotUseOrYouWillBeFired: PurePanel } = notification;
+ return ;
+ },
};
interface NodeProps {
@@ -48,7 +76,7 @@ const defaultNode = ({ excludes = [], includes }: NodeProps) => {
{components
.filter(
(name) =>
- ![...defaultBlackList, ...excludes].includes(name) && name[0] === name[0].toUpperCase(),
+ ![...defaultBlackList, ...excludes].includes(name) && (name[0] === name[0].toUpperCase() || ['notification', 'message'].includes(name)),
)
.map((compName) => {
const Comp = antd[compName];
diff --git a/tests/index.test.tsx b/tests/index.test.tsx
index 2e4960b..ec06760 100644
--- a/tests/index.test.tsx
+++ b/tests/index.test.tsx
@@ -69,6 +69,8 @@ describe('Static-Style-Extract', () => {
excludes: ['Card']
});
expect(cssText).toContain('.ant-btn');
+ expect(cssText).toContain('.ant-notification');
+ expect(cssText).toContain('.ant-message');
expect(cssText).not.toContain('.ant-card');
})
});