Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 3 additions & 7 deletions src/drawer/index.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ import classNames from 'classnames';
import { omit } from 'lodash-es';
import RcDrawer, { DrawerProps as AntdDrawerProps } from 'rc-drawer';

import { isAlertObjectProps } from '../utils';
import motionProps from './motion';
import './style.scss';

Expand Down Expand Up @@ -67,11 +68,6 @@ const getWidthFromSize = (size: DrawerProps['size']) => {
return `${(DrawerSize.Default / 1440) * 100}%`;
};

const isValidBanner = (banner: DrawerProps['banner']): banner is AlertProps['message'] => {
if (typeof banner === 'object') return React.isValidElement(banner);
return true;
};

const Drawer = <T extends readOnlyTab>(props: DrawerProps<T>) => {
const drawerPrefixCls = 'dtc-drawer';

Expand Down Expand Up @@ -148,9 +144,9 @@ const Drawer = <T extends readOnlyTab>(props: DrawerProps<T>) => {
)}
{banner && (
<Alert
message={isValidBanner(banner) ? banner : (banner as any).message}
message={isAlertObjectProps(banner) ? banner.message : banner}
banner
{...(isValidBanner(banner) ? {} : omit(banner, 'message'))}
{...(isAlertObjectProps(banner) ? omit(banner, 'message') : {})}
/>
)}
{isTabMode(props) && (
Expand Down
10 changes: 3 additions & 7 deletions src/modal/modal.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ import { omit } from 'lodash-es';

import Float, { type IFloatProps } from '../float';
import useMergeOption, { type MergeOption } from '../useMergeOption';
import { isAlertObjectProps } from '../utils';
import Handler from './handle';
import './index.scss';

Expand All @@ -29,11 +30,6 @@ const getWidthFromSize = (size: IModalProps['size']) => {
return 520;
};

const isValidBanner = (banner: IModalProps['banner']): banner is AlertProps['message'] => {
if (typeof banner === 'object') return React.isValidElement(banner);
return true;
};

export default function InternalModal({
bodyStyle,
banner,
Expand Down Expand Up @@ -137,9 +133,9 @@ export default function InternalModal({
{banner && (
<Alert
className="dtc-modal-alert"
message={isValidBanner(banner) ? banner : banner.message}
message={isAlertObjectProps(banner) ? banner.message : banner}
banner
{...(isValidBanner(banner) ? {} : omit(banner, 'message'))}
{...(isAlertObjectProps(banner) ? omit(banner, 'message') : {})}
/>
)}
<section className="dtc-modal-body">{children}</section>
Expand Down
13 changes: 12 additions & 1 deletion src/utils/index.ts
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
import React from 'react';
import { TooltipProps } from 'antd';
import { AlertProps, TooltipProps } from 'antd';

export type LabelTooltipType = TooltipProps | TooltipProps['title'];

Expand All @@ -11,3 +11,14 @@ export function toTooltipProps(tooltip: LabelTooltipType): TooltipProps | null {
title: tooltip,
};
}

type BannerPropType = AlertProps['message'] | Omit<AlertProps, 'banner'>;

export function isAlertObjectProps(banner: BannerPropType): banner is Omit<AlertProps, 'banner'> {
return (
typeof banner === 'object' &&
banner !== null &&
!React.isValidElement(banner) &&
'message' in banner
);
}
Loading