snapshot of https://github.com/shadcn-ui/ui.git (main, 2026-09-15; blobs > 131072B omitted)

apps/v4/lib/format-code.ts

@@ -0,0 +1,142 @@
+import { promises as fsPromises } from "fs"
+import path from "path"
+import {
+  createStyleMap,
+  transformIcons,
+  transformMenu,
+  transformRender,
+  transformStyle,
+} from "shadcn/utils"
+import { Project, ScriptKind, type SourceFile } from "ts-morph"
+
+import { BASES } from "@/registry/bases"
+
+function getStyleFromStyleName(styleName: string) {
+  const parts = styleName.split("-")
+  return parts.length > 1 ? parts.slice(1).join("-") : styleName
+}
+
+function buildDisplayConfig(styleName: string) {
+  return {
+    $schema: "https://ui.shadcn.com/schema.json",
+    style: styleName,
+    rsc: true,
+    tsx: true,
+    tailwind: {
+      config: "",
+      css: "",
+      baseColor: "neutral",
+      cssVariables: true,
+      prefix: "",
+    },
+    iconLibrary: "lucide",
+    aliases: {
+      components: "@/components",
+      utils: "@/lib/utils",
+      ui: "@/components/ui",
+      lib: "@/lib",
+      hooks: "@/hooks",
+    },
+    resolvedPaths: {
+      cwd: "/",
+      tailwindConfig: "",
+      tailwindCss: "",
+      utils: "@/lib/utils",
+      components: "@/components",
+      lib: "@/lib",
+      hooks: "@/hooks",
+      ui: "@/components/ui",
+    },
+  }
+}
+
+type DisplayTransformer = (opts: {
+  filename: string
+  raw: string
+  sourceFile: SourceFile
+  config: ReturnType<typeof buildDisplayConfig>
+}) => Promise<unknown>
+
+const styleMapCache = new Map<string, Record<string, string>>()
+
+async function getStyleMap(styleName: string) {
+  const style = getStyleFromStyleName(styleName)
+
+  if (styleMapCache.has(style)) {
+    return styleMapCache.get(style)!
+  }
+
+  try {
+    const cssPath = path.join(
+      process.cwd(),
+      `registry/styles/style-${style}.css`
+    )
+    const css = await fsPromises.readFile(cssPath, "utf-8")
+    const styleMap = createStyleMap(css)
+    styleMapCache.set(style, styleMap)
+    return styleMap
+  } catch {
+    return {}
+  }
+}
+
+export async function formatCode(code: string, styleName: string) {
+  code = code.replaceAll(`@/registry/${styleName}/`, "@/components/")
+
+  // Always rewrite the legacy tree so src-based sources render clean imports
+  // regardless of the styleName in play.
+  code = code.replaceAll("@/registry/new-york-v4/", "@/components/")
+
+  for (const base of BASES) {
+    code = code.replaceAll(`@/registry/bases/${base.name}/`, "@/components/")
+    code = code.replaceAll(
+      `@/examples/${base.name}/ui-rtl/`,
+      "@/components/ui/"
+    )
+    code = code.replaceAll(`@/examples/${base.name}/ui/`, "@/components/ui/")
+    code = code.replaceAll(`@/examples/${base.name}/lib/`, "@/lib/")
+    code = code.replaceAll(`@/examples/${base.name}/hooks/`, "@/hooks/")
+  }
+
+  code = code.replace(
+    /@\/styles\/([\w-]+)\/(ui-rtl|ui)\/([\w-]+)/g,
+    (match, _styleName, type, component) => {
+      if (type === "ui" || type === "ui-rtl") {
+        return `@/components/ui/${component}`
+      }
+
+      return match
+    }
+  )
+
+  code = code.replaceAll("export default", "export")
+
+  try {
+    const styleMap = await getStyleMap(styleName)
+    const transformed = await transformStyle(code, { styleMap })
+    const config = buildDisplayConfig(styleName)
+    const project = new Project({ compilerOptions: {} })
+    const sourceFile = project.createSourceFile("component.tsx", transformed, {
+      scriptKind: ScriptKind.TSX,
+    })
+
+    const transformers: DisplayTransformer[] = [
+      transformIcons as DisplayTransformer,
+      transformMenu as DisplayTransformer,
+      transformRender as DisplayTransformer,
+    ]
+    for (const transformer of transformers) {
+      await transformer({
+        filename: "component.tsx",
+        raw: transformed,
+        sourceFile,
+        config,
+      })
+    }
+
+    return sourceFile.getText()
+  } catch (error) {
+    console.error("Transform failed:", error)
+    return code
+  }
+}