preskok-ui 0.0.5 → 0.0.6

package/README.md

@@ -3,10 +3,11 @@
 Thin CLI wrapper around shadcn with Preskok registry wiring.
 
 ```bash
+npx preskok-ui@latest init
 npx preskok-ui@latest init button
 ```
 
-The CLI delegates to `shadcn@latest`. `init` installs the Preskok base and optional components in one pass, and `add` resolves bare component names through the Preskok registry.
+The CLI delegates to `shadcn@latest`. `init` installs the Preskok base for new projects, only registers the Preskok namespace in projects that already have `components.json`, and can add optional components in one pass. `add` resolves bare component names through the Preskok registry.
 
 ```bash
 npx preskok-ui@latest registry

package/bin/preskok-ui.mjs

@@ -1,12 +1,44 @@
 #!/usr/bin/env node
 import { spawnSync } from "node:child_process"
+import { existsSync, readFileSync, unlinkSync, writeFileSync } from "node:fs"
+import path from "node:path"
 
-const REGISTRY = "@preskok=https://ui-three-mu.vercel.app/r/{name}.json"
-const DEFAULT_REGISTRY_ITEM = "https://ui-three-mu.vercel.app/r/default.json"
+const REGISTRY_BASE_URL = (
+  process.env.PRESKOK_REGISTRY_URL ?? "https://ui-three-mu.vercel.app"
+).replace(/\/$/, "")
+const REGISTRY = `@preskok=${REGISTRY_BASE_URL}/r/{name}.json`
+const DEFAULT_REGISTRY_ITEM = `${REGISTRY_BASE_URL}/r/default.json`
 const CWD_OPTION_VALUES = new Set(["-c", "--cwd"])
+const ADD_OPTION_VALUES = new Set(["-c", "--cwd", "-p", "--path", "--diff"])
+const DEPENDENCY_SECTIONS = [
+  "dependencies",
+  "devDependencies",
+  "optionalDependencies",
+  "peerDependencies",
+]
+const INIT_OPTION_VALUES = new Set([
+  "-t",
+  "--template",
+  "-b",
+  "--base",
+  "-p",
+  "--preset",
+  "-c",
+  "--cwd",
+  "-n",
+  "--name",
+])
 
 const args = process.argv.slice(2)
 const [command, ...commandArgs] = args
+const LOCKFILES = [
+  "package-lock.json",
+  "npm-shrinkwrap.json",
+  "pnpm-lock.yaml",
+  "yarn.lock",
+  "bun.lock",
+  "bun.lockb",
+]
 
 if (!command || command === "-h" || command === "--help") {
   printHelp()
@@ -14,20 +46,54 @@ if (!command || command === "-h" || command === "--help") {
 }
 
 if (command === "add") {
-  const registryStatus = runShadcn(["registry", "add", REGISTRY])
+  const registryStatus = runShadcn([
+    "registry",
+    "add",
+    REGISTRY,
+    ...getCommandOptions(commandArgs, CWD_OPTION_VALUES),
+  ])
 
   if (registryStatus !== 0) {
     process.exit(registryStatus)
   }
 
   process.exit(
-    runShadcn(["add", "--overwrite", "--yes", ...toPreskokItems(commandArgs)])
+    runPreskokAdd(toPreskokCommandItems(commandArgs, ADD_OPTION_VALUES))
   )
 }
 
 if (command === "init") {
+  const { initArgs, items } = splitInitArgs(commandArgs)
+  const cwd = getCwd(initArgs)
+
+  if (!existsSync(path.join(cwd, "components.json"))) {
+    const initStatus = runShadcn(["init", DEFAULT_REGISTRY_ITEM, ...initArgs])
+
+    if (initStatus !== 0) {
+      process.exit(initStatus)
+    }
+  }
+
+  const registryStatus = runShadcn([
+    "registry",
+    "add",
+    REGISTRY,
+    ...getCommandOptions(initArgs, CWD_OPTION_VALUES),
+  ])
+
+  if (registryStatus !== 0) {
+    process.exit(registryStatus)
+  }
+
+  if (items.length === 0) {
+    process.exit(0)
+  }
+
   process.exit(
-    runShadcn(["init", DEFAULT_REGISTRY_ITEM, ...toPreskokInitItems(commandArgs)])
+    runPreskokAdd([
+      ...getCommandOptions(initArgs, CWD_OPTION_VALUES),
+      ...toPreskokItems(items),
+    ])
   )
 }
 
@@ -39,7 +105,10 @@ if (command === "registry") {
 
 if (command === "view") {
   process.exit(
-    runShadcn(["view", ...toPreskokCommandItems(commandArgs, CWD_OPTION_VALUES)])
+    runShadcn([
+      "view",
+      ...toPreskokCommandItems(commandArgs, CWD_OPTION_VALUES),
+    ])
   )
 }
 
@@ -76,6 +145,14 @@ function runShadcn(shadcnArgs) {
   return result.status ?? 1
 }
 
+function runPreskokAdd(addArgs) {
+  const packageState = capturePackageState(addArgs)
+  const status = runShadcn(["add", "--yes", ...addArgs])
+  restorePackageState(packageState)
+
+  return status
+}
+
 function toPreskokItems(values) {
   return values.map(toPreskokItem)
 }
@@ -106,46 +183,190 @@ function toPreskokCommandItems(values, optionsWithValues) {
   })
 }
 
-function toPreskokInitItems(values) {
-  const initOptionsWithValues = new Set([
-    "-t",
-    "--template",
-    "-b",
-    "--base",
-    "-p",
-    "--preset",
-    "-c",
-    "--cwd",
-    "-n",
-    "--name",
-  ])
+function getCommandOptions(values, optionsWithValues) {
+  let skipNext = false
+  const options = []
 
+  for (const value of values) {
+    if (skipNext) {
+      skipNext = false
+      options.push(value)
+      continue
+    }
+
+    if (optionsWithValues.has(value)) {
+      skipNext = true
+      options.push(value)
+      continue
+    }
+
+    if (hasInlineOptionValue(value, optionsWithValues)) {
+      options.push(value)
+    }
+  }
+
+  return options
+}
+
+function splitInitArgs(values) {
   let skipNext = false
+  const initArgs = []
+  const items = []
 
-  return values.map((value) => {
+  for (const value of values) {
     if (skipNext) {
       skipNext = false
-      return value
+      initArgs.push(value)
+      continue
     }
 
-    if (initOptionsWithValues.has(value)) {
+    if (INIT_OPTION_VALUES.has(value)) {
       skipNext = true
-      return value
+      initArgs.push(value)
+      continue
+    }
+
+    if (
+      hasInlineOptionValue(value, INIT_OPTION_VALUES) ||
+      value.startsWith("-")
+    ) {
+      initArgs.push(value)
+      continue
     }
 
-    return toPreskokItems([value])[0]
+    items.push(value)
+  }
+
+  return { initArgs, items }
+}
+
+function getCwd(values) {
+  for (let index = 0; index < values.length; index++) {
+    const value = values[index]
+
+    if (value === "-c" || value === "--cwd") {
+      return path.resolve(values[index + 1] ?? process.cwd())
+    }
+
+    if (value.startsWith("--cwd=")) {
+      return path.resolve(value.slice("--cwd=".length))
+    }
+  }
+
+  return process.cwd()
+}
+
+function hasInlineOptionValue(value, options) {
+  return [...options].some((option) => value.startsWith(`${option}=`))
+}
+
+function capturePackageState(values) {
+  const cwd = getCwd(values)
+  const packageJsonPath = path.join(cwd, "package.json")
+
+  if (!existsSync(packageJsonPath)) {
+    return null
+  }
+
+  const packageJsonContent = readFileSync(packageJsonPath, "utf8")
+  const packageJson = JSON.parse(packageJsonContent)
+  const lockfiles = new Map()
+
+  for (const lockfile of LOCKFILES) {
+    const lockfilePath = path.join(cwd, lockfile)
+
+    if (existsSync(lockfilePath)) {
+      lockfiles.set(lockfile, readFileSync(lockfilePath))
+    }
+  }
+
+  return {
+    cwd,
+    lockfiles,
+    packageJson,
+    packageJsonContent,
+    packageJsonPath,
+  }
+}
+
+function restorePackageState(packageState) {
+  if (!packageState || !existsSync(packageState.packageJsonPath)) {
+    return
+  }
+
+  const packageJson = JSON.parse(
+    readFileSync(packageState.packageJsonPath, "utf8")
+  )
+  const beforeDependencyNames = getDependencyNames(packageState.packageJson)
+  const afterDependencyNames = getDependencyNames(packageJson)
+  const hasNewDependencies = [...afterDependencyNames].some((dependency) => {
+    return !beforeDependencyNames.has(dependency)
   })
+
+  if (!hasNewDependencies) {
+    writeFileSync(packageState.packageJsonPath, packageState.packageJsonContent)
+    restoreLockfiles(packageState)
+    return
+  }
+
+  restoreExistingDependencySpecs(packageState.packageJson, packageJson)
+  writeFileSync(
+    packageState.packageJsonPath,
+    `${JSON.stringify(packageJson, null, 2)}\n`
+  )
+}
+
+function getDependencyNames(packageJson) {
+  const names = new Set()
+
+  for (const section of DEPENDENCY_SECTIONS) {
+    for (const dependency of Object.keys(packageJson[section] ?? {})) {
+      names.add(dependency)
+    }
+  }
+
+  return names
+}
+
+function restoreExistingDependencySpecs(sourcePackageJson, targetPackageJson) {
+  for (const section of DEPENDENCY_SECTIONS) {
+    const sourceDependencies = sourcePackageJson[section] ?? {}
+    const targetDependencies = targetPackageJson[section] ?? {}
+
+    for (const [dependency, version] of Object.entries(sourceDependencies)) {
+      if (targetDependencies[dependency] !== undefined) {
+        targetDependencies[dependency] = version
+      }
+    }
+  }
+}
+
+function restoreLockfiles(packageState) {
+  for (const lockfile of LOCKFILES) {
+    const lockfilePath = path.join(packageState.cwd, lockfile)
+    const content = packageState.lockfiles.get(lockfile)
+
+    if (content) {
+      writeFileSync(lockfilePath, content)
+      continue
+    }
+
+    if (existsSync(lockfilePath)) {
+      unlinkSync(lockfilePath)
+    }
+  }
 }
 
 function printHelp() {
   console.log(`Usage: preskok-ui <command> [options]
 
 Commands:
-  init [items...]     run shadcn init with the Preskok base and optional items
+  init [items...]     run shadcn init, register Preskok, and optionally add items
   add <items...>      register Preskok and add items by name
   registry            register the @preskok namespace in components.json
 
 Examples:
+  npx preskok-ui@latest init
   npx preskok-ui@latest init button
   npx preskok-ui@latest add button
   npx preskok-ui@latest view button

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "preskok-ui",
-  "version": "0.0.5",
+  "version": "0.0.6",
   "license": "MIT",
   "type": "module",
   "bin": {