shadcn-ui/packages/cli/src/index.ts

#!/usr/bin/env node
import { existsSync, promises as fs } from "fs"
import path from "path"
import { Command } from "commander"
import { execa } from "execa"
import ora from "ora"
import prompts from "prompts"

import { Component, getAvailableComponents } from "./utils/get-components"
import { getPackageInfo } from "./utils/get-package-info"
import { getPackageManager } from "./utils/get-package-manager"
import { logger } from "./utils/logger"

process.on("SIGINT", () => process.exit(0))
process.on("SIGTERM", () => process.exit(0))

async function main() {
  const packageInfo = await getPackageInfo()

  const program = new Command()
    .name("@shadcn/ui")
    .description("Add @shadcn/ui components to your project")
    .version(
      packageInfo.version || "1.0.0",
      "-v, --version",
      "display the version number"
    )

  program
    .command("add")
    .description("add components to your project")
    .action(async () => {
      const { components, dir } = await promptForAddOptions()
      if (!components?.length) {
        logger.warn("No components selected. Nothing to install.")
        process.exit(0)
      }

      // Create componentPath directory if it doesn't exist.
      const destinationDir = path.resolve(dir)
      if (!existsSync(destinationDir)) {
        const spinner = ora(`Creating ${dir}...`).start()
        await fs.mkdir(destinationDir, { recursive: true })
        spinner.succeed()
      }

      const packageManager = getPackageManager()

      logger.success(`Installing components...`)
      for (const component of components) {
        const componentSpinner = ora(`${component.name}...`).start()

        // Write the files.
        for (const file of component.files) {
          const filePath = path.resolve(dir, file.name)
          await fs.writeFile(filePath, file.content)
        }

        // Install dependencies.
        if (component.dependencies?.length) {
          const dependencies = component.dependencies.join(" ")
          await execa(packageManager, [
            packageManager === "npm" ? "install" : "add",
            dependencies,
          ])
        }
        componentSpinner.succeed(component.name)
      }
    })

  program.parse()
}

type AddOptions = {
  components: Component[]
  dir: string
}

async function promptForAddOptions() {
  const availableComponents = await getAvailableComponents()

  if (!availableComponents?.length) {
    logger.error(
      "An error occurred while fetching components. Please try again."
    )
    process.exit(0)
  }

  const options = await prompts([
    {
      type: "multiselect",
      name: "components",
      message: "Which component(s) would you like to add?",
      hint: "Space to select. A to select all. I to invert selection.",
      instructions: false,

      choices: availableComponents.map((component) => ({
        title: component.name,
        value: component,
      })),
    },
    {
      type: "text",
      name: "dir",
      message: "Where would you like to install the component(s)?",
      initial: "./components/ui",
    },
  ])

  return options as AddOptions
}

main()