@fernado03/zoo-flow 0.9.0 → 0.10.0

package/bin/zoo-flow.js

@@ -2,12 +2,14 @@
 
 import fs from "node:fs";
 import path from "node:path";
+import readline from "node:readline";
 import { fileURLToPath } from "node:url";
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
 const packageRoot = path.resolve(__dirname, "..");
 const templateRoot = path.join(packageRoot, "templates", "full");
+const claudeCodeTemplateRoot = path.join(packageRoot, "templates", "claude-code");
 
 const COMMAND_POLICY = {
   routed: {
@@ -55,30 +57,74 @@ const BUILT_IN_MODE_NAMES = new Set([
   "orchestrator"
 ]);
 
+const CLAUDE_CODE_GITIGNORE_MARKER = "# Zoo Flow — Claude Code config (never committed)";
+const CLAUDE_CODE_GITIGNORE_ENTRIES = [".claude/", ".zoo-flow/"];
+
+function detectPlatform(projectRoot) {
+  if (pathExists(path.join(projectRoot, "CLAUDE.md"))) return "claude-code";
+  if (pathExists(path.join(projectRoot, ".roomodes"))) return "zoo-code";
+  return null;
+}
+
+function getTemplateRoot(platform) {
+  return platform === "claude-code" ? claudeCodeTemplateRoot : templateRoot;
+}
+
+async function promptPlatform() {
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout
+  });
+
+  return new Promise((resolve) => {
+    rl.question(
+      "\nZoo Flow — Workflow control plane for AI coding assistants\n\nInstall for which platform?\n  1. Claude Code\n  2. Zoo Code\n\nEnter number (1 or 2): ",
+      (answer) => {
+        rl.close();
+        const choice = answer.trim();
+        if (choice === "1") resolve("claude-code");
+        else if (choice === "2") resolve("zoo-code");
+        else {
+          console.log("\nInvalid choice. Please run again and enter 1 or 2.\n");
+          process.exit(1);
+        }
+      }
+    );
+  });
+}
+
 const command = process.argv[2];
 
 const HELP = `
-Zoo Flow
+Zoo Flow — Workflow control plane for AI coding assistants
 
 Usage:
   npx @fernado03/zoo-flow@latest init
+  npx @fernado03/zoo-flow@latest init --platform=claude-code
+  npx @fernado03/zoo-flow@latest init --platform=zoo-code
   npx @fernado03/zoo-flow@latest init --force
   npx @fernado03/zoo-flow@latest update
   npx @fernado03/zoo-flow@latest update --dry-run
   npx @fernado03/zoo-flow@latest update --force
   npx @fernado03/zoo-flow@latest doctor
   npx @fernado03/zoo-flow@latest doctor --template-only
+  npx @fernado03/zoo-flow@latest doctor --template-only --platform=claude-code
 
 Commands:
-  init              Install Zoo Flow into the current project
+  init              Install Zoo Flow into the current project (prompts for platform)
   update            Back up current config and copy the latest template
   doctor            Validate Zoo Flow in the current project
   doctor --template-only   Validate this package's bundled template
 
 Options:
-  --force           Overwrite existing .roomodes and .roo after backup
+  --platform=<platform>   Install for 'claude-code' or 'zoo-code' (skip prompt)
+  --force           Overwrite existing config after backup
   --dry-run         Print what update would do without changing files
   --template-only   Validate this package's template instead of current project
+
+Supported platforms:
+  Claude Code       Installs CLAUDE.md, .claude/commands/, .claude/skills/
+  Zoo Code          Installs .roomodes, .roo/commands/, .roo/skills/, .roo/rules/
 `;
 
 function exitWithError(message) {
@@ -147,6 +193,22 @@ function appendZooFlowToGitignore(projectRoot) {
   return true;
 }
 
+function appendClaudeCodeToGitignore(projectRoot) {
+  const gi = path.join(projectRoot, ".gitignore");
+
+  if (pathExists(gi)) {
+    const content = fs.readFileSync(gi, "utf8");
+    if (content.includes(CLAUDE_CODE_GITIGNORE_MARKER)) return false;
+    const addition = `\n${CLAUDE_CODE_GITIGNORE_MARKER}\n${CLAUDE_CODE_GITIGNORE_ENTRIES.join("\n")}\n`;
+    fs.appendFileSync(gi, addition);
+    return true;
+  }
+
+  const fresh = `${CLAUDE_CODE_GITIGNORE_MARKER}\n${CLAUDE_CODE_GITIGNORE_ENTRIES.join("\n")}\n`;
+  fs.writeFileSync(gi, fresh);
+  return true;
+}
+
 function migrateRootContextDocs(projectRoot) {
   const moved = [];
   const targetFlowDir = path.join(projectRoot, ".zoo-flow");
@@ -543,25 +605,44 @@ function validateTemplate(rootDir) {
   return failures;
 }
 
-function install() {
+async function install() {
   const args = new Set(process.argv.slice(3));
   const force = args.has("--force");
   const projectRoot = process.cwd();
 
-  const sourceRoomodes = path.join(templateRoot, ".roomodes");
-  const sourceRoo = path.join(templateRoot, ".roo");
+  // Check if platform was specified via --platform flag
+  let platform = null;
+  for (const arg of process.argv.slice(3)) {
+    if (arg.startsWith("--platform=")) {
+      platform = arg.substring(11);
+      if (platform !== "claude-code" && platform !== "zoo-code") {
+        exitWithError(`Invalid platform: ${platform}. Must be 'claude-code' or 'zoo-code'.`);
+      }
+    }
+  }
 
-  if (!pathExists(sourceRoomodes) || !pathExists(sourceRoo)) {
-    exitWithError("Bundled template is missing templates/full/.roomodes or templates/full/.roo/");
+  // If no platform specified, prompt
+  if (!platform) {
+    platform = await promptPlatform();
   }
 
-  const targetRoomodes = path.join(projectRoot, ".roomodes");
-  const targetRoo = path.join(projectRoot, ".roo");
-  const hasExistingConfig = pathExists(targetRoomodes) || pathExists(targetRoo);
+  const root = getTemplateRoot(platform);
 
-  if (hasExistingConfig && !force) {
-    console.log(`
-Zoo Flow found existing config in this project.
+  if (platform === "zoo-code") {
+    const sourceRoomodes = path.join(root, ".roomodes");
+    const sourceRoo = path.join(root, ".roo");
+
+    if (!pathExists(sourceRoomodes) || !pathExists(sourceRoo)) {
+      exitWithError("Bundled template is missing templates/full/.roomodes or templates/full/.roo/");
+    }
+
+    const targetRoomodes = path.join(projectRoot, ".roomodes");
+    const targetRoo = path.join(projectRoot, ".roo");
+    const hasExistingConfig = pathExists(targetRoomodes) || pathExists(targetRoo);
+
+    if (hasExistingConfig && !force) {
+      console.log(`
+Zoo Flow found existing Zoo Code config in this project.
 
 Existing:
 ${pathExists(targetRoomodes) ? "  - .roomodes\n" : ""}${pathExists(targetRoo) ? "  - .roo/\n" : ""}
@@ -569,38 +650,38 @@ Run again with --force to back up and overwrite:
 
   npx @fernado03/zoo-flow@latest init --force
 `);
-    process.exit(0);
-  }
+      process.exit(0);
+    }
 
-  const backupDir = path.join(projectRoot, ".zoo-flow-backup", makeTimestamp());
-  let didBackup = false;
+    const backupDir = path.join(projectRoot, ".zoo-flow-backup", makeTimestamp());
+    let didBackup = false;
 
-  if (hasExistingConfig) {
-    fs.mkdirSync(backupDir, { recursive: true });
-    didBackup = backupIfExists(projectRoot, backupDir, ".roomodes") || didBackup;
-    didBackup = backupIfExists(projectRoot, backupDir, ".roo") || didBackup;
+    if (hasExistingConfig) {
+      fs.mkdirSync(backupDir, { recursive: true });
+      didBackup = backupIfExists(projectRoot, backupDir, ".roomodes") || didBackup;
+      didBackup = backupIfExists(projectRoot, backupDir, ".roo") || didBackup;
 
-    removeRecursive(targetRoomodes);
-    removeRecursive(targetRoo);
-  }
+      removeRecursive(targetRoomodes);
+      removeRecursive(targetRoo);
+    }
 
-  copyRecursive(sourceRoomodes, targetRoomodes);
-  copyRecursive(sourceRoo, targetRoo);
+    copyRecursive(sourceRoomodes, targetRoomodes);
+    copyRecursive(sourceRoo, targetRoo);
 
-  const didAddGitignore = appendZooFlowToGitignore(projectRoot);
-  const didAddZooFlow = copyZooFlowIfMissing(projectRoot);
+    const didAddGitignore = appendZooFlowToGitignore(projectRoot);
+    const didAddZooFlow = copyZooFlowIfMissing(projectRoot);
 
-  const copiedLines = ["  - .roomodes", "  - .roo/"];
-  if (didAddZooFlow) {
-    copiedLines.push("  - .zoo-flow/CONTEXT.md");
-    copiedLines.push("  - .zoo-flow/docs/adr/0001-record-architecture-decisions.md");
-  }
-  if (didAddGitignore) {
-    copiedLines.push("  - appended .roo/, .roomodes, .zoo-flow/ to .gitignore");
-  }
+    const copiedLines = ["  - .roomodes", "  - .roo/"];
+    if (didAddZooFlow) {
+      copiedLines.push("  - .zoo-flow/CONTEXT.md");
+      copiedLines.push("  - .zoo-flow/docs/adr/0001-record-architecture-decisions.md");
+    }
+    if (didAddGitignore) {
+      copiedLines.push("  - appended .roo/, .roomodes, .zoo-flow/ to .gitignore");
+    }
 
-  console.log(`
-Zoo Flow installed.
+    console.log(`
+Zoo Flow installed for Zoo Code.
 
 Copied:
 ${copiedLines.join("\n")}
@@ -615,93 +696,256 @@ Next:
 
 When workflow choices appear, type the number manually, e.g. 1.
 `);
+  } else {
+    // Claude Code installation
+    const sourceClaudeMd = path.join(root, "CLAUDE.md");
+    const sourceClaude = path.join(root, ".claude");
+    const sourceZooFlow = path.join(root, ".zoo-flow");
+
+    if (!pathExists(sourceClaudeMd) || !pathExists(sourceClaude)) {
+      exitWithError("Bundled template is missing templates/claude-code/CLAUDE.md or templates/claude-code/.claude/");
+    }
+
+    const targetClaudeMd = path.join(projectRoot, "CLAUDE.md");
+    const targetClaude = path.join(projectRoot, ".claude");
+    const targetZooFlow = path.join(projectRoot, ".zoo-flow");
+    const hasExistingConfig = pathExists(targetClaudeMd) || pathExists(targetClaude);
+
+    if (hasExistingConfig && !force) {
+      console.log(`
+Zoo Flow found existing Claude Code config in this project.
+
+Existing:
+${pathExists(targetClaudeMd) ? "  - CLAUDE.md\n" : ""}${pathExists(targetClaude) ? "  - .claude/\n" : ""}
+Run again with --force to back up and overwrite:
+
+  npx @fernado03/zoo-flow@latest init --force
+`);
+      process.exit(0);
+    }
+
+    const backupDir = path.join(projectRoot, ".zoo-flow-backup", makeTimestamp());
+    let didBackup = false;
+
+    if (hasExistingConfig) {
+      fs.mkdirSync(backupDir, { recursive: true });
+      didBackup = backupIfExists(projectRoot, backupDir, "CLAUDE.md") || didBackup;
+      didBackup = backupIfExists(projectRoot, backupDir, ".claude") || didBackup;
+
+      removeRecursive(targetClaudeMd);
+      removeRecursive(targetClaude);
+    }
+
+    copyRecursive(sourceClaudeMd, targetClaudeMd);
+    copyRecursive(sourceClaude, targetClaude);
+
+    // Copy .zoo-flow if missing
+    let didAddZooFlow = false;
+    if (!pathExists(targetZooFlow) && pathExists(sourceZooFlow)) {
+      copyRecursive(sourceZooFlow, targetZooFlow);
+      didAddZooFlow = true;
+    }
+
+    // Append to .gitignore
+    const didAddGitignore = appendClaudeCodeToGitignore(projectRoot);
+
+    const copiedLines = ["  - CLAUDE.md", "  - .claude/"];
+    if (didAddZooFlow) {
+      copiedLines.push("  - .zoo-flow/CONTEXT.md");
+      copiedLines.push("  - .zoo-flow/docs/adr/0001-record-architecture-decisions.md");
+    }
+    if (didAddGitignore) {
+      copiedLines.push("  - appended .claude/, .zoo-flow/ to .gitignore");
+    }
+
+    console.log(`
+Zoo Flow installed for Claude Code.
+
+Copied:
+${copiedLines.join("\n")}
+${didBackup ? `\nBackup:\n  ${backupDir}\n` : ""}
+Next:
+  1. Open .zoo-flow/START_HERE.md (first-run guide)
+  2. Restart Claude Code (or open in a new session)
+  3. Try a small request, e.g.:
+       change a harmless comment in README
+  4. Or use a slash command like /tweak or /fix
+
+When workflow choices appear, type the number manually, e.g. 1.
+`);
+  }
 }
 
 function doctor() {
   const args = new Set(process.argv.slice(3));
   const templateOnly = args.has("--template-only");
-  const rootToCheck = templateOnly ? templateRoot : process.cwd();
-  const failures = validateTemplate(rootToCheck);
   const optionalInfo = [];
 
-  if (!templateOnly) {
-    // Installed-project checks
-    const projectRoot = process.cwd();
-
-    // .gitignore check
-    const gi = path.join(projectRoot, ".gitignore");
-    if (pathExists(gi)) {
-      const giContent = fs.readFileSync(gi, "utf8");
-      if (!giContent.includes(ZOO_FLOW_GITIGNORE_MARKER)) {
-        optionalInfo.push("info: .gitignore does not contain Zoo Flow entries (.roomodes, .roo/, .zoo-flow/, docs/agents/, AGENTS.md)");
+  // Platform detection for --template-only
+  let platform = null;
+  for (const arg of process.argv.slice(3)) {
+    if (arg.startsWith("--platform=")) {
+      platform = arg.substring(11);
+      if (platform !== "claude-code" && platform !== "zoo-code") {
+        exitWithError(`Invalid platform: ${platform}. Must be 'claude-code' or 'zoo-code'.`);
       }
-    } else {
-      optionalInfo.push("info: .gitignore missing — run `npx @fernado03/zoo-flow@latest init` to create it");
     }
+  }
 
-    // START_HERE.md check
-    const startHere = path.join(projectRoot, ".zoo-flow", "START_HERE.md");
-    if (!pathExists(startHere)) {
-      optionalInfo.push("info: .zoo-flow/START_HERE.md missing — run `npx @fernado03/zoo-flow@latest init` to install starter docs");
-    }
+  let rootToCheck;
+  let detectedPlatform;
 
-    // CONTEXT.md check
-    const contextMd = path.join(projectRoot, ".zoo-flow", "CONTEXT.md");
-    if (!pathExists(contextMd)) {
-      optionalInfo.push("info: .zoo-flow/CONTEXT.md missing — run `/scaffold-context` to create project context");
-    }
+  if (templateOnly) {
+    detectedPlatform = platform || "zoo-code";
+    rootToCheck = getTemplateRoot(detectedPlatform);
+  } else {
+    const projectRoot = process.cwd();
+    detectedPlatform = detectPlatform(projectRoot);
 
-    // Project-profile check
-    const profile = path.join(projectRoot, ".zoo-flow", "project-profile.json");
-    if (!pathExists(profile)) {
-      optionalInfo.push("info: .zoo-flow/project-profile.json missing — run `/setup-matt-pocock-skills` to configure");
+    if (!detectedPlatform) {
+      exitWithError("No Zoo Flow installation found. Run `npx @fernado03/zoo-flow@latest init` first.");
     }
 
-    // LESSONS check
-    const lessons = path.join(projectRoot, ".zoo-flow", "LESSONS.md");
-    if (!pathExists(lessons)) {
-      optionalInfo.push("info: .zoo-flow/LESSONS.md missing — optional, created on demand");
-    }
+    rootToCheck = projectRoot;
+  }
+
+  if (detectedPlatform === "zoo-code") {
+    // Zoo Code validation (existing)
+    const failures = validateTemplate(rootToCheck);
+
+    if (!templateOnly) {
+      const projectRoot = process.cwd();
+
+      const gi = path.join(projectRoot, ".gitignore");
+      if (pathExists(gi)) {
+        const giContent = fs.readFileSync(gi, "utf8");
+        if (!giContent.includes(ZOO_FLOW_GITIGNORE_MARKER)) {
+          optionalInfo.push("info: .gitignore does not contain Zoo Flow entries (.roomodes, .roo/, .zoo-flow/, docs/agents/, AGENTS.md)");
+        }
+      } else {
+        optionalInfo.push("info: .gitignore missing — run `npx @fernado03/zoo-flow@latest init` to create it");
+      }
 
-    // Verify three modes exist in .roomodes
-    const roomodesPath = path.join(projectRoot, ".roomodes");
-    if (pathExists(roomodesPath)) {
-      try {
-        const roomodes = readJson(roomodesPath);
-        const slugs = (roomodes.customModes || []).map((m) => m.slug);
-        const required = ["custom-orchestrator", "system-architect", "code-tweaker"];
-        for (const slug of required) {
-          if (!slugs.includes(slug)) {
-            failures.push(`.roomodes missing required mode slug: ${slug}`);
+      const startHere = path.join(projectRoot, ".zoo-flow", "START_HERE.md");
+      if (!pathExists(startHere)) {
+        optionalInfo.push("info: .zoo-flow/START_HERE.md missing — run `npx @fernado03/zoo-flow@latest init` to install starter docs");
+      }
+
+      const contextMd = path.join(projectRoot, ".zoo-flow", "CONTEXT.md");
+      if (!pathExists(contextMd)) {
+        optionalInfo.push("info: .zoo-flow/CONTEXT.md missing — run `/scaffold-context` to create project context");
+      }
+
+      const profile = path.join(projectRoot, ".zoo-flow", "project-profile.json");
+      if (!pathExists(profile)) {
+        optionalInfo.push("info: .zoo-flow/project-profile.json missing — run `/setup-matt-pocock-skills` to configure");
+      }
+
+      const lessons = path.join(projectRoot, ".zoo-flow", "LESSONS.md");
+      if (!pathExists(lessons)) {
+        optionalInfo.push("info: .zoo-flow/LESSONS.md missing — optional, created on demand");
+      }
+
+      const roomodesPath = path.join(projectRoot, ".roomodes");
+      if (pathExists(roomodesPath)) {
+        try {
+          const roomodes = readJson(roomodesPath);
+          const slugs = (roomodes.customModes || []).map((m) => m.slug);
+          const required = ["custom-orchestrator", "system-architect", "code-tweaker"];
+          for (const slug of required) {
+            if (!slugs.includes(slug)) {
+              failures.push(`.roomodes missing required mode slug: ${slug}`);
+            }
           }
+        } catch (error) {
+          failures.push(`.roomodes parse error: ${error.message}`);
         }
-      } catch (error) {
-        failures.push(`.roomodes parse error: ${error.message}`);
+      }
+
+      if (!pathExists(path.join(projectRoot, ".roo", "commands"))) {
+        failures.push(".roo/commands/ missing — run `npx @fernado03/zoo-flow@latest update`");
+      }
+      if (!pathExists(path.join(projectRoot, ".roo", "rules"))) {
+        failures.push(".roo/rules/ missing — run `npx @fernado03/zoo-flow@latest update`");
       }
     }
 
-    // .roo/commands and .roo/rules exist
-    if (!pathExists(path.join(projectRoot, ".roo", "commands"))) {
-      failures.push(".roo/commands/ missing — run `npx @fernado03/zoo-flow@latest update`");
+    if (failures.length > 0) {
+      console.error("\nZoo Flow doctor found problems (Zoo Code):\n");
+      for (const failure of failures) {
+        console.error(`- ${failure}`);
+      }
+      if (!templateOnly) {
+        console.error("\nRun `npx @fernado03/zoo-flow@latest update --dry-run` to preview replacing local Zoo Flow config.");
+      }
+      console.error("");
+      process.exit(1);
+    }
+  } else {
+    // Claude Code validation
+    const failures = [];
+    const projectRoot = templateOnly ? rootToCheck : rootToCheck;
+
+    if (!pathExists(path.join(projectRoot, "CLAUDE.md"))) {
+      failures.push("Missing CLAUDE.md");
     }
-    if (!pathExists(path.join(projectRoot, ".roo", "rules"))) {
-      failures.push(".roo/rules/ missing — run `npx @fernado03/zoo-flow@latest update`");
+    if (!pathExists(path.join(projectRoot, ".claude", "commands"))) {
+      failures.push("Missing .claude/commands/");
+    }
+    if (!pathExists(path.join(projectRoot, ".claude", "skills"))) {
+      failures.push("Missing .claude/skills/");
     }
-  }
 
-  if (failures.length > 0) {
-    console.error("\nZoo Flow doctor found problems:\n");
-    for (const failure of failures) {
-      console.error(`- ${failure}`);
+    // Validate command files exist and reference valid skill paths
+    const commandsDir = path.join(projectRoot, ".claude", "commands");
+    const skillRefRegex = /`?\.claude\/skills\/[^\s`]+SKILL\.md`?/g;
+    if (pathExists(commandsDir)) {
+      for (const entry of fs.readdirSync(commandsDir)) {
+        if (!entry.endsWith(".md")) continue;
+        const commandFile = path.join(commandsDir, entry);
+        const text = fs.readFileSync(commandFile, "utf8");
+        const matches = text.matchAll(skillRefRegex);
+        for (const match of matches) {
+          const skillPath = match[0].replace(/`/g, "");
+          const skillAbsolute = path.join(projectRoot, skillPath);
+          if (!pathExists(skillAbsolute)) {
+            failures.push(`Command .claude/commands/${entry} references missing skill: ${skillPath}`);
+          }
+        }
+      }
     }
+
     if (!templateOnly) {
-      console.error("\nRun `npx @fernado03/zoo-flow@latest update --dry-run` to preview replacing local Zoo Flow config.");
+      const gi = path.join(projectRoot, ".gitignore");
+      if (pathExists(gi)) {
+        const giContent = fs.readFileSync(gi, "utf8");
+        if (!giContent.includes(CLAUDE_CODE_GITIGNORE_MARKER)) {
+          optionalInfo.push("info: .gitignore does not contain Zoo Flow Claude Code entries (.claude/, .zoo-flow/)");
+        }
+      } else {
+        optionalInfo.push("info: .gitignore missing — run `npx @fernado03/zoo-flow@latest init` to create it");
+      }
+
+      const startHere = path.join(projectRoot, ".zoo-flow", "START_HERE.md");
+      if (!pathExists(startHere)) {
+        optionalInfo.push("info: .zoo-flow/START_HERE.md missing — run `npx @fernado03/zoo-flow@latest init` to install starter docs");
+      }
+    }
+
+    if (failures.length > 0) {
+      console.error("\nZoo Flow doctor found problems (Claude Code):\n");
+      for (const failure of failures) {
+        console.error(`- ${failure}`);
+      }
+      if (!templateOnly) {
+        console.error("\nRun `npx @fernado03/zoo-flow@latest update --dry-run` to preview replacing local Zoo Flow config.");
+      }
+      console.error("");
+      process.exit(1);
     }
-    console.error("");
-    process.exit(1);
   }
 
-  const targetName = templateOnly ? "bundled template" : "current project";
+  const targetName = templateOnly ? `bundled template (${detectedPlatform})` : `current project (${detectedPlatform})`;
   console.log(`Zoo Flow doctor passed for ${targetName}: ${rootToCheck}`);
 
   if (optionalInfo.length > 0) {
@@ -718,19 +962,9 @@ function update() {
   const dryRun = args.has("--dry-run");
   const projectRoot = process.cwd();
 
-  const sourceRoomodes = path.join(templateRoot, ".roomodes");
-  const sourceRoo = path.join(templateRoot, ".roo");
-
-  if (!pathExists(sourceRoomodes) || !pathExists(sourceRoo)) {
-    exitWithError("Bundled template is missing templates/full/.roomodes or templates/full/.roo/");
-  }
+  const platform = detectPlatform(projectRoot);
 
-  const targetRoomodes = path.join(projectRoot, ".roomodes");
-  const targetRoo = path.join(projectRoot, ".roo");
-  const hasRoomodes = pathExists(targetRoomodes);
-  const hasRoo = pathExists(targetRoo);
-
-  if (!hasRoomodes && !hasRoo) {
+  if (!platform) {
     console.log(`
 Zoo Flow is not installed in this project yet.
 
@@ -740,15 +974,29 @@ Run:
     process.exit(0);
   }
 
-  if (dryRun) {
-    const rootContext = path.join(projectRoot, "CONTEXT.md");
-    const rootAdr = path.join(projectRoot, "docs", "adr");
-    const wouldMigrate =
-      (pathExists(rootContext) ? ["CONTEXT.md"] : [])
-        .concat(pathExists(rootAdr) ? ["docs/adr/"] : []);
+  if (platform === "zoo-code") {
+    const root = getTemplateRoot("zoo-code");
+    const sourceRoomodes = path.join(root, ".roomodes");
+    const sourceRoo = path.join(root, ".roo");
 
-    console.log(`
-Zoo Flow update dry run.
+    if (!pathExists(sourceRoomodes) || !pathExists(sourceRoo)) {
+      exitWithError("Bundled template is missing templates/full/.roomodes or templates/full/.roo/");
+    }
+
+    const targetRoomodes = path.join(projectRoot, ".roomodes");
+    const targetRoo = path.join(projectRoot, ".roo");
+    const hasRoomodes = pathExists(targetRoomodes);
+    const hasRoo = pathExists(targetRoo);
+
+    if (dryRun) {
+      const rootContext = path.join(projectRoot, "CONTEXT.md");
+      const rootAdr = path.join(projectRoot, "docs", "adr");
+      const wouldMigrate =
+        (pathExists(rootContext) ? ["CONTEXT.md"] : [])
+          .concat(pathExists(rootAdr) ? ["docs/adr/"] : []);
+
+      console.log(`
+Zoo Flow update dry run (Zoo Code).
 
 Would back up:
 ${hasRoomodes ? "  - .roomodes\n" : ""}${hasRoo ? "  - .roo/\n" : ""}
@@ -761,27 +1009,27 @@ Would append .roo/, .roomodes, .zoo-flow/ to .gitignore (idempotent).
 Run this to update:
   npx @fernado03/zoo-flow@latest update
 `);
-    process.exit(0);
-  }
+      process.exit(0);
+    }
 
-  const backupDir = path.join(projectRoot, ".zoo-flow-backup", makeTimestamp());
-  fs.mkdirSync(backupDir, { recursive: true });
+    const backupDir = path.join(projectRoot, ".zoo-flow-backup", makeTimestamp());
+    fs.mkdirSync(backupDir, { recursive: true });
 
-  let didBackup = false;
-  didBackup = backupIfExists(projectRoot, backupDir, ".roomodes") || didBackup;
-  didBackup = backupIfExists(projectRoot, backupDir, ".roo") || didBackup;
+    let didBackup = false;
+    didBackup = backupIfExists(projectRoot, backupDir, ".roomodes") || didBackup;
+    didBackup = backupIfExists(projectRoot, backupDir, ".roo") || didBackup;
 
-  removeRecursive(targetRoomodes);
-  removeRecursive(targetRoo);
+    removeRecursive(targetRoomodes);
+    removeRecursive(targetRoo);
 
-  copyRecursive(sourceRoomodes, targetRoomodes);
-  copyRecursive(sourceRoo, targetRoo);
+    copyRecursive(sourceRoomodes, targetRoomodes);
+    copyRecursive(sourceRoo, targetRoo);
 
-  const moved = migrateRootContextDocs(projectRoot);
-  appendZooFlowToGitignore(projectRoot);
+    const moved = migrateRootContextDocs(projectRoot);
+    appendZooFlowToGitignore(projectRoot);
 
-  console.log(`
-Zoo Flow updated.
+    console.log(`
+Zoo Flow updated (Zoo Code).
 
 Replaced:
   - .roomodes
@@ -792,6 +1040,65 @@ ${didBackup ? `Backup:\n  ${backupDir}\n\nTo restore the previous config:\n  rm
   2. Open Zoo Code
   3. Confirm the three custom modes still appear
 `);
+  } else {
+    // Claude Code update
+    const root = getTemplateRoot("claude-code");
+    const sourceClaudeMd = path.join(root, "CLAUDE.md");
+    const sourceClaude = path.join(root, ".claude");
+
+    if (!pathExists(sourceClaudeMd) || !pathExists(sourceClaude)) {
+      exitWithError("Bundled template is missing templates/claude-code/CLAUDE.md or templates/claude-code/.claude/");
+    }
+
+    const targetClaudeMd = path.join(projectRoot, "CLAUDE.md");
+    const targetClaude = path.join(projectRoot, ".claude");
+    const hasClaudeMd = pathExists(targetClaudeMd);
+    const hasClaude = pathExists(targetClaude);
+
+    if (dryRun) {
+      console.log(`
+Zoo Flow update dry run (Claude Code).
+
+Would back up:
+${hasClaudeMd ? "  - CLAUDE.md\n" : ""}${hasClaude ? "  - .claude/\n" : ""}
+Would replace with latest template:
+  - CLAUDE.md
+  - .claude/
+
+Would append .claude/, .zoo-flow/ to .gitignore (idempotent).
+
+Run this to update:
+  npx @fernado03/zoo-flow@latest update
+`);
+      process.exit(0);
+    }
+
+    const backupDir = path.join(projectRoot, ".zoo-flow-backup", makeTimestamp());
+    fs.mkdirSync(backupDir, { recursive: true });
+
+    let didBackup = false;
+    didBackup = backupIfExists(projectRoot, backupDir, "CLAUDE.md") || didBackup;
+    didBackup = backupIfExists(projectRoot, backupDir, ".claude") || didBackup;
+
+    removeRecursive(targetClaudeMd);
+    removeRecursive(targetClaude);
+
+    copyRecursive(sourceClaudeMd, targetClaudeMd);
+    copyRecursive(sourceClaude, targetClaude);
+
+    appendClaudeCodeToGitignore(projectRoot);
+
+    console.log(`
+Zoo Flow updated (Claude Code).
+
+Replaced:
+  - CLAUDE.md
+  - .claude/
+${didBackup ? `Backup:\n  ${backupDir}\n\nTo restore the previous config:\n  rm -rf CLAUDE.md .claude\n  cp -R ${backupDir}/. .\n` : ""}Next:
+  1. Restart Claude Code (or open in a new session)
+  2. Confirm slash commands are available
+`);
+  }
 }
 
 if (!command || command === "--help" || command === "-h") {
@@ -800,7 +1107,10 @@ if (!command || command === "--help" || command === "-h") {
 }
 
 if (command === "init") {
-  install();
+  install().catch((error) => {
+    console.error(`\nError: ${error.message}\n`);
+    process.exit(1);
+  });
 } else if (command === "update") {
   update();
 } else if (command === "doctor") {