import { Badge, Icon, Spin, Tabs } from 'antd'; import React, { Component } from 'react'; import classNames from 'classnames'; import NoticeList, { NoticeIconTabProps } from './NoticeList'; import HeaderDropdown from '../HeaderDropdown'; import styles from './index.less'; const { TabPane } = Tabs; export interface NoticeIconData { avatar?: string | React.ReactNode; title?: React.ReactNode; description?: React.ReactNode; datetime?: React.ReactNode; extra?: React.ReactNode; style?: React.CSSProperties; key?: string | number; read?: boolean; } export interface NoticeIconProps { count?: number; bell?: React.ReactNode; className?: string; loading?: boolean; onClear?: (tabName: string, tabKey: string) => void; onItemClick?: (item: NoticeIconData, tabProps: NoticeIconTabProps) => void; onViewMore?: (tabProps: NoticeIconTabProps, e: MouseEvent) => void; onTabChange?: (tabTile: string) => void; style?: React.CSSProperties; onPopupVisibleChange?: (visible: boolean) => void; popupVisible?: boolean; clearText?: string; viewMoreText?: string; clearClose?: boolean; children: React.ReactElement[]; } export default class NoticeIcon extends Component { public static Tab: typeof NoticeList = NoticeList; static defaultProps = { onItemClick: (): void => {}, onPopupVisibleChange: (): void => {}, onTabChange: (): void => {}, onClear: (): void => {}, onViewMore: (): void => {}, loading: false, clearClose: false, emptyImage: 'https://gw.alipayobjects.com/zos/rmsportal/wAhyIChODzsoKIOBHcBk.svg', }; state = { visible: false, }; onItemClick = (item: NoticeIconData, tabProps: NoticeIconTabProps): void => { const { onItemClick } = this.props; if (onItemClick) { onItemClick(item, tabProps); } }; onClear = (name: string, key: string): void => { const { onClear } = this.props; if (onClear) { onClear(name, key); } }; onTabChange = (tabType: string): void => { const { onTabChange } = this.props; if (onTabChange) { onTabChange(tabType); } }; onViewMore = (tabProps: NoticeIconTabProps, event: MouseEvent): void => { const { onViewMore } = this.props; if (onViewMore) { onViewMore(tabProps, event); } }; getNotificationBox(): React.ReactNode { const { children, loading, clearText, viewMoreText } = this.props; if (!children) { return null; } const panes = React.Children.map( children, (child: React.ReactElement): React.ReactNode => { if (!child) { return null; } const { list, title, count, tabKey, showClear, showViewMore } = child.props; const len = list && list.length ? list.length : 0; const msgCount = count || count === 0 ? count : len; const tabTitle: string = msgCount > 0 ? `${title} (${msgCount})` : title; return ( this.onClear(title, tabKey)} onClick={(item): void => this.onItemClick(item, child.props)} onViewMore={(event): void => this.onViewMore(child.props, event)} showClear={showClear} showViewMore={showViewMore} title={title} {...child.props} /> ); }, ); return ( {panes} ); } handleVisibleChange = (visible: boolean): void => { const { onPopupVisibleChange } = this.props; this.setState({ visible }); if (onPopupVisibleChange) { onPopupVisibleChange(visible); } }; render(): React.ReactNode { const { className, count, popupVisible, bell } = this.props; const { visible } = this.state; const noticeButtonClass = classNames(className, styles.noticeButton); const notificationBox = this.getNotificationBox(); const NoticeBellIcon = bell || ; const trigger = ( {NoticeBellIcon} ); if (!notificationBox) { return trigger; } const popoverProps: { visible?: boolean; } = {}; if ('popupVisible' in this.props) { popoverProps.visible = popupVisible; } return ( {trigger} ); } }