|
| 1 | +'use client'; |
| 2 | + |
| 3 | +import * as React from 'react'; |
| 4 | +import { ToolbarButton } from '@mui/x-charts/Toolbar'; |
| 5 | +import { useChartContext } from '@mui/x-charts/context/ChartProvider'; |
| 6 | +import { useIsHydrated } from '@mui/x-charts/hooks/useIsHydrated'; |
| 7 | +import materialSlots, { type ChartsSlots } from '@mui/x-charts/material'; |
| 8 | +import { useChartsLocalization } from '@mui/x-charts/hooks'; |
| 9 | +import { UseChartProExportSignature } from '../internals/plugins/useChartProExport'; |
| 10 | + |
| 11 | +type ChartsToolbarExportButtonSlots = Partial< |
| 12 | + Pick<ChartsSlots, 'baseMenu' | 'baseMenuItem' | 'baseTooltip' | 'exportIcon'> |
| 13 | +>; |
| 14 | + |
| 15 | +type ChartsToolbarExportButtonSlotProps = { |
| 16 | + [K in keyof Required<ChartsToolbarExportButtonSlots>]: React.ComponentProps< |
| 17 | + Required<ChartsToolbarExportButtonSlots>[K] |
| 18 | + >; |
| 19 | +}; |
| 20 | + |
| 21 | +interface ChartsToolbarExportButtonProps { |
| 22 | + slots?: ChartsToolbarExportButtonSlots; |
| 23 | + slotProps?: ChartsToolbarExportButtonSlotProps; |
| 24 | +} |
| 25 | + |
| 26 | +export function ChartsToolbarExportButton({ slots, slotProps }: ChartsToolbarExportButtonProps) { |
| 27 | + const buttonRef = React.useRef<HTMLButtonElement>(null); |
| 28 | + const [menuOpen, setMenuOpen] = React.useState(false); |
| 29 | + const handleClose = () => setMenuOpen(false); |
| 30 | + const { publicAPI } = useChartContext<[UseChartProExportSignature]>(); |
| 31 | + const isHydrated = useIsHydrated(); |
| 32 | + const ExportIcon = slots?.exportIcon ?? materialSlots.exportIcon; |
| 33 | + const Tooltip = slots?.baseTooltip ?? materialSlots.baseTooltip; |
| 34 | + const Menu = slots?.baseMenu ?? materialSlots.baseMenu; |
| 35 | + const MenuItem = slots?.baseMenuItem ?? materialSlots.baseMenuItem; |
| 36 | + const { localeText } = useChartsLocalization(); |
| 37 | + const canExport = publicAPI.exportAsPrint || publicAPI.exportAsImage; |
| 38 | + |
| 39 | + if (!canExport) { |
| 40 | + return null; |
| 41 | + } |
| 42 | + |
| 43 | + return ( |
| 44 | + <React.Fragment> |
| 45 | + <Tooltip title={localeText.export}> |
| 46 | + <ToolbarButton ref={buttonRef} onClick={() => setMenuOpen(true)}> |
| 47 | + <ExportIcon {...slotProps?.exportIcon} /> |
| 48 | + </ToolbarButton> |
| 49 | + </Tooltip> |
| 50 | + {isHydrated && ( |
| 51 | + <Menu anchorEl={() => buttonRef.current} open={menuOpen} onClose={handleClose}> |
| 52 | + {publicAPI.exportAsPrint ? ( |
| 53 | + <MenuItem |
| 54 | + onClick={() => { |
| 55 | + publicAPI.exportAsPrint(); |
| 56 | + handleClose(); |
| 57 | + }} |
| 58 | + > |
| 59 | + {localeText.print} |
| 60 | + </MenuItem> |
| 61 | + ) : null} |
| 62 | + {publicAPI.exportAsImage ? ( |
| 63 | + <MenuItem |
| 64 | + onClick={() => { |
| 65 | + publicAPI.exportAsImage(); |
| 66 | + handleClose(); |
| 67 | + }} |
| 68 | + > |
| 69 | + {localeText.exportAsImage} |
| 70 | + </MenuItem> |
| 71 | + ) : null} |
| 72 | + </Menu> |
| 73 | + )} |
| 74 | + </React.Fragment> |
| 75 | + ); |
| 76 | +} |
0 commit comments