@keeroklab/cli 0.2.0 → 0.4.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@keeroklab/cli",
-  "version": "0.2.0",
+  "version": "0.4.0",
   "description": "CLI wrapper for Keerok Lab — live Claude Code supervision",
   "type": "module",
   "bin": {
@@ -33,6 +33,7 @@
   "dependencies": {
     "chalk": "^5.3.0",
     "commander": "^12.0.0",
+    "node-pty": "^1.1.0",
     "ws": "^8.16.0"
   },
   "engines": {

package/src/index.js

@@ -39,13 +39,20 @@ program
         console.log(`\n📋 Instructions:\n${config.instructions}\n`);
       }
 
-      // Step 2: Connect WebSocket
+      // Step 2: Connect WebSocket and wait for session config
+      let resolveConfig;
+      const configReady = new Promise((resolve, reject) => {
+        resolveConfig = resolve;
+        setTimeout(() => reject(new Error('Timed out waiting for session config (5s)')), 5000);
+      });
+
       const ws = createWsClient(config.ws_url, {
         onSessionConfig: (data) => {
           // API key received — inject into environment
           if (data.api_key) {
             process.env.ANTHROPIC_API_KEY = data.api_key;
           }
+          resolveConfig(data);
         },
         onBudgetWarning: (data) => {
           console.log(`\n⚠️  Budget warning: $${data.remaining_usd.toFixed(2)} remaining (${data.percentage.toFixed(0)}% used)\n`);
@@ -70,9 +77,12 @@ program
         },
       });
 
-      // Wait for WS connection
+      // Wait for WS connection, then wait for session_config (API key)
       await ws.waitForConnection();
-      console.log('🔗 WebSocket connected\n');
+      console.log('🔗 WebSocket connected');
+      console.log('⏳ Waiting for session config...');
+      await configReady;
+      console.log('🔑 Session config received\n');
       console.log('─'.repeat(50));
       console.log(`Starting: ${options.command}`);
       console.log('─'.repeat(50) + '\n');

package/src/pty-capture.js

@@ -1,10 +1,9 @@
 /**
- * pty-capture.js — Spawns a process and captures output.
- * Uses child_process.spawn with pipes to capture terminal output
- * and forward it to both stdout and the WebSocket handler.
+ * pty-capture.js — Spawns a process in a real pseudo-terminal.
+ * Uses node-pty so interactive TUI apps (like claude) work properly.
  */
 
-import { spawn } from 'child_process';
+import pty from 'node-pty';
 import os from 'os';
 
 export function spawnPty(command, handlers) {
@@ -12,32 +11,25 @@ export function spawnPty(command, handlers) {
   const cols = process.stdout.columns || 80;
   const rows = process.stdout.rows || 24;
 
-  const child = spawn(shell, ['-c', command], {
-    stdio: ['pipe', 'pipe', 'pipe'],
+  const child = pty.spawn(shell, ['-c', command], {
+    name: 'xterm-256color',
+    cols,
+    rows,
     cwd: process.cwd(),
     env: {
       ...process.env,
-      TERM: 'xterm-256color',
       FORCE_COLOR: '1',
-      COLUMNS: String(cols),
-      LINES: String(rows),
     },
   });
 
-  // Forward stdout to both terminal and handler
-  child.stdout.on('data', (data) => {
+  // Forward PTY output to both local terminal and handler
+  child.onData((data) => {
     process.stdout.write(data);
-    handlers.onData?.(data.toString());
+    handlers.onData?.(data);
   });
 
-  // Forward stderr to both terminal and handler
-  child.stderr.on('data', (data) => {
-    process.stderr.write(data);
-    handlers.onData?.(data.toString());
-  });
-
-  child.on('exit', (code) => {
-    handlers.onExit?.(code ?? 0);
+  child.onExit(({ exitCode }) => {
+    handlers.onExit?.(exitCode ?? 0);
   });
 
   // Forward stdin to child
@@ -46,15 +38,14 @@ export function spawnPty(command, handlers) {
   }
   process.stdin.resume();
   process.stdin.on('data', (data) => {
-    if (child.stdin.writable) {
-      child.stdin.write(data);
-    }
+    child.write(data.toString());
   });
 
   // Handle terminal resize
   process.stdout.on('resize', () => {
     const newCols = process.stdout.columns;
     const newRows = process.stdout.rows;
+    child.resize(newCols, newRows);
     handlers.onResize?.(newCols, newRows);
   });
 
@@ -66,12 +57,10 @@ export function spawnPty(command, handlers) {
       child.kill();
     },
     write(data) {
-      if (child.stdin.writable) {
-        child.stdin.write(data);
-      }
+      child.write(data);
     },
-    resize() {
-      // child_process doesn't support resize, but we notify the handler
+    resize(cols, rows) {
+      child.resize(cols, rows);
     },
   };
 }