chore: handoff checkpoint on main

Author: jonathanpeterwu

.env.example

@@ -9,6 +9,10 @@ NPM_TOKEN=npm_XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
 # Get from https://github.com/settings/tokens
 GITHUB_TOKEN=ghp_XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
 
+# Greptile API Key (for AI code review via MCP)
+# Get from https://app.greptile.com/settings
+GREPTILE_API_KEY=
+
 # Linear API Key (for Linear integration)
 LINEAR_API_KEY=lin_api_XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
 

package-lock.json

@@ -151,6 +151,7 @@
   "optionalDependencies": {
     "blessed": "^0.1.81",
     "blessed-contrib": "^4.11.0",
-    "chokidar": "^5.0.0"
+    "chokidar": "^5.0.0",
+    "node-pty": "^1.1.0"
   }
 }

package.json

@@ -30,6 +30,7 @@ import {
   loadModelRouterConfig,
   type ModelProvider,
 } from '../core/models/model-router.js';
+import { launchWrapper } from '../features/sweep/pty-wrapper.js';
 import { FallbackMonitor } from '../core/models/fallback-monitor.js';
 
 // __filename and __dirname are provided by esbuild banner for ESM compatibility
@@ -43,6 +44,8 @@ interface ClaudeSMConfig {
   defaultNotifyOnDone: boolean;
   defaultWhatsApp: boolean;
   defaultModelRouting: boolean;
+  defaultSweep: boolean;
+  defaultGreptile: boolean;
 }
 
 interface ClaudeConfig {
@@ -65,6 +68,8 @@ interface ClaudeConfig {
   useModelRouting: boolean;
   forceProvider?: ModelProvider;
   useThinkingMode: boolean;
+  useSweep: boolean;
+  useGreptile: boolean;
 }
 
 const DEFAULT_SM_CONFIG: ClaudeSMConfig = {
@@ -76,6 +81,8 @@ const DEFAULT_SM_CONFIG: ClaudeSMConfig = {
   defaultNotifyOnDone: true,
   defaultWhatsApp: false,
   defaultModelRouting: false,
+  defaultSweep: true,
+  defaultGreptile: true,
 };
 
 function getConfigPath(): string {
@@ -129,6 +136,8 @@ class ClaudeSM {
       sessionStartTime: Date.now(),
       useModelRouting: this.smConfig.defaultModelRouting,
       useThinkingMode: false,
+      useSweep: this.smConfig.defaultSweep,
+      useGreptile: this.smConfig.defaultGreptile,
     };
 
     this.stackmemoryPath = this.findStackMemory();
@@ -213,6 +222,60 @@ class ClaudeSM {
     return null;
   }
 
+  private ensureGreptileMcp(): void {
+    const apiKey = process.env['GREPTILE_API_KEY'];
+    const ghToken = process.env['GITHUB_TOKEN'];
+
+    if (!apiKey) {
+      console.log(
+        chalk.gray('   Greptile: disabled (set GREPTILE_API_KEY in .env)')
+      );
+      return;
+    }
+
+    // Check if already registered
+    try {
+      const result = execSync('claude mcp list 2>/dev/null', {
+        encoding: 'utf-8',
+      });
+      if (result.includes('greptile')) {
+        console.log(chalk.gray('   Greptile: MCP server registered'));
+        return;
+      }
+    } catch {
+      // claude mcp list not available, try to add anyway
+    }
+
+    // Register Greptile MCP server via HTTP transport
+    try {
+      const cmd = [
+        'claude mcp add',
+        '--transport http',
+        'greptile',
+        'https://api.greptile.com/mcp',
+        `--header "Authorization: Bearer ${apiKey}"`,
+      ];
+      if (ghToken) {
+        cmd.push(`--header "X-GitHub-Token: ${ghToken}"`);
+      }
+      execSync(cmd.join(' '), { stdio: 'ignore' });
+      console.log(chalk.cyan('   Greptile: MCP server registered'));
+    } catch {
+      // Fallback: register via stdio transport with npx
+      try {
+        const envArgs = [`GREPTILE_API_KEY=${apiKey}`];
+        if (ghToken) envArgs.push(`GITHUB_TOKEN=${ghToken}`);
+        execSync(
+          `claude mcp add greptile -- env ${envArgs.join(' ')} npx greptile-mcp-server`,
+          { stdio: 'ignore' }
+        );
+        console.log(chalk.cyan('   Greptile: MCP server registered (stdio)'));
+      } catch {
+        console.log(chalk.gray('   Greptile: failed to register MCP server'));
+      }
+    }
+  }
+
   private setupWorktree(): string | null {
     if (!this.config.useWorktree || !this.isGitRepo()) {
       return null;
@@ -674,6 +737,18 @@ class ClaudeSM {
         case '--no-model-routing':
           this.config.useModelRouting = false;
           break;
+        case '--sweep':
+          this.config.useSweep = true;
+          break;
+        case '--no-sweep':
+          this.config.useSweep = false;
+          break;
+        case '--greptile':
+          this.config.useGreptile = true;
+          break;
+        case '--no-greptile':
+          this.config.useGreptile = false;
+          break;
         default:
           claudeArgs.push(arg);
       }
@@ -856,6 +931,11 @@ class ClaudeSM {
       }
     }
 
+    // Ensure Greptile MCP server is registered if enabled
+    if (this.config.useGreptile) {
+      this.ensureGreptileMcp();
+    }
+
     // Start WhatsApp services if enabled
     if (this.config.useWhatsApp) {
       console.log(
@@ -865,6 +945,31 @@ class ClaudeSM {
     }
 
     console.log();
+
+    // Launch via Sweep PTY wrapper if enabled
+    if (this.config.useSweep) {
+      const claudeBin = this.resolveClaudeBin();
+      if (!claudeBin) {
+        console.error(chalk.red('Claude CLI not found.'));
+        process.exit(1);
+        return;
+      }
+      console.log(
+        chalk.cyan('[Sweep] Launching Claude with prediction bar...')
+      );
+      console.log(chalk.gray('─'.repeat(42)));
+      try {
+        await launchWrapper({
+          claudeBin,
+          claudeArgs,
+        });
+      } catch (error) {
+        console.error(chalk.red((error as Error).message));
+        process.exit(1);
+      }
+      return;
+    }
+
     console.log(chalk.gray('Starting Claude...'));
     console.log(chalk.gray('─'.repeat(42)));
 
@@ -1041,6 +1146,12 @@ configCmd
     console.log(
       `  defaultModelRouting: ${config.defaultModelRouting ? chalk.green('true') : chalk.gray('false')}`
     );
+    console.log(
+      `  defaultSweep:        ${config.defaultSweep ? chalk.green('true') : chalk.gray('false')}`
+    );
+    console.log(
+      `  defaultGreptile:     ${config.defaultGreptile ? chalk.green('true') : chalk.gray('false')}`
+    );
     console.log(chalk.gray(`\nConfig: ${getConfigPath()}`));
   });
 
@@ -1062,14 +1173,16 @@ configCmd
       whatsapp: 'defaultWhatsApp',
       'model-routing': 'defaultModelRouting',
       modelrouting: 'defaultModelRouting',
+      sweep: 'defaultSweep',
+      greptile: 'defaultGreptile',
     };
 
     const configKey = keyMap[key];
     if (!configKey) {
       console.log(chalk.red(`Unknown key: ${key}`));
       console.log(
         chalk.gray(
-          'Valid keys: worktree, sandbox, chrome, tracing, remote, notify-done, whatsapp'
+          'Valid keys: worktree, sandbox, chrome, tracing, remote, notify-done, whatsapp, sweep, greptile'
         )
       );
       process.exit(1);
@@ -1185,6 +1298,31 @@ configCmd
     console.log(chalk.green('Model routing disabled by default'));
   });
 
+configCmd
+  .command('greptile-on')
+  .description(
+    'Enable Greptile AI code review by default (requires GREPTILE_API_KEY)'
+  )
+  .action(() => {
+    const config = loadSMConfig();
+    config.defaultGreptile = true;
+    saveSMConfig(config);
+    console.log(chalk.green('Greptile enabled by default'));
+    if (!process.env['GREPTILE_API_KEY']) {
+      console.log(chalk.gray('Set GREPTILE_API_KEY in .env to activate'));
+    }
+  });
+
+configCmd
+  .command('greptile-off')
+  .description('Disable Greptile AI code review by default')
+  .action(() => {
+    const config = loadSMConfig();
+    config.defaultGreptile = false;
+    saveSMConfig(config);
+    console.log(chalk.green('Greptile disabled by default'));
+  });
+
 // Main command (default action when no subcommand)
 program
   .option('-w, --worktree', 'Create isolated worktree for this instance')
@@ -1215,6 +1353,10 @@ program
   .option('--ollama', 'Force Ollama provider for this session')
   .option('--model-routing', 'Enable model routing')
   .option('--no-model-routing', 'Disable model routing')
+  .option('--sweep', 'Enable Sweep next-edit predictions (PTY wrapper)')
+  .option('--no-sweep', 'Disable Sweep predictions')
+  .option('--greptile', 'Enable Greptile AI code review (MCP server)')
+  .option('--no-greptile', 'Disable Greptile integration')
   .helpOption('-h, --help', 'Display help')
   .allowUnknownOption(true)
   .action(async (_options) => {

src/cli/claude-sm.ts

@@ -11,6 +11,7 @@ import { join } from 'path';
 import {
   createServerManager,
   createPredictionClient,
+  launchWrapper,
   SweepServerConfig,
   DEFAULT_SERVER_CONFIG,
 } from '../../features/sweep/index.js';
@@ -285,6 +286,25 @@ Predictions are triggered after file edits via Claude Code hooks.
       }
     });
 
+  // Wrap command
+  cmd
+    .command('wrap')
+    .description('Launch Claude Code with Sweep prediction status bar')
+    .option('--claude-bin <path>', 'Path to claude binary')
+    .allowUnknownOption(true)
+    .action(async (options, command) => {
+      try {
+        const claudeArgs = command.args || [];
+        await launchWrapper({
+          claudeBin: options.claudeBin,
+          claudeArgs,
+        });
+      } catch (error) {
+        console.error(chalk.red((error as Error).message));
+        process.exit(1);
+      }
+    });
+
   return cmd;
 }
 

src/cli/commands/sweep.ts

@@ -16,6 +16,7 @@ export interface FeatureFlags {
   aiSummaries: boolean;
   skills: boolean;
   ralph: boolean;
+  greptile: boolean;
 }
 
 /**
@@ -68,6 +69,12 @@ export function isFeatureEnabled(feature: keyof FeatureFlags): boolean {
       // Ralph enabled by default in development (unless explicitly disabled)
       // For npm package users, must be explicitly enabled
       return process.env['STACKMEMORY_RALPH'] !== 'false';
+    case 'greptile':
+      // Greptile enabled when API key is available
+      return (
+        process.env['STACKMEMORY_GREPTILE'] !== 'false' &&
+        !!process.env['GREPTILE_API_KEY']
+      );
     default:
       return false;
   }
@@ -85,6 +92,7 @@ export function getFeatureFlags(): FeatureFlags {
     aiSummaries: isFeatureEnabled('aiSummaries'),
     skills: isFeatureEnabled('skills'),
     ralph: isFeatureEnabled('ralph'),
+    greptile: isFeatureEnabled('greptile'),
   };
 }
 
@@ -115,5 +123,8 @@ export function logFeatureStatus(): void {
     console.log(
       `  Ralph: ${flags.ralph ? 'enabled' : 'disabled (set STACKMEMORY_RALPH=true)'}`
     );
+    console.log(
+      `  Greptile: ${flags.greptile ? 'enabled' : 'disabled (no GREPTILE_API_KEY)'}`
+    );
   }
 }

src/core/config/feature-flags.ts

@@ -9,3 +9,11 @@ export * from './types.js';
 export * from './prompt-builder.js';
 export * from './prediction-client.js';
 export * from './sweep-server-manager.js';
+export * from './state-watcher.js';
+export * from './status-bar.js';
+export * from './tab-interceptor.js';
+export {
+  PtyWrapper,
+  launchWrapper,
+  type PtyWrapperConfig,
+} from './pty-wrapper.js';