/* @jsxImportSource @opentui/react */ import { useTheme } from "@/hooks/use-theme"; export type LineChartDataPoint = | number | { label?: string; value: number; }; export interface LineChartProps { data: LineChartDataPoint[]; width?: number; height?: number; title?: string; color?: string; showAxes?: boolean; } const getValue = (d: LineChartDataPoint): number => typeof d === "number" ? d : d.value; const getLabel = (d: LineChartDataPoint): string => typeof d === "number" ? "" : (d.label ?? ""); const normalize = ( value: number, min: number, max: number, rows: number ): number => { if (max === min) { return Math.floor(rows / 2); } return Math.round(((value - min) / (max - min)) * (rows - 1)); }; const PLOT_CHAR = "●"; const _CONNECT_H = "─"; const CONNECT_UP = "╱"; const CONNECT_DOWN = "╲"; const _CONNECT_FLAT = "─"; const AXIS_V = "│"; const AXIS_H = "─"; const AXIS_CORNER = "└"; const AXIS_TICK_V = "┤"; const _AXIS_TICK_H = "┬"; export const LineChart = ({ data, width = 40, height = 10, title, color, showAxes = true, }: LineChartProps) => { const theme = useTheme(); const resolvedColor = color ?? theme.colors.primary; if (data.length === 0) { return No data; } const values = data.map(getValue); const minVal = Math.min(...values); const maxVal = Math.max(...values); const yAxisWidth = showAxes ? String(Math.round(maxVal)).length + 2 : 0; const chartWidth = Math.max(4, width - yAxisWidth); const numPoints = data.length; const sampledIndices = Array.from({ length: chartWidth }, (_, i) => Math.round((i / (chartWidth - 1)) * (numPoints - 1)) ); const sampledValues = sampledIndices.map((si) => values[si] ?? 0); const sampledData = sampledIndices.map((si) => data[si]); const grid: string[][] = Array.from({ length: height }, () => Array.from({ length: chartWidth }, () => " ") ); const normalizedRows = sampledValues.map( (v) => height - 1 - normalize(v, minVal, maxVal, height) ); for (let col = 0; col < chartWidth; col += 1) { const row = normalizedRows[col]; grid[row][col] = PLOT_CHAR; if (col < chartWidth - 1) { const nextRow = normalizedRows[col + 1] ?? row; if (nextRow < row) { let r = row - 1; while (r > nextRow) { grid[r][col] = AXIS_V; r -= 1; } if (grid[nextRow][col] === " ") { grid[nextRow][col] = CONNECT_UP; } } else { let r = row + 1; while (r < nextRow) { grid[r][col] = AXIS_V; r += 1; } if (grid[nextRow][col] === " ") { grid[nextRow][col] = CONNECT_DOWN; } } } } const yLabels = Array.from({ length: height }, (_, i) => { const rowVal = minVal + ((height - 1 - i) / (height - 1)) * (maxVal - minVal); if (i === 0 || i === Math.floor(height / 2) || i === height - 1) { return String(Math.round(rowVal)); } return ""; }); return ( {title && ( {title} )} {grid.map((row, rowIdx) => ( {showAxes && ( {`${String(yLabels[rowIdx] ?? "").padStart(yAxisWidth - 1)}${AXIS_TICK_V}`} )} {row.map((cell, colIdx) => { const isPlot = cell === PLOT_CHAR; return ( {cell} ); })} ))} {showAxes && ( {`${" ".repeat(yAxisWidth - 1)}${AXIS_CORNER}${AXIS_H.repeat(chartWidth)}`} )} {showAxes && ( {" ".repeat(yAxisWidth)} {sampledData.map((d, idx) => { const lbl = getLabel(d); if ( idx === 0 || idx === Math.floor(chartWidth / 2) || idx === chartWidth - 1 ) { return ( {lbl || String(idx)} ); } return ; })} )} ); };