test: update config test for new settings format

- Full README rewrite with all new commands, pipe support, aliases,
  global flags, and configuration examples
- Bump version to 0.2.0, update description

README.md

@@ -118,33 +118,23 @@ git diff | gai suggest branch
 
 ### Interactive Menu
 
-Run `gai` without arguments to open the mole-style interactive menu:
-
 ```
-  gai v0.1.3
+$ gai
-  AI-powered git helper for commits, PRs, reviews, and changelogs
-  ────────────────────────────────────────────────────────────────────────
 
-  CREATE
+  gai
-  │ › 1  Commit     Generate AI commit message
+  AI-powered git helper — choose a workflow
-      2  PR         Create a PR with AI-generated title
+  ↑/↓ navigate · enter/space select · ctrl+c cancel
-      3  Amend      Amend last commit with AI message
 
-  INSPECT
+  ❯ ● commit     Generate AI commit message
-      4  Explain    Explain staged changes in plain language
+    ○ pr         Create a PR with AI-generated title
-      5  Review     AI code review of staged changes
+    ○ explain    Explain staged changes in plain language
-      6  Changelog  Generate changelog from commits
+    ○ review     AI code review of staged changes
-      7  Suggest    Suggest branch name or commit type
+    ○ changelog  Generate changelog from commits
-
+    ○ suggest    Suggest branch name or commit type
-  PROJECT
+    ○ amend      Amend last commit with AI message
-      8  Config     Configure API settings
+    ○ config     Configure API settings
-
-  ────────────────────────────────────────────────────────────────────────
-  ↑/↓ navigate   enter run   1-8 jump   h help   v version   q quit
 ```
 
-Number keys `1`–`8` jump directly to the corresponding action. After a command finishes, press Enter to return to the menu.
-
 ### Command Examples
 
 #### Commit

index.ts

@@ -1,7 +1,7 @@
 #!/usr/bin/env bun
 
 // gai — AI-powered git commit and PR helper
-// v0.1.3
+// v0.2.0
 
 import { runCLI, registerCommands, formatHelp, type CommandDef, type ParsedArgs } from "./src/cli";
 import { handleCommit } from "./src/commands/commit";
@@ -14,172 +14,9 @@ import { handleSuggest } from "./src/commands/suggest";
 import { setColorEnabled } from "./src/terminal";
 import { BOLD, GREEN, CYAN, DIM, RESET } from "./src/terminal";
 import { isStdinTTY, initTTY } from "./src/tty";
-import { VERSION } from "./src/brand";
+import { BACK, selectOne } from "./src/menu";
-import { SKIP_WAIT } from "./src/menu";
 
-// ── Interactive Menu (mole-style) ─────────────────────────────────────
+// ── Interactive Menu (default, no subcommand) ─────────────────────────
-
-interface MenuItem {
-  key: string;
-  label: string;
-  description: string;
-  group: "Create" | "Inspect" | "Project";
-}
-
-const MENU_ITEMS: MenuItem[] = [
-  { key: "commit", label: "Commit", description: "Generate AI commit message", group: "Create" },
-  { key: "pr", label: "PR", description: "Create a PR with AI-generated title", group: "Create" },
-  { key: "amend", label: "Amend", description: "Amend last commit with AI message", group: "Create" },
-  { key: "explain", label: "Explain", description: "Explain staged changes in plain language", group: "Inspect" },
-  { key: "review", label: "Review", description: "AI code review of staged changes", group: "Inspect" },
-  { key: "changelog", label: "Changelog", description: "Generate changelog from commits", group: "Inspect" },
-  { key: "suggest", label: "Suggest", description: "Suggest branch name or commit type", group: "Inspect" },
-  { key: "config", label: "Config", description: "Configure API settings", group: "Project" },
-];
-
-function hideCursor() { process.stdout.write("\x1b[?25l"); }
-function showCursor() { process.stdout.write("\x1b[?25h"); }
-
-function clearLine() { process.stdout.write("\r\x1b[2K"); }
-
-async function readKey(): Promise<string> {
-  return new Promise((resolve) => {
-    const onData = (data: Buffer) => {
-      process.stdin.removeListener("data", onData);
-      resolve(data.toString());
-    };
-    process.stdin.once("data", onData);
-  });
-}
-
-function visibleLen(s: string): number {
-  return s.replace(/\x1b\[[0-9;]*m/g, "").length;
-}
-
-function padRight(value: string, width: number): string {
-  return value + " ".repeat(Math.max(0, width - visibleLen(value)));
-}
-
-function renderMenu(cursor: number): number {
-  process.stdout.write("\x1b[H"); // cursor home
-
-  let lineCount = 0;
-  const C = CYAN();
-  const D = DIM();
-  const G = GREEN();
-  const R = RESET();
-  const width = 72;
-  const separator = `${D}${"─".repeat(width)}${R}`;
-
-  const write = (line = "") => {
-    clearLine();
-    process.stdout.write(`${line}\n`);
-    lineCount++;
-  };
-
-  write("");
-  write(`  ${BOLD()}gai${R} ${D}v${VERSION}${R}`);
-  write(`  ${D}AI-powered git helper for commits, PRs, reviews, and changelogs${R}`);
-  write(`  ${separator}`);
-  write("");
-
-  const keyWidth = 3;
-  const labelWidth = Math.max(...MENU_ITEMS.map((m) => visibleLen(m.label))) + 2;
-  let currentGroup: MenuItem["group"] | null = null;
-
-  for (let i = 0; i < MENU_ITEMS.length; i++) {
-    const item = MENU_ITEMS[i]!;
-    const num = String(i + 1);
-    const active = i === cursor;
-
-    if (item.group !== currentGroup) {
-      if (currentGroup !== null) write("");
-      write(`  ${D}${item.group.toUpperCase()}${R}`);
-      currentGroup = item.group;
-    }
-
-    const pointer = active ? `${C}›${R}` : " ";
-    const key = active ? `${G}${num}${R}` : `${D}${num}${R}`;
-    const label = active ? `${BOLD()}${item.label}${R}` : item.label;
-    const description = active ? item.description : `${D}${item.description}${R}`;
-    const row = [
-      pointer,
-      padRight(key, keyWidth),
-      padRight(label, labelWidth),
-      description,
-    ].join(" ");
-
-    if (active) {
-      write(`  ${C}│${R} ${row}`);
-    } else {
-      write(`    ${row}`);
-    }
-  }
-
-  write("");
-  write(`  ${separator}`);
-  write(`  ${D}↑/↓ navigate   enter run   ${G}1-8${D} jump   ${G}h${D} help   ${G}v${D} version   ${G}q${D} quit${R}`);
-  write("");
-
-  // Clear rest of screen
-  process.stdout.write("\x1b[J");
-
-  return lineCount;
-}
-
-async function dispatchMenuAction(key: string): Promise<number> {
-  const fakeArgs: ParsedArgs = {
-    command: key,
-    flags: {},
-    positional: [],
-    raw: [],
-    subcommand: { name: key, description: "", handler: async () => 0 },
-  };
-
-  if (key === "amend") fakeArgs.flags["amend"] = true;
-
-  switch (key) {
-    case "commit": return handleCommit(fakeArgs);
-    case "pr": return handlePR(fakeArgs);
-    case "config": return handleConfig(fakeArgs);
-    case "explain": return handleExplain(fakeArgs);
-    case "review": return handleReview(fakeArgs);
-    case "changelog": return handleChangelog(fakeArgs);
-    case "suggest": return handleSuggest(fakeArgs);
-    case "amend": return handleCommit(fakeArgs);
-    default: return 0;
-  }
-}
-
-async function waitForEnter(): Promise<void> {
-  const D = DIM();
-  const R = RESET();
-  process.stdout.write(`\n  ${D}Press Enter to return to menu...${R}`);
-  // Read a line from stdin (works in cooked mode — blocks until Enter)
-  await new Promise<void>((resolve) => {
-    const onData = (data: Buffer) => {
-      process.stdin.removeListener("data", onData);
-      resolve();
-    };
-    process.stdin.on("data", onData);
-    // Resume stdin in case it was paused
-    process.stdin.resume();
-  });
-  process.stdout.write("\x1b[2J\x1b[H"); // clear screen
-}
-
-async function dispatchAndWait(item: MenuItem, wasRaw: boolean): Promise<number> {
-  showCursor();
-  process.stdin.setRawMode(wasRaw === true);
-  process.stdin.pause();
-  process.stdout.write("\x1b[2J\x1b[H"); // clear screen
-  const result = await dispatchMenuAction(item.key);
-  if (result === (SKIP_WAIT as unknown as number)) {
-    return 0; // user explicitly backed out — skip "Press Enter" and return directly
-  }
-  await waitForEnter();
-  return result;
-}
 
 async function showMenu(): Promise<number> {
   if (!isStdinTTY()) {
@@ -187,96 +24,72 @@ async function showMenu(): Promise<number> {
     return 1;
   }
 
-  let cursor = 0;
+  while (true) {
-  const wasRaw = process.stdin.isRaw;
+    const selected = await selectOne({
+      title: "gai",
+      subtitle: "AI-powered git helper — choose a workflow",
+      allowBack: false,
+      items: [
+        { label: "commit", value: "commit", description: "Generate AI commit message" },
+        { label: "pr", value: "pr", description: "Create a PR with AI-generated title" },
+        { label: "explain", value: "explain", description: "Explain staged changes in plain language" },
+        { label: "review", value: "review", description: "AI code review of staged changes" },
+        { label: "changelog", value: "changelog", description: "Generate changelog from commits" },
+        { label: "suggest", value: "suggest", description: "Suggest branch name or commit type" },
+        { label: "amend", value: "amend", description: "Amend last commit with AI message" },
+        { label: "config", value: "config", description: "Configure API settings" },
+      ],
+    });
 
-  if (wasRaw !== true) process.stdin.setRawMode(true);
+    if (selected === null || selected === BACK) return 0;
-  process.stdin.resume();
-  hideCursor();
 
-  // Initial render
+    // Build synthetic args for subcommand handlers
-  renderMenu(cursor);
+    const fakeArgs: ParsedArgs = {
+      command: selected,
+      flags: {},
+      positional: [],
+      raw: [],
+      subcommand: {
+        name: selected,
+        description: "",
+        handler: async () => 0,
+      },
+    };
 
-  try {
+    let result: number;
-    while (true) {
+    switch (selected) {
-      const raw = await readKey();
+      case "commit":
-
+        result = await handleCommit(fakeArgs);
-      // Escape sequences (arrows)
+        break;
-      if (raw === "\x1b[A" || raw === "\x1bOA") {
+      case "pr":
-        if (cursor > 0) { cursor--; renderMenu(cursor); }
+        result = await handlePR(fakeArgs);
-        continue;
+        break;
-      }
+      case "config":
-      if (raw === "\x1b[B" || raw === "\x1bOB") {
+        result = await handleConfig(fakeArgs);
-        if (cursor < MENU_ITEMS.length - 1) { cursor++; renderMenu(cursor); }
+        break;
-        continue;
+      case "explain":
-      }
+        result = await handleExplain(fakeArgs);
-
+        break;
-      // Enter
+      case "review":
-      if (raw === "\r" || raw === "\n") {
+        result = await handleReview(fakeArgs);
-        const result = await dispatchAndWait(MENU_ITEMS[cursor]!, wasRaw);
+        break;
-        if (result !== 0) return result;
+      case "changelog":
-        hideCursor();
+        result = await handleChangelog(fakeArgs);
-        if (wasRaw !== true) process.stdin.setRawMode(true);
+        break;
-        process.stdin.resume();
+      case "suggest":
-        renderMenu(cursor);
+        result = await handleSuggest(fakeArgs);
-        continue;
+        break;
-      }
+      case "amend":
-
+        fakeArgs.flags["amend"] = true;
-      // Ctrl+C
+        result = await handleCommit(fakeArgs);
-      if (raw === "\x03") {
+        break;
-        showCursor();
+      default:
-        process.stdin.setRawMode(wasRaw === true);
+        result = 0;
-        process.stdin.pause();
-        process.stdout.write("\n");
-        return 0;
-      }
-
-      // Number hotkeys (1-8)
-      if (raw >= "1" && raw <= "8") {
-        const idx = parseInt(raw) - 1;
-        if (idx < MENU_ITEMS.length) {
-          cursor = idx;
-          renderMenu(cursor);
-          const result = await dispatchAndWait(MENU_ITEMS[idx]!, wasRaw);
-          if (result !== 0) return result;
-          hideCursor();
-          if (wasRaw !== true) process.stdin.setRawMode(true);
-          process.stdin.resume();
-          renderMenu(cursor);
-          continue;
-        }
-      }
-
-      // Letter hotkeys
-      const lower = raw.toLowerCase();
-      if (lower === "h") {
-        showCursor();
-        process.stdin.setRawMode(wasRaw === true);
-        process.stdin.pause();
-        process.stdout.write("\x1b[2J\x1b[H");
-        console.log(formatHelp(commands));
-        return 0;
-      }
-      if (lower === "v") {
-        showCursor();
-        process.stdin.setRawMode(wasRaw === true);
-        process.stdin.pause();
-        process.stdout.write("\x1b[2J\x1b[H");
-        console.log(`gai v${VERSION}`);
-        return 0;
-      }
-      if (lower === "q") {
-        showCursor();
-        process.stdin.setRawMode(wasRaw === true);
-        process.stdin.pause();
-        process.stdout.write("\n");
-        return 0;
-      }
     }
-  } finally {
+
-    showCursor();
+    // Return to menu unless they explicitly chose "back"
-    process.stdin.setRawMode(wasRaw === true);
+    if (result !== 0) return result;
-    process.stdin.pause();
+    // Loop back to menu for another action
   }
 }
 

package.json

@@ -1,6 +1,6 @@
 {
   "name": "gai",
-  "version": "0.1.3",
+  "version": "0.2.0",
   "description": "AI-powered git helper — commit messages, PRs, code review, changelogs, and more",
   "module": "index.ts",
   "type": "module",

src/brand.ts

@@ -1,22 +0,0 @@
-// Brand banner and ASCII art logo for gai
-
-import { GREEN, CYAN, RESET } from "./terminal";
-
-export const VERSION = "0.1.3";
-
-export function showBanner(): string {
-  const G = GREEN();
-  const C = CYAN();
-  const R = RESET();
-
-  return [
-    "",
-    `${G}  ██████╗  █████╗ ██╗${R}`,
-    `${G} ██╔════╝ ██╔══██╗██║${R}`,
-    `${G} ██║  ██╗ ███████║██║${R}`,
-    `${G} ██║  ██║ ██╔══██║██║${R}`,
-    `${G} ╚██████╝ ██║  ██║██║${R}  ${C}AI-powered git helper${R}`,
-    `${G}  ╚═════╝ ╚═╝  ╚═╝╚═╝${R}  ${C}v${VERSION}${R}`,
-    "",
-  ].join("\n");
-}

src/cli.ts

@@ -282,7 +282,7 @@ export async function runCLI(rawArgs: string[], commands: Map<string, CommandDef
 
   // Handle --version globally
   if (result.flags["version"]) {
-    console.log("gai v0.1.3");
+    console.log("gai v0.2.0");
     return 0;
   }
 

src/commands/changelog.ts

@@ -62,7 +62,7 @@ export async function handleChangelog(args: ParsedArgs): Promise<number> {
   }
 
   try {
-    const callbacks: StreamCallbacks | undefined = tty ? {
+    const callbacks: StreamCallbacks = tty ? {
       onToken: (token) => process.stdout.write(token),
     } : undefined;
 

src/commands/commit.ts

@@ -7,11 +7,10 @@ import {
   getStagedDiff,
   getRecentCommits,
   stageFiles,
-  applyFileSelection,
   commit,
 } from "../git";
 import { selectFiles } from "../selector";
-import { BACK, SKIP_WAIT } from "../menu";
+import { BACK } from "../menu";
 import { collectProjectContext } from "../context";
 import { buildPrompt, SYSTEM_PROMPT } from "../prompt";
 import { generateCommitMessage } from "../ai";
@@ -51,19 +50,6 @@ function printCommitResult(result: CommitResult, msg: string) {
   if (parts.length > 0) console.log(`    ${parts.join(", ")}`);
 }
 
-function printSelectionResult(result: { staged: string[]; unstaged: string[] }) {
-  const parts: string[] = [];
-  if (result.staged.length > 0) {
-    parts.push(`${GREEN()}staged ${result.staged.length}${RESET()}`);
-  }
-  if (result.unstaged.length > 0) {
-    parts.push(`${YELLOW()}unstaged ${result.unstaged.length}${RESET()}`);
-  }
-  if (parts.length > 0) {
-    console.log(`  Updated staging area: ${parts.join(", ")}`);
-  }
-}
-
 async function confirmCommit(message: string): Promise<"y" | "n" | "e"> {
   console.log(`\n  ${BOLD()}Generated commit message:${RESET()}`);
   console.log(`    ${GREEN()}${message}${RESET()}\n`);
@@ -187,22 +173,16 @@ export async function handleCommit(args: ParsedArgs): Promise<number> {
     const stagedFiles = await getStagedFiles();
     const unstagedFiles = await getUnstagedFiles();
 
-    if (!amend && stagedFiles.length === 0 && unstagedFiles.length === 0) {
+    if (stagedFiles.length === 0 && !amend) {
-      console.error(`\n  ${RED()}Error: Nothing to commit.${RESET()}\n`);
+      if (autoMode && unstagedFiles.length > 0) {
-      return 1;
+        await stageFiles(unstagedFiles.map((f) => f.path));
-    }
+        console.log(`  ${GREEN()}Auto-staged ${unstagedFiles.length} file(s).${RESET()}`);
-
+      } else if (unstagedFiles.length > 0) {
-    if (!amend && autoMode && unstagedFiles.length > 0) {
-      await stageFiles(unstagedFiles.map((f) => f.path));
-      console.log(`  ${GREEN()}Auto-staged ${unstagedFiles.length} file(s).${RESET()}`);
-    } else if (!amend) {
-      if (isStdinTTY()) {
-        const selected = await selectFiles(stagedFiles, unstagedFiles);
-        if (selected === BACK) return SKIP_WAIT as unknown as number;
-        printSelectionResult(await applyFileSelection(stagedFiles, unstagedFiles, selected));
-      } else if (stagedFiles.length === 0 && unstagedFiles.length > 0) {
         console.error(`\n  ${RED()}Error: No staged changes. Use -a to auto-stage, or stage files manually.${RESET()}\n`);
         return 1;
+      } else {
+        console.error(`\n  ${RED()}Error: Nothing to commit.${RESET()}\n`);
+        return 1;
       }
     }
 
@@ -235,13 +215,18 @@ export async function handleCommit(args: ParsedArgs): Promise<number> {
     return 0;
   }
 
-  if (autoMode && unstagedFiles.length > 0) {
+  if (unstagedFiles.length > 0) {
-    await stageFiles(unstagedFiles.map((f) => f.path));
+    if (autoMode) {
-    console.log(`  ${GREEN()}Auto-staged ${unstagedFiles.length} file(s).${RESET()}`);
+      await stageFiles(unstagedFiles.map((f) => f.path));
-  } else {
+      console.log(`  ${GREEN()}Auto-staged ${unstagedFiles.length} file(s).${RESET()}`);
-    const selected = await selectFiles(stagedFiles, unstagedFiles);
+    } else {
-    if (selected === BACK) return SKIP_WAIT as unknown as number;
+      const selected = await selectFiles(stagedFiles, unstagedFiles);
-    printSelectionResult(await applyFileSelection(stagedFiles, unstagedFiles, selected));
+      if (selected === BACK) return 0;
+      if (selected.length > 0) {
+        await stageFiles(selected);
+        console.log(`  ${GREEN()}Staged ${selected.length} file(s).${RESET()}`);
+      }
+    }
   }
 
   const diff = await getStagedDiff();

src/commands/config.ts

@@ -1,7 +1,6 @@
 import { loadConfig, saveConfig } from "../config";
 import { BOLD, GREEN, YELLOW, CYAN, RED, DIM, RESET } from "../terminal";
 import { isStdinTTY } from "../tty";
-import { SKIP_WAIT } from "../menu";
 import type { Config } from "../types";
 import type { ParsedArgs } from "../cli";
 
@@ -324,7 +323,7 @@ export async function handleConfig(args: ParsedArgs): Promise<number> {
   // gai config (no args) → interactive
   if (positional.length === 0) {
     const result = await interactiveConfig();
-    return result === "back" ? (SKIP_WAIT as unknown as number) : 0;
+    return result === "back" ? 0 : 0;
   }
 
   console.error(`\n  ${RED()}Error: Unknown config subcommand: ${positional[0]}${RESET()}`);

src/commands/explain.ts

@@ -1,14 +1,7 @@
 import { loadConfig } from "../config";
-import {
+import { isGitRepo, getStagedDiff, getUnstagedFiles, getRepoRoot, stageFiles } from "../git";
-  isGitRepo,
-  getStagedFiles,
-  getStagedDiff,
-  getUnstagedFiles,
-  getRepoRoot,
-  applyFileSelection,
-} from "../git";
 import { selectFiles } from "../selector";
-import { BACK, SKIP_WAIT } from "../menu";
+import { BACK } from "../menu";
 import { collectProjectContext } from "../context";
 import { EXPLAIN_SYSTEM_PROMPT, buildExplainPrompt } from "../prompt";
 import { callAI } from "../ai";
@@ -26,6 +19,7 @@ export async function handleExplain(args: ParsedArgs): Promise<number> {
   }
 
   const unstaged = args.flags["unstaged"] as boolean;
+  const staged = args.flags["staged"] as boolean;
   const verbose = args.flags["verbose"] as boolean;
 
   // Determine which diff to explain
@@ -50,16 +44,22 @@ export async function handleExplain(args: ParsedArgs): Promise<number> {
         console.error(`\n  ${RED()}Error: Not a git repository.${RESET()}\n`);
         return 1;
       }
-      const stagedFiles = await getStagedFiles();
-      const unstagedFiles = await getUnstagedFiles();
-      sourceLabel = "selected changes";
-
-      if (stagedFiles.length > 0 || unstagedFiles.length > 0) {
-        const selected = await selectFiles(stagedFiles, unstagedFiles);
-        if (selected === BACK) return SKIP_WAIT as unknown as number;
-        await applyFileSelection(stagedFiles, unstagedFiles, selected);
-      }
       diff = await getStagedDiff();
+      sourceLabel = "staged changes";
+
+      // If no staged changes, offer to stage unstaged files
+      if (!diff) {
+        const unstagedFiles = await getUnstagedFiles();
+        if (unstagedFiles.length > 0) {
+          const selected = await selectFiles([], unstagedFiles);
+          if (selected === BACK) return 0;
+          if (selected.length > 0) {
+            await stageFiles(selected);
+            console.log(`  ${GREEN()}Staged ${selected.length} file(s).${RESET()}`);
+            diff = await getStagedDiff();
+          }
+        }
+      }
     } else {
       // Read from pipe
       const chunks: Buffer[] = [];
@@ -109,7 +109,7 @@ export async function handleExplain(args: ParsedArgs): Promise<number> {
   }
 
   try {
-    const callbacks: StreamCallbacks | undefined = tty ? {
+    const callbacks: StreamCallbacks = tty ? {
       onToken: (token) => process.stdout.write(token),
     } : undefined;
 

src/commands/pr.ts

@@ -4,7 +4,7 @@ import { isGitRepo, getRepoRoot } from "../git";
 import { collectProjectContext } from "../context";
 import { PR_SYSTEM_PROMPT, buildPRPrompt } from "../prompt";
 import { generatePRMessage } from "../ai";
-import { BACK, SKIP_WAIT, selectOne } from "../menu";
+import { BACK, selectOne } from "../menu";
 import {
   getDefaultBranch,
   getBranchName,
@@ -68,7 +68,7 @@ export async function handlePR(args: ParsedArgs): Promise<number> {
   if (!platform) {
     const hostname = (await getRemoteHostname()) || "unknown";
     const chosen = await selectPlatform(hostname);
-    if (chosen === BACK) return SKIP_WAIT as unknown as number;
+    if (chosen === BACK) return 0;
     if (!chosen) {
       console.log("  Aborted.");
       return 0;
@@ -98,7 +98,7 @@ export async function handlePR(args: ParsedArgs): Promise<number> {
       items: [{ label: "Back", value: "back" as const }],
     });
     if (choice === null) process.exit(0);
-    return SKIP_WAIT as unknown as number;
+    return 0;
   }
 
   console.log(`  ${commits.length} commit${commits.length > 1 ? "s" : ""} on this branch`);

src/commands/review.ts

@@ -1,14 +1,7 @@
 import { loadConfig } from "../config";
-import {
+import { isGitRepo, getStagedDiff, getUnstagedFiles, getRepoRoot, stageFiles } from "../git";
-  isGitRepo,
-  getStagedFiles,
-  getStagedDiff,
-  getUnstagedFiles,
-  getRepoRoot,
-  applyFileSelection,
-} from "../git";
 import { selectFiles } from "../selector";
-import { BACK, SKIP_WAIT } from "../menu";
+import { BACK } from "../menu";
 import { collectProjectContext } from "../context";
 import { REVIEW_SYSTEM_PROMPT, buildReviewPrompt } from "../prompt";
 import { callAI } from "../ai";
@@ -60,16 +53,22 @@ export async function handleReview(args: ParsedArgs): Promise<number> {
       console.error(`\n  ${RED()}Error: Not a git repository.${RESET()}\n`);
       return 1;
     }
-    const stagedFiles = await getStagedFiles();
-    const unstagedFiles = await getUnstagedFiles();
-    sourceLabel = "selected changes";
-
-    if (stagedFiles.length > 0 || unstagedFiles.length > 0) {
-      const selected = await selectFiles(stagedFiles, unstagedFiles);
-      if (selected === BACK) return SKIP_WAIT as unknown as number;
-      await applyFileSelection(stagedFiles, unstagedFiles, selected);
-    }
     diff = await getStagedDiff();
+    sourceLabel = "staged changes";
+
+    // If no staged changes, offer to stage unstaged files
+    if (!diff) {
+      const unstagedFiles = await getUnstagedFiles();
+      if (unstagedFiles.length > 0) {
+        const selected = await selectFiles([], unstagedFiles);
+        if (selected === BACK) return 0;
+        if (selected.length > 0) {
+          await stageFiles(selected);
+          console.log(`  ${GREEN()}Staged ${selected.length} file(s).${RESET()}`);
+          diff = await getStagedDiff();
+        }
+      }
+    }
   }
 
   if (!diff) {
@@ -111,7 +110,7 @@ export async function handleReview(args: ParsedArgs): Promise<number> {
   }
 
   try {
-    const callbacks: StreamCallbacks | undefined = tty ? {
+    const callbacks: StreamCallbacks = tty ? {
       onToken: (token) => process.stdout.write(token),
     } : undefined;
 

src/commands/suggest.ts

@@ -1,13 +1,7 @@
 import { loadConfig } from "../config";
-import {
+import { isGitRepo, getStagedDiff, getUnstagedFiles, stageFiles } from "../git";
-  isGitRepo,
-  getStagedFiles,
-  getStagedDiff,
-  getUnstagedFiles,
-  applyFileSelection,
-} from "../git";
 import { selectFiles } from "../selector";
-import { BACK, SKIP_WAIT } from "../menu";
+import { BACK } from "../menu";
 import {
   SUGGEST_SYSTEM_PROMPT,
   buildSuggestBranchPrompt,
@@ -57,15 +51,21 @@ export async function handleSuggest(args: ParsedArgs): Promise<number> {
         diff = "";
       }
     } else {
-      const stagedFiles = await getStagedFiles();
-      const unstagedFiles = await getUnstagedFiles();
-
-      if (stagedFiles.length > 0 || unstagedFiles.length > 0) {
-        const selected = await selectFiles(stagedFiles, unstagedFiles);
-        if (selected === BACK) return SKIP_WAIT as unknown as number;
-        await applyFileSelection(stagedFiles, unstagedFiles, selected);
-      }
       diff = await getStagedDiff();
+
+      // If no staged changes, offer to stage unstaged files
+      if (!diff) {
+        const unstagedFiles = await getUnstagedFiles();
+        if (unstagedFiles.length > 0) {
+          const selected = await selectFiles([], unstagedFiles);
+          if (selected === BACK) return 0;
+          if (selected.length > 0) {
+            await stageFiles(selected);
+            console.log(`  ${GREEN()}Staged ${selected.length} file(s).${RESET()}`);
+            diff = await getStagedDiff();
+          }
+        }
+      }
     }
   }
 

src/git.ts

@@ -98,33 +98,6 @@ export async function stageFiles(paths: string[]): Promise<void> {
   await Bun.$`git add -- ${paths}`;
 }
 
-export async function unstageFiles(paths: string[]): Promise<void> {
-  if (paths.length === 0) return;
-  try {
-    await Bun.$`git restore --staged -- ${paths}`.quiet();
-  } catch {
-    await Bun.$`git rm --cached -r -- ${paths}`.quiet();
-  }
-}
-
-export async function applyFileSelection(
-  stagedFiles: FileEntry[],
-  unstagedFiles: FileEntry[],
-  selectedPaths: string[],
-): Promise<{ staged: string[]; unstaged: string[] }> {
-  const selected = new Set(selectedPaths);
-  const stagedPaths = new Set(stagedFiles.map((file) => file.path));
-  const unstagedPaths = new Set(unstagedFiles.map((file) => file.path));
-
-  const toUnstage = [...stagedPaths].filter((path) => !selected.has(path));
-  const toStage = [...selected].filter((path) => unstagedPaths.has(path));
-
-  await unstageFiles(toUnstage);
-  await stageFiles(toStage);
-
-  return { staged: toStage, unstaged: toUnstage };
-}
-
 export async function commit(
   message: string,
 ): Promise<{ branch: string; hash: string; files: number; insertions: number; deletions: number }> {

src/menu.ts

@@ -15,10 +15,6 @@ const BACKSPACE = "\x7f";
 export const BACK = Symbol("prompt-back");
 export type PromptBack = typeof BACK;
 
-// Sent by command handlers to skip the "Press Enter to return" wait in the
-// interactive menu when the user explicitly backed out of a sub-menu.
-export const SKIP_WAIT = Symbol("skip-wait");
-
 export interface Choice<T> {
   label: string;
   value: T;
@@ -225,7 +221,6 @@ export async function selectMany<T>(
     if (!options.selectAllLabel) return;
     items[0]!.selected = items.slice(1).every((item) => item.selected);
   };
-  syncSelectAll();
 
   const toggle = (index: number) => {
     const item = items[index]!;

src/selector.ts

@@ -1,54 +1,44 @@
 import type { FileEntry } from "./types";
+import { BOLD, GREEN, YELLOW, RESET } from "./terminal";
 import { isStdinTTY } from "./tty";
 import { BACK, selectMany } from "./menu";
 import type { PromptBack } from "./menu";
 
-function mergeFiles(stagedFiles: FileEntry[], unstagedFiles: FileEntry[]) {
-  const files = new Map<string, FileEntry & { staged: boolean; unstaged: boolean }>();
-
-  for (const file of stagedFiles) {
-    files.set(file.path, { ...file, staged: true, unstaged: false });
-  }
-
-  for (const file of unstagedFiles) {
-    const existing = files.get(file.path);
-    if (existing) {
-      existing.unstaged = true;
-      existing.label = existing.label === file.label
-        ? existing.label
-        : `${existing.label}, ${file.label}`;
-    } else {
-      files.set(file.path, { ...file, staged: false, unstaged: true });
-    }
-  }
-
-  return [...files.values()];
-}
-
 export async function selectFiles(
   stagedFiles: FileEntry[],
   unstagedFiles: FileEntry[],
 ): Promise<string[] | PromptBack> {
-  const files = mergeFiles(stagedFiles, unstagedFiles);
+  if (unstagedFiles.length === 0) return [];
-  if (files.length === 0) return [];
 
-  if (!isStdinTTY()) return stagedFiles.map((file) => file.path);
+  if (stagedFiles.length > 0) {
+    process.stdout.write(`\n  ${BOLD()}Staged files (will be included):${RESET()}\n`);
+    for (const f of stagedFiles) {
+      process.stdout.write(`    ${GREEN()}✓${RESET()} ${f.path} (${YELLOW()}${f.label}${RESET()})\n`);
+    }
+  }
+
+  if (!isStdinTTY()) return [];
 
   const selected = await selectMany({
-    title: "Select files for this action",
+    title: "Select files to stage",
-    subtitle: `${stagedFiles.length} staged, ${unstagedFiles.length} unstaged`,
+    subtitle: `${unstagedFiles.length} unstaged file${unstagedFiles.length > 1 ? "s" : ""} available`,
     selectAllLabel: "Select all",
     cancelMessage: "Aborted.",
-    items: files.map((f) => ({
+    items: unstagedFiles.map((f) => ({
       label: f.path,
       value: f.path,
       description: f.label,
-      selected: f.staged,
     })),
   });
 
   if (selected === null) process.exit(1);
   if (selected === BACK) return BACK;
 
+  if (selected.length > 0) {
+    process.stdout.write(
+      `  ${GREEN()}Staged ${selected.length} file(s):${RESET()} ${selected.join(", ")}\n`,
+    );
+  }
+
   return selected;
 }