@astryxdesign/lab 0.1.4-canary.37c19d3 → 0.1.4-canary.39ffd21

package/src/ChartV2/ChartAxis.tsx

@@ -0,0 +1,209 @@
+// Copyright (c) Meta Platforms, Inc. and affiliates.
+
+/**
+ * @file ChartAxis.tsx (v2)
+ * @output Renders an axis (top, right, bottom, left) using the chart's scales
+ * @position Child of Chart v2; reads scales from chart context
+ *
+ * The axis edge line and tick marks are independently toggleable so callers
+ * can compose: line-only, line + ticks, or line + ticks + grid (paired with
+ * `<ChartGrid>`). To stay consistent with most charts, the bottom axis
+ * draws its edge line by default and other positions don't.
+ *
+ * Ticks use CSS transitions for smooth sliding during streaming updates.
+ */
+
+import {useMemo} from 'react';
+import * as stylex from '@stylexjs/stylex';
+import {colorVars} from '@astryxdesign/core/theme/tokens.stylex';
+import {useChart} from '../Chart/ChartContext';
+import {isBandScale} from '../Chart/utils';
+import type {ScaleLinear, ScaleTime} from 'd3-scale';
+
+export interface ChartAxisProps {
+  /** Which edge to render the axis on */
+  position: 'top' | 'right' | 'bottom' | 'left';
+  /** Number of ticks (approximate — d3 decides final count) */
+  tickCount?: number;
+  /** Maximum number of tick labels to show. Labels are evenly skipped when exceeded. */
+  maxTicks?: number;
+  /** Custom tick formatter */
+  tickFormat?: (value: unknown) => string;
+  /** Truncate labels to this many characters (appends "\u2026"). */
+  truncate?: number;
+  /** Enable smooth transitions for streaming (default: true) */
+  animated?: boolean;
+  /**
+   * Show the axis edge line. Defaults to `true` for `position="bottom"`,
+   * `false` otherwise. Forced to `true` when `showTicks` is enabled.
+   * Pair with `<ChartGrid vertical />` and pass `showAxisLine` on the
+   * corresponding side axis to keep the visual edge grounded.
+   */
+  showAxisLine?: boolean;
+  /** Show perpendicular tick marks at each tick label. Default: `false`. */
+  showTicks?: boolean;
+}
+
+const styles = stylex.create({
+  axisLine: {
+    stroke: colorVars['--color-text-primary'],
+    strokeWidth: 1,
+  },
+  tickMark: {
+    stroke: colorVars['--color-text-primary'],
+    strokeWidth: 1,
+  },
+});
+
+const TICK_SIZE = 6;
+
+/**
+ * Chart axis. Always draws tick labels; the axis edge line and tick marks
+ * are independently toggleable. Pair with `<ChartGrid>` for grid lines.
+ *
+ * @example
+ * ```
+ * <ChartAxis position="bottom" />               // edge line + labels
+ * <ChartAxis position="left" />                 // labels only (no edge line)
+ * <ChartAxis position="left" showAxisLine showTicks />
+ * ```
+ */
+export function ChartAxis({
+  position,
+  tickCount = 5,
+  maxTicks,
+  tickFormat,
+  truncate,
+  animated = true,
+  showAxisLine = position === 'bottom',
+  showTicks = false,
+}: ChartAxisProps) {
+  // Tick marks need an axis line to anchor against; force it on when ticks
+  // are enabled so the two visuals stay coherent.
+  const renderAxisLine = showAxisLine || showTicks;
+  const {width, height, xScale, yScale} = useChart();
+
+  const isHorizontal = position === 'top' || position === 'bottom';
+  const scale = isHorizontal ? xScale : yScale;
+
+  const ticks = useMemo(() => {
+    let allTicks: {value: unknown; offset: number}[];
+    if (isBandScale(scale)) {
+      allTicks = scale.domain().map(d => ({
+        value: d,
+        offset: (scale(d) ?? 0) + scale.bandwidth() / 2,
+      }));
+    } else {
+      const linearScale = scale as
+        | ScaleLinear<number, number>
+        | ScaleTime<number, number>;
+      allTicks = linearScale.ticks(tickCount).map(d => ({
+        value: d,
+        offset: linearScale(d as number & Date),
+      }));
+    }
+
+    // Auto-skip: show every Nth label if maxTicks is exceeded
+    if (maxTicks && allTicks.length > maxTicks) {
+      const step = Math.ceil(allTicks.length / maxTicks);
+      allTicks = allTicks.filter((_, i) => i % step === 0);
+    }
+
+    return allTicks;
+  }, [scale, tickCount, maxTicks]);
+
+  const transform =
+    position === 'bottom'
+      ? `translate(0,${height})`
+      : position === 'right'
+        ? `translate(${width},0)`
+        : undefined;
+
+  // For the bottom axis, draw the axis line at y=0 when the domain spans
+  // negative values (so the axis line still represents zero, not the chart edge).
+  const axisLineY = (() => {
+    if (position !== 'bottom') {
+      return 0;
+    }
+    const domain = yScale.domain();
+    if (domain[0] < 0 && domain[1] > 0) {
+      return yScale(0) - height;
+    }
+    return 0;
+  })();
+
+  const baseFormat = tickFormat ?? String;
+  const format = truncate
+    ? (value: unknown) => {
+        const str = baseFormat(value);
+        return str.length > truncate ? str.slice(0, truncate) + '\u2026' : str;
+      }
+    : baseFormat;
+
+  const tickTransition = animated
+    ? 'transform 150ms linear, opacity 150ms ease'
+    : undefined;
+
+  return (
+    <g transform={transform}>
+      {renderAxisLine && (
+        <line
+          x1={0}
+          x2={isHorizontal ? width : 0}
+          y1={isHorizontal ? axisLineY : 0}
+          y2={isHorizontal ? axisLineY : height}
+          {...stylex.props(styles.axisLine)}
+        />
+      )}
+      {ticks.map(({value, offset}) => {
+        const label = format(value);
+        const isVisible = isHorizontal
+          ? offset >= -10 && offset <= width + 10
+          : offset >= -10 && offset <= height + 10;
+
+        if (isHorizontal) {
+          const y = position === 'bottom' ? TICK_SIZE : -TICK_SIZE;
+          return (
+            <g
+              key={label}
+              style={{
+                transform: `translateX(${offset}px)`,
+                transition: tickTransition,
+                opacity: isVisible ? 1 : 0,
+              }}>
+              {showTicks && <line y2={y} {...stylex.props(styles.tickMark)} />}
+              <text
+                y={position === 'bottom' ? TICK_SIZE + 12 : -(TICK_SIZE + 4)}
+                textAnchor="middle"
+                fill="var(--color-text-secondary)"
+                fontSize={12}>
+                {label}
+              </text>
+            </g>
+          );
+        }
+        // Vertical axis
+        const x = position === 'left' ? -TICK_SIZE : TICK_SIZE;
+        return (
+          <g
+            key={label}
+            style={{
+              transform: `translateY(${offset}px)`,
+              transition: tickTransition,
+              opacity: isVisible ? 1 : 0,
+            }}>
+            {showTicks && <line x2={x} {...stylex.props(styles.tickMark)} />}
+            <text
+              x={position === 'left' ? -(TICK_SIZE + 4) : TICK_SIZE + 4}
+              dy="0.32em"
+              textAnchor={position === 'left' ? 'end' : 'start'}
+              fill="var(--color-text-secondary)"
+              fontSize={12}>
+              {label}
+            </text>
+          </g>
+        );
+      })}
+    </g>
+  );
+}

package/src/ChartV2/ChartGrid.tsx

@@ -0,0 +1,99 @@
+// Copyright (c) Meta Platforms, Inc. and affiliates.
+
+/**
+ * @file ChartGrid.tsx (v2)
+ * @output Grid lines behind chart marks
+ * @position Child of Chart v2; reads scales from chart context
+ */
+
+import {useMemo} from 'react';
+import * as stylex from '@stylexjs/stylex';
+import {colorVars} from '@astryxdesign/core/theme/tokens.stylex';
+import {useChart} from '../Chart/ChartContext';
+import {isBandScale} from '../Chart/utils';
+import type {ScaleLinear, ScaleTime} from 'd3-scale';
+
+export interface ChartGridProps {
+  /** Show horizontal grid lines. Default: `true`. */
+  horizontal?: boolean;
+  /** Show vertical grid lines. Default: `false`. */
+  vertical?: boolean;
+}
+
+const styles = stylex.create({
+  gridLine: {
+    stroke: colorVars['--color-border-emphasized'],
+    strokeWidth: 1,
+  },
+});
+
+/**
+ * Grid lines behind chart marks.
+ *
+ * Pair with `<ChartAxis>` so users always see an axis line beside the
+ * grid — the components are designed to be used together.
+ *
+ * @example
+ * ```
+ * <ChartGrid />                 // horizontal grid lines (default)
+ * <ChartGrid horizontal vertical />
+ * ```
+ */
+export function ChartGrid({
+  horizontal = true,
+  vertical = false,
+}: ChartGridProps) {
+  const {width, height, xScale, yScale} = useChart();
+
+  // Skip the y=0 line when emphasizing it via the axis. Without this we'd
+  // double-draw on top of the axis line.
+  const hLines = useMemo(() => {
+    if (!horizontal) {
+      return [];
+    }
+    return yScale
+      .ticks(5)
+      .filter(tick => tick !== 0)
+      .map(tick => yScale(tick));
+  }, [horizontal, yScale]);
+
+  const vLines = useMemo(() => {
+    if (!vertical) {
+      return [];
+    }
+    if (isBandScale(xScale)) {
+      return xScale
+        .domain()
+        .map(d => (xScale(d) ?? 0) + xScale.bandwidth() / 2);
+    }
+    const linear = xScale as
+      | ScaleLinear<number, number>
+      | ScaleTime<number, number>;
+    return linear.ticks(5).map(d => linear(d as number & Date));
+  }, [vertical, xScale]);
+
+  return (
+    <g>
+      {hLines.map(y => (
+        <line
+          key={`h-${y}`}
+          x1={0}
+          x2={width}
+          y1={y}
+          y2={y}
+          {...stylex.props(styles.gridLine)}
+        />
+      ))}
+      {vLines.map(x => (
+        <line
+          key={`v-${x}`}
+          x1={x}
+          x2={x}
+          y1={0}
+          y2={height}
+          {...stylex.props(styles.gridLine)}
+        />
+      ))}
+    </g>
+  );
+}

package/src/ChartV2/ChartLegend.tsx

@@ -0,0 +1,69 @@
+// Copyright (c) Meta Platforms, Inc. and affiliates.
+
+/**
+ * @file ChartLegend.tsx
+ * @output Standalone chart legend component
+ * @position Can be used inside Chart via the `legend` prop, or independently
+ *
+ * @example
+ * // Via Chart
+ * <Chart legend={{position: 'top', alignment: 'start'}} ... />
+ *
+ * // Standalone
+ * <ChartLegend items={[{label: 'Revenue', color: '#3b82f6'}]} />
+ */
+
+import {Text} from '@astryxdesign/core';
+import {VStack, HStack} from '@astryxdesign/core';
+import {ChartSwatch, swatchVariantForType} from './ChartSwatch';
+import type {LegendItem} from './legend';
+
+export type {LegendItem};
+
+export type LegendPosition = 'top' | 'bottom' | 'start' | 'end';
+export type LegendAlignment = 'start' | 'center' | 'end';
+
+export interface ChartLegendProps {
+  /** Legend items to display */
+  items?: LegendItem[];
+  /** Position of the legend relative to the chart. Default: 'bottom' */
+  position?: LegendPosition;
+  /** Alignment of the legend within its position. Default: 'start' */
+  alignment?: LegendAlignment;
+}
+
+export function ChartLegend({
+  items = [],
+  position = 'bottom',
+  alignment = 'start',
+}: ChartLegendProps) {
+  if (items.length === 0) {
+    return null;
+  }
+
+  const isVertical = position === 'start' || position === 'end';
+
+  const legendItems = items.map(item => (
+    <HStack key={item.label} gap={2} vAlign="center">
+      <ChartSwatch
+        color={item.color}
+        variant={swatchVariantForType(item.type)}
+      />
+      <Text type="supporting">{item.label}</Text>
+    </HStack>
+  ));
+
+  if (isVertical) {
+    return (
+      <VStack gap={2} hAlign={alignment}>
+        {legendItems}
+      </VStack>
+    );
+  }
+
+  return (
+    <HStack gap={4} justify={alignment} vAlign="center" wrap="wrap">
+      {legendItems}
+    </HStack>
+  );
+}

package/src/ChartV2/ChartSwatch.doc.mjs

@@ -0,0 +1,17 @@
+// Copyright (c) Meta Platforms, Inc. and affiliates.
+
+/** @type {import('../../core/src/docs-types').ComponentDoc} */
+
+export const docs = {
+  name: 'ChartSwatch',
+  displayName: 'Chart Swatch',
+  group: 'ChartV2',
+  category: 'Data Visualization',
+  isHiddenFromOverview: true,
+
+  usage: {
+    description: '',
+  },
+
+  props: [],
+};

package/src/ChartV2/ChartSwatch.tsx

@@ -0,0 +1,86 @@
+// Copyright (c) Meta Platforms, Inc. and affiliates.
+
+/**
+ * @file ChartSwatch.tsx
+ * @output Color swatch primitive for legends, tooltips, and standalone use
+ * @position Composable primitive — used by ChartLegend (and future
+ *           ChartTooltip), or independently anywhere a series indicator
+ *           is needed.
+ *
+ * @example
+ * ```
+ * <ChartSwatch color="#3b82f6" variant="square" />
+ * <ChartSwatch color="#f59e0b" variant="line" />
+ * ```
+ */
+
+import {memo} from 'react';
+import * as stylex from '@stylexjs/stylex';
+
+/**
+ * Visual variant — drives shape.
+ * - `'square'` — square chip (bar series)
+ * - `'line'` — short stroke (line, dot, area, and other non-bar series)
+ */
+export type ChartSwatchVariant = 'square' | 'line';
+
+export interface ChartSwatchProps {
+  /** Fill color — typically a series color token */
+  color: string;
+  /** Visual variant. Default: `'square'`. */
+  variant?: ChartSwatchVariant;
+}
+
+const styles = stylex.create({
+  base: {
+    flexShrink: 0,
+  },
+  square: {
+    width: 8,
+    height: 8,
+    borderRadius: 2,
+    marginInline: 1,
+  },
+  line: {
+    width: 10,
+    height: 3,
+    borderRadius: 1.5,
+  },
+});
+
+/**
+ * Color swatch primitive.
+ *
+ * Used by ChartLegend (and ChartTooltip) to render the colored
+ * indicator next to a series label. Exported as a primitive so consumers
+ * can build their own custom legend/tooltip layouts while keeping visuals
+ * consistent.
+ */
+function ChartSwatchImpl({color, variant = 'square'}: ChartSwatchProps) {
+  return (
+    <div
+      aria-hidden
+      {...stylex.props(styles.base, styles[variant])}
+      style={{backgroundColor: color}}
+    />
+  );
+}
+
+/**
+ * Memoized — swatches are static for a given (color, variant) tuple
+ * and can render many times in a tooltip/legend with several series.
+ */
+export const ChartSwatch = memo(ChartSwatchImpl);
+
+/**
+ * Map a SeriesDef.type string to its swatch variant.
+ *
+ * Bar series get a square chip; everything else (line, dot, area, and any
+ * other non-bar mark) gets a short stroke. Centralized so legend, tooltip,
+ * and any custom chrome stay consistent.
+ */
+export function swatchVariantForType(
+  type: string | undefined,
+): ChartSwatchVariant {
+  return type === 'bar' ? 'square' : 'line';
+}