codebuff save-agent command

Author: jahooma

npm-app/src/cli-handlers/save-agent.ts

@@ -0,0 +1,94 @@
+import * as fs from 'fs'
+import * as path from 'path'
+
+import { codebuffConfigFile } from '@codebuff/common/json-config/constants'
+import { green, red, yellow, cyan, gray } from 'picocolors'
+
+import { getProjectRoot } from '../project-files'
+import { loadRawCodebuffConfig } from '../json-config/parser'
+
+/**
+ * Add one or more agent IDs to the addedSpawnableAgents field in codebuff.json
+ * @param agentIds Array of agent IDs to add
+ */
+export async function handleSaveAgent(agentIds: string[]): Promise<void> {
+  if (!agentIds || agentIds.length === 0) {
+    console.log(
+      red(
+        'Agent ID is required. Usage: codebuff save-agent <agent-id> [agent-id2] ...',
+      ),
+    )
+    console.log(
+      gray(
+        'Example: codebuff save-agent codebuff/file-picker@1.0.0 codebuff/reviewer@1.0.0',
+      ),
+    )
+    return
+  }
+
+  const projectRoot = getProjectRoot()
+  const configPath = path.join(projectRoot, codebuffConfigFile)
+
+  try {
+    // Use the safe configuration loader that handles JSONC
+    const config = loadRawCodebuffConfig()
+
+    // Initialize addedSpawnableAgents if it doesn't exist
+    if (!config.addedSpawnableAgents) {
+      config.addedSpawnableAgents = []
+    }
+
+    // Track which agents were added vs already existed
+    const newAgents: string[] = []
+    const existingAgents: string[] = []
+
+    for (const agentId of agentIds) {
+      if (config.addedSpawnableAgents.includes(agentId)) {
+        existingAgents.push(agentId)
+      } else {
+        config.addedSpawnableAgents.push(agentId)
+        newAgents.push(agentId)
+      }
+    }
+
+    // Write the updated configuration back to file
+    const configJson = JSON.stringify(config, null, 2)
+    fs.writeFileSync(configPath, configJson + '\n')
+
+    // Provide feedback to the user
+    if (newAgents.length > 0) {
+      console.log(
+        green(
+          `✅ Successfully added ${newAgents.length} agent(s) to codebuff.json:`,
+        ),
+      )
+      newAgents.forEach((agentId) => {
+        console.log(cyan(`  + ${agentId}`))
+      })
+    }
+
+    if (existingAgents.length > 0) {
+      console.log(
+        yellow(
+          `⚠️  ${existingAgents.length} agent(s) were already in the list:`,
+        ),
+      )
+      existingAgents.forEach((agentId) => {
+        console.log(gray(`  • ${agentId}`))
+      })
+    }
+
+    if (newAgents.length > 0) {
+      console.log()
+      console.log(
+        gray('These agents are now available for spawning by the base agent.'),
+      )
+    }
+  } catch (error) {
+    console.log(
+      red(
+        `Error updating codebuff.json: ${error instanceof Error ? error.message : String(error)}`,
+      ),
+    )
+  }
+}

npm-app/src/index.ts

@@ -10,6 +10,7 @@ import { CLI } from './cli'
 import { cliArguments, cliOptions } from './cli-definitions'
 import { handlePublish } from './cli-handlers/publish'
 import { handleInitAgents } from './cli-handlers/init-agents'
+import { handleSaveAgent } from './cli-handlers/save-agent'
 import { npmAppVersion, backendUrl } from './config'
 import { createTemplateProject } from './create-template-project'
 import { printModeLog, setPrintMode } from './display/print-mode'
@@ -122,6 +123,7 @@ Examples:
   $ codebuff --trace                          # Enable subagent trace logging to .agents/traces/*.log
   $ codebuff --create nextjs my-app           # Create and scaffold a new Next.js project
   $ codebuff init-agents                      # Create example agent files in .agents directory
+  $ codebuff save-agent my-agent-id           # Add agent ID to spawnable agents list
   $ codebuff publish my-agent                 # Publish agent template to store
   $ codebuff --agent file-picker "find relevant files for authentication"
   $ codebuff --agent reviewer --params '{"focus": "security"}' "review this code"
@@ -159,6 +161,13 @@ For all commands and options, run 'codebuff' and then type 'help'.
     process.exit(0)
   }
 
+  // Handle save-agent command
+  if (args[0] === 'save-agent') {
+    const agentIds = args.slice(1)
+    await handleSaveAgent(agentIds)
+    process.exit(0)
+  }
+
   // Handle deprecated --pro flag
   if (options.pro) {
     console.error(
@@ -225,8 +234,10 @@ For all commands and options, run 'codebuff' and then type 'help'.
   // Remove the first argument if it's the compiled binary path which bun weirdly injects (starts with /$bunfs)
   const filteredArgs = args[0]?.startsWith('/$bunfs') ? args.slice(1) : args
 
-  // If first arg is a command like 'publish', don't treat it as initial input
-  const isCommand = filteredArgs[0] === 'publish'
+  // If first arg is a command like 'publish' or 'save-agent', don't treat it as initial input
+  const isCommand = ['publish', 'init-agents', 'save-agent'].includes(
+    filteredArgs[0],
+  )
   const initialInput = isCommand ? '' : filteredArgs.join(' ')
 
   codebuff({

npm-app/src/json-config/parser.ts

@@ -123,3 +123,21 @@ export function loadCodebuffConfig(): CodebuffConfig {
     return getDefaultConfig()
   }
 }
+
+export function loadRawCodebuffConfig(): Partial<CodebuffConfig> {
+  const projectPath = getProjectRoot()
+  const configPathPrimary = path.join(projectPath, codebuffConfigFile)
+  const configPathBackup = path.join(projectPath, codebuffConfigFileBackup)
+  const configPath = existsSync(configPathBackup)
+    ? configPathBackup
+    : existsSync(configPathPrimary)
+      ? configPathPrimary
+      : null
+
+  if (configPath === null) {
+    return {}
+  }
+
+  const jsoncContent = readFileSync(configPath, 'utf-8')
+  return parseJsonc(jsoncContent)
+}