345 lines
11 KiB
TypeScript
345 lines
11 KiB
TypeScript
'use client';
|
|
|
|
import * as React from 'react';
|
|
import type { LegendPayload, TooltipPayloadEntry } from 'recharts';
|
|
import * as RechartsPrimitive from 'recharts';
|
|
|
|
import { cn } from '@/lib/utils.js';
|
|
|
|
type VerticalAlignmentType = 'top' | 'bottom' | 'middle';
|
|
type ValueType = number | string | ReadonlyArray<number | string>;
|
|
type NameType = number | string;
|
|
|
|
// Format: { THEME_NAME: CSS_SELECTOR }
|
|
const THEMES = { light: '', dark: '.dark' } as const;
|
|
|
|
export type ChartConfig = {
|
|
[k in string]: {
|
|
label?: React.ReactNode;
|
|
icon?: React.ComponentType;
|
|
} & ({ color?: string; theme?: never } | { color?: never; theme: Record<keyof typeof THEMES, string> });
|
|
};
|
|
|
|
type ChartContextProps = {
|
|
config: ChartConfig;
|
|
};
|
|
|
|
const ChartContext = React.createContext<ChartContextProps | null>(null);
|
|
|
|
function useChart() {
|
|
const context = React.useContext(ChartContext);
|
|
|
|
if (!context) {
|
|
throw new Error('useChart must be used within a <ChartContainer />');
|
|
}
|
|
|
|
return context;
|
|
}
|
|
|
|
function ChartContainer({
|
|
id,
|
|
className,
|
|
children,
|
|
config,
|
|
...props
|
|
}: React.ComponentProps<'div'> & {
|
|
config: ChartConfig;
|
|
children: React.ComponentProps<typeof RechartsPrimitive.ResponsiveContainer>['children'];
|
|
}) {
|
|
const uniqueId = React.useId();
|
|
const chartId = `chart-${id || uniqueId.replace(/:/g, '')}`;
|
|
|
|
return (
|
|
<ChartContext.Provider value={{ config }}>
|
|
<div
|
|
data-slot="chart"
|
|
data-chart={chartId}
|
|
className={cn(
|
|
"flex aspect-video justify-center text-xs [&_.recharts-cartesian-axis-tick_text]:fill-muted-foreground [&_.recharts-cartesian-grid_line[stroke='#ccc']]:stroke-border/50 [&_.recharts-curve.recharts-tooltip-cursor]:stroke-border [&_.recharts-dot[stroke='#fff']]:stroke-transparent [&_.recharts-layer]:outline-hidden [&_.recharts-polar-grid_[stroke='#ccc']]:stroke-border [&_.recharts-radial-bar-background-sector]:fill-muted [&_.recharts-rectangle.recharts-tooltip-cursor]:fill-muted [&_.recharts-reference-line_[stroke='#ccc']]:stroke-border [&_.recharts-sector]:outline-hidden [&_.recharts-sector[stroke='#fff']]:stroke-transparent [&_.recharts-surface]:outline-hidden",
|
|
className,
|
|
)}
|
|
{...props}>
|
|
<ChartStyle id={chartId} config={config} />
|
|
<RechartsPrimitive.ResponsiveContainer>{children}</RechartsPrimitive.ResponsiveContainer>
|
|
</div>
|
|
</ChartContext.Provider>
|
|
);
|
|
}
|
|
|
|
const ChartStyle = ({ id, config }: { id: string; config: ChartConfig }) => {
|
|
const colorConfig = Object.entries(config).filter(([, config]) => config.theme || config.color);
|
|
|
|
if (!colorConfig.length) {
|
|
return null;
|
|
}
|
|
|
|
return (
|
|
<style
|
|
dangerouslySetInnerHTML={{
|
|
__html: Object.entries(THEMES)
|
|
.map(
|
|
([theme, prefix]) => `
|
|
${prefix} [data-chart=${id}] {
|
|
${colorConfig
|
|
.map(([key, itemConfig]) => {
|
|
const themeKey = theme === 'light' || theme === 'dark' ? theme : undefined;
|
|
const color = (itemConfig.theme && themeKey ? itemConfig.theme[themeKey] : undefined) || itemConfig.color;
|
|
return color ? ` --color-${key}: ${color};` : null;
|
|
})
|
|
.join('\n')}
|
|
}
|
|
`,
|
|
)
|
|
.join('\n'),
|
|
}}
|
|
/>
|
|
);
|
|
};
|
|
|
|
const ChartTooltip = RechartsPrimitive.Tooltip;
|
|
|
|
function cssVarsToStyle(vars: Record<string, string | undefined>): React.CSSProperties {
|
|
const style: Record<string, string | undefined> = {};
|
|
for (const [key, val] of Object.entries(vars)) {
|
|
style[key] = val;
|
|
}
|
|
return style;
|
|
}
|
|
|
|
function dataKeyToString(dataKey: unknown): string {
|
|
if (typeof dataKey === 'string' || typeof dataKey === 'number') {
|
|
return String(dataKey);
|
|
}
|
|
return 'value';
|
|
}
|
|
|
|
function getPayloadFill(item: TooltipPayloadEntry<ValueType, NameType>): string | undefined {
|
|
const p: unknown = item.payload;
|
|
if (typeof p === 'object' && p !== null && 'fill' in p) {
|
|
const fill = (p as { fill: unknown }).fill;
|
|
return typeof fill === 'string' ? fill : undefined;
|
|
}
|
|
return undefined;
|
|
}
|
|
|
|
function ChartTooltipContent({
|
|
active,
|
|
payload,
|
|
className,
|
|
indicator = 'dot',
|
|
hideLabel = false,
|
|
hideIndicator = false,
|
|
label,
|
|
labelFormatter,
|
|
labelClassName,
|
|
formatter,
|
|
color,
|
|
nameKey,
|
|
labelKey,
|
|
}: {
|
|
active?: boolean;
|
|
payload?: ReadonlyArray<TooltipPayloadEntry<ValueType, NameType>>;
|
|
label?: string | number;
|
|
labelFormatter?: (
|
|
label: React.ReactNode,
|
|
payload: ReadonlyArray<TooltipPayloadEntry<ValueType, NameType>>,
|
|
) => React.ReactNode;
|
|
formatter?: (
|
|
value: ValueType,
|
|
name: NameType,
|
|
item: TooltipPayloadEntry<ValueType, NameType>,
|
|
index: number,
|
|
payload: ReadonlyArray<TooltipPayloadEntry<ValueType, NameType>>,
|
|
) => React.ReactNode;
|
|
} & React.ComponentProps<'div'> & {
|
|
hideLabel?: boolean;
|
|
hideIndicator?: boolean;
|
|
indicator?: 'line' | 'dot' | 'dashed';
|
|
nameKey?: string;
|
|
labelKey?: string;
|
|
labelClassName?: string;
|
|
color?: string;
|
|
}) {
|
|
const { config } = useChart();
|
|
|
|
const tooltipLabel = React.useMemo(() => {
|
|
if (hideLabel || !payload?.length) {
|
|
return null;
|
|
}
|
|
|
|
const [item] = payload;
|
|
const key = labelKey || dataKeyToString(item?.dataKey) || String(item?.name) || 'value';
|
|
const itemConfig = getPayloadConfigFromPayload(config, item, key);
|
|
const value = !labelKey && typeof label === 'string' ? config[label]?.label || label : itemConfig?.label;
|
|
|
|
if (labelFormatter) {
|
|
return <div className={cn('font-medium', labelClassName)}>{labelFormatter(value, payload)}</div>;
|
|
}
|
|
|
|
if (!value) {
|
|
return null;
|
|
}
|
|
|
|
return <div className={cn('font-medium', labelClassName)}>{value}</div>;
|
|
}, [label, labelFormatter, payload, hideLabel, labelClassName, config, labelKey]);
|
|
|
|
if (!active || !payload?.length) {
|
|
return null;
|
|
}
|
|
|
|
const nestLabel = payload.length === 1 && indicator !== 'dot';
|
|
|
|
return (
|
|
<div
|
|
className={cn(
|
|
'grid min-w-[8rem] items-start gap-1.5 rounded-lg border border-border/50 bg-background px-2.5 py-1.5 text-xs shadow-xl',
|
|
className,
|
|
)}>
|
|
{!nestLabel ? tooltipLabel : null}
|
|
<div className="grid gap-1.5">
|
|
{payload
|
|
.filter(item => item.type !== 'none')
|
|
.map((item, index) => {
|
|
const key = nameKey || String(item.name) || dataKeyToString(item.dataKey) || 'value';
|
|
const itemConfig = getPayloadConfigFromPayload(config, item, key);
|
|
const indicatorColor = color || getPayloadFill(item) || item.color;
|
|
|
|
return (
|
|
<div
|
|
key={dataKeyToString(item.dataKey)}
|
|
className={cn(
|
|
'flex w-full flex-wrap items-stretch gap-2 [&>svg]:h-2.5 [&>svg]:w-2.5 [&>svg]:text-muted-foreground',
|
|
indicator === 'dot' && 'items-center',
|
|
)}>
|
|
{formatter && item?.value !== undefined && item.name ? (
|
|
formatter(item.value, item.name, item, index, payload)
|
|
) : (
|
|
<>
|
|
{itemConfig?.icon ? (
|
|
<itemConfig.icon />
|
|
) : (
|
|
!hideIndicator && (
|
|
<div
|
|
className={cn('shrink-0 rounded-[2px] border-(--color-border) bg-(--color-bg)', {
|
|
'h-2.5 w-2.5': indicator === 'dot',
|
|
'w-1': indicator === 'line',
|
|
'w-0 border-[1.5px] border-dashed bg-transparent': indicator === 'dashed',
|
|
'my-0.5': nestLabel && indicator === 'dashed',
|
|
})}
|
|
style={cssVarsToStyle({
|
|
'--color-bg': indicatorColor,
|
|
'--color-border': indicatorColor,
|
|
})}
|
|
/>
|
|
)
|
|
)}
|
|
<div
|
|
className={cn(
|
|
'flex flex-1 justify-between leading-none',
|
|
nestLabel ? 'items-end' : 'items-center',
|
|
)}>
|
|
<div className="grid gap-1.5">
|
|
{nestLabel ? tooltipLabel : null}
|
|
<span className="text-muted-foreground">{itemConfig?.label || item.name}</span>
|
|
</div>
|
|
{item.value && (
|
|
<span className="font-mono font-medium text-foreground tabular-nums">
|
|
{item.value.toLocaleString()}
|
|
</span>
|
|
)}
|
|
</div>
|
|
</>
|
|
)}
|
|
</div>
|
|
);
|
|
})}
|
|
</div>
|
|
</div>
|
|
);
|
|
}
|
|
|
|
const ChartLegend = RechartsPrimitive.Legend;
|
|
|
|
function ChartLegendContent({
|
|
className,
|
|
hideIcon = false,
|
|
payload,
|
|
verticalAlign = 'bottom',
|
|
nameKey,
|
|
}: React.ComponentProps<'div'> & {
|
|
hideIcon?: boolean;
|
|
nameKey?: string;
|
|
payload?: ReadonlyArray<LegendPayload>;
|
|
verticalAlign?: VerticalAlignmentType;
|
|
}) {
|
|
const { config } = useChart();
|
|
|
|
if (!payload?.length) {
|
|
return null;
|
|
}
|
|
|
|
return (
|
|
<div className={cn('flex items-center justify-center gap-4', verticalAlign === 'top' ? 'pb-3' : 'pt-3', className)}>
|
|
{payload
|
|
.filter(item => item.type !== 'none')
|
|
.map(item => {
|
|
const key = nameKey || dataKeyToString(item.dataKey) || 'value';
|
|
const itemConfig = getPayloadConfigFromPayload(config, item, key);
|
|
|
|
return (
|
|
<div
|
|
key={item.value}
|
|
className={cn('flex items-center gap-1.5 [&>svg]:h-3 [&>svg]:w-3 [&>svg]:text-muted-foreground')}>
|
|
{itemConfig?.icon && !hideIcon ? (
|
|
<itemConfig.icon />
|
|
) : (
|
|
<div
|
|
className="h-2 w-2 shrink-0 rounded-[2px]"
|
|
style={{
|
|
backgroundColor: item.color,
|
|
}}
|
|
/>
|
|
)}
|
|
{itemConfig?.label}
|
|
</div>
|
|
);
|
|
})}
|
|
</div>
|
|
);
|
|
}
|
|
|
|
function getRecordValue(obj: unknown, key: string): unknown {
|
|
if (typeof obj === 'object' && obj !== null && key in obj) {
|
|
return Object.getOwnPropertyDescriptor(obj, key)?.value;
|
|
}
|
|
return undefined;
|
|
}
|
|
|
|
// Helper to extract item config from a payload.
|
|
function getPayloadConfigFromPayload(config: ChartConfig, payload: unknown, key: string) {
|
|
if (typeof payload !== 'object' || payload === null) {
|
|
return undefined;
|
|
}
|
|
|
|
const payloadPayload =
|
|
'payload' in payload && typeof payload.payload === 'object' && payload.payload !== null
|
|
? payload.payload
|
|
: undefined;
|
|
|
|
let configLabelKey: string = key;
|
|
|
|
const payloadValue = getRecordValue(payload, key);
|
|
if (typeof payloadValue === 'string') {
|
|
configLabelKey = payloadValue;
|
|
} else {
|
|
const nestedValue = getRecordValue(payloadPayload, key);
|
|
if (typeof nestedValue === 'string') {
|
|
configLabelKey = nestedValue;
|
|
}
|
|
}
|
|
|
|
return configLabelKey in config ? config[configLabelKey] : config[key];
|
|
}
|
|
|
|
export { ChartContainer, ChartLegend, ChartLegendContent, ChartStyle, ChartTooltip, ChartTooltipContent };
|