@inferencesh/ui 0.0.0-stage → 0.1.2

package/src/components/chart.tsx

@@ -0,0 +1,403 @@
+"use client"
+
+import * as React from "react"
+import * as RechartsPrimitive from "recharts"
+
+import { cn } from "cn"
+import { safeCssColor, safeCssIdent } from "../lib/safe-content"
+
+// 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> }
+  )
+}
+
+// Tooltip payload type for recharts v3
+type TooltipPayloadItem = {
+  name?: string
+  dataKey?: string | number
+  value?: number | string
+  type?: string
+  color?: string
+  payload?: Record<string, unknown>
+  fill?: string
+}
+
+// Legend payload type for recharts v3
+type LegendPayloadItem = {
+  value?: string
+  dataKey?: string | number
+  type?: string
+  color?: 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(
+          "[&_.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-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 flex w-full aspect-video justify-center text-xs [&_.recharts-dot[stroke='#fff']]:stroke-transparent [&_.recharts-layer]:outline-hidden [&_.recharts-sector]:outline-hidden [&_.recharts-sector[stroke='#fff']]:stroke-transparent [&_.recharts-surface]:outline-hidden [&_.recharts-responsive-container]:!flex-1 [&_.recharts-responsive-container]:!min-w-0",
+          className
+        )}
+        {...props}
+      >
+        <ChartStyle id={chartId} config={config} />
+        <RechartsPrimitive.ResponsiveContainer
+          width="100%"
+          height="100%"
+          initialDimension={{ width: 320, height: 200 }}
+        >
+          {children}
+        </RechartsPrimitive.ResponsiveContainer>
+      </div>
+    </ChartContext.Provider>
+  )
+}
+
+const ChartStyle = ({ id, config }: { id: string; config: ChartConfig }) => {
+  // The id, series keys and colours can come from data (an agent's chart
+  // widget): written into a <style> element unchecked, a value could close
+  // the rule and style the whole page (safe-content).
+  const colorConfig = Object.entries(config).filter(
+    ([key, config]) => safeCssIdent(key) && (config.theme || config.color)
+  )
+
+  if (!colorConfig.length || !safeCssIdent(id)) {
+    return null
+  }
+
+  return (
+    <style
+      dangerouslySetInnerHTML={{
+        __html: Object.entries(THEMES)
+          .map(
+            ([theme, prefix]) => `
+${prefix} [data-chart=${id}] {
+${colorConfig
+  .map(([key, itemConfig]) => {
+    const color =
+      itemConfig.theme?.[theme as keyof typeof itemConfig.theme] ||
+      itemConfig.color
+    const safe = safeCssColor(color)
+    return safe ? `  --color-${key}: ${safe};` : null
+  })
+  .join("\n")}
+}
+`
+          )
+          .join("\n"),
+      }}
+    />
+  )
+}
+
+const ChartTooltip = RechartsPrimitive.Tooltip
+
+interface ChartTooltipContentProps extends React.ComponentProps<"div"> {
+  active?: boolean
+  payload?: TooltipPayloadItem[]
+  label?: string | number
+  labelFormatter?: (label: unknown, payload: TooltipPayloadItem[]) => React.ReactNode
+  formatter?: (
+    value: number | string,
+    name: string,
+    item: TooltipPayloadItem,
+    index: number,
+    payload: Record<string, unknown>
+  ) => React.ReactNode
+  hideLabel?: boolean
+  hideIndicator?: boolean
+  indicator?: "line" | "dot" | "dashed"
+  nameKey?: string
+  labelKey?: string
+  labelClassName?: string
+  color?: string
+}
+
+function ChartTooltipContent({
+  active,
+  payload,
+  className,
+  indicator = "dot",
+  hideLabel = false,
+  hideIndicator = false,
+  label,
+  labelFormatter,
+  labelClassName,
+  formatter,
+  color,
+  nameKey,
+  labelKey,
+}: ChartTooltipContentProps) {
+  const { config } = useChart()
+
+  const tooltipLabel = React.useMemo(() => {
+    if (hideLabel || !payload?.length) {
+      return null
+    }
+
+    const [item] = payload
+    const key = `${labelKey || item?.dataKey || item?.name || "value"}`
+    const itemConfig = getPayloadConfigFromPayload(config, item, key)
+    const value =
+      !labelKey && typeof label === "string"
+        ? config[label as keyof typeof config]?.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)}><span>{value}</span></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(
+        "border-border/50 bg-background grid w-fit items-start gap-1.5 rounded-lg border px-2.5 py-1.5 text-xs shadow-xl",
+        className
+      )}
+    >
+      {!nestLabel ? tooltipLabel : null}
+      <div className="grid gap-1.5">
+        {payload
+          .filter((item: TooltipPayloadItem) => item.type !== "none")
+          .map((item: TooltipPayloadItem, index: number) => {
+            const key = `${nameKey || item.name || item.dataKey || "value"}`
+            const itemConfig = getPayloadConfigFromPayload(config, item, key)
+            const indicatorColor = color || (item.payload?.fill as string) || item.color
+
+            return (
+              <div
+                key={item.dataKey}
+                className={cn(
+                  "[&>svg]:text-muted-foreground flex w-full flex-wrap items-stretch gap-2 [&>svg]:h-2.5 [&>svg]:w-2.5",
+                  indicator === "dot" && "items-center"
+                )}
+              >
+                {formatter && item?.value !== undefined && item.name ? (
+                  formatter(item.value, item.name, item, index, item.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={
+                            {
+                              "--color-bg": indicatorColor,
+                              "--color-border": indicatorColor,
+                            } as React.CSSProperties
+                          }
+                        />
+                      )
+                    )}
+                    <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 != null && (
+                        <span className="text-foreground font-mono font-medium tabular-nums">
+                          {item.value.toLocaleString()}
+                        </span>
+                      )}
+                    </div>
+                  </>
+                )}
+              </div>
+            )
+          })}
+      </div>
+    </div>
+  )
+}
+
+const ChartLegend = RechartsPrimitive.Legend
+
+interface ChartLegendContentProps extends React.ComponentProps<"div"> {
+  payload?: LegendPayloadItem[]
+  verticalAlign?: "top" | "bottom" | "middle"
+  hideIcon?: boolean
+  nameKey?: string
+}
+
+function ChartLegendContent({
+  className,
+  hideIcon = false,
+  payload,
+  verticalAlign = "bottom",
+  nameKey,
+}: ChartLegendContentProps) {
+  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: LegendPayloadItem) => item.type !== "none")
+        .map((item: LegendPayloadItem) => {
+          const key = `${nameKey || item.dataKey || "value"}`
+          const itemConfig = getPayloadConfigFromPayload(config, item, key)
+
+          return (
+            <div
+              key={item.value}
+              className={cn(
+                "[&>svg]:text-muted-foreground flex items-center gap-1.5 [&>svg]:h-3 [&>svg]:w-3"
+              )}
+            >
+              {itemConfig?.icon && !hideIcon ? (
+                <itemConfig.icon />
+              ) : (
+                <div
+                  className="h-2 w-2 shrink-0 rounded-[2px]"
+                  style={{
+                    backgroundColor: item.color,
+                  }}
+                />
+              )}
+              {itemConfig?.label}
+            </div>
+          )
+        })}
+    </div>
+  )
+}
+
+// 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
+
+  if (
+    key in payload &&
+    typeof payload[key as keyof typeof payload] === "string"
+  ) {
+    configLabelKey = payload[key as keyof typeof payload] as string
+  } else if (
+    payloadPayload &&
+    key in payloadPayload &&
+    typeof payloadPayload[key as keyof typeof payloadPayload] === "string"
+  ) {
+    configLabelKey = payloadPayload[
+      key as keyof typeof payloadPayload
+    ] as string
+  }
+
+  return configLabelKey in config
+    ? config[configLabelKey]
+    : config[key as keyof typeof config]
+}
+
+export {
+  ChartContext,
+  ChartContainer,
+  ChartTooltip,
+  ChartTooltipContent,
+  ChartLegend,
+  ChartLegendContent,
+  ChartStyle,
+}

package/src/components/circuit.tsx

@@ -0,0 +1,276 @@
+import { useEffect, useRef } from "react";
+
+type Segment = [[number, number], [number, number]];
+
+function linesIntersect(
+  a1: [number, number],
+  a2: [number, number],
+  b1: [number, number],
+  b2: [number, number]
+): boolean {
+  const det = (x1: number, y1: number, x2: number, y2: number) => x1 * y2 - x2 * y1;
+
+  const [ax, ay] = a1;
+  const [bx, by] = a2;
+  const [cx, cy] = b1;
+  const [dx, dy] = b2;
+
+  const d1 = det(bx - ax, by - ay, cx - ax, cy - ay);
+  const d2 = det(bx - ax, by - ay, dx - ax, dy - ay);
+  const d3 = det(dx - cx, dy - cy, ax - cx, ay - cy);
+  const d4 = det(dx - cx, dy - cy, bx - cx, by - cy);
+
+  return d1 * d2 < 0 && d3 * d4 < 0;
+}
+
+export function CircuitBackground({
+  cellSize = 24,
+  strokeWidth = 4.5,
+  maxPathLength = 12,
+  minPathLength = 2,
+  attempts = 10,
+  className = "",
+}: {
+  cellSize?: number;
+  strokeWidth?: number;
+  maxPathLength?: number;
+  minPathLength?: number;
+  attempts?: number;
+  className?: string;
+}) {
+  const containerRef = useRef<HTMLDivElement>(null);
+  const svgRef = useRef<string | null>(null);
+
+  useEffect(() => {
+    if (!containerRef.current || svgRef.current) return;
+
+    const container = containerRef.current;
+    const { offsetWidth, offsetHeight } = container;
+    const width = offsetWidth + cellSize * 4
+    const height = offsetHeight + cellSize * 4
+    const cols = Math.floor(width / cellSize) + 2;
+    const rows = Math.floor(height / cellSize) + 2;
+
+    const colors = ["#FF569D", "#5415ff", "#FFF14C", "#00D08A"];
+    const xmlns = "http://www.w3.org/2000/svg";
+    const svg: string[] = [
+      `<svg xmlns="${xmlns}" width="${width}" height="${height}" viewBox="${cellSize} ${cellSize} ${width} ${height}" fill="none">`,
+      `<rect width="200%" height="200%" fill="hsl(0 0% 5.9%)"/>`,
+    ];
+
+    const visited = new Set<string>();
+    const segments: Segment[] = [];
+
+    const isInside = (x: number, y: number) => x >= 0 && x <= cols && y >= 0 && y <= rows;
+    const vecKey = ([dx, dy]: number[]) => `${dx},${dy}`;
+    const pointKey = (x: number, y: number) => `${x},${y}`;
+
+    const getTurnOptions = ([dx, dy]: number[]): number[][] => {
+      const norm = (n: number) => (n < 0 ? -1 : n > 0 ? 1 : 0);
+      dx = norm(dx);
+      dy = norm(dy);
+
+      if (dx === 0 && dy === 0) return [];
+
+      const options: [number, number][] = [];
+
+      if (dx === 0 || dy === 0) {
+        // orthogonal: yield (straight + 2 diagonals)
+        options.push([dx, dy]);
+        options.push([dx + dy, dy + dx]);
+        options.push([dx - dy, dy - dx]);
+      } else {
+        // diagonal: yield (straight + x-only + y-only)
+        options.push([dx, dy]);
+        options.push([dx, 0]);
+        options.push([0, dy]);
+      }
+
+      return options.filter(([x, y]) =>
+        Math.abs(x) <= 1 && Math.abs(y) <= 1 &&
+        !(x === 0 && y === 0)
+      );
+    };
+
+    const directions: number[][] = [
+      [1, 0], [-1, 0], [0, 1], [0, -1],
+      [1, 1], [-1, 1], [1, -1], [-1, -1],
+    ];
+
+    const randomInt = (n: number) => Math.floor(Math.random() * n);
+    const midpoint = (a: number, b: number) => (a + b) / 2;
+
+    const realAttempts = cols * rows * attempts;
+    for (let i = 0; i < realAttempts; i++) {
+      const x = randomInt(cols);
+      const y = randomInt(rows);
+      const startKey = pointKey(x, y);
+      if (visited.has(startKey)) continue;
+
+      let color = colors[randomInt(colors.length)];
+      const hex = color.slice(1);
+      const r = parseInt(hex.slice(0, 2), 16);
+      const g = parseInt(hex.slice(2, 4), 16);
+      const b = parseInt(hex.slice(4, 6), 16);
+      const brightness = 0.7 + randomInt(30) / 100; // brightness between 0.5 and 1.0
+      const brightR = Math.round(r * brightness);
+      const brightG = Math.round(g * brightness);
+      const brightB = Math.round(b * brightness);
+      color = `rgb(${brightR}, ${brightG}, ${brightB})`;
+
+      const path: [number, number][] = [[x, y]];
+      visited.add(pointKey(x, y));
+      let head = [x, y];
+      let tail = [x, y];
+      let headDir: [number, number] | null = null;
+      let tailDir: [number, number] | null = null;
+
+      const getDirection = (path: [number, number][], fromHead: boolean): [number, number] | null => {
+        if (fromHead && path.length >= 2) {
+          const len = path.length;
+          const [x1, y1] = path[len - 2];
+          const [x2, y2] = path[len - 1];
+          return [x2 - x1, y2 - y1];
+        }
+
+        if (!fromHead && path.length >= 2) {
+          const [x1, y1] = path[0];
+          const [x2, y2] = path[1];
+          return [x1 - x2, y1 - y2];
+        }
+
+        return null;
+      }
+
+      const grow = (fromHead: boolean) => {
+        const origin = fromHead ? head : tail;
+        let dir = fromHead ? headDir : tailDir;
+
+        if (!dir) {
+          if (fromHead && path.length === 1) {
+            for (const d of directions.sort(() => 0.5 - Math.random())) {
+              const [dx, dy] = d;
+              const nx = origin[0] + dx;
+              const ny = origin[1] + dy;
+              const point = pointKey(nx, ny);
+              if (!isInside(nx, ny) || visited.has(point)) continue;
+
+              const x1 = origin[0] * cellSize + cellSize / 2;
+              const y1 = origin[1] * cellSize + cellSize / 2;
+              const x2 = nx * cellSize + cellSize / 2;
+              const y2 = ny * cellSize + cellSize / 2;
+              const intersects = segments.some(
+                ([a, b]) => linesIntersect([x1, y1], [x2, y2], a, b)
+              );
+              if (intersects) continue;
+
+              segments.push([[x1, y1], [x2, y2]]);
+              visited.add(point);
+              path.push([nx, ny]);
+              head = [nx, ny];
+              headDir = [dx, dy];
+              return true;
+            }
+            return false;
+          } else if (!fromHead && path.length >= 2) {
+            dir = getDirection(path, fromHead);
+            if (!dir) return false;
+            tailDir = dir;
+          } else {
+            return false;
+          }
+        }
+
+        const options = getTurnOptions(dir);
+        for (const [dx, dy] of options.sort(() => 0.5 - Math.random())) {
+          const nx = origin[0] + dx;
+          const ny = origin[1] + dy;
+          const point = pointKey(nx, ny);
+          if (!isInside(nx, ny) || visited.has(point)) continue;
+
+          const x1 = origin[0] * cellSize + cellSize / 2;
+          const y1 = origin[1] * cellSize + cellSize / 2;
+          const x2 = nx * cellSize + cellSize / 2;
+          const y2 = ny * cellSize + cellSize / 2;
+          const intersects = segments.some(
+            ([a, b]) => linesIntersect([x1, y1], [x2, y2], a, b)
+          );
+          if (intersects) continue;
+
+          segments.push([[x1, y1], [x2, y2]]);
+          visited.add(point);
+
+          if (fromHead) {
+            path.push([nx, ny]);
+            head = [nx, ny];
+            headDir = [dx, dy];
+          } else {
+            path.unshift([nx, ny]);
+            tail = [nx, ny];
+            tailDir = [dx, dy];
+          }
+          return true;
+        }
+
+        return false;
+      };
+
+      let headStuck = false;
+      let tailStuck = false;
+      
+      while (path.length < maxPathLength && !(headStuck && tailStuck)) {
+        const fromHead = Math.random() < 0.5;
+        const grown = grow(fromHead);
+      
+        if (fromHead) headStuck = !grown;
+        else tailStuck = !grown;
+      }
+
+      if (path.length < minPathLength && path.length > 1) continue;
+
+      let d = "";
+      for (let j = 0; j < path.length; j++) {
+        const [gx, gy] = path[j];
+        const px = gx * cellSize + cellSize / 2;
+        const py = gy * cellSize + cellSize / 2;
+        if (j === 0) {
+          d += `M ${px} ${py}`;
+        } else {
+          d += ` L ${px} ${py}`;
+        }
+      }
+
+      if (path.length < minPathLength) {
+        color = "#333333";
+      }
+      svg.push(
+        `<path d="${d}" stroke="${color}" stroke-width="${strokeWidth}" stroke-linecap="round" stroke-linejoin="round" fill="none"/>`
+      );
+
+      const circleR = strokeWidth * 1.8;
+      const dotR = strokeWidth * 0.9;
+      const [sx, sy] = path[0];
+      const [ex, ey] = path[path.length - 1];
+      const sxpx = sx * cellSize + cellSize / 2;
+      const sypx = sy * cellSize + cellSize / 2;
+      const expx = ex * cellSize + cellSize / 2;
+      const eypx = ey * cellSize + cellSize / 2;
+
+      svg.push(
+        `<circle cx="${sxpx}" cy="${sypx}" r="${circleR}" fill="${color}"/>`,
+        `<circle cx="${sxpx}" cy="${sypx}" r="${dotR}" fill="hsl(0 0% 5.9%)"/>`,
+        `<circle cx="${expx}" cy="${eypx}" r="${circleR}" fill="${color}"/>`,
+        `<circle cx="${expx}" cy="${eypx}" r="${dotR}" fill="hsl(0 0% 5.9%)"/>`
+      );
+    }
+
+    svg.push(`</svg>`);
+    svgRef.current = svg.join("");
+    container.style.backgroundImage = `url("data:image/svg+xml;utf8,${encodeURIComponent(
+      svgRef.current
+    )}")`;
+    container.style.backgroundSize = "cover";
+  }, [cellSize, strokeWidth, maxPathLength, minPathLength, attempts]);
+
+  return <div ref={containerRef} className={className} />;
+}