upstream-shadcn-ui/apps/v4/lib/registry-health/dry-run.ts

import { spawn, type ChildProcess } from "node:child_process"
import { promises as fs } from "node:fs"
import os from "node:os"
import path from "node:path"
import { stripVTControlCharacters } from "node:util"

import { DEFAULT_STYLE, type RegistryDryRunResult } from "./monitor"
import { REGISTRY_DRY_RUN_FAILURE_MESSAGE_MAX_LENGTH } from "./schema"

const MAX_CAPTURED_OUTPUT_LENGTH = 4000

function getFailureMessage(output: string) {
  const message = stripVTControlCharacters(output).replace(/\s+/g, " ").trim()
  return message
    ? message.slice(-REGISTRY_DRY_RUN_FAILURE_MESSAGE_MAX_LENGTH)
    : undefined
}

async function runLocalCliDryRun({
  namespace,
  item,
  registryUrl,
  cliPath,
  timeoutMs = 60_000,
  killGraceMs = 5_000,
}: {
  namespace: string
  item: string
  registryUrl: string
  cliPath: string
  timeoutMs?: number
  killGraceMs?: number
}): Promise<RegistryDryRunResult> {
  const startedAt = Date.now()
  const temporaryDirectory = await fs.mkdtemp(
    path.join(os.tmpdir(), "shadcn-registry-health-")
  )

  try {
    await fs.mkdir(path.join(temporaryDirectory, "app"), { recursive: true })
    await Promise.all([
      fs.writeFile(
        path.join(temporaryDirectory, "components.json"),
        JSON.stringify(
          {
            $schema: "https://ui.shadcn.com/schema.json",
            style: DEFAULT_STYLE,
            rsc: true,
            tsx: true,
            tailwind: {
              config: "",
              css: "app/globals.css",
              baseColor: "neutral",
              cssVariables: true,
              prefix: "",
            },
            iconLibrary: "lucide",
            aliases: {
              components: "@/components",
              utils: "@/lib/utils",
              ui: "@/components/ui",
              lib: "@/lib",
              hooks: "@/hooks",
            },
            registries: {
              [namespace]: registryUrl,
            },
          },
          null,
          2
        )
      ),
      fs.writeFile(path.join(temporaryDirectory, "app/globals.css"), ""),
      fs.writeFile(
        path.join(temporaryDirectory, "package.json"),
        JSON.stringify({ private: true })
      ),
      fs.writeFile(
        path.join(temporaryDirectory, "tsconfig.json"),
        JSON.stringify(
          {
            compilerOptions: {
              baseUrl: ".",
              paths: {
                "@/*": ["./*"],
              },
              jsx: "preserve",
            },
          },
          null,
          2
        )
      ),
    ])

    return await new Promise<RegistryDryRunResult>((resolve) => {
      const child: ChildProcess = spawn(
        process.execPath,
        [
          cliPath,
          "add",
          `${namespace}/${item}`,
          "--dry-run",
          "--yes",
          "--cwd",
          temporaryDirectory,
        ],
        {
          cwd: temporaryDirectory,
          env: getDryRunEnvironment(temporaryDirectory),
          stdio: ["ignore", "pipe", "pipe"],
        }
      )
      let completed = false
      let timedOut = false
      let output = ""
      let forceKillTimeout: NodeJS.Timeout | undefined
      const captureOutput = (chunk: Buffer | string) => {
        const value = typeof chunk === "string" ? chunk : chunk.toString("utf8")
        output = `${output}${value}`.slice(-MAX_CAPTURED_OUTPUT_LENGTH)
      }
      const finish = (result: RegistryDryRunResult) => {
        if (completed) return
        completed = true
        clearTimeout(timeout)
        if (forceKillTimeout) clearTimeout(forceKillTimeout)
        resolve(result)
      }
      const timeout = setTimeout(() => {
        timedOut = true
        child.kill("SIGTERM")
        forceKillTimeout = setTimeout(() => {
          if (!completed) child.kill("SIGKILL")
        }, killGraceMs)
      }, timeoutMs)

      child.stdout?.on("data", captureOutput)
      child.stderr?.on("data", captureOutput)
      child.on("error", (error) => {
        captureOutput(error.message)
        const failureMessage = getFailureMessage(output)
        finish({
          success: false,
          failureCode: timedOut ? "timeout" : "spawn_error",
          ...(failureMessage ? { failureMessage } : {}),
          durationMs: Date.now() - startedAt,
        })
      })
      child.on("close", (code, signal) => {
        const failureCode = timedOut
          ? "timeout"
          : code === 0
            ? undefined
            : signal
              ? "terminated"
              : `exit_${code}`
        const failureMessage =
          code === 0 ? undefined : getFailureMessage(output)

        finish({
          success: !timedOut && code === 0,
          failureCode,
          ...(failureMessage ? { failureMessage } : {}),
          durationMs: Date.now() - startedAt,
        })
      })
    })
  } finally {
    await fs.rm(temporaryDirectory, { recursive: true, force: true })
  }
}

function getDryRunEnvironment(temporaryDirectory: string) {
  return {
    PATH: process.env.PATH,
    TMPDIR: os.tmpdir(),
    XDG_CACHE_HOME: temporaryDirectory,
    XDG_CONFIG_HOME: temporaryDirectory,
    npm_config_userconfig: path.join(temporaryDirectory, ".npmrc"),
    NODE_ENV: "production",
    CI: "1",
    NO_COLOR: "1",
  } satisfies NodeJS.ProcessEnv
}

export { getDryRunEnvironment, runLocalCliDryRun }