{ "$schema": "https://ui.shadcn.com/schema/registry-item.json", "name": "csv-viewer", "title": "CSV Viewer", "description": "A fixed-window virtualized CSV/TSV table: sortable columns, sticky header, row numbers, horizontal scroll. Renders straight from raw rows (no table row model) so it stays flat in memory on huge files.", "dependencies": [ "lucide-react" ], "registryDependencies": [ "@retab/csv", "@retab/utils", "button", "dropdown-menu", "@retab/viewer-controls", "@retab/use-keyed-layout-effect", "@retab/use-mount-effect" ], "files": [ { "path": "registry/new-york-v4/ui/csv-viewer.tsx", "content": "\"use client\";\n\nimport * as React from \"react\";\n\nimport { useKeyedMountEffect } from \"@/hooks/use-keyed-mount-effect\";\nimport type { CsvDialect, CsvTable } from \"@/lib/csv\";\nimport {\n createViewerResource,\n type ViewerResource,\n} from \"@/lib/viewer-resource\";\n\nimport {\n CsvViewerFrame,\n CsvViewerHeader,\n csvViewerStatusNode,\n} from \"./csv-viewer-chrome\";\nimport {\n csvViewerDownloadActions,\n defaultCsvDownloadName,\n} from \"./csv-viewer-download\";\nimport { CsvGrid, type CsvGridHandle } from \"./csv-viewer-grid\";\nimport {\n csvViewerExportFileName,\n csvViewerSortResetKey,\n isCsvDocumentSource,\n resolveCsvViewerDialect,\n type CsvDocumentSource,\n type CsvTableSource,\n type CsvViewerSource,\n} from \"./csv-viewer-resource\";\nimport { useCsvResourceState, type CsvCellAddress } from \"./csv-viewer-state\";\nimport type { CsvViewerHandle, CsvViewerProps } from \"./csv-viewer-types\";\nimport {\n useViewerControlsRegistration,\n type ViewerControlsState,\n} from \"./viewer-controls\";\nimport { joinEffectKey } from \"@/lib/effect-key\";\n\nexport type {\n CsvScrollOptions,\n CsvViewerHandle,\n CsvViewerProps,\n} from \"./csv-viewer-types\";\n\nexport type CsvResourceContentProps = Omit & {\n resource?: ViewerResource | null;\n source?: CsvViewerSource;\n};\ntype CsvResourceContentInternalProps = CsvResourceContentProps & {\n frame?: boolean;\n};\n\nexport type CsvViewerProviderProps = {\n children: React.ReactNode;\n resource: ViewerResource;\n};\n\nexport type CsvViewerGridProps = Omit;\n\nexport const CsvViewer = React.forwardRef(\n function CsvViewer({ source, ...props }, ref) {\n const resource = React.useMemo(\n () =>\n source && isCsvDocumentSource(source)\n ? createViewerResource(source)\n : null,\n [source],\n );\n return (\n \n );\n },\n);\n\nconst CsvViewerResourceContext = React.createContext(\n null,\n);\n\nexport function CsvViewerProvider({\n children,\n resource,\n}: CsvViewerProviderProps) {\n return (\n \n {children}\n \n );\n}\n\nfunction useCsvViewerResource(): ViewerResource {\n const resource = React.useContext(CsvViewerResourceContext);\n if (!resource) {\n throw new Error(\"CsvViewerGrid must be used within CsvViewerProvider.\");\n }\n return resource;\n}\n\nexport const CsvViewerDocument = React.forwardRef<\n CsvViewerHandle,\n CsvViewerProps\n>(function CsvViewerDocument({ source, ...props }, ref) {\n const resource = React.useMemo(\n () =>\n source && isCsvDocumentSource(source)\n ? createViewerResource(source)\n : null,\n [source],\n );\n return (\n \n );\n});\n\nexport const CsvViewerGrid = React.forwardRef<\n CsvViewerHandle,\n CsvViewerGridProps\n>(function CsvViewerGrid(props, ref) {\n const resource = useCsvViewerResource();\n return (\n \n );\n});\n\nexport const CsvResourceContent = React.forwardRef<\n CsvViewerHandle,\n CsvResourceContentInternalProps\n>(function CsvResourceContent(\n {\n source,\n resource = null,\n dialect: dialectProp,\n className,\n controls = true,\n height = 480,\n fillHeight = false,\n frame = true,\n activeCell,\n isolateStyles = false,\n },\n ref,\n) {\n const [retryVersion, setRetryVersion] = React.useState(0);\n const dialect = React.useMemo(\n () =>\n resolveCsvViewerDialect({\n dialect: dialectProp,\n source,\n resource,\n }),\n [dialectProp, source, resource],\n );\n const resourceState = useCsvResourceState({\n source,\n resource,\n dialect,\n retryVersion,\n });\n const gridRef = React.useRef(null);\n const [zoom, setZoom] = React.useState(1);\n const columns = resourceState.columns;\n const sourceRows = resourceState.sourceRows;\n const rowStore = resourceState.rowStore;\n const rowCount = rowStore.rowCount;\n const canExportTable =\n resourceState.status === \"ready\" || resourceState.status === \"empty\";\n const sortResetKey = csvViewerSortResetKey({\n dialect,\n source,\n resource,\n });\n const exportFileName = csvViewerExportFileName({\n dialect,\n source,\n resource,\n fallback: defaultCsvDownloadName,\n });\n const downloadActions = React.useMemo(() => {\n return csvViewerDownloadActions({\n resource,\n columns,\n sourceRows,\n dialect,\n fileName: exportFileName,\n canExportTable,\n });\n }, [canExportTable, columns, dialect, exportFileName, resource, sourceRows]);\n\n React.useImperativeHandle(\n ref ?? null,\n () => ({\n scrollToCell: (cellAddress, options) => {\n gridRef.current?.scrollToCell(cellAddress, options);\n },\n getViewportElement: () => gridRef.current?.getViewportElement() ?? null,\n }),\n [],\n );\n\n const statusNode = React.useMemo(\n () =>\n csvViewerStatusNode({\n resourceState,\n resource,\n rowCount,\n showDownload: controls,\n onRetry: () => setRetryVersion((version) => version + 1),\n }),\n [controls, resource, resourceState, rowCount],\n );\n const zoomOut = React.useCallback(\n () => setZoom((value) => clampCsvViewerZoom(value / 1.2)),\n [],\n );\n const zoomIn = React.useCallback(\n () => setZoom((value) => clampCsvViewerZoom(value * 1.2)),\n [],\n );\n const resetZoom = React.useCallback(() => setZoom(1), []);\n useCsvControlsRegistration({\n columnCount: columns.length,\n downloadActions,\n isLoading: resourceState.status === \"loading\",\n onResetZoom: resetZoom,\n onZoomIn: zoomIn,\n onZoomOut: zoomOut,\n rowCount,\n zoom,\n });\n\n return (\n \n \n \n \n );\n});\n\nexport {\n type CsvCellAddress,\n type CsvDialect,\n type CsvDocumentSource,\n type CsvTableSource,\n type CsvTable,\n type CsvViewerSource,\n};\n\nfunction useCsvControlsRegistration({\n columnCount,\n downloadActions,\n isLoading,\n onResetZoom,\n onZoomIn,\n onZoomOut,\n rowCount,\n zoom,\n}: {\n columnCount: number;\n downloadActions: ViewerControlsState[\"downloads\"];\n isLoading: boolean;\n onResetZoom: () => void;\n onZoomIn: () => void;\n onZoomOut: () => void;\n rowCount: number;\n zoom: number;\n}) {\n const onControlsChange = useViewerControlsRegistration();\n const rowLabel = `${rowCount.toLocaleString()} row${rowCount === 1 ? \"\" : \"s\"}`;\n const columnLabel = `${columnCount} column${columnCount === 1 ? \"\" : \"s\"}`;\n const controlsState = React.useMemo(\n () => ({\n title: isLoading ? `${rowLabel} loading` : rowLabel,\n subtitle: isLoading ? null : columnLabel,\n loading: isLoading,\n zoom: {\n scale: zoom,\n onZoomOut,\n onZoomIn,\n onFit: onResetZoom,\n fitLabel: \"Reset zoom\",\n },\n downloads: downloadActions,\n }),\n [\n columnLabel,\n downloadActions,\n isLoading,\n onResetZoom,\n onZoomIn,\n onZoomOut,\n rowLabel,\n zoom,\n ],\n );\n\n useKeyedMountEffect(\n joinEffectKey([\"csv-controls\", onControlsChange, controlsState]),\n () => {\n if (!onControlsChange) return;\n onControlsChange(controlsState);\n return () => onControlsChange(null);\n },\n );\n}\n\nfunction clampCsvViewerZoom(zoom: number) {\n return Math.max(0.1, Math.min(5, zoom));\n}\n", "type": "registry:ui", "target": "@ui/csv-viewer.tsx" }, { "path": "registry/new-york-v4/ui/file-viewer-core.ts", "content": "import {\n detectCategory,\n extensionOf,\n extractName,\n resolveViewerDescriptor,\n textPayloadKey,\n type FileCategory,\n type ViewerDescriptor,\n type ViewerSource,\n} from \"@/lib/viewer-source\";\n\nexport type { FileCategory, ViewerSource };\nexport type FileViewerFallbackSize = { width: number; height: number };\n\nexport interface FileViewerCoreProps {\n source: ViewerSource;\n category?: FileCategory;\n className?: string;\n /** Intrinsic first-frame size for image/TIFF loading skeletons. */\n fallbackFrameSize?: FileViewerFallbackSize;\n /** Intrinsic first-slide size for PPTX loading skeletons. */\n fallbackSlideSize?: FileViewerFallbackSize;\n isolateStyles?: boolean;\n}\n\nexport type FileDescriptor = ViewerDescriptor;\n\nexport function resolveFileDescriptor({\n source,\n category,\n}: FileViewerCoreProps): FileDescriptor {\n return resolveViewerDescriptor({\n source,\n category,\n });\n}\n\nexport function descriptorResetKey(descriptor: FileDescriptor): string {\n return [\n descriptorIdentityResetKey(descriptor),\n descriptor.displayName,\n descriptor.mimeType ?? \"\",\n descriptor.category,\n ].join(\"\\u0000\");\n}\n\nfunction descriptorIdentityResetKey(descriptor: FileDescriptor): string {\n if (\n descriptor.source.kind === \"text\" &&\n descriptor.source.identityKey == null\n ) {\n return textPayloadKey(descriptor.source.text);\n }\n return descriptor.identityKey;\n}\n\nexport function isProseTextDescriptor(descriptor: FileDescriptor): boolean {\n if (descriptor.category !== \"text\") return false;\n\n const extension = extensionOf(descriptor.fileName);\n if (extension === \"txt\" || extension === \"text\") return true;\n return !extension && descriptor.mimeType?.toLowerCase() === \"text/plain\";\n}\n\nexport { detectCategory, extensionOf, extractName };\n", "type": "registry:ui", "target": "@ui/file-viewer-core.ts" }, { "path": "registry/new-york-v4/ui/use-is-client.ts", "content": "\"use client\";\n\nimport * as React from \"react\";\n\nconst emptySubscribe = () => () => {};\nconst getClientSnapshot = () => true;\nconst getServerSnapshot = () => false;\n\n/**\n * SSR gate: `false` on the server (and during hydration's first pass),\n * `true` on the client.\n *\n * This must stay a synchronous external-store read, NOT the\n * `useState(false)` + mount-effect flip. The flip pattern makes every\n * viewer mount its Suspense boundary in a later update; when two such\n * boundaries suspend on pending resources in the same flush as other\n * commit-phase updates (viewer sidebar/geometry registration), React 19's\n * retry lanes desynchronize and re-attempt each other's boundary on every\n * commit — an unbounded synchronous suspend/retry loop that starves the\n * event loop (jsdom tests OOM; browsers busy-spin until the resource\n * resolves). With the store read, client renders suspend on mount, which\n * never enters that loop. Regression-guarded in\n * tests/pdf-viewer-thumbnails.test.tsx (\"shares one document resource…\").\n */\nexport function useIsClient(): boolean {\n return React.useSyncExternalStore(\n emptySubscribe,\n getClientSnapshot,\n getServerSnapshot,\n );\n}\n", "type": "registry:ui", "target": "@ui/use-is-client.ts" }, { "path": "registry/new-york-v4/lib/viewer-source.ts", "content": "export type FileCategory =\n | \"pdf\"\n | \"docx\"\n | \"xlsx\"\n | \"pptx\"\n | \"csv\"\n | \"image\"\n | \"markdown\"\n | \"html\"\n | \"email\"\n | \"text\"\n | \"unsupported\";\n\nexport type ViewerSource = UrlViewerSource | TextSource | BlobViewerSource;\n\nexport interface UrlViewerSource {\n kind: \"url\";\n url: string;\n fileName?: string;\n mimeType?: string;\n downloadUrl?: string;\n identityKey?: string;\n}\n\nexport interface TextSource {\n kind: \"text\";\n text: string;\n fileName?: string;\n mimeType?: string;\n identityKey?: string;\n}\n\nexport interface BlobViewerSource {\n kind: \"blob\";\n blob: Blob;\n identityKey: string;\n fileName?: string;\n mimeType?: string;\n downloadUrl?: string;\n}\n\nexport interface ViewerDescriptor {\n source: ViewerSource;\n category: FileCategory;\n identityKey: string;\n displayName: string;\n fileName: string;\n mimeType?: string;\n}\n\nconst EXTENSION_CATEGORY: Record = {\n pdf: \"pdf\",\n docx: \"docx\",\n xlsx: \"xlsx\",\n xls: \"xlsx\",\n xlsm: \"xlsx\",\n pptx: \"pptx\",\n csv: \"csv\",\n tsv: \"csv\",\n png: \"image\",\n jpg: \"image\",\n jpeg: \"image\",\n gif: \"image\",\n webp: \"image\",\n avif: \"image\",\n bmp: \"image\",\n svg: \"image\",\n ico: \"image\",\n tif: \"image\",\n tiff: \"image\",\n md: \"markdown\",\n markdown: \"markdown\",\n mdx: \"text\",\n html: \"html\",\n htm: \"html\",\n eml: \"email\",\n txt: \"text\",\n text: \"text\",\n log: \"text\",\n json: \"text\",\n jsonl: \"text\",\n json5: \"text\",\n ndjson: \"text\",\n xml: \"text\",\n yaml: \"text\",\n yml: \"text\",\n toml: \"text\",\n ini: \"text\",\n env: \"text\",\n js: \"text\",\n mjs: \"text\",\n cjs: \"text\",\n jsx: \"text\",\n ts: \"text\",\n tsx: \"text\",\n css: \"text\",\n scss: \"text\",\n less: \"text\",\n py: \"text\",\n rb: \"text\",\n go: \"text\",\n rs: \"text\",\n java: \"text\",\n kt: \"text\",\n c: \"text\",\n h: \"text\",\n cpp: \"text\",\n cc: \"text\",\n cs: \"text\",\n php: \"text\",\n sh: \"text\",\n bash: \"text\",\n zsh: \"text\",\n sql: \"text\",\n graphql: \"text\",\n proto: \"text\",\n lua: \"text\",\n r: \"text\",\n swift: \"text\",\n scala: \"text\",\n pl: \"text\",\n vue: \"text\",\n svelte: \"text\",\n};\n\nexport function extensionOf(name: string): string | null {\n const clean = name.split(/[?#]/)[0];\n const base = clean.split(\"/\").pop() ?? clean;\n const dot = base.lastIndexOf(\".\");\n return dot > 0 ? base.slice(dot + 1).toLowerCase() : null;\n}\n\nexport function extractName(url: string): string {\n const clean = url.split(/[?#]/)[0];\n return clean.split(\"/\").pop() || \"file\";\n}\n\nexport function detectCategory(\n fileName: string,\n mimeType?: string,\n): FileCategory {\n const ext = extensionOf(fileName);\n if (ext && EXTENSION_CATEGORY[ext]) return EXTENSION_CATEGORY[ext];\n if (mimeType) {\n const fromMime = categoryFromMime(mimeType);\n if (fromMime) return fromMime;\n }\n return \"unsupported\";\n}\n\nexport function resolveViewerDescriptor({\n source,\n category,\n}: {\n source: ViewerSource;\n category?: FileCategory;\n}): ViewerDescriptor {\n const resolvedMimeType =\n source.mimeType ??\n (source.kind === \"blob\" && source.blob.type ? source.blob.type : undefined);\n const displayName = source.fileName ?? defaultDisplayName(source);\n const fileName = source.fileName ?? defaultFileName(source);\n const resolvedCategory =\n category ?? detectCategory(displayName, resolvedMimeType);\n\n return {\n source,\n category: resolvedCategory,\n identityKey: source.identityKey ?? defaultIdentityKey(source),\n displayName,\n fileName,\n mimeType: resolvedMimeType,\n };\n}\n\nfunction categoryFromMime(mimeType: string): FileCategory | null {\n const mime = mimeType.toLowerCase().split(\";\")[0].trim();\n if (mime === \"application/pdf\") return \"pdf\";\n if (mime.includes(\"wordprocessingml\")) return \"docx\";\n if (mime.includes(\"spreadsheet\") || mime.includes(\"ms-excel\")) return \"xlsx\";\n if (mime.includes(\"presentation\") || mime.includes(\"ms-powerpoint\")) {\n return \"pptx\";\n }\n if (mime === \"text/csv\" || mime === \"text/tab-separated-values\") return \"csv\";\n if (mime === \"text/markdown\") return \"markdown\";\n if (mime === \"text/html\") return \"html\";\n if (mime === \"message/rfc822\" || mime === \"message/global\") {\n return \"email\";\n }\n if (mime.startsWith(\"image/\")) return \"image\";\n if (mime === \"application/json\" || mime === \"application/xml\") return \"text\";\n if (mime.startsWith(\"text/\")) return \"text\";\n return null;\n}\n\nfunction defaultDisplayName(source: ViewerSource) {\n if (source.kind === \"url\") return source.url;\n if (source.kind === \"text\") return \"text.txt\";\n return \"file\";\n}\n\nfunction defaultFileName(source: ViewerSource) {\n if (source.kind === \"url\") return extractName(source.url);\n if (source.kind === \"text\") return \"text.txt\";\n return \"file\";\n}\n\nfunction defaultIdentityKey(source: ViewerSource) {\n if (source.kind === \"url\") return `url:${source.url}`;\n if (source.kind === \"text\") return textPayloadIdentityKey(source.text);\n return source.identityKey;\n}\n\nexport function textPayloadIdentityKey(text: string) {\n return textPayloadKey(text);\n}\n\nexport function textPayloadKey(text: string) {\n return `text:${text.length}:${hashString(text)}`;\n}\n\nfunction hashString(text: string) {\n let hash = 0x811c9dc5;\n for (let index = 0; index < text.length; index += 1) {\n hash ^= text.charCodeAt(index);\n hash = Math.imul(hash, 0x01000193);\n }\n return (hash >>> 0).toString(36);\n}\n", "type": "registry:lib", "target": "@lib/viewer-source.ts" }, { "path": "registry/new-york-v4/lib/viewer-resource.ts", "content": "import {\n createBlobDownloadAction,\n createHrefDownloadAction,\n createTextDownloadAction,\n type ViewerDownloadAction,\n} from \"@/lib/viewer-download-actions\";\nimport {\n isAbortError,\n ResourceError,\n type ResourceTooLargeReason,\n} from \"@/lib/viewer-errors\";\nimport {\n resolveViewerDescriptor,\n textPayloadKey,\n type BlobViewerSource,\n type FileCategory,\n type TextSource,\n type UrlViewerSource,\n type ViewerDescriptor,\n type ViewerSource,\n} from \"@/lib/viewer-source\";\n\nexport interface ResourceReadOptions {\n cache?: RequestCache;\n signal?: AbortSignal;\n}\n\nexport interface TextReadOptions extends ResourceReadOptions {\n maxBytes?: number;\n maxLines?: number;\n}\n\nexport interface ByteRange {\n start: number;\n end: number;\n}\n\nexport interface ByteRangeResult {\n buffer: ArrayBuffer;\n contentRange?: {\n start: number;\n end: number;\n total: number | null;\n };\n isComplete: boolean;\n}\n\nexport interface ViewerResourceKeys {\n readonly load: string;\n readonly presentation: string;\n readonly resource: string;\n}\n\nexport type ViewerResourcePayload =\n | { kind: \"url\"; url: string }\n | { kind: \"blob\"; blob: Blob }\n | { kind: \"text\"; text: string };\n\nexport interface ViewerResourceContent {\n readonly key: string;\n readonly sourceKind: ViewerSource[\"kind\"];\n readonly directUrl: string | null;\n readonly mimeType?: string;\n readonly payload: ViewerResourcePayload;\n readBlob(options?: ResourceReadOptions): Promise;\n readBytes(options?: ResourceReadOptions): Promise;\n readText(options?: TextReadOptions): Promise;\n readStream(\n options?: ResourceReadOptions,\n ): Promise>;\n readRange(\n range: ByteRange,\n options?: ResourceReadOptions,\n ): Promise;\n}\n\nexport type ViewerContentIdentity = Pick<\n ViewerResourceContent,\n \"key\" | \"sourceKind\"\n>;\n\nexport type ViewerContentDirectUrl = ViewerContentIdentity &\n Pick;\n\nexport type ViewerContentPayload = ViewerContentIdentity &\n Pick;\n\nexport type ViewerContentMime = ViewerContentIdentity &\n Pick;\n\nexport type ViewerContentBlob = ViewerContentIdentity &\n Pick;\n\nexport type ViewerContentBytes = ViewerContentIdentity &\n Pick;\n\nexport type ViewerContentText = ViewerContentIdentity &\n Pick;\n\nexport type ViewerContentStream = ViewerContentIdentity &\n Pick;\n\nexport type ViewerContentRange = ViewerContentIdentity &\n Pick;\n\nexport interface ViewerResource {\n readonly descriptor: ViewerDescriptor;\n readonly sourceKind: ViewerSource[\"kind\"];\n readonly keys: ViewerResourceKeys;\n readonly identityKey: string;\n readonly fileName: string;\n readonly mimeType?: string;\n readonly content: ViewerResourceContent;\n readonly originalDownload: ViewerDownloadAction;\n}\n\nconst URL_RESOURCE_REGISTRY_MAX = 128;\nconst TEXT_RESOURCE_REGISTRY_MAX = 64;\n// LF, CR, CRLF, LINE SEPARATOR (U+2028), and PARAGRAPH SEPARATOR (U+2029) — the\n// ECMAScript LineTerminator set, matching what a browser breaks on in a\n// `white-space: pre` block. Kept in sync with text-viewer-resource's splitter.\nconst TEXT_LINE_BREAK_PATTERN = /\\r\\n|[\\n\\r\\u2028\\u2029]/g;\n\nconst urlViewerResourceRegistry = new Map();\nconst urlViewerResourceContentRegistry = new Map<\n string,\n ViewerResourceContent\n>();\nconst textViewerResourceRegistry = new Map();\nconst textViewerResourceContentRegistry = new Map<\n string,\n ViewerResourceContent\n>();\nlet blobViewerResourceRegistry = new WeakMap<\n Blob,\n Map\n>();\nlet blobViewerResourceContentRegistry = new WeakMap<\n Blob,\n Map\n>();\nconst blobObjectKeys = new WeakMap();\nlet nextBlobObjectKey = 0;\n\nexport function createViewerResource(\n source: ViewerSource,\n category?: FileCategory,\n): ViewerResource {\n const descriptor = resolveViewerDescriptor({ source, category });\n const keys = viewerResourceKeys(source, descriptor);\n\n if (source.kind === \"url\") {\n return internUrlResource(source, descriptor, keys);\n }\n if (source.kind === \"blob\") {\n return internBlobResource(source, descriptor, keys);\n }\n return internTextResource(source, descriptor, keys);\n}\n\nexport function clearViewerResourceRegistryForTests() {\n urlViewerResourceRegistry.clear();\n urlViewerResourceContentRegistry.clear();\n textViewerResourceRegistry.clear();\n textViewerResourceContentRegistry.clear();\n blobViewerResourceRegistry = new WeakMap();\n blobViewerResourceContentRegistry = new WeakMap();\n}\n\nfunction internUrlResource(\n source: UrlViewerSource,\n descriptor: ViewerDescriptor,\n keys: ViewerResourceKeys,\n): ViewerResource {\n const cached = urlViewerResourceRegistry.get(keys.resource);\n if (cached) return cached;\n\n const resource = createUrlResource(source, descriptor, keys);\n urlViewerResourceRegistry.set(keys.resource, resource);\n pruneUrlResourceRegistry();\n return resource;\n}\n\nfunction internBlobResource(\n source: BlobViewerSource,\n descriptor: ViewerDescriptor,\n keys: ViewerResourceKeys,\n): ViewerResource {\n let resources = blobViewerResourceRegistry.get(source.blob);\n if (!resources) {\n resources = new Map();\n blobViewerResourceRegistry.set(source.blob, resources);\n }\n\n const cached = resources.get(keys.resource);\n if (cached) return cached;\n\n const resource = createBlobResource(source, descriptor, keys);\n resources.set(keys.resource, resource);\n return resource;\n}\n\nfunction internTextResource(\n source: TextSource,\n descriptor: ViewerDescriptor,\n keys: ViewerResourceKeys,\n): ViewerResource {\n const cached = textViewerResourceRegistry.get(keys.resource);\n if (cached) return cached;\n\n const resource = createTextResource(source, descriptor, keys);\n textViewerResourceRegistry.set(keys.resource, resource);\n pruneTextResourceRegistry();\n return resource;\n}\n\nfunction pruneUrlResourceRegistry() {\n while (urlViewerResourceRegistry.size > URL_RESOURCE_REGISTRY_MAX) {\n const firstKey = urlViewerResourceRegistry.keys().next().value;\n if (!firstKey) return;\n urlViewerResourceRegistry.delete(firstKey);\n }\n}\n\nfunction pruneUrlResourceContentRegistry() {\n while (urlViewerResourceContentRegistry.size > URL_RESOURCE_REGISTRY_MAX) {\n const firstKey = urlViewerResourceContentRegistry.keys().next().value;\n if (!firstKey) return;\n urlViewerResourceContentRegistry.delete(firstKey);\n }\n}\n\nfunction pruneTextResourceRegistry() {\n while (textViewerResourceRegistry.size > TEXT_RESOURCE_REGISTRY_MAX) {\n const firstKey = textViewerResourceRegistry.keys().next().value;\n if (!firstKey) return;\n textViewerResourceRegistry.delete(firstKey);\n }\n}\n\nfunction pruneTextResourceContentRegistry() {\n while (textViewerResourceContentRegistry.size > TEXT_RESOURCE_REGISTRY_MAX) {\n const firstKey = textViewerResourceContentRegistry.keys().next().value;\n if (!firstKey) return;\n textViewerResourceContentRegistry.delete(firstKey);\n }\n}\n\nfunction viewerResourceKeys(\n source: ViewerSource,\n descriptor: ViewerDescriptor,\n): ViewerResourceKeys {\n const load = viewerResourceLoadKey(source, descriptor);\n const presentation = viewerResourcePresentationKey(source, descriptor);\n return {\n load,\n presentation,\n resource: [load, presentation].join(\"\\u0000\"),\n };\n}\n\nfunction viewerResourceLoadKey(\n source: ViewerSource,\n descriptor: ViewerDescriptor,\n) {\n return [\n source.kind,\n source.identityKey ?? \"\",\n sourceMimeType(source) ?? \"\",\n directLoadCacheKey(source),\n payloadCacheKey(source, descriptor),\n ].join(\"\\u0000\");\n}\n\nfunction viewerResourcePresentationKey(\n source: ViewerSource,\n descriptor: ViewerDescriptor,\n) {\n return [\n descriptor.category,\n descriptor.displayName,\n descriptor.fileName,\n descriptor.mimeType ?? \"\",\n downloadCacheKey(source),\n ].join(\"\\u0000\");\n}\n\nfunction directLoadCacheKey(source: ViewerSource) {\n return source.kind === \"url\" ? source.url : \"\";\n}\n\nfunction downloadCacheKey(source: ViewerSource) {\n if (source.kind === \"text\") return \"\";\n return source.downloadUrl ?? \"\";\n}\n\nfunction payloadCacheKey(source: ViewerSource, descriptor: ViewerDescriptor) {\n if (source.kind === \"url\") return \"\";\n if (source.kind === \"blob\") return blobObjectKey(source.blob);\n return source.identityKey ? \"\" : descriptor.identityKey;\n}\n\nexport function viewerResourceRenderKey(resource: ViewerResource): string {\n const load = [\n resource.sourceKind,\n resource.identityKey,\n resource.mimeType ?? resource.content.mimeType ?? \"\",\n resource.content.directUrl ?? \"\",\n viewerContentRenderKey(resource.content),\n ].join(\"\\u0000\");\n\n return [load, resource.keys.presentation].join(\"\\u0000\");\n}\n\nexport function viewerContentRenderKey(content: ViewerResourceContent): string {\n if (content.payload.kind === \"text\")\n return textPayloadKey(content.payload.text);\n return content.key;\n}\n\nfunction sourceMimeType(source: ViewerSource) {\n if (source.kind === \"blob\") return source.mimeType ?? source.blob.type;\n return source.mimeType;\n}\n\nfunction blobObjectKey(blob: Blob) {\n let key = blobObjectKeys.get(blob);\n if (!key) {\n nextBlobObjectKey += 1;\n key = `blob-object:${nextBlobObjectKey}`;\n blobObjectKeys.set(blob, key);\n }\n return key;\n}\n\nexport function blobSource(\n bytes: Blob | ArrayBuffer | Uint8Array,\n metadata: {\n identityKey: string;\n fileName?: string;\n mimeType?: string;\n downloadUrl?: string;\n },\n): BlobViewerSource {\n const blob =\n bytes instanceof Blob\n ? bytes\n : new Blob(\n [bytes instanceof ArrayBuffer ? bytes : new Uint8Array(bytes)],\n {\n type: metadata.mimeType ?? \"\",\n },\n );\n return {\n kind: \"blob\",\n blob,\n fileName: metadata.fileName,\n mimeType: metadata.mimeType ?? blob.type,\n downloadUrl: metadata.downloadUrl,\n identityKey: metadata.identityKey,\n };\n}\n\nfunction createUrlResource(\n source: UrlViewerSource,\n descriptor: ViewerDescriptor,\n keys: ViewerResourceKeys,\n): ViewerResource {\n const content = internUrlResourceContent(source, keys);\n const originalDownload = createHrefDownloadAction({\n id: \"download-original\",\n label: \"Download\",\n href: source.downloadUrl ?? source.url,\n fileName: descriptor.fileName,\n });\n\n return resourceBase(source, descriptor, keys, { content, originalDownload });\n}\n\nfunction internUrlResourceContent(\n source: UrlViewerSource,\n keys: ViewerResourceKeys,\n): ViewerResourceContent {\n const cached = urlViewerResourceContentRegistry.get(keys.load);\n if (cached) return cached;\n\n const content = resourceContentBase(source, keys, {\n directUrl: source.url,\n payload: { kind: \"url\", url: source.url },\n readBlob: async ({ cache, signal } = {}) => {\n const response = await fetchResource(\n source.url,\n cache ? { cache, signal } : { signal },\n );\n validateFullContentResponse(response);\n return readResponseBlob(response);\n },\n readBytes: async ({ cache, signal } = {}) => {\n const response = await fetchResource(\n source.url,\n cache ? { cache, signal } : { signal },\n );\n validateFullContentResponse(response);\n return readResponseArrayBuffer(response);\n },\n readText: async ({ cache, signal, maxBytes, maxLines } = {}) => {\n const response = await fetchResource(\n source.url,\n cache ? { cache, signal } : { signal },\n );\n return readBoundedResponseText(response, { maxBytes, maxLines });\n },\n readStream: async ({ cache, signal } = {}) => {\n const response = await fetchResource(\n source.url,\n cache ? { cache, signal } : { signal },\n );\n validateFullContentResponse(response);\n if (!response.body) {\n if (response.status === 204 || response.status === 205) {\n return emptyByteStream();\n }\n throw new ResourceError({\n kind: \"unsupported_capability\",\n message: \"This response cannot be streamed.\",\n });\n }\n return response.body;\n },\n readRange: async (range, { cache, signal } = {}) => {\n validateByteRange(range);\n const { start, end } = range;\n const init = {\n signal,\n headers: { Range: `bytes=${start}-${end}` },\n };\n const response = await fetchResource(\n source.url,\n cache ? { ...init, cache } : init,\n );\n const buffer = await readResponseArrayBuffer(response);\n const contentRange = parseContentRange(\n response.headers.get(\"content-range\"),\n );\n validateUrlRangeResponse({\n bufferLength: buffer.byteLength,\n contentRange,\n range,\n status: response.status,\n });\n return {\n buffer,\n contentRange,\n isComplete: isByteRangeComplete({\n bufferLength: buffer.byteLength,\n contentRange,\n requestedLength: end - start + 1,\n status: response.status,\n }),\n };\n },\n });\n urlViewerResourceContentRegistry.set(keys.load, content);\n pruneUrlResourceContentRegistry();\n return content;\n}\n\nfunction emptyByteStream() {\n return new ReadableStream({\n start(controller) {\n controller.close();\n },\n });\n}\n\nfunction createBlobResource(\n source: BlobViewerSource,\n descriptor: ViewerDescriptor,\n keys: ViewerResourceKeys,\n): ViewerResource {\n const blob = source.blob;\n const content = internBlobResourceContent(source, keys);\n const originalDownload = source.downloadUrl\n ? createHrefDownloadAction({\n id: \"download-original\",\n label: \"Download\",\n href: source.downloadUrl,\n fileName: descriptor.fileName,\n })\n : createBlobDownloadAction({\n id: \"download-original\",\n label: \"Download\",\n blob,\n fileName: descriptor.fileName,\n });\n\n return resourceBase(source, descriptor, keys, { content, originalDownload });\n}\n\nfunction internBlobResourceContent(\n source: BlobViewerSource,\n keys: ViewerResourceKeys,\n): ViewerResourceContent {\n const blob = source.blob;\n let contents = blobViewerResourceContentRegistry.get(blob);\n if (!contents) {\n contents = new Map();\n blobViewerResourceContentRegistry.set(blob, contents);\n }\n\n const cached = contents.get(keys.load);\n if (cached) return cached;\n\n const content = resourceContentBase(source, keys, {\n directUrl: null,\n payload: { kind: \"blob\", blob },\n readBlob: async () => blob,\n readBytes: async () => blob.arrayBuffer(),\n readText: async ({ maxBytes, maxLines } = {}) =>\n readBoundedBlobText(blob, { maxBytes, maxLines }),\n readStream: async () => blob.stream(),\n readRange: async (range) => {\n validateByteRange(range);\n const { start, end } = range;\n validateKnownByteRangeStart(start, blob.size);\n const rangeBlob = blob.slice(start, end + 1);\n return {\n buffer: await rangeBlob.arrayBuffer(),\n contentRange: {\n start,\n end: Math.min(end, blob.size - 1),\n total: blob.size,\n },\n isComplete: end >= blob.size - 1,\n };\n },\n });\n contents.set(keys.load, content);\n return content;\n}\n\nfunction createTextResource(\n source: TextSource,\n descriptor: ViewerDescriptor,\n keys: ViewerResourceKeys,\n): ViewerResource {\n const content = internTextResourceContent(source, keys);\n const originalDownload = createTextDownloadAction({\n id: \"download-original\",\n label: \"Download\",\n text: source.text,\n fileName: descriptor.fileName,\n mimeType: descriptor.mimeType,\n });\n\n return resourceBase(source, descriptor, keys, { content, originalDownload });\n}\n\nfunction internTextResourceContent(\n source: TextSource,\n keys: ViewerResourceKeys,\n): ViewerResourceContent {\n const cached = textViewerResourceContentRegistry.get(keys.load);\n if (cached) return cached;\n\n const content = resourceContentBase(source, keys, {\n directUrl: null,\n payload: { kind: \"text\", text: source.text },\n readBlob: async () =>\n new Blob([source.text], {\n type: \"text/plain;charset=utf-8\",\n }),\n readBytes: async () =>\n typedArrayBuffer(new TextEncoder().encode(source.text)),\n readText: async ({ maxBytes, maxLines } = {}) =>\n readBoundedInlineText(source.text, { maxBytes, maxLines }),\n readStream: async () => new Blob([source.text]).stream(),\n readRange: async (range) => {\n validateByteRange(range);\n const { start, end } = range;\n const buffer = new TextEncoder().encode(source.text);\n validateKnownByteRangeStart(start, buffer.byteLength);\n const slice = buffer.slice(start, end + 1);\n return {\n buffer: typedArrayBuffer(slice),\n contentRange: {\n start,\n end: Math.min(end, buffer.byteLength - 1),\n total: buffer.byteLength,\n },\n isComplete: end >= buffer.byteLength - 1,\n };\n },\n });\n textViewerResourceContentRegistry.set(keys.load, content);\n pruneTextResourceContentRegistry();\n return content;\n}\n\nfunction resourceBase(\n source: ViewerSource,\n descriptor: ViewerDescriptor,\n keys: ViewerResourceKeys,\n options: {\n content: ViewerResourceContent;\n originalDownload: ViewerDownloadAction;\n },\n): ViewerResource {\n const { content, originalDownload } = options;\n return Object.freeze({\n descriptor,\n sourceKind: source.kind,\n keys,\n identityKey: descriptor.identityKey,\n fileName: descriptor.fileName,\n mimeType: descriptor.mimeType,\n content,\n originalDownload,\n });\n}\n\nfunction resourceContentBase(\n source: ViewerSource,\n keys: ViewerResourceKeys,\n methods: Omit,\n): ViewerResourceContent {\n return Object.freeze({\n key: keys.load,\n sourceKind: source.kind,\n mimeType: sourceMimeType(source),\n ...methods,\n });\n}\n\nfunction typedArrayBuffer(bytes: Uint8Array): ArrayBuffer {\n return bytes.buffer.slice(\n bytes.byteOffset,\n bytes.byteOffset + bytes.byteLength,\n ) as ArrayBuffer;\n}\n\nasync function fetchResource(\n input: RequestInfo | URL,\n init?: RequestInit,\n): Promise {\n let response: Response;\n try {\n response = await fetch(input, init);\n } catch (error) {\n if (isAbortError(error)) {\n throw new ResourceError({\n kind: \"aborted\",\n message: \"Loading was cancelled.\",\n cause: error,\n });\n }\n throw new ResourceError({\n kind: \"fetch_failed\",\n message: \"Could not fetch this resource.\",\n cause: error,\n });\n }\n\n if (!response.ok && response.status !== 206) {\n throw new ResourceError({\n kind: \"http_error\",\n message: `Failed to load resource: ${response.status}`,\n status: response.status,\n });\n }\n\n return response;\n}\n\nasync function readBoundedResponseText(\n response: Response,\n bounds: { maxBytes?: number; maxLines?: number },\n) {\n validateFullContentResponse(response);\n\n const maxBytes = bounds.maxBytes;\n if (\n isContentLengthOverLimit(response.headers.get(\"content-length\"), maxBytes)\n ) {\n throw tooLarge(\"bytes\");\n }\n\n const body = response.body;\n if (!body) {\n const buffer = await readResponseArrayBuffer(response);\n if (maxBytes != null && buffer.byteLength > maxBytes) {\n throw tooLarge(\"bytes\");\n }\n const text = new TextDecoder().decode(buffer);\n assertLineLimit(text, bounds.maxLines);\n return text;\n }\n\n const reader = body.getReader();\n const decoder = new TextDecoder();\n const lineLimitTracker = createLineLimitTracker(bounds.maxLines);\n let receivedBytes = 0;\n let text = \"\";\n\n while (true) {\n const { done, value } = await readResponseStreamChunk(reader);\n if (done) break;\n receivedBytes += value.byteLength;\n if (maxBytes != null && receivedBytes > maxBytes) {\n await cancelReaderSilently(reader);\n throw tooLarge(\"bytes\");\n }\n const chunkText = decoder.decode(value, { stream: true });\n try {\n lineLimitTracker.push(chunkText);\n } catch (error) {\n await cancelReaderSilently(reader);\n throw error;\n }\n text += chunkText;\n }\n\n const finalText = decoder.decode();\n lineLimitTracker.push(finalText);\n text += finalText;\n return text;\n}\n\nfunction isContentLengthOverLimit(\n contentLength: string | null,\n maxBytes: number | undefined,\n) {\n if (maxBytes == null || contentLength == null) return false;\n\n const normalizedLength = contentLength.trim().replace(/^0+(?=\\d)/, \"\");\n if (!/^\\d+$/.test(normalizedLength)) return false;\n\n const maxLength = String(maxBytes);\n return (\n normalizedLength.length > maxLength.length ||\n (normalizedLength.length === maxLength.length &&\n normalizedLength > maxLength)\n );\n}\n\nfunction validateFullContentResponse(response: Response) {\n if (response.status !== 206) return;\n\n const contentRange = parseContentRange(response.headers.get(\"content-range\"));\n if (\n contentRange?.total != null &&\n contentRange.start === 0 &&\n contentRange.end === contentRange.total - 1\n ) {\n return;\n }\n\n throw new ResourceError({\n kind: \"partial_content\",\n message: \"Full response returned partial content.\",\n status: response.status,\n });\n}\n\nasync function readResponseStreamChunk(\n reader: ReadableStreamDefaultReader,\n) {\n try {\n return await reader.read();\n } catch (error) {\n throw resourceReadError(error);\n }\n}\n\nasync function readResponseArrayBuffer(response: Response) {\n try {\n return await response.arrayBuffer();\n } catch (error) {\n throw resourceReadError(error);\n }\n}\n\nasync function readResponseBlob(response: Response) {\n try {\n return await response.blob();\n } catch (error) {\n throw resourceReadError(error);\n }\n}\n\nfunction resourceReadError(error: unknown) {\n if (isAbortError(error)) {\n return new ResourceError({\n kind: \"aborted\",\n message: \"Loading was cancelled.\",\n cause: error,\n });\n }\n return new ResourceError({\n kind: \"fetch_failed\",\n message: \"Could not read this resource.\",\n cause: error,\n });\n}\n\nasync function readBoundedBlobText(\n blob: Blob,\n bounds: { maxBytes?: number; maxLines?: number },\n) {\n if (bounds.maxBytes != null && blob.size > bounds.maxBytes) {\n throw tooLarge(\"bytes\");\n }\n const text = await blob.text();\n assertLineLimit(text, bounds.maxLines);\n return text;\n}\n\nfunction readBoundedInlineText(\n text: string,\n { maxBytes, maxLines }: { maxBytes?: number; maxLines?: number },\n) {\n // For inline sources the string *is* the resource, so its UTF-8 byte length\n // is the authoritative size to measure against maxBytes.\n if (\n maxBytes != null &&\n new TextEncoder().encode(text).byteLength > maxBytes\n ) {\n throw tooLarge(\"bytes\");\n }\n assertLineLimit(text, maxLines);\n return text;\n}\n\n// Used after a transferred-byte check has already enforced maxBytes (URL/blob).\n// Re-encoding the decoded text here would double-count: invalid UTF-8 decodes to\n// U+FFFD (3 bytes each), inflating the measured size past the real wire bytes\n// and falsely rejecting small resources as \"too large\".\nfunction assertLineLimit(text: string, maxLines: number | undefined) {\n if (\n maxLines != null &&\n text.split(TEXT_LINE_BREAK_PATTERN).length > maxLines\n ) {\n throw tooLarge(\"lines\");\n }\n}\n\nfunction tooLarge(reason: ResourceTooLargeReason) {\n return new ResourceError({\n kind: \"too_large\",\n tooLargeReason: reason,\n message: `Resource exceeds ${reason} limit.`,\n });\n}\n\nasync function cancelReaderSilently(\n reader: ReadableStreamDefaultReader,\n) {\n try {\n await reader.cancel();\n } catch {\n // Preserve the user-facing load failure; cancellation is best-effort cleanup.\n }\n}\n\nfunction validateByteRange({ start, end }: ByteRange) {\n if (\n !Number.isSafeInteger(start) ||\n !Number.isSafeInteger(end) ||\n start < 0 ||\n end < start\n ) {\n throw new ResourceError({\n kind: \"invalid_range\",\n message: \"Byte range must use non-negative integer bounds.\",\n });\n }\n}\n\nfunction validateKnownByteRangeStart(start: number, total: number) {\n if (start > 0 && start >= total) {\n throw new ResourceError({\n kind: \"invalid_range\",\n message: \"Byte range starts past the available resource.\",\n });\n }\n}\n\nfunction validateUrlRangeResponse({\n bufferLength,\n contentRange,\n range,\n status,\n}: {\n bufferLength: number;\n contentRange: ByteRangeResult[\"contentRange\"];\n range: ByteRange;\n status: number;\n}) {\n if (status === 200) {\n if (range.start !== 0 || bufferLength > range.end - range.start + 1) {\n throw new ResourceError({\n kind: \"invalid_range\",\n message: \"Full response does not match the requested byte range.\",\n });\n }\n return;\n }\n if (status !== 206) {\n throw new ResourceError({\n kind: \"invalid_range\",\n message: \"Range response must return full or partial content.\",\n });\n }\n if (!contentRange) {\n throw new ResourceError({\n kind: \"invalid_range\",\n message: \"Partial content response is missing a valid byte range.\",\n });\n }\n const declaredLength = contentRange.end - contentRange.start + 1;\n if (\n contentRange.start !== range.start ||\n contentRange.end < contentRange.start ||\n contentRange.end > range.end ||\n (contentRange.total != null && contentRange.end >= contentRange.total) ||\n declaredLength !== bufferLength\n ) {\n throw new ResourceError({\n kind: \"invalid_range\",\n message: \"Response byte range does not match the requested range.\",\n });\n }\n}\n\nfunction isByteRangeComplete({\n bufferLength,\n contentRange,\n requestedLength,\n status,\n}: {\n bufferLength: number;\n contentRange: ByteRangeResult[\"contentRange\"];\n requestedLength: number;\n status: number;\n}) {\n if (status === 200) return true;\n if (contentRange?.total != null) {\n if (contentRange.total <= 0) return true;\n return contentRange.end >= contentRange.total - 1;\n }\n if (contentRange) return false;\n return bufferLength < requestedLength;\n}\n\nfunction isStandaloneLineBreak(character: string) {\n const code = character.charCodeAt(0);\n return code === 0x0a || code === 0x2028 || code === 0x2029;\n}\n\nfunction createLineLimitTracker(maxLines: number | undefined) {\n let lineCount = 1;\n let previousWasCR = false;\n\n return {\n push(text: string) {\n if (maxLines == null || text.length === 0) return;\n\n for (const character of text) {\n if (previousWasCR) {\n previousWasCR = false;\n if (character === \"\\n\") continue;\n }\n\n if (character === \"\\r\") {\n lineCount += 1;\n previousWasCR = true;\n } else if (isStandaloneLineBreak(character)) {\n // LF, plus LINE/PARAGRAPH SEPARATOR (U+2028/U+2029); none pair with CR.\n lineCount += 1;\n }\n\n if (lineCount > maxLines) {\n throw tooLarge(\"lines\");\n }\n }\n },\n };\n}\n\nfunction parseContentRange(value: string | null) {\n if (!value) return undefined;\n const match = value.match(/^bytes\\s+(\\d+)-(\\d+)\\/(\\d+|\\*)\\s*$/i);\n if (!match) return undefined;\n const start = parseContentRangeNumber(match[1]);\n const end = parseContentRangeNumber(match[2]);\n const total =\n match[3] === \"*\" ? null : parseContentRangeNumber(match[3] ?? \"\");\n if (start == null || end == null || total === undefined) return undefined;\n return {\n start,\n end,\n total,\n };\n}\n\nfunction parseContentRangeNumber(value: string) {\n const number = Number(value);\n return Number.isSafeInteger(number) ? number : undefined;\n}\n", "type": "registry:lib", "target": "@lib/viewer-resource.ts" }, { "path": "registry/new-york-v4/lib/viewer-download-actions.ts", "content": "export type ViewerDownloadOrigin = \"original\" | \"derived\";\n\nexport type ViewerDownloadPayload =\n | { kind: \"href\"; href: string }\n | { kind: \"blob\"; blob: Blob }\n | { kind: \"text\"; text: string; mimeType?: string }\n | { kind: \"none\" };\n\nexport interface ViewerDownloadAction {\n id: string;\n label: string;\n fileName: string;\n origin: ViewerDownloadOrigin;\n isDisabled?: boolean;\n getPayload: (options?: {\n signal?: AbortSignal;\n }) => ViewerDownloadPayload | Promise;\n}\n\nexport type ViewerDownloadErrorKind =\n | \"disabled\"\n | \"aborted\"\n | \"payload_failed\"\n | \"unsupported\";\n\nexport class ViewerDownloadError extends Error {\n readonly kind: ViewerDownloadErrorKind;\n readonly actionId: string;\n override readonly cause?: unknown;\n\n constructor({\n actionId,\n kind,\n message,\n cause,\n }: {\n actionId: string;\n kind: ViewerDownloadErrorKind;\n message: string;\n cause?: unknown;\n }) {\n super(message);\n this.name = \"ViewerDownloadError\";\n this.actionId = actionId;\n this.kind = kind;\n this.cause = cause;\n }\n}\n\nexport function createHrefDownloadAction({\n id,\n label = \"Download\",\n href,\n fileName,\n origin = \"original\",\n}: {\n id: string;\n label?: string;\n href: string;\n fileName: string;\n origin?: ViewerDownloadOrigin;\n}): ViewerDownloadAction {\n return {\n id,\n label,\n fileName,\n origin,\n getPayload: () => ({ kind: \"href\", href }),\n };\n}\n\nexport function createBlobDownloadAction({\n id,\n label = \"Download\",\n blob,\n fileName,\n origin = \"original\",\n}: {\n id: string;\n label?: string;\n blob: Blob;\n fileName: string;\n origin?: ViewerDownloadOrigin;\n}): ViewerDownloadAction {\n return {\n id,\n label,\n fileName,\n origin,\n getPayload: () => ({ kind: \"blob\", blob }),\n };\n}\n\nexport function createTextDownloadAction({\n id,\n label = \"Download\",\n text,\n fileName,\n mimeType,\n origin = \"original\",\n}: {\n id: string;\n label?: string;\n text: string;\n fileName: string;\n mimeType?: string;\n origin?: ViewerDownloadOrigin;\n}): ViewerDownloadAction {\n return {\n id,\n label,\n fileName,\n origin,\n getPayload: () => ({ kind: \"text\", text, mimeType }),\n };\n}\n\nexport function createDisabledDownloadAction({\n id,\n label = \"Download\",\n fileName,\n origin = \"original\",\n}: {\n id: string;\n label?: string;\n fileName: string;\n origin?: ViewerDownloadOrigin;\n}): ViewerDownloadAction {\n return {\n id,\n label,\n fileName,\n origin,\n isDisabled: true,\n getPayload: () => ({ kind: \"none\" }),\n };\n}\n", "type": "registry:lib", "target": "@lib/viewer-download-actions.ts" }, { "path": "registry/new-york-v4/lib/viewer-errors.ts", "content": "export type ViewerFormat =\n | \"pdf\"\n | \"image\"\n | \"text\"\n | \"csv\"\n | \"docx\"\n | \"xlsx\"\n | \"pptx\"\n | \"file\";\n\nexport type ViewerErrorDomain =\n | \"resource\"\n | \"format\"\n | \"state\"\n | \"unsupported\"\n | \"unknown\";\n\nexport type ResourceErrorKind =\n | \"fetch_failed\"\n | \"http_error\"\n | \"aborted\"\n | \"invalid_range\"\n | \"partial_content\"\n | \"too_large\"\n | \"unsupported_capability\"\n | \"unknown\";\n\nexport type ResourceTooLargeReason = \"bytes\" | \"lines\";\n\nexport class ResourceError extends Error {\n readonly domain = \"resource\";\n readonly kind: ResourceErrorKind;\n readonly status?: number;\n readonly tooLargeReason?: ResourceTooLargeReason;\n override readonly cause?: unknown;\n\n constructor({\n kind,\n message,\n status,\n tooLargeReason,\n cause,\n }: {\n kind: ResourceErrorKind;\n message: string;\n status?: number;\n tooLargeReason?: ResourceTooLargeReason;\n cause?: unknown;\n }) {\n super(message);\n this.name = \"ResourceError\";\n this.kind = kind;\n this.status = status;\n this.tooLargeReason = tooLargeReason;\n this.cause = cause;\n }\n}\n\nexport type ViewerFormatErrorKind =\n | \"bounds\"\n | \"decode_failed\"\n | \"disposed\"\n | \"index_out_of_range\"\n | \"load_failed\"\n | \"parse_failed\"\n | \"render_failed\"\n | \"worker_failed\"\n | \"unknown\";\n\nexport interface ViewerFormatErrorMapperOptions {\n kind: ViewerFormatErrorKind;\n message: string;\n}\n\nexport class ViewerFormatError extends Error {\n readonly domain = \"format\";\n readonly format: ViewerFormat;\n readonly kind: ViewerFormatErrorKind;\n override readonly cause?: unknown;\n\n constructor({\n format,\n kind,\n message,\n cause,\n }: {\n format: ViewerFormat;\n kind: ViewerFormatErrorKind;\n message: string;\n cause?: unknown;\n }) {\n super(message);\n this.name = \"ViewerFormatError\";\n this.format = format;\n this.kind = kind;\n this.cause = cause;\n }\n}\n\nexport type ViewerStateErrorKind =\n | \"invalid_bounds\"\n | \"invalid_target\"\n | \"out_of_range\"\n | \"stale_resource\"\n | \"unknown\";\n\nexport class ViewerStateError extends Error {\n readonly domain = \"state\";\n readonly format?: ViewerFormat;\n readonly kind: ViewerStateErrorKind;\n override readonly cause?: unknown;\n\n constructor({\n format,\n kind,\n message,\n cause,\n }: {\n format?: ViewerFormat;\n kind: ViewerStateErrorKind;\n message: string;\n cause?: unknown;\n }) {\n super(message);\n this.name = \"ViewerStateError\";\n this.format = format;\n this.kind = kind;\n this.cause = cause;\n }\n}\n\nexport class ViewerUnsupportedError extends Error {\n readonly domain = \"unsupported\";\n readonly format?: ViewerFormat;\n readonly sourceKind?: string;\n override readonly cause?: unknown;\n\n constructor({\n format,\n sourceKind,\n message,\n cause,\n }: {\n format?: ViewerFormat;\n sourceKind?: string;\n message: string;\n cause?: unknown;\n }) {\n super(message);\n this.name = \"ViewerUnsupportedError\";\n this.format = format;\n this.sourceKind = sourceKind;\n this.cause = cause;\n }\n}\n\nexport interface ViewerErrorInfo {\n domain: ViewerErrorDomain;\n format?: ViewerFormat;\n kind: string;\n message: string;\n status?: number;\n isRetryable: boolean;\n isDownloadUseful: boolean;\n userMessage: string;\n cause?: unknown;\n}\n\nexport interface ViewerErrorContext {\n format?: ViewerFormat;\n sourceKind?: \"url\" | \"blob\" | \"text\";\n canDownload?: boolean;\n retry?: \"auto\" | \"always\" | \"never\";\n}\n\nexport function isAbortError(error: unknown): boolean {\n return (\n (error instanceof DOMException && error.name === \"AbortError\") ||\n (error instanceof Error && error.name === \"AbortError\")\n );\n}\n\nexport function isResourceError(error: unknown): error is ResourceError {\n return (\n error instanceof ResourceError ||\n isErrorLike(error, \"ResourceError\", \"resource\")\n );\n}\n\nexport function isViewerFormatError(\n error: unknown,\n): error is ViewerFormatError {\n return (\n error instanceof ViewerFormatError ||\n isErrorLike(error, \"ViewerFormatError\", \"format\")\n );\n}\n\nexport function isViewerStateError(error: unknown): error is ViewerStateError {\n return (\n error instanceof ViewerStateError ||\n isErrorLike(error, \"ViewerStateError\", \"state\")\n );\n}\n\nexport function isViewerUnsupportedError(\n error: unknown,\n): error is ViewerUnsupportedError {\n return (\n error instanceof ViewerUnsupportedError ||\n isErrorLike(error, \"ViewerUnsupportedError\", \"unsupported\")\n );\n}\n\nexport function toViewerErrorInfo(\n error: unknown,\n context: ViewerErrorContext = {},\n): ViewerErrorInfo {\n const canDownload = context.canDownload ?? true;\n\n if (isResourceError(error)) {\n return {\n domain: \"resource\",\n format: context.format,\n kind: error.kind,\n message: error.message,\n status: error.status,\n isRetryable: retryable(\n context,\n resourceErrorDefaultRetry(error, context),\n ),\n isDownloadUseful: canDownload && error.kind !== \"aborted\",\n userMessage: resourceErrorUserMessage(error),\n cause: error.cause,\n };\n }\n\n if (isViewerFormatError(error)) {\n const format = error.format ?? context.format;\n return {\n domain: \"format\",\n format,\n kind: error.kind,\n message: error.message,\n isRetryable: retryable(\n context,\n formatErrorDefaultRetry(error, context, format),\n ),\n isDownloadUseful: canDownload,\n userMessage: formatErrorUserMessage(format, error.kind, error),\n cause: error.cause,\n };\n }\n\n if (isViewerStateError(error)) {\n return {\n domain: \"state\",\n format: error.format ?? context.format,\n kind: error.kind,\n message: error.message,\n isRetryable: retryable(context, false),\n isDownloadUseful: canDownload,\n userMessage: stateErrorUserMessage(error.kind),\n cause: error.cause,\n };\n }\n\n if (isViewerUnsupportedError(error)) {\n return {\n domain: \"unsupported\",\n format: error.format ?? context.format,\n kind: \"unsupported\",\n message: error.message,\n isRetryable: retryable(context, false),\n isDownloadUseful: canDownload,\n userMessage: \"This file cannot be previewed here.\",\n cause: error.cause,\n };\n }\n\n if (isAbortError(error)) {\n return {\n domain: \"resource\",\n format: context.format,\n kind: \"aborted\",\n message: \"Loading was cancelled.\",\n isRetryable: retryable(context, false),\n isDownloadUseful: false,\n userMessage: \"Loading was cancelled.\",\n cause: error,\n };\n }\n\n const message = error instanceof Error ? error.message : String(error);\n return {\n domain: \"unknown\",\n format: context.format,\n kind: \"unknown\",\n message,\n isRetryable: retryable(context, unknownErrorDefaultRetry(context)),\n isDownloadUseful: canDownload,\n userMessage: fallbackUserMessage(context.format),\n cause: error,\n };\n}\n\nfunction isErrorLike(error: unknown, name: string, domain: ViewerErrorDomain) {\n if (!error || typeof error !== \"object\") return false;\n const candidate = error as {\n name?: unknown;\n domain?: unknown;\n kind?: unknown;\n };\n return (\n (candidate.name === name || candidate.domain === domain) &&\n typeof candidate.kind === \"string\"\n );\n}\n\nfunction retryable(context: ViewerErrorContext, fallback: boolean) {\n if (context.retry === \"always\") return true;\n if (context.retry === \"never\") return false;\n return fallback;\n}\n\nfunction resourceErrorDefaultRetry(\n error: ResourceError,\n context: ViewerErrorContext,\n) {\n if (error.kind === \"aborted\") return false;\n if (error.kind === \"invalid_range\") return false;\n if (error.kind === \"too_large\") return false;\n if (error.kind === \"unsupported_capability\") return false;\n return context.sourceKind === \"url\";\n}\n\nfunction formatErrorDefaultRetry(\n error: ViewerFormatError,\n context: ViewerErrorContext,\n format: ViewerFormat | undefined,\n) {\n if (format === \"text\" && error.kind === \"bounds\") return false;\n if (error.kind === \"disposed\") return false;\n if (error.kind === \"index_out_of_range\") return false;\n if (format === \"docx\") return true;\n return context.sourceKind === \"url\";\n}\n\nfunction unknownErrorDefaultRetry(context: ViewerErrorContext) {\n if (context.format === \"docx\") return true;\n return context.sourceKind === \"url\";\n}\n\nfunction resourceErrorUserMessage(error: ResourceError) {\n if (error.kind === \"http_error\") {\n return error.status\n ? `Failed to load file: ${error.status}.`\n : \"Couldn't load this file.\";\n }\n if (error.kind === \"fetch_failed\") return \"Couldn't load this file.\";\n if (error.kind === \"aborted\") return \"Loading was cancelled.\";\n if (error.kind === \"invalid_range\") return \"This source range is invalid.\";\n if (error.kind === \"too_large\") {\n return error.tooLargeReason === \"lines\"\n ? \"This file has too many lines to preview.\"\n : \"This file is too large to preview.\";\n }\n if (error.kind === \"partial_content\") {\n return \"This source returned partial content and cannot be previewed here.\";\n }\n if (error.kind === \"unsupported_capability\") {\n return \"This source cannot be previewed here.\";\n }\n return \"Couldn't load this file.\";\n}\n\nfunction formatErrorUserMessage(\n format: ViewerFormat | undefined,\n kind: string,\n error?: unknown,\n) {\n if (format === \"pdf\") return \"Couldn't load this PDF.\";\n if (format === \"image\") {\n if (kind === \"index_out_of_range\")\n return \"This image page is out of range.\";\n if (kind === \"decode_failed\") return \"Couldn't decode this image.\";\n return \"Couldn't load this image.\";\n }\n if (format === \"text\") {\n if (kind === \"render_failed\") return \"Couldn't render this text file.\";\n if (kind === \"bounds\") {\n const boundsError = error as {\n reason?: unknown;\n boundName?: unknown;\n };\n if (boundsError.reason === \"lines\") {\n return \"This text file has too many lines to preview.\";\n }\n if (boundsError.reason === \"bytes\") {\n return \"This text file is too large to preview.\";\n }\n if (typeof boundsError.boundName === \"string\") {\n return \"Text viewer bounds are invalid.\";\n }\n }\n return \"Couldn't load this text file.\";\n }\n if (format === \"csv\") return \"Couldn't parse this table.\";\n if (format === \"docx\") return \"Couldn't render this document.\";\n if (format === \"xlsx\") return \"Couldn't parse this spreadsheet.\";\n if (format === \"pptx\") {\n if (kind === \"render_failed\") return \"Couldn't render this slide.\";\n return \"Couldn't load this presentation.\";\n }\n return \"Couldn't load this file.\";\n}\n\nfunction stateErrorUserMessage(kind: ViewerStateErrorKind) {\n if (kind === \"invalid_bounds\") return \"Viewer bounds are invalid.\";\n if (kind === \"invalid_target\") return \"The requested target is invalid.\";\n if (kind === \"out_of_range\") return \"The requested item is out of range.\";\n if (kind === \"stale_resource\") return \"This viewer state is no longer valid.\";\n return \"Couldn't load this file.\";\n}\n\nfunction fallbackUserMessage(format: ViewerFormat | undefined) {\n if (format === \"pdf\") return \"Couldn't load this PDF.\";\n if (format === \"image\") return \"Couldn't load this image.\";\n if (format === \"text\") return \"Couldn't load this text file.\";\n if (format === \"csv\") return \"Couldn't parse this table.\";\n if (format === \"docx\") return \"Couldn't load this document.\";\n if (format === \"xlsx\") return \"Couldn't load this spreadsheet.\";\n if (format === \"pptx\") return \"Couldn't load this presentation.\";\n return \"Couldn't load this file.\";\n}\n", "type": "registry:lib", "target": "@lib/viewer-errors.ts" }, { "path": "registry/new-york-v4/ui/csv-viewer-chrome.tsx", "content": "\"use client\";\n\nimport * as React from \"react\";\n\nimport { cn } from \"@/lib/utils\";\nimport type { ViewerDownloadAction } from \"@/lib/viewer-download-actions\";\nimport type { ViewerResource } from \"@/lib/viewer-resource\";\n\nimport type { CsvResourceState } from \"./csv-viewer-state\";\nimport { ViewerErrorState } from \"./viewer-error\";\nimport { ViewerControls } from \"./viewer-controls\";\n\nexport const CSV_VIEWER_BASE_FONT_SIZE = 13;\n\nexport function CsvViewerFrame({\n className,\n frame,\n fillHeight,\n zoom,\n children,\n}: {\n className?: string;\n frame: boolean;\n fillHeight: boolean;\n zoom: number;\n children: React.ReactNode;\n}) {\n return (\n \n {children}\n \n );\n}\n\nexport function CsvViewerHeader({\n controls,\n rowCount,\n columnCount,\n isLoading,\n zoom,\n downloadActions,\n onZoomChange,\n}: {\n controls: boolean;\n rowCount: number;\n columnCount: number;\n isLoading: boolean;\n zoom: number;\n downloadActions: ViewerDownloadAction[];\n onZoomChange: (zoom: number) => void;\n}) {\n const rowLabel = `${rowCount.toLocaleString()} row${rowCount === 1 ? \"\" : \"s\"}`;\n const columnLabel = `${columnCount} column${columnCount === 1 ? \"\" : \"s\"}`;\n\n return controls ? (\n onZoomChange(clampCsvZoom(zoom / 1.2)),\n onZoomIn: () => onZoomChange(clampCsvZoom(zoom * 1.2)),\n onFit: () => onZoomChange(1),\n fitLabel: \"Reset zoom\",\n }}\n downloads={downloadActions}\n />\n ) : null;\n}\n\nfunction clampCsvZoom(zoom: number) {\n return Math.max(0.1, Math.min(5, zoom));\n}\n\nexport function csvViewerStatusNode({\n resourceState,\n resource,\n rowCount,\n showDownload,\n onRetry,\n}: {\n resourceState: CsvResourceState;\n resource: ViewerResource | null;\n rowCount: number;\n showDownload: boolean;\n onRetry: () => void;\n}): React.ReactNode {\n if (resourceState.status === \"error\") {\n return (\n \n );\n }\n\n if (rowCount === 0 && resourceState.status !== \"loading\") {\n return (\n
\n No rows\n
\n );\n }\n\n return null;\n}\n", "type": "registry:ui", "target": "@ui/csv-viewer-chrome.tsx" }, { "path": "registry/new-york-v4/ui/viewer-error.tsx", "content": "\"use client\";\n\nimport * as React from \"react\";\nimport { RotateCcw } from \"lucide-react\";\n\nimport { cn } from \"@/lib/utils\";\nimport type { ViewerDownloadAction } from \"@/lib/viewer-download-actions\";\nimport {\n toViewerErrorInfo,\n type ViewerErrorContext,\n type ViewerFormat,\n} from \"@/lib/viewer-errors\";\n\nimport { Button } from \"./button\";\nimport { ViewerDownloadButton } from \"./viewer-download\";\n\nexport interface ViewerErrorStateProps extends ViewerErrorContext {\n error: unknown;\n download?: ViewerDownloadAction | null;\n className?: string;\n bare?: boolean;\n variant?: \"card\" | \"document\" | \"inline\";\n onRetry?: () => void;\n}\n\nexport function ViewerErrorState({\n error,\n format,\n sourceKind,\n canDownload,\n retry,\n download,\n className,\n bare = false,\n variant = \"card\",\n onRetry,\n}: ViewerErrorStateProps) {\n const info = toViewerErrorInfo(error, {\n format,\n sourceKind,\n canDownload: canDownload ?? Boolean(download && !download.isDisabled),\n retry,\n });\n const showRetry = info.isRetryable && onRetry;\n const showDownload =\n info.isDownloadUseful && download != null && !download.isDisabled;\n\n return (\n \n

{info.userMessage}

\n {showRetry || showDownload ? (\n
\n {showRetry ? (\n \n ) : null}\n {showDownload ? (\n \n ) : null}\n
\n ) : null}\n \n );\n}\n\nexport interface ViewerErrorBoundaryProps extends ViewerErrorContext {\n children: React.ReactNode;\n resetKey?: unknown;\n download?: ViewerDownloadAction | null;\n className?: string;\n bare?: boolean;\n variant?: \"card\" | \"document\" | \"inline\";\n mapError?: (error: unknown) => unknown;\n onRetry?: (error: unknown) => void;\n onCaughtError?: (error: unknown, errorInfo: React.ErrorInfo) => void;\n}\n\nexport class ViewerErrorBoundary extends React.Component<\n ViewerErrorBoundaryProps,\n { error: unknown | null; retryKey: number }\n> {\n state: Readonly<{ error: unknown | null; retryKey: number }> = {\n error: null,\n retryKey: 0,\n };\n\n componentDidUpdate(previousProps: ViewerErrorBoundaryProps) {\n if (\n previousProps.resetKey !== this.props.resetKey &&\n this.state.error != null\n ) {\n this.setState({ error: null });\n }\n }\n\n static getDerivedStateFromError(error: unknown) {\n return { error };\n }\n\n componentDidCatch(error: unknown, errorInfo: React.ErrorInfo) {\n this.props.onCaughtError?.(error, errorInfo);\n }\n\n render() {\n if (this.state.error != null) {\n const error = this.props.mapError\n ? this.props.mapError(this.state.error)\n : this.state.error;\n\n return (\n {\n this.setState((state) => ({\n error: null,\n retryKey: state.retryKey + 1,\n }));\n this.props.onRetry?.(this.state.error);\n }}\n />\n );\n }\n\n return (\n \n {this.props.children}\n \n );\n }\n}\n\nfunction errorStateClassName({\n bare,\n variant,\n}: {\n bare: boolean;\n variant: \"card\" | \"document\" | \"inline\";\n}) {\n if (variant === \"inline\") {\n return \"flex h-24 flex-col items-center justify-center gap-3 px-3 text-center text-xs text-muted-foreground\";\n }\n return cn(\n \"flex min-h-64 flex-col items-center justify-center gap-3 p-6 text-center text-sm text-muted-foreground\",\n bare ? \"bg-muted/20\" : \"rounded-xl border bg-muted/30\",\n variant === \"document\" && \"min-h-full\",\n );\n}\n\nexport type { ViewerFormat };\n", "type": "registry:ui", "target": "@ui/viewer-error.tsx" }, { "path": "registry/new-york-v4/ui/csv-viewer-download.ts", "content": "import {\n isTabDelimited,\n normalizeCsvDelimiter,\n type CsvDialect,\n} from \"@/lib/csv\";\nimport type { ViewerDownloadAction } from \"@/lib/viewer-download-actions\";\nimport type { ViewerResource } from \"@/lib/viewer-resource\";\n\nexport function escapeDelimitedField(value: string, delimiter: string): string {\n const text = value ?? \"\";\n return text.includes(delimiter) || /[\"\\r\\n]/.test(text)\n ? `\"${text.replace(/\"/g, '\"\"')}\"`\n : text;\n}\n\nexport function serializeCsvTable({\n columns,\n sourceRows,\n dialect,\n}: {\n columns: string[];\n sourceRows: string[][];\n dialect: CsvDialect;\n}): string {\n const delimiter =\n normalizeCsvDelimiter(dialect.delimiter) ?? dialect.delimiter;\n const lines = [\n columns\n .map((value) => escapeDelimitedField(value, delimiter))\n .join(delimiter),\n ];\n for (const sourceRow of sourceRows) {\n const row = fitExportRow(sourceRow, columns.length);\n lines.push(\n row\n .map((value) => escapeDelimitedField(value, delimiter))\n .join(delimiter),\n );\n }\n return lines.join(\"\\r\\n\");\n}\n\nfunction fitExportRow(row: string[], columnCount: number): string[] {\n const out = row.slice(0, columnCount);\n while (out.length < columnCount) out.push(\"\");\n return out;\n}\n\nexport function defaultCsvDownloadName(dialect: CsvDialect): string {\n return isTabDelimited(dialect) ? \"data.tsv\" : \"data.csv\";\n}\n\nexport function createCsvExportAction({\n columns,\n sourceRows,\n dialect,\n fileName,\n isDisabled,\n}: {\n columns: string[];\n sourceRows: string[][];\n dialect: CsvDialect;\n fileName: string;\n isDisabled?: boolean;\n}): ViewerDownloadAction {\n return {\n id: \"csv-export-table\",\n label: \"Export table\",\n fileName,\n origin: \"derived\",\n isDisabled,\n getPayload: () => ({\n kind: \"text\",\n text: serializeCsvTable({ columns, sourceRows, dialect }),\n mimeType: isTabDelimited(dialect)\n ? \"text/tab-separated-values;charset=utf-8\"\n : \"text/csv;charset=utf-8\",\n }),\n };\n}\n\nexport function csvViewerDownloadActions({\n resource,\n columns,\n sourceRows,\n dialect,\n fileName,\n canExportTable,\n}: {\n resource: ViewerResource | null;\n columns: string[];\n sourceRows: string[][];\n dialect: CsvDialect;\n fileName: string;\n canExportTable: boolean;\n}): ViewerDownloadAction[] {\n const actions: ViewerDownloadAction[] = [];\n if (resource) {\n actions.push({\n ...resource.originalDownload,\n label: \"Download original\",\n });\n }\n actions.push(\n createCsvExportAction({\n columns,\n sourceRows,\n dialect,\n fileName,\n isDisabled: !canExportTable,\n }),\n );\n return actions;\n}\n", "type": "registry:ui", "target": "@ui/csv-viewer-download.ts" }, { "path": "registry/new-york-v4/ui/csv-viewer-grid.tsx", "content": "\"use client\";\n\nimport * as React from \"react\";\nimport { ChevronDown, ChevronUp } from \"lucide-react\";\n\nimport { useKeyedMountEffect } from \"@/hooks/use-keyed-mount-effect\";\nimport { cn } from \"@/lib/utils\";\n\nimport {\n useCsvRowPatcher,\n type CsvRowPatchState,\n} from \"./csv-viewer-row-patcher\";\nimport { csvCellClassName } from \"./csv-viewer-cell-classes\";\nimport type { CsvRowStore } from \"./csv-row-store\";\nimport {\n CSV_SCROLLBAR_CSS,\n HeaderAwareScrollbar,\n} from \"./csv-viewer-scrollbar\";\nimport {\n sortCsvRowsInWorker,\n sortCsvRowsOnMainThread,\n} from \"./csv-viewer-sort-worker\";\nimport type { CsvCellAddress } from \"./csv-viewer-state\";\nimport { CsvStyleScope } from \"./csv-viewer-style-scope\";\nimport { fixedGridColumnWidths } from \"./fixed-grid-columns\";\nimport { getFixedGridCanvasStyle } from \"./fixed-grid-layout\";\nimport {\n FixedGridNativeFindIndex,\n type FixedGridNativeFindCellAddress,\n} from \"./fixed-grid-native-find-index\";\nimport { FixedGridRowWindow } from \"./fixed-grid-row-window\";\nimport { getFixedGridRowStyle } from \"./fixed-grid-row-style\";\nimport { buildVirtualGridTemplate } from \"./fixed-grid-template\";\nimport { FixedGridViewport } from \"./fixed-grid-viewport\";\nimport {\n useFixedGridVirtualization,\n useFixedRowPool,\n type FixedGridColumnItem,\n type FixedGridRowPoolSlot,\n} from \"./fixed-grid-virtualization\";\nimport { joinEffectKey } from \"@/lib/effect-key\";\n\ntype Row = string[] | undefined;\n\nexport interface CsvGridHandle {\n scrollToCell: (\n cellAddress: CsvCellAddress,\n options?: { behavior?: ScrollBehavior },\n ) => void;\n getViewportElement: () => HTMLDivElement | null;\n}\n\nexport interface CsvGridProps {\n columns: string[];\n rowStore: CsvRowStore;\n activeCell: CsvCellAddress | null;\n height: number;\n fillHeight: boolean;\n isolateStyles: boolean;\n scale: number;\n sortResetKey: unknown;\n statusNode: React.ReactNode;\n}\n\nconst CSV_TABLE_LABEL = \"CSV data\";\nconst COLUMN_WIDTH = 180;\nconst COLUMN_OVERSCAN = 2;\nconst JUMP_COLUMN_OVERSCAN = 0;\nconst ROW_HEIGHT = 33;\nconst ROW_NUMBER_WIDTH = 56;\nconst ROW_OVERSCAN = 4;\nconst JUMP_ROW_OVERSCAN = 0;\nconst SMALL_TABLE_ROW_LIMIT = 200;\nconst SMALL_TABLE_COLUMN_LIMIT = 8;\nconst WORKER_SORT_ROW_THRESHOLD = 20_000;\nconst CSV_NATIVE_FIND_MAX_INDEXED_CELLS = 1_000_000;\n\nexport const CsvGrid = React.forwardRef(\n function CsvGrid(\n {\n columns,\n rowStore,\n activeCell,\n height,\n fillHeight,\n isolateStyles,\n scale,\n sortResetKey,\n statusNode,\n },\n ref,\n ) {\n const [sort, setSort] = React.useState<{\n columnIndex: number;\n descending: boolean;\n } | null>(null);\n const columnShapeKey = React.useMemo(\n () => columns.join(\"\\u0000\"),\n [columns],\n );\n\n useKeyedMountEffect(\n joinEffectKey([\"csv-sort-reset\", columnShapeKey, sortResetKey]),\n () => {\n setSort(null);\n },\n );\n\n const toggleSort = React.useCallback((columnIndex: number) => {\n setSort((current) =>\n !current || current.columnIndex !== columnIndex\n ? { columnIndex, descending: false }\n : current.descending\n ? null\n : { columnIndex, descending: true },\n );\n }, []);\n\n const rowOrder = useCsvSortedRowOrder({ rowStore, sort });\n\n const displayIndexByRowIndex = React.useMemo | null>(() => {\n if (!rowOrder) return null;\n const map = new Map();\n rowOrder.forEach((rowIndex, displayRowIndex) => {\n map.set(rowIndex, displayRowIndex);\n });\n return map;\n }, [rowOrder]);\n\n const rowAt = React.useCallback(\n (displayRowIndex: number): Row => {\n const sourceRowIndex = rowOrder?.[displayRowIndex] ?? displayRowIndex;\n return rowStore.getRow(sourceRowIndex);\n },\n [rowStore, rowOrder],\n );\n\n const rowIndexAt = React.useCallback(\n (displayRowIndex: number): number =>\n rowOrder?.[displayRowIndex] ?? displayRowIndex,\n [rowOrder],\n );\n\n const viewportRef = React.useRef(null);\n const rowOffsetRef = React.useRef(null);\n const rowWindowRef = React.useRef(null);\n const [viewportElement, setViewportElement] =\n React.useState(null);\n const setViewportRef = React.useCallback((node: HTMLDivElement | null) => {\n viewportRef.current = node;\n setViewportElement((current) => (current === node ? current : node));\n }, []);\n const columnCount = columns.length;\n const rowCount = rowStore.rowCount;\n const columnOffset = 1;\n const shouldVirtualizeRows = rowCount > SMALL_TABLE_ROW_LIMIT;\n const shouldVirtualizeColumns = columnCount > SMALL_TABLE_COLUMN_LIMIT;\n const effectiveRowHeight = Math.max(1, Math.round(ROW_HEIGHT * scale));\n const effectiveColumnWidth = Math.max(1, Math.round(COLUMN_WIDTH * scale));\n const effectiveRowNumberWidth = Math.round(ROW_NUMBER_WIDTH * scale);\n const columnItemsRef = React.useRef([]);\n const viewportClientHeightRef = React.useRef(0);\n const getRowPatchState = React.useCallback(\n (): CsvRowPatchState => ({\n activeCell: activeCell ?? null,\n columnItems: columnItemsRef.current,\n effectiveRowHeight,\n viewportHeight: viewportClientHeightRef.current,\n rowStore,\n rowOrder,\n shouldVirtualizeRows,\n }),\n [\n activeCell,\n effectiveRowHeight,\n rowStore,\n rowOrder,\n shouldVirtualizeRows,\n ],\n );\n const rowPatcher = useCsvRowPatcher({\n rowOffsetRef,\n rowWindowRef,\n getState: getRowPatchState,\n });\n const rowScrollStrategy = React.useMemo(\n () => ({ handleViewport: rowPatcher.patch }),\n [rowPatcher],\n );\n\n const {\n virtualRows,\n totalRowSize,\n totalColumnSize,\n columnItems,\n virtualRowWindow,\n leftPad,\n rightPad,\n scrollToCell,\n viewportClientHeight,\n } = useFixedGridVirtualization({\n rowCount,\n columnCount,\n rowSize: effectiveRowHeight,\n columnSize: effectiveColumnWidth,\n rowOverscan: ROW_OVERSCAN,\n columnOverscan: COLUMN_OVERSCAN,\n jumpRowOverscan: JUMP_ROW_OVERSCAN,\n jumpColumnOverscan: JUMP_COLUMN_OVERSCAN,\n minimumRenderedRows: 1,\n rowScrollStrategy,\n scrollRef: viewportRef,\n scrollElement: viewportElement,\n virtualizeColumns: shouldVirtualizeColumns,\n });\n\n columnItemsRef.current = columnItems;\n viewportClientHeightRef.current = viewportClientHeight;\n\n React.useImperativeHandle(\n ref ?? null,\n () => ({\n scrollToCell: (cellAddress, options) => {\n const behavior = options?.behavior ?? \"smooth\";\n const displayRowIndex =\n displayIndexByRowIndex?.get(cellAddress.rowIndex) ??\n cellAddress.rowIndex;\n scrollToCell({\n rowIndex: displayRowIndex,\n columnIndex: cellAddress.columnIndex,\n behavior,\n align: \"center\",\n });\n },\n getViewportElement: () => viewportRef.current,\n }),\n [displayIndexByRowIndex, scrollToCell],\n );\n\n const gridTemplate = React.useMemo(\n () =>\n buildVirtualGridTemplate({\n leadingWidth: effectiveRowNumberWidth,\n leftPad,\n columnWidths: fixedGridColumnWidths(columnItems),\n rightPad,\n }),\n [effectiveRowNumberWidth, leftPad, columnItems, rightPad],\n );\n const totalWidth = effectiveRowNumberWidth + totalColumnSize;\n const minimumRowPoolSize =\n Math.ceil(viewportClientHeight / effectiveRowHeight) +\n ROW_OVERSCAN * 2 +\n 2;\n const rowPoolSlots = useFixedRowPool({\n minimumPoolSize: minimumRowPoolSize,\n rowCount,\n virtualRows: virtualRowWindow.items,\n });\n const shouldIndexNativeFind =\n rowCount > 0 &&\n columnCount > 0 &&\n (shouldVirtualizeRows || shouldVirtualizeColumns);\n const nativeFindCellText = React.useCallback(\n (displayRowIndex: number, columnIndex: number) =>\n rowAt(displayRowIndex)?.[columnIndex] ?? \"\",\n [rowAt],\n );\n const scrollToNativeFindCell = React.useCallback(\n ({ rowIndex, columnIndex }: FixedGridNativeFindCellAddress) => {\n scrollToCell({\n rowIndex,\n columnIndex,\n behavior: \"auto\",\n align: \"center\",\n });\n },\n [scrollToCell],\n );\n\n useKeyedMountEffect(\n joinEffectKey([\n \"csv-row-patcher\",\n rowPatcher,\n virtualRows,\n columnItems,\n shouldVirtualizeRows,\n ]),\n () => {\n // After React commits the canonical row window, push that window's\n // visibility, position, and text back onto the pooled DOM so any stale\n // state left by the imperative scroll patcher (a `hidden` row React's\n // reconciler never re-showed, or a cyclic-column cell it never rewrote)\n // is cleared. Falls back to a plain cache invalidation when row\n // virtualization is inactive.\n if (shouldVirtualizeRows) {\n rowPatcher.resync(virtualRows);\n } else {\n rowPatcher.invalidate();\n }\n },\n );\n\n return (\n \n \n \n \n \n \n \n \n \n {columnItems.map((item) => (\n toggleSort(item.index)}\n />\n ))}\n \n \n\n {statusNode ? (\n statusNode\n ) : shouldVirtualizeRows ? (\n \n {rowPoolSlots.map((slot) => (\n \n ))}\n \n ) : (\n
\n {Array.from({ length: rowCount }, (_, displayRowIndex) => (\n \n ))}\n
\n )}\n \n \n \n \n \n );\n },\n);\n\nfunction useCsvSortedRowOrder({\n rowStore,\n sort,\n}: {\n rowStore: CsvRowStore;\n sort: { columnIndex: number; descending: boolean } | null;\n}): number[] | null {\n const [workerRowOrder, setWorkerRowOrder] = React.useState(\n null,\n );\n const sourceRows = React.useMemo(\n () => (sort ? rowStore.materializeRows() : null),\n [rowStore, sort],\n );\n const shouldUseWorker =\n !!sort &&\n !!sourceRows &&\n sourceRows.length >= WORKER_SORT_ROW_THRESHOLD &&\n typeof Worker !== \"undefined\";\n\n const workerSortKey =\n sort && sourceRows && shouldUseWorker\n ? joinEffectKey([\"csv-worker-sort\", shouldUseWorker, sort, sourceRows])\n : null;\n useKeyedMountEffect(workerSortKey, () => {\n if (!sort || !sourceRows || !shouldUseWorker) {\n setWorkerRowOrder(null);\n return;\n }\n setWorkerRowOrder(null);\n\n const controller = new AbortController();\n void sortCsvRowsInWorker({\n sourceRows,\n columnIndex: sort.columnIndex,\n descending: sort.descending,\n signal: controller.signal,\n }).then(\n (rowOrder) => setWorkerRowOrder(rowOrder),\n (error) => {\n if (controller.signal.aborted) return;\n setWorkerRowOrder(\n sortCsvRowsOnMainThread({\n sourceRows,\n columnIndex: sort.columnIndex,\n descending: sort.descending,\n }),\n );\n },\n );\n\n return () => controller.abort();\n });\n\n return React.useMemo(() => {\n if (!sort || !sourceRows) return null;\n if (shouldUseWorker) return workerRowOrder;\n return sortCsvRowsOnMainThread({\n sourceRows,\n columnIndex: sort.columnIndex,\n descending: sort.descending,\n });\n }, [shouldUseWorker, sort, sourceRows, workerRowOrder]);\n}\n\nfunction Spacer({ width }: { width: number }) {\n return
;\n}\n\nfunction HeaderCell({\n name,\n columnIndex,\n height,\n sorted,\n onToggle,\n}: {\n name: string;\n columnIndex: number;\n height: number;\n sorted: \"asc\" | \"desc\" | false;\n onToggle: () => void;\n}) {\n return (\n \n \n {name}\n {sorted ? (\n sorted === \"asc\" ? (\n \n ) : (\n \n )\n ) : null}\n \n
\n );\n}\n\nfunction CsvRowSlot({\n slot,\n sourceRowCount,\n rowAt,\n rowIndexAt,\n gridTemplate,\n rowHeight,\n columnOffset,\n columnItems,\n leftPad,\n rightPad,\n activeCell,\n}: {\n slot: FixedGridRowPoolSlot;\n sourceRowCount: number;\n rowAt: (displayRowIndex: number) => Row;\n rowIndexAt: (displayRowIndex: number) => number;\n gridTemplate: string;\n rowHeight: number;\n columnOffset: number;\n columnItems: FixedGridColumnItem[];\n leftPad: number;\n rightPad: number;\n activeCell: CsvCellAddress | null;\n}) {\n const fallbackDisplayRowIndex =\n sourceRowCount > 0 ? Math.min(slot.slotIndex, sourceRowCount - 1) : 0;\n const displayRowIndex = slot.virtualRow?.index ?? fallbackDisplayRowIndex;\n const rowIndex = rowIndexAt(displayRowIndex);\n\n return (\n