@automagik/genie 0.260202.1607 → 0.260202.1901

package/src/lib/worker-registry.ts

@@ -0,0 +1,229 @@
+/**
+ * Worker Registry - Manages worker state persistence
+ *
+ * Tracks Claude workers bound to beads issues, storing state in
+ * ~/.config/term/workers.json
+ */
+
+import { mkdir, readFile, writeFile } from 'fs/promises';
+import { join, dirname } from 'path';
+import { homedir } from 'os';
+
+// ============================================================================
+// Types
+// ============================================================================
+
+export type WorkerState =
+  | 'spawning'    // Worker being created
+  | 'working'     // Actively producing output
+  | 'idle'        // At prompt, waiting for input
+  | 'permission'  // Waiting for permission approval
+  | 'question'    // Waiting for question answer
+  | 'done'        // Task completed, ready for close
+  | 'error';      // Encountered error
+
+export interface Worker {
+  /** Unique worker ID (usually matches taskId, e.g., "bd-42") */
+  id: string;
+  /** tmux pane ID (e.g., "%16") */
+  paneId: string;
+  /** tmux session name */
+  session: string;
+  /** Path to git worktree, null if using shared repo */
+  worktree: string | null;
+  /** Beads issue ID this worker is bound to */
+  taskId: string;
+  /** Task title from beads */
+  taskTitle?: string;
+  /** Associated wish slug (if from decompose) */
+  wishSlug?: string;
+  /** Execution group number within wish */
+  groupNumber?: number;
+  /** ISO timestamp when worker was started */
+  startedAt: string;
+  /** Current worker state */
+  state: WorkerState;
+  /** Last state change timestamp */
+  lastStateChange: string;
+  /** Repository path where worker operates */
+  repoPath: string;
+}
+
+export interface WorkerRegistry {
+  workers: Record<string, Worker>;
+  lastUpdated: string;
+}
+
+// ============================================================================
+// Configuration
+// ============================================================================
+
+const CONFIG_DIR = join(homedir(), '.config', 'term');
+const REGISTRY_FILE = join(CONFIG_DIR, 'workers.json');
+
+// ============================================================================
+// Private Functions
+// ============================================================================
+
+async function ensureConfigDir(): Promise<void> {
+  await mkdir(CONFIG_DIR, { recursive: true });
+}
+
+async function loadRegistry(): Promise<WorkerRegistry> {
+  try {
+    const content = await readFile(REGISTRY_FILE, 'utf-8');
+    return JSON.parse(content);
+  } catch {
+    return { workers: {}, lastUpdated: new Date().toISOString() };
+  }
+}
+
+async function saveRegistry(registry: WorkerRegistry): Promise<void> {
+  await ensureConfigDir();
+  registry.lastUpdated = new Date().toISOString();
+  await writeFile(REGISTRY_FILE, JSON.stringify(registry, null, 2));
+}
+
+// ============================================================================
+// Public API
+// ============================================================================
+
+/**
+ * Register a new worker in the registry
+ */
+export async function register(worker: Worker): Promise<void> {
+  const registry = await loadRegistry();
+  registry.workers[worker.id] = worker;
+  await saveRegistry(registry);
+}
+
+/**
+ * Unregister (remove) a worker from the registry
+ */
+export async function unregister(id: string): Promise<void> {
+  const registry = await loadRegistry();
+  delete registry.workers[id];
+  await saveRegistry(registry);
+}
+
+/**
+ * Get a worker by ID
+ */
+export async function get(id: string): Promise<Worker | null> {
+  const registry = await loadRegistry();
+  return registry.workers[id] || null;
+}
+
+/**
+ * List all workers
+ */
+export async function list(): Promise<Worker[]> {
+  const registry = await loadRegistry();
+  return Object.values(registry.workers);
+}
+
+/**
+ * Update a worker's state
+ */
+export async function updateState(id: string, state: WorkerState): Promise<void> {
+  const registry = await loadRegistry();
+  const worker = registry.workers[id];
+  if (worker) {
+    worker.state = state;
+    worker.lastStateChange = new Date().toISOString();
+    await saveRegistry(registry);
+  }
+}
+
+/**
+ * Update multiple worker fields
+ */
+export async function update(id: string, updates: Partial<Worker>): Promise<void> {
+  const registry = await loadRegistry();
+  const worker = registry.workers[id];
+  if (worker) {
+    Object.assign(worker, updates);
+    if (updates.state) {
+      worker.lastStateChange = new Date().toISOString();
+    }
+    await saveRegistry(registry);
+  }
+}
+
+/**
+ * Find worker by tmux pane ID
+ */
+export async function findByPane(paneId: string): Promise<Worker | null> {
+  const workers = await list();
+  // Normalize pane ID (with or without % prefix)
+  const normalizedPaneId = paneId.startsWith('%') ? paneId : `%${paneId}`;
+  return workers.find(w => w.paneId === normalizedPaneId) || null;
+}
+
+/**
+ * Find worker by beads task ID
+ */
+export async function findByTask(taskId: string): Promise<Worker | null> {
+  const workers = await list();
+  return workers.find(w => w.taskId === taskId) || null;
+}
+
+/**
+ * Find workers by wish slug
+ */
+export async function findByWish(wishSlug: string): Promise<Worker[]> {
+  const workers = await list();
+  return workers.filter(w => w.wishSlug === wishSlug);
+}
+
+/**
+ * Check if a worker exists for a given task
+ */
+export async function hasWorkerForTask(taskId: string): Promise<boolean> {
+  const worker = await findByTask(taskId);
+  return worker !== null;
+}
+
+/**
+ * Get workers in a specific state
+ */
+export async function getByState(state: WorkerState): Promise<Worker[]> {
+  const workers = await list();
+  return workers.filter(w => w.state === state);
+}
+
+/**
+ * Calculate elapsed time for a worker
+ */
+export function getElapsedTime(worker: Worker): { ms: number; formatted: string } {
+  const startTime = new Date(worker.startedAt).getTime();
+  const ms = Date.now() - startTime;
+
+  const minutes = Math.floor(ms / 60000);
+  const hours = Math.floor(minutes / 60);
+
+  let formatted: string;
+  if (hours > 0) {
+    formatted = `${hours}h ${minutes % 60}m`;
+  } else if (minutes > 0) {
+    formatted = `${minutes}m`;
+  } else {
+    formatted = '<1m';
+  }
+
+  return { ms, formatted };
+}
+
+/**
+ * Get the config directory path
+ */
+export function getConfigDir(): string {
+  return CONFIG_DIR;
+}
+
+/**
+ * Get the registry file path
+ */
+export function getRegistryPath(): string {
+  return REGISTRY_FILE;
+}

package/src/term-commands/close.ts

@@ -0,0 +1,256 @@
+/**
+ * Close command - Close beads issue and cleanup worker
+ *
+ * Usage:
+ *   term close <bd-id>     - Close issue, cleanup worktree, kill worker
+ *
+ * Options:
+ *   --no-sync              - Skip bd sync
+ *   --keep-worktree        - Don't remove the worktree
+ *   --merge                - Merge worktree changes to main branch
+ *   -y, --yes              - Skip confirmation
+ */
+
+import { $ } from 'bun';
+import { confirm } from '@inquirer/prompts';
+import * as tmux from '../lib/tmux.js';
+import * as registry from '../lib/worker-registry.js';
+import * as beadsRegistry from '../lib/beads-registry.js';
+import { WorktreeManager } from '../lib/worktree.js';
+import { join } from 'path';
+import { homedir } from 'os';
+
+// Use beads registry when enabled
+const useBeads = beadsRegistry.isBeadsRegistryEnabled();
+
+// ============================================================================
+// Types
+// ============================================================================
+
+export interface CloseOptions {
+  noSync?: boolean;
+  keepWorktree?: boolean;
+  merge?: boolean;
+  yes?: boolean;
+}
+
+// ============================================================================
+// Configuration
+// ============================================================================
+
+const WORKTREE_BASE = join(homedir(), '.local', 'share', 'term', 'worktrees');
+
+// ============================================================================
+// Helper Functions
+// ============================================================================
+
+/**
+ * Run bd command
+ */
+async function runBd(args: string[]): Promise<{ stdout: string; exitCode: number }> {
+  try {
+    const result = await $`bd ${args}`.quiet();
+    return { stdout: result.stdout.toString().trim(), exitCode: 0 };
+  } catch (error: any) {
+    return { stdout: error.stdout?.toString().trim() || '', exitCode: error.exitCode || 1 };
+  }
+}
+
+/**
+ * Close beads issue
+ */
+async function closeBeadsIssue(taskId: string): Promise<boolean> {
+  const { exitCode } = await runBd(['close', taskId]);
+  return exitCode === 0;
+}
+
+/**
+ * Sync beads to git
+ */
+async function syncBeads(): Promise<boolean> {
+  const { exitCode } = await runBd(['sync']);
+  return exitCode === 0;
+}
+
+/**
+ * Merge worktree branch to main
+ */
+async function mergeToMain(
+  repoPath: string,
+  branchName: string
+): Promise<boolean> {
+  try {
+    // Get current branch
+    const currentResult = await $`git -C ${repoPath} branch --show-current`.quiet();
+    const currentBranch = currentResult.stdout.toString().trim();
+
+    if (currentBranch === branchName) {
+      console.log(`⚠️  Already on branch ${branchName}. Skipping merge.`);
+      return true;
+    }
+
+    // Checkout main and merge
+    console.log(`   Switching to ${currentBranch}...`);
+    await $`git -C ${repoPath} checkout ${currentBranch}`.quiet();
+
+    console.log(`   Merging ${branchName}...`);
+    await $`git -C ${repoPath} merge ${branchName} --no-edit`.quiet();
+
+    return true;
+  } catch (error: any) {
+    console.error(`⚠️  Merge failed: ${error.message}`);
+    return false;
+  }
+}
+
+/**
+ * Remove worktree
+ * Uses bd worktree when beads registry is enabled
+ * Falls back to WorktreeManager otherwise
+ */
+async function removeWorktree(taskId: string, repoPath: string): Promise<boolean> {
+  // Try bd worktree first when beads is enabled
+  if (useBeads) {
+    try {
+      const removed = await beadsRegistry.removeWorktree(taskId);
+      if (removed) return true;
+      // Fall through to WorktreeManager if bd worktree fails
+    } catch {
+      // Fall through
+    }
+  }
+
+  // Fallback to WorktreeManager
+  try {
+    const manager = new WorktreeManager({
+      baseDir: WORKTREE_BASE,
+      repoPath,
+    });
+
+    if (await manager.worktreeExists(taskId)) {
+      await manager.removeWorktree(taskId);
+      return true;
+    }
+    return true; // Already doesn't exist
+  } catch (error: any) {
+    console.error(`⚠️  Failed to remove worktree: ${error.message}`);
+    return false;
+  }
+}
+
+/**
+ * Kill worker pane
+ */
+async function killWorkerPane(paneId: string): Promise<boolean> {
+  try {
+    await tmux.killPane(paneId);
+    return true;
+  } catch {
+    return false; // Pane may already be gone
+  }
+}
+
+// ============================================================================
+// Main Command
+// ============================================================================
+
+export async function closeCommand(
+  taskId: string,
+  options: CloseOptions = {}
+): Promise<void> {
+  try {
+    // Find worker in registry (check both during transition)
+    let worker = useBeads
+      ? await beadsRegistry.findByTask(taskId)
+      : null;
+    if (!worker) {
+      worker = await registry.findByTask(taskId);
+    }
+
+    if (!worker) {
+      console.log(`ℹ️  No active worker for ${taskId}. Closing issue only.`);
+    }
+
+    // Confirm with user
+    if (!options.yes) {
+      const confirmed = await confirm({
+        message: `Close ${taskId}${worker ? ` and kill worker (pane ${worker.paneId})` : ''}?`,
+        default: true,
+      });
+
+      if (!confirmed) {
+        console.log('Cancelled.');
+        return;
+      }
+    }
+
+    // 1. Close beads issue
+    console.log(`📝 Closing ${taskId}...`);
+    const closed = await closeBeadsIssue(taskId);
+    if (!closed) {
+      console.error(`❌ Failed to close ${taskId}. Check \`bd show ${taskId}\`.`);
+      // Continue with cleanup anyway
+    } else {
+      console.log(`   ✅ Issue closed`);
+    }
+
+    // 2. Sync beads (unless --no-sync)
+    if (!options.noSync) {
+      console.log(`🔄 Syncing beads...`);
+      const synced = await syncBeads();
+      if (synced) {
+        console.log(`   ✅ Synced to git`);
+      } else {
+        console.log(`   ⚠️  Sync failed (non-fatal)`);
+      }
+    }
+
+    // 3. Handle worktree
+    if (worker?.worktree && !options.keepWorktree) {
+      // Merge if requested
+      if (options.merge) {
+        console.log(`🔀 Merging changes...`);
+        const merged = await mergeToMain(worker.repoPath, taskId);
+        if (merged) {
+          console.log(`   ✅ Merged to main`);
+        }
+      }
+
+      // Remove worktree
+      console.log(`🌳 Removing worktree...`);
+      const removed = await removeWorktree(taskId, worker.repoPath);
+      if (removed) {
+        console.log(`   ✅ Worktree removed`);
+      }
+    }
+
+    // 4. Kill worker pane
+    if (worker) {
+      console.log(`💀 Killing worker pane...`);
+      await killWorkerPane(worker.paneId);
+      console.log(`   ✅ Pane killed`);
+
+      // 5. Unregister worker from both registries
+      if (useBeads) {
+        try {
+          // Unbind work from agent
+          await beadsRegistry.unbindWork(worker.id);
+          // Set agent state to done
+          await beadsRegistry.setAgentState(worker.id, 'done');
+          // Delete agent bead
+          await beadsRegistry.deleteAgent(worker.id);
+        } catch {
+          // Non-fatal if beads cleanup fails
+        }
+      }
+      await registry.unregister(worker.id);
+      console.log(`   ✅ Worker unregistered`);
+    }
+
+    console.log(`\n✅ ${taskId} closed successfully`);
+
+  } catch (error: any) {
+    console.error(`❌ Error: ${error.message}`);
+    process.exit(1);
+  }
+}

package/src/term-commands/daemon.ts

@@ -0,0 +1,176 @@
+/**
+ * Daemon command - Manage beads daemon
+ *
+ * Usage:
+ *   term daemon start      - Start beads daemon (auto-commit, auto-sync)
+ *   term daemon stop       - Stop beads daemon
+ *   term daemon status     - Show daemon status
+ *   term daemon restart    - Restart daemon
+ *
+ * Options:
+ *   --auto-commit         - Enable auto-commit (default: true for start)
+ *   --auto-push           - Enable auto-push to remote
+ *   --json                - Output as JSON
+ */
+
+import * as beadsRegistry from '../lib/beads-registry.js';
+
+// ============================================================================
+// Types
+// ============================================================================
+
+export interface DaemonStartOptions {
+  autoCommit?: boolean;
+  autoPush?: boolean;
+}
+
+export interface DaemonStatusOptions {
+  json?: boolean;
+}
+
+// ============================================================================
+// Commands
+// ============================================================================
+
+/**
+ * Start the beads daemon
+ */
+export async function startCommand(options: DaemonStartOptions = {}): Promise<void> {
+  try {
+    // Check if already running
+    const status = await beadsRegistry.checkDaemonStatus();
+    if (status.running) {
+      console.log('ℹ️  Daemon is already running');
+      if (status.pid) {
+        console.log(`   PID: ${status.pid}`);
+      }
+      return;
+    }
+
+    console.log('🚀 Starting beads daemon...');
+    const started = await beadsRegistry.startDaemon({
+      autoCommit: options.autoCommit !== false, // Default to true
+      autoPush: options.autoPush,
+    });
+
+    if (started) {
+      console.log('   ✅ Daemon started');
+
+      // Show updated status
+      const newStatus = await beadsRegistry.checkDaemonStatus();
+      if (newStatus.pid) {
+        console.log(`   PID: ${newStatus.pid}`);
+      }
+      if (newStatus.autoCommit) {
+        console.log('   Auto-commit: enabled');
+      }
+      if (newStatus.autoPush) {
+        console.log('   Auto-push: enabled');
+      }
+    } else {
+      console.error('❌ Failed to start daemon');
+      console.log('   Check `bd daemon status` for details');
+      process.exit(1);
+    }
+  } catch (error: any) {
+    console.error(`❌ Error: ${error.message}`);
+    process.exit(1);
+  }
+}
+
+/**
+ * Stop the beads daemon
+ */
+export async function stopCommand(): Promise<void> {
+  try {
+    // Check if running
+    const status = await beadsRegistry.checkDaemonStatus();
+    if (!status.running) {
+      console.log('ℹ️  Daemon is not running');
+      return;
+    }
+
+    console.log('🛑 Stopping beads daemon...');
+    const stopped = await beadsRegistry.stopDaemon();
+
+    if (stopped) {
+      console.log('   ✅ Daemon stopped');
+    } else {
+      console.error('❌ Failed to stop daemon');
+      process.exit(1);
+    }
+  } catch (error: any) {
+    console.error(`❌ Error: ${error.message}`);
+    process.exit(1);
+  }
+}
+
+/**
+ * Show daemon status
+ */
+export async function statusCommand(options: DaemonStatusOptions = {}): Promise<void> {
+  try {
+    const status = await beadsRegistry.checkDaemonStatus();
+
+    if (options.json) {
+      console.log(JSON.stringify(status, null, 2));
+      return;
+    }
+
+    console.log('Beads Daemon Status');
+    console.log('───────────────────');
+    console.log(`Running: ${status.running ? '✅ yes' : '❌ no'}`);
+
+    if (status.running) {
+      if (status.pid) {
+        console.log(`PID: ${status.pid}`);
+      }
+      if (status.lastSync) {
+        console.log(`Last sync: ${status.lastSync}`);
+      }
+      if (status.autoCommit !== undefined) {
+        console.log(`Auto-commit: ${status.autoCommit ? 'enabled' : 'disabled'}`);
+      }
+      if (status.autoPush !== undefined) {
+        console.log(`Auto-push: ${status.autoPush ? 'enabled' : 'disabled'}`);
+      }
+    } else {
+      console.log('\nRun `term daemon start` to start the daemon');
+    }
+  } catch (error: any) {
+    console.error(`❌ Error: ${error.message}`);
+    process.exit(1);
+  }
+}
+
+/**
+ * Restart the beads daemon
+ */
+export async function restartCommand(options: DaemonStartOptions = {}): Promise<void> {
+  try {
+    // Check if running and stop
+    const status = await beadsRegistry.checkDaemonStatus();
+    if (status.running) {
+      console.log('🛑 Stopping beads daemon...');
+      await beadsRegistry.stopDaemon();
+      console.log('   ✅ Stopped');
+    }
+
+    // Start with new options
+    console.log('🚀 Starting beads daemon...');
+    const started = await beadsRegistry.startDaemon({
+      autoCommit: options.autoCommit !== false,
+      autoPush: options.autoPush,
+    });
+
+    if (started) {
+      console.log('   ✅ Daemon restarted');
+    } else {
+      console.error('❌ Failed to restart daemon');
+      process.exit(1);
+    }
+  } catch (error: any) {
+    console.error(`❌ Error: ${error.message}`);
+    process.exit(1);
+  }
+}