add support for --build-missing flag and enhance registry handling

Author: LadyBluenotes

packages/installer/src/cli.ts

@@ -2,6 +2,7 @@ import fs from 'node:fs'
 import path from 'node:path'
 import os from 'node:os'
 import { fileURLToPath } from 'node:url'
+import { spawn } from 'node:child_process'
 import yaml from 'yaml'
 
 type Catalog = {
@@ -39,6 +40,10 @@ function arg(name: string) {
   return idx >= 0 ? process.argv[idx + 1] : undefined
 }
 
+function hasFlag(name: string) {
+  return process.argv.includes(`--${name}`)
+}
+
 function readJson<T>(p: string): T {
   return JSON.parse(fs.readFileSync(p, 'utf8'))
 }
@@ -81,6 +86,25 @@ function readCatalog(repoRoot: string): Catalog {
   return yaml.parse(fs.readFileSync(p, 'utf8'))
 }
 
+async function runNodeScript(scriptPath: string, args: string[], cwd: string) {
+  const fullArgs = [scriptPath, ...args]
+  const display = `${process.execPath} ${fullArgs.join(' ')}`
+  console.log(`Running: ${display}`)
+
+  await new Promise<void>((resolve, reject) => {
+    const child = spawn(process.execPath, fullArgs, {
+      cwd,
+      stdio: 'inherit',
+    })
+
+    child.on('error', reject)
+    child.on('close', (code) => {
+      if (code === 0) return resolve()
+      reject(new Error(`Command failed with exit code ${code}: ${display}`))
+    })
+  })
+}
+
 function getAllDeps(pkg: any): Record<string, string> {
   return {
     ...(pkg.dependencies ?? {}),
@@ -213,7 +237,7 @@ function resolveExactFromPackageJson(
   )
 }
 
-function main() {
+async function main() {
   const cmd = process.argv[2]
   if (cmd !== 'install') throw new Error(`Unknown command: ${cmd}`)
 
@@ -234,6 +258,8 @@ function main() {
 
   const repoRoot = getRepoRootFromThisFile()
 
+  const buildMissing = hasFlag('build-missing')
+
   const manifestPath = path.join(repoRoot, 'registry', 'manifest.json')
   if (!fs.existsSync(manifestPath)) {
     throw new Error(
@@ -244,7 +270,7 @@ function main() {
   }
 
   const catalog = readCatalog(repoRoot)
-  const manifest = readJson<Manifest>(manifestPath)
+  let manifest = readJson<Manifest>(manifestPath)
 
   // Validate catalog schema minimally (fail fast with clear errors)
   for (const [id, lib] of Object.entries<any>(catalog.libraries ?? {})) {
@@ -269,6 +295,16 @@ function main() {
   const prevLock = readJsonIfExists<any>(outLockPath)
   const prevInstalled: Record<string, any> = prevLock?.installed ?? {}
 
+  if (
+    buildMissing &&
+    prevLock?.registry?.type &&
+    prevLock.registry.type !== 'local'
+  ) {
+    throw new Error(
+      `--build-missing is only supported for local registries (found: ${prevLock.registry.type}).`,
+    )
+  }
+
   // Always install the top-level tanstack agent if present
   const tanstackAgent = manifest.agents.find((a) => a.id === 'tanstack')
   if (tanstackAgent) {
@@ -292,14 +328,56 @@ function main() {
       )
     }
 
-    const skill = manifest.skills.find(
+    let skill = manifest.skills.find(
       (s) => s.lib === libId && s.version === exactVersion,
     )
     if (!skill) {
-      throw new Error(
-        `Registry missing ${libId}@${exactVersion}.\n` +
-          `Build it in TanStack/agents and rebuild manifest.`,
+      if (!buildMissing) {
+        throw new Error(
+          `Registry missing ${libId}@${exactVersion}.\n` +
+            `Build it in TanStack/agents and rebuild manifest.`,
+        )
+      }
+
+      const skillcCliPath = path.join(
+        repoRoot,
+        'packages',
+        'skillc',
+        'dist',
+        'cli.js',
+      )
+
+      if (!fs.existsSync(skillcCliPath)) {
+        throw new Error(
+          `skillc CLI not found at: ${skillcCliPath}\n` +
+            `Build the repo first (pnpm build:all) or run skillc from source.`,
+        )
+      }
+
+      console.log(
+        `Registry missing ${libId}@${exactVersion}. Building with skillc...`,
+      )
+
+      await runNodeScript(
+        skillcCliPath,
+        ['build-lib', '--lib', libId, '--version', exactVersion],
+        repoRoot,
       )
+
+      console.log('Rebuilding registry manifest...')
+      await runNodeScript(skillcCliPath, ['build-manifest'], repoRoot)
+
+      manifest = readJson<Manifest>(manifestPath)
+      skill = manifest.skills.find(
+        (s) => s.lib === libId && s.version === exactVersion,
+      )
+
+      if (!skill) {
+        throw new Error(
+          `Registry still missing ${libId}@${exactVersion} after build.\n` +
+            `Check skillc output for details.`,
+        )
+      }
     }
 
     const agentFileName = `tanstack-${libId}.md`
@@ -364,4 +442,7 @@ function main() {
   console.log(`Installed: ${libs.join(', ') || '(none)'}`)
 }
 
-main()
+main().catch((error) => {
+  console.error(error instanceof Error ? error.message : error)
+  process.exit(1)
+})