Aurora AI Sources

1"use client"
2 
3import * as React from "react"
4import { ChevronDown, FileText } from "lucide-react"
5import { Badge } from "@/registry/aurora/ui/badge"
6import { Button } from "@/registry/aurora/ui/button"
7import { Source, type SourceItem, type SourceProps } from "@/registry/aurora/blocks/ai/elements/source"
8 
9// ---------------------------------------------------------------------------
10// Types (architecture source of truth — keep the existing registry API)
11// ---------------------------------------------------------------------------
12 
13export interface SourcesProps extends Omit<React.HTMLAttributes<HTMLDivElement>, "title"> {
14  title?: React.ReactNode
15  /** Render the count badge + chevron and allow the body to collapse. */
16  collapsible?: boolean
17  /** Control the collapsed state (uncontrolled defaults to expanded). */
18  open?: boolean
19  defaultOpen?: boolean
20  onOpenChange?: (open: boolean) => void
21}
22 
23export type { SourceItem, SourceProps }
24 
25// ---------------------------------------------------------------------------
26// Shared surface helpers (ported from CD injected CSS)
27// ---------------------------------------------------------------------------
28 
29function panelStyle(style?: React.CSSProperties): React.CSSProperties {
30  return {
31    background: "var(--aurora-surface-raised)",
32    border: "1px solid var(--aurora-border-strong)",
33    borderRadius: "var(--aurora-radius-1)",
34    boxShadow: "var(--aurora-shadow-medium), var(--aurora-highlight-medium)",
35    ...style,
36  }
37}
38 
39 
40// ---------------------------------------------------------------------------
41// Count badge — small neutral pill next to the title
42// ---------------------------------------------------------------------------
43 
44function CountBadge({ count }: { count: number }) {
45  return (
46    <Badge
47      tone="neutral"
48      fill="outline"
49      className="min-w-[22px] justify-center px-1.5 tabular-nums"
50    >
51      {count}
52    </Badge>
53  )
54}
55 
56// ---------------------------------------------------------------------------
57// Sources — bordered panel with file icon, title, count badge, collapsible body
58// ---------------------------------------------------------------------------
59 
60const Sources = (
61    { ref,
62      className,
63      title = "Sources",
64      collapsible = false,
65      open: openProp,
66      defaultOpen = true,
67      onOpenChange,
68      style,
69      children,
70      ...props
71    }: SourcesProps & { ref?: React.Ref<HTMLDivElement> }
72  ) => {
73// … truncated