feat: auto-install validation library deps

Author: onmax

src/wizard/shelve-setup.ts

@@ -1,15 +1,19 @@
 import type { Nuxt } from '@nuxt/schema'
 import type { ConsolaInstance } from 'consola'
+import { exec } from 'node:child_process'
 import { existsSync, readFileSync, writeFileSync } from 'node:fs'
 import { homedir } from 'node:os'
 import { join, resolve } from 'node:path'
 import process from 'node:process'
+import { promisify } from 'node:util'
 import { consola } from 'consola'
 import { builders, loadFile, writeFile } from 'magicast'
 import { getDefaultExportOptions } from 'magicast/helpers'
 import { $fetch } from 'ofetch'
 import { getPrefix, toCamelCase } from '../utils/transform'
 
+const execAsync = promisify(exec)
+
 const SHELVE_RC_PATH = join(homedir(), '.shelve')
 const DEFAULT_URL = 'https://app.shelve.cloud'
 
@@ -86,6 +90,34 @@ function detectValidationLibrary(rootDir: string): ValidationLibrary | null {
   catch { return null }
 }
 
+function detectPackageManager(rootDir: string): 'pnpm' | 'yarn' | 'npm' {
+  if (existsSync(join(rootDir, 'pnpm-lock.yaml')))
+    return 'pnpm'
+  if (existsSync(join(rootDir, 'yarn.lock')))
+    return 'yarn'
+  return 'npm'
+}
+
+async function installValidationLibrary(rootDir: string, library: ValidationLibrary, logger: ConsolaInstance): Promise<void> {
+  const pm = detectPackageManager(rootDir)
+  const packages = library === 'valibot'
+    ? ['valibot', '@valibot/to-json-schema']
+    : ['zod']
+
+  const cmd = pm === 'npm'
+    ? `npm install ${packages.join(' ')}`
+    : `${pm} add ${packages.join(' ')}`
+
+  logger.info(`Installing ${packages.join(', ')}...`)
+  try {
+    await execAsync(cmd, { cwd: rootDir })
+    logger.success(`Installed ${library} dependencies`)
+  }
+  catch (error) {
+    logger.error(`Failed to install dependencies: ${error}`)
+  }
+}
+
 function countPrefixes(keys: string[]): Map<string, number> {
   const counts = new Map<string, number>()
   for (const key of keys) {
@@ -377,6 +409,7 @@ export async function runShelveWizard(nuxt: Nuxt): Promise<void> {
 
   // 1. Detect or ask for validation library
   let library: ValidationLibrary | null = detectValidationLibrary(nuxt.options.rootDir)
+  let needsInstall = false
   if (library) {
     logger.info(`Detected ${library} for schema validation`)
   }
@@ -394,6 +427,7 @@ export async function runShelveWizard(nuxt: Nuxt): Promise<void> {
     }
     else {
       library = choice as ValidationLibrary
+      needsInstall = true
     }
   }
 
@@ -431,14 +465,19 @@ export async function runShelveWizard(nuxt: Nuxt): Promise<void> {
     }
   }
 
-  // 3. Generate schema (only if library selected)
-  let schemaCode: { imports: string, schema: string } | null = null
+  // 3. Install validation library if needed
+  if (library && needsInstall) {
+    await installValidationLibrary(nuxt.options.rootDir, library, logger)
+  }
+
+  // 4. Generate schema (only if library selected)
+  let schemaCode: { imports: string, schemaExpr: string } | null = null
   if (library) {
     const keys = variables.length > 0 ? variables.map(v => v.key) : null
     schemaCode = generateSchemaCode(keys, library)
   }
 
-  // 4. Only update config if we have something to add
+  // 5. Only update config if we have something to add
   if (!schemaCode && !shelveConfig) {
     logger.info('No configuration changes needed')
     return