src/menu.ts

@@ -1,4 +1,5 @@
 import { BOLD, GREEN, CYAN, DIM, RESET } from "./terminal";
+import { isStdinTTY } from "./tty";
 
 const UP = "\x1b[A";
 const DOWN = "\x1b[B";
@@ -38,13 +39,8 @@ interface MultiPromptOptions<T> extends BasePromptOptions {
   doneLabel?: string;
 }
 
-function hideCursor() {
-  process.stdout.write("\x1b[?25l");
-}
-
-function showCursor() {
-  process.stdout.write("\x1b[?25h");
-}
+function hideCursor() { process.stdout.write("\x1b[?25l"); }
+function showCursor() { process.stdout.write("\x1b[?25h"); }
 
 function moveUp(lines: number) {
   if (lines > 0) process.stdout.write(`\x1b[${lines}A`);
@@ -65,10 +61,10 @@ function padLabel(label: string, width: number) {
 function controls(mode: "single" | "multi", showBackHint = true) {
   if (mode === "single") {
     const backHint = showBackHint ? " · ←/backspace back" : "";
-    return `${DIM}↑/↓ navigate · enter/space select${backHint} · ctrl+c cancel${RESET}`;
+    return `${DIM()}↑/↓ navigate · enter/space select${backHint} · ctrl+c cancel${RESET()}`;
   }
   const backHint = showBackHint ? " · ←/backspace back" : "";
-  return `${DIM}↑/↓ navigate · space toggle · enter confirm${backHint} · ctrl+c cancel${RESET}`;
+  return `${DIM()}↑/↓ navigate · space toggle · enter confirm${backHint} · ctrl+c cancel${RESET()}`;
 }
 
 function renderPrompt(lines: string[], previousLines: number) {
@@ -79,10 +75,7 @@ function renderPrompt(lines: string[], previousLines: number) {
     }
     moveUp(previousLines);
   }
-
-  for (const line of lines) {
-    process.stdout.write(`${line}\n`);
-  }
+  for (const line of lines) process.stdout.write(`${line}\n`);
   moveUp(lines.length);
   return lines.length;
 }
@@ -98,28 +91,19 @@ function clearPrompt(lines: number) {
 function normalizeKey(key: string, escapeBuf: string) {
   if (key === UP || key === ALT_UP) return { action: "up", escapeBuf: "" };
   if (key === DOWN || key === ALT_DOWN) return { action: "down", escapeBuf: "" };
-  if (key === LEFT || key === ALT_LEFT || key === BACKSPACE) {
-    return { action: "back", escapeBuf: "" };
-  }
+  if (key === LEFT || key === ALT_LEFT || key === BACKSPACE) return { action: "back", escapeBuf: "" };
   if (key === SPACE) return { action: "space", escapeBuf: "" };
   if (key === ENTER) return { action: "enter", escapeBuf: "" };
   if (key === CTRL_C) return { action: "cancel", escapeBuf: "" };
 
-  if (key === "\x1b" || key.startsWith("\x1b[")) {
-    return { action: null, escapeBuf: key };
-  }
+  if (key === "\x1b" || key.startsWith("\x1b[")) return { action: null, escapeBuf: key };
 
   if (escapeBuf) {
     const next = escapeBuf + key;
     if (next === UP || next === ALT_UP) return { action: "up", escapeBuf: "" };
     if (next === DOWN || next === ALT_DOWN) return { action: "down", escapeBuf: "" };
-    if (next === LEFT || next === ALT_LEFT) {
-      return { action: "back", escapeBuf: "" };
-    }
-    return {
-      action: null,
-      escapeBuf: /^[A-Za-z~]$/.test(key) || next.length > 8 ? "" : next,
-    };
+    if (next === LEFT || next === ALT_LEFT) return { action: "back", escapeBuf: "" };
+    return { action: null, escapeBuf: /^[A-Za-z~]$/.test(key) || next.length > 8 ? "" : next };
   }
 
   return { action: null, escapeBuf: "" };
@@ -130,41 +114,35 @@ function createLines<T>(
   mode: "single" | "multi",
   cursor: number,
 ) {
-  const labelWidth = Math.max(
-    ...options.items.map((item) => visibleLength(item.label)),
-    0,
-  ) + 2;
-
+  const labelWidth = Math.max(...options.items.map((item) => visibleLength(item.label)), 0) + 2;
   const lines = [
     "",
-    `  ${BOLD}${options.title}${RESET}`,
+    `  ${BOLD()}${options.title}${RESET()}`,
   ];
 
-  if (options.subtitle) lines.push(`  ${DIM}${options.subtitle}${RESET}`);
+  if (options.subtitle) lines.push(`  ${DIM()}${options.subtitle}${RESET()}`);
   lines.push(`  ${controls(mode, options.allowBack !== false)}`, "");
 
   for (let i = 0; i < options.items.length; i++) {
     const item = options.items[i]!;
     const active = i === cursor;
-    const pointer = active ? `${CYAN}❯${RESET}` : " ";
+    const pointer = active ? `${CYAN()}❯${RESET()}` : " ";
     const marker = mode === "single"
-      ? active ? `${GREEN}●${RESET}` : `${DIM}○${RESET}`
-      : item.selected ? `${GREEN}◼${RESET}` : `${DIM}□${RESET}`;
-    const label = active ? `${BOLD}${item.label}${RESET}` : item.label;
+      ? active ? `${GREEN()}●${RESET()}` : `${DIM()}○${RESET()}`
+      : item.selected ? `${GREEN()}◼${RESET()}` : `${DIM()}□${RESET()}`;
+    const label = active ? `${BOLD()}${item.label}${RESET()}` : item.label;
     const description = item.description
-      ? active ? item.description : `${DIM}${item.description}${RESET}`
+      ? active ? item.description : `${DIM()}${item.description}${RESET()}`
       : "";
 
-    lines.push(
-      `  ${pointer} ${marker} ${padLabel(label, labelWidth)}${description}`,
-    );
+    lines.push(`  ${pointer} ${marker} ${padLabel(label, labelWidth)}${description}`);
   }
 
   return lines;
 }
 
 function ensureTTY(title: string) {
-  if (process.stdin.isTTY !== true) {
+  if (!isStdinTTY()) {
     throw new Error(`${title} requires a TTY.`);
   }
 }
@@ -184,12 +162,8 @@ export async function selectOne<T>(
   hideCursor();
 
   const render = () => {
-    renderedLines = renderPrompt(
-      createLines(options, "single", cursor),
-      renderedLines,
-    );
+    renderedLines = renderPrompt(createLines(options, "single", cursor), renderedLines);
   };
-
   render();
 
   return new Promise((resolve) => {
@@ -210,16 +184,10 @@ export async function selectOne<T>(
       escapeBuf = result.escapeBuf;
 
       if (result.action === "cancel") return finish(null);
-      if (result.action === "back" && options.allowBack !== false) {
-        return finish(BACK);
-      }
-      if (result.action === "up" && cursor > 0) {
-        cursor--;
-        render();
-      } else if (result.action === "down" && cursor < options.items.length - 1) {
-        cursor++;
-        render();
-      } else if (result.action === "space" || result.action === "enter") {
+      if (result.action === "back" && options.allowBack !== false) return finish(BACK);
+      if (result.action === "up" && cursor > 0) { cursor--; render(); }
+      else if (result.action === "down" && cursor < options.items.length - 1) { cursor++; render(); }
+      else if (result.action === "space" || result.action === "enter") {
         finish(options.items[cursor]!.value);
       }
     };
@@ -257,11 +225,8 @@ export async function selectMany<T>(
   const toggle = (index: number) => {
     const item = items[index]!;
     item.selected = !item.selected;
-
     if (options.selectAllLabel && index === 0) {
-      for (let i = 1; i < items.length; i++) {
-        items[i]!.selected = item.selected;
-      }
+      for (let i = 1; i < items.length; i++) items[i]!.selected = item.selected;
     } else {
       syncSelectAll();
     }
@@ -269,19 +234,10 @@ export async function selectMany<T>(
 
   const render = () => {
     renderedLines = renderPrompt(
-      createLines(
-        {
-          title: options.title,
-          subtitle: options.subtitle,
-          items,
-        },
-        "multi",
-        cursor,
-      ),
+      createLines({ title: options.title, subtitle: options.subtitle, items }, "multi", cursor),
       renderedLines,
     );
   };
-
   render();
 
   return new Promise((resolve) => {
@@ -302,24 +258,12 @@ export async function selectMany<T>(
       escapeBuf = result.escapeBuf;
 
       if (result.action === "cancel") return finish(null);
-      if (result.action === "back" && options.allowBack !== false) {
-        return finish(BACK);
-      }
-      if (result.action === "up" && cursor > 0) {
-        cursor--;
-        render();
-      } else if (result.action === "down" && cursor < items.length - 1) {
-        cursor++;
-        render();
-      } else if (result.action === "space") {
-        toggle(cursor);
-        render();
-      } else if (result.action === "enter") {
-        finish(
-          items
-            .filter((item) => item.selected && item.value !== null)
-            .map((item) => item.value as T),
-        );
+      if (result.action === "back" && options.allowBack !== false) return finish(BACK);
+      if (result.action === "up" && cursor > 0) { cursor--; render(); }
+      else if (result.action === "down" && cursor < items.length - 1) { cursor++; render(); }
+      else if (result.action === "space") { toggle(cursor); render(); }
+      else if (result.action === "enter") {
+        finish(items.filter((item) => item.selected && item.value !== null).map((item) => item.value as T));
       }
     };
 

src/selector.ts

@@ -1,5 +1,6 @@
 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";
 
@@ -10,13 +11,13 @@ export async function selectFiles(
   if (unstagedFiles.length === 0) return [];
 
   if (stagedFiles.length > 0) {
-    process.stdout.write(`\n  ${BOLD}Staged files (will be included):${RESET}\n`);
+    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`);
+      process.stdout.write(`    ${GREEN()}✓${RESET()} ${f.path} (${YELLOW()}${f.label}${RESET()})\n`);
     }
   }
 
-  if (process.stdin.isTTY !== true) return [];
+  if (!isStdinTTY()) return [];
 
   const selected = await selectMany({
     title: "Select files to stage",
@@ -35,7 +36,7 @@ export async function selectFiles(
 
   if (selected.length > 0) {
     process.stdout.write(
-      `  ${GREEN}Staged ${selected.length} file(s):${RESET} ${selected.join(", ")}\n`,
+      `  ${GREEN()}Staged ${selected.length} file(s):${RESET()} ${selected.join(", ")}\n`,
     );
   }