118 lines
2.7 KiB
TypeScript
118 lines
2.7 KiB
TypeScript
import * as React from "react"
|
|
import { cn } from "cn"
|
|
|
|
import { formatCode } from "@/lib/format-code"
|
|
import { highlightCode } from "@/lib/highlight-code"
|
|
import { readFileFromRoot } from "@/lib/read-file"
|
|
import { getDemoItem, getRegistryItem } from "@/lib/registry"
|
|
import { CodeCollapsibleWrapper } from "@/components/code-collapsible-wrapper"
|
|
import { CopyButton } from "@/components/copy-button"
|
|
import { getIconForLanguageExtension } from "@/components/icons"
|
|
|
|
export async function ComponentSource({
|
|
name,
|
|
src,
|
|
title,
|
|
language,
|
|
collapsible = true,
|
|
className,
|
|
styleName = "base-nova",
|
|
maxLines,
|
|
}: React.ComponentProps<"div"> & {
|
|
name?: string
|
|
src?: string
|
|
title?: string
|
|
language?: string
|
|
collapsible?: boolean
|
|
styleName?: string
|
|
maxLines?: number
|
|
}) {
|
|
if (!name && !src) {
|
|
return null
|
|
}
|
|
|
|
let code: string | undefined
|
|
|
|
if (name) {
|
|
const item =
|
|
(await getDemoItem(name, styleName)) ??
|
|
(await getRegistryItem(name, styleName))
|
|
code = item?.files?.[0]?.content
|
|
}
|
|
|
|
if (src) {
|
|
code = await readFileFromRoot(src)
|
|
}
|
|
|
|
if (!code) {
|
|
return null
|
|
}
|
|
|
|
code = await formatCode(code, styleName)
|
|
code = code.replaceAll("/* eslint-disable react/no-children-prop */\n", "")
|
|
|
|
// Truncate code if maxLines is set.
|
|
if (maxLines) {
|
|
code = code.split("\n").slice(0, maxLines).join("\n")
|
|
}
|
|
|
|
const lang = language ?? title?.split(".").pop() ?? "tsx"
|
|
const highlightedCode = await highlightCode(code, lang)
|
|
|
|
if (!collapsible) {
|
|
return (
|
|
<div className={cn("relative", className)}>
|
|
<ComponentCode
|
|
code={code}
|
|
highlightedCode={highlightedCode}
|
|
language={lang}
|
|
title={title}
|
|
/>
|
|
</div>
|
|
)
|
|
}
|
|
|
|
return (
|
|
<CodeCollapsibleWrapper className={className}>
|
|
<ComponentCode
|
|
code={code}
|
|
highlightedCode={highlightedCode}
|
|
language={lang}
|
|
title={title}
|
|
/>
|
|
</CodeCollapsibleWrapper>
|
|
)
|
|
}
|
|
|
|
function ComponentCode({
|
|
code,
|
|
highlightedCode,
|
|
language,
|
|
title,
|
|
}: {
|
|
code: string
|
|
highlightedCode: string
|
|
language: string
|
|
title: string | undefined
|
|
}) {
|
|
return (
|
|
<figure data-rehype-pretty-code-figure="" className="[&>pre]:max-h-96">
|
|
{title && (
|
|
<figcaption
|
|
data-rehype-pretty-code-title=""
|
|
className="flex items-center gap-2 text-code-foreground [&_svg]:size-4 [&_svg]:text-code-foreground [&_svg]:opacity-70"
|
|
data-language={language}
|
|
>
|
|
{getIconForLanguageExtension(language)}
|
|
{title}
|
|
</figcaption>
|
|
)}
|
|
<CopyButton value={code} />
|
|
<div
|
|
data-not-typeset
|
|
dangerouslySetInnerHTML={{ __html: highlightedCode }}
|
|
/>
|
|
</figure>
|
|
)
|
|
}
|