bonzai-burn 1.0.3 → 1.0.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bonzai-burn",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "Git branch-based cleanup tool with btrim and brevert commands",
   "type": "module",
   "main": "src/index.js",

package/src/btrim.js

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
-import { execSync } from 'child_process';
-import { existsSync, mkdirSync, readFileSync, writeFileSync } from 'fs';
+import { execSync, spawn } from 'child_process';
+import fs from 'fs';
 import { join } from 'path';
 
 const BONZAI_DIR = 'bonzai';
@@ -16,17 +16,37 @@ Define your cleanup requirements below. btrim will follow these instructions.
 - Clean up console.log statements
 `;
 
+function initializeBonzai() {
+  const bonzaiPath = join(process.cwd(), BONZAI_DIR);
+  const specsPath = join(bonzaiPath, SPECS_FILE);
+
+  // Check if bonzai/ folder exists
+  if (!fs.existsSync(bonzaiPath)) {
+    // Create bonzai/ folder
+    fs.mkdirSync(bonzaiPath, { recursive: true });
+    console.log(`📁 Created ${BONZAI_DIR}/ folder`);
+  }
+
+  // Generate bonzai/specs.md with template
+  if (!fs.existsSync(specsPath)) {
+    fs.writeFileSync(specsPath, DEFAULT_SPECS);
+    console.log(`📝 Created ${BONZAI_DIR}/${SPECS_FILE}`);
+    console.log(`\n⚠️  Please edit ${BONZAI_DIR}/${SPECS_FILE} to define your cleanup rules before running btrim.\n`);
+    process.exit(0);
+  }
+}
+
 function ensureBonzaiDir() {
   const bonzaiPath = join(process.cwd(), BONZAI_DIR);
   const specsPath = join(bonzaiPath, SPECS_FILE);
 
-  if (!existsSync(bonzaiPath)) {
-    mkdirSync(bonzaiPath, { recursive: true });
+  if (!fs.existsSync(bonzaiPath)) {
+    fs.mkdirSync(bonzaiPath, { recursive: true });
     console.log(`📁 Created ${BONZAI_DIR}/ folder\n`);
   }
 
-  if (!existsSync(specsPath)) {
-    writeFileSync(specsPath, DEFAULT_SPECS);
+  if (!fs.existsSync(specsPath)) {
+    fs.writeFileSync(specsPath, DEFAULT_SPECS);
     console.log(`📝 Created ${BONZAI_DIR}/${SPECS_FILE} - edit this file to define your cleanup specs\n`);
   }
 
@@ -34,7 +54,7 @@ function ensureBonzaiDir() {
 }
 
 function loadSpecs(specsPath) {
-  const content = readFileSync(specsPath, 'utf-8');
+  const content = fs.readFileSync(specsPath, 'utf-8');
   return `You are a code cleanup assistant. Follow these specifications:\n\n${content}`;
 }
 
@@ -46,21 +66,132 @@ function execVisible(command) {
   execSync(command, { stdio: 'inherit' });
 }
 
+function executeClaude(requirements) {
+  // Check if Claude CLI exists
+  try {
+    execSync('which claude', { encoding: 'utf-8', stdio: 'pipe' });
+  } catch (error) {
+    throw new Error(
+      'Claude Code CLI not found.\n' +
+      'Install it with: npm install -g @anthropic-ai/claude-code'
+    );
+  }
+
+  // Track token usage
+  let totalInputTokens = 0;
+  let totalOutputTokens = 0;
+  let lastToolName = '';
+
+  // Execute Claude with streaming JSON output
+
+  return new Promise((resolve, reject) => {
+    const args = [
+      '-p', requirements,
+      '--allowedTools', 'Read,Write,Edit,Bash',
+      '--permission-mode', 'dontAsk',
+      '--output-format', 'stream-json'
+    ];
+
+    const claude = spawn('claude', args, {
+      stdio: ['inherit', 'pipe', 'pipe']
+    });
+
+    let buffer = '';
+
+    claude.stdout.on('data', (data) => {
+      buffer += data.toString();
+
+      // Process complete JSON lines
+      const lines = buffer.split('\n');
+      buffer = lines.pop(); // Keep incomplete line in buffer
+
+      for (const line of lines) {
+        if (!line.trim()) continue;
+
+        try {
+          const event = JSON.parse(line);
+
+          // Track tokens from assistant messages
+          if (event.type === 'assistant' && event.message?.usage) {
+            const usage = event.message.usage;
+            if (usage.input_tokens) totalInputTokens += usage.input_tokens;
+            if (usage.output_tokens) totalOutputTokens += usage.output_tokens;
+          }
+
+          // Show tool usage updates
+          if (event.type === 'content_block_start' && event.content_block?.type === 'tool_use') {
+            lastToolName = event.content_block.name || '';
+          }
+
+          if (event.type === 'content_block_stop' && lastToolName) {
+            const icon = getToolIcon(lastToolName);
+            console.log(`  ${icon} ${lastToolName}`);
+            lastToolName = '';
+          }
+
+          // Show result events with file info
+          if (event.type === 'result') {
+            if (event.usage) {
+              // Final usage stats
+              totalInputTokens = event.usage.input_tokens || totalInputTokens;
+              totalOutputTokens = event.usage.output_tokens || totalOutputTokens;
+            }
+          }
+
+        } catch (e) {
+          // Not valid JSON, skip
+        }
+      }
+    });
+
+    claude.stderr.on('data', (data) => {
+      // Show errors but don't clutter with minor stderr
+      const msg = data.toString().trim();
+      if (msg && !msg.includes('ExperimentalWarning')) {
+        console.error(msg);
+      }
+    });
+
+    claude.on('close', (code) => {
+      // Print token summary
+      console.log(`\n📊 Tokens: ${totalInputTokens.toLocaleString()} in / ${totalOutputTokens.toLocaleString()} out`);
+
+      if (code === 0) {
+        resolve();
+      } else {
+        reject(new Error(`Claude exited with code ${code}`));
+      }
+    });
+
+    claude.on('error', (err) => {
+      reject(new Error(`Failed to execute Claude: ${err.message}`));
+    });
+  });
+}
+
+function getToolIcon(toolName) {
+  const icons = {
+    'Read': '📖',
+    'Write': '✏️',
+    'Edit': '🔧',
+    'Bash': '💻',
+    'Glob': '🔍',
+    'Grep': '🔎'
+  };
+  return icons[toolName] || '🔹';
+}
+
 async function burn() {
   try {
+    // Initialize bonzai folder and specs.md on first execution
+    initializeBonzai();
+    
     // Ensure bonzai directory and specs file exist
     const specsPath = ensureBonzaiDir();
     const specs = loadSpecs(specsPath);
 
-    // Check if Claude CLI exists
+    // Check if Claude CLI exists and execute
     console.log('🔍 Checking for Claude Code CLI...');
-    try {
-      exec('which claude');
-    } catch {
-      console.error('❌ Claude Code CLI not found');
-      console.error('Install: npm install -g @anthropic-ai/claude-code');
-      process.exit(1);
-    }
 
     // Check if in git repo
     try {
@@ -114,7 +245,7 @@ async function burn() {
     const startTime = Date.now();
 
     // Execute Claude with specs from bonzai/specs.md
-    execVisible(`claude -p "${specs.replace(/"/g, '\\"')}" --allowedTools "Read,Write,Edit,Bash" --permission-mode dontAsk`);
+    await executeClaude(specs);
 
     const duration = Math.round((Date.now() - startTime) / 1000);
 
@@ -135,6 +266,9 @@ async function burn() {
 
   } catch (error) {
     console.error('❌ Burn failed:', error.message);
+    if (error.message.includes('Claude Code CLI not found')) {
+      console.error('\n' + error.message);
+    }
     process.exit(1);
   }
 }