[Init] Initial commit - NetMesh terminal manager
Some checks failed
build-packages / resolve bundled mosh-client (push) Has been cancelled
build-packages / resolve bundled et-client (push) Has been cancelled
build-packages / build-macos (push) Has been cancelled
build-packages / build-windows (push) Has been cancelled
build-packages / build-linux-x64 (push) Has been cancelled
build-packages / build-linux-arm64 (push) Has been cancelled
build-packages / release (push) Has been cancelled
build-packages / update Nix release metadata (push) Has been cancelled
build-packages / bump homebrew tap (push) Has been cancelled
test / lint-and-test (push) Has been cancelled
AI automation / Route event (push) Has been cancelled
AI automation / Hand reopened issue to maintainers (push) Has been cancelled
AI automation / Clean source issue state (push) Has been cancelled
AI automation / Reconcile handoffs (push) Has been cancelled
AI automation / Classify issue (push) Has been cancelled
AI automation / Claude Code smoke (push) Has been cancelled
AI automation / Review issue follow-up (push) Has been cancelled
AI automation / Publish issue follow-up (push) Has been cancelled
AI automation / Implement with Claude Code (push) Has been cancelled
AI automation / Publish implement PR (push) Has been cancelled
AI automation / Continue queued issue comments (push) Has been cancelled
AI automation / Codex review loop (push) Has been cancelled
AI automation / Publish Codex fix (push) Has been cancelled
AI automation / Clear Codex dispatch marker (push) Has been cancelled
AI automation / Own PR re-request Codex (push) Has been cancelled
AI automation / External PR re-request Codex (push) Has been cancelled
AI automation / Poll Codex reaction / retry (push) Has been cancelled
build-et-binaries / build-linux-x64 (push) Has been cancelled
build-et-binaries / build-linux-arm64 (push) Has been cancelled
build-et-binaries / build-macos-universal (push) Has been cancelled
build-et-binaries / build-windows-x64 (push) Has been cancelled
build-et-binaries / release (push) Has been cancelled
Some checks failed
build-packages / resolve bundled mosh-client (push) Has been cancelled
build-packages / resolve bundled et-client (push) Has been cancelled
build-packages / build-macos (push) Has been cancelled
build-packages / build-windows (push) Has been cancelled
build-packages / build-linux-x64 (push) Has been cancelled
build-packages / build-linux-arm64 (push) Has been cancelled
build-packages / release (push) Has been cancelled
build-packages / update Nix release metadata (push) Has been cancelled
build-packages / bump homebrew tap (push) Has been cancelled
test / lint-and-test (push) Has been cancelled
AI automation / Route event (push) Has been cancelled
AI automation / Hand reopened issue to maintainers (push) Has been cancelled
AI automation / Clean source issue state (push) Has been cancelled
AI automation / Reconcile handoffs (push) Has been cancelled
AI automation / Classify issue (push) Has been cancelled
AI automation / Claude Code smoke (push) Has been cancelled
AI automation / Review issue follow-up (push) Has been cancelled
AI automation / Publish issue follow-up (push) Has been cancelled
AI automation / Implement with Claude Code (push) Has been cancelled
AI automation / Publish implement PR (push) Has been cancelled
AI automation / Continue queued issue comments (push) Has been cancelled
AI automation / Codex review loop (push) Has been cancelled
AI automation / Publish Codex fix (push) Has been cancelled
AI automation / Clear Codex dispatch marker (push) Has been cancelled
AI automation / Own PR re-request Codex (push) Has been cancelled
AI automation / External PR re-request Codex (push) Has been cancelled
AI automation / Poll Codex reaction / retry (push) Has been cancelled
build-et-binaries / build-linux-x64 (push) Has been cancelled
build-et-binaries / build-linux-arm64 (push) Has been cancelled
build-et-binaries / build-macos-universal (push) Has been cancelled
build-et-binaries / build-windows-x64 (push) Has been cancelled
build-et-binaries / release (push) Has been cancelled
This commit is contained in:
819
components/terminal/PromptDetector.test.ts
Normal file
819
components/terminal/PromptDetector.test.ts
Normal file
@@ -0,0 +1,819 @@
|
||||
import test from "node:test";
|
||||
import assert from "node:assert/strict";
|
||||
|
||||
import { detectPrompt, getAlignedPrompt } from "./autocomplete/promptDetector.ts";
|
||||
import { resolveAutocompleteQueryInput } from "./autocomplete/terminalAutocompletePrompt.ts";
|
||||
import { getSnippetSuggestions } from "./autocomplete/snippetCompleter.ts";
|
||||
import { stringCellWidth } from "./autocomplete/terminalStringCellWidth.ts";
|
||||
import { getCommandToRecordOnEnter } from "./autocomplete/useTerminalAutocomplete.ts";
|
||||
|
||||
function createFakeTerm(lineText: string, cursorX: number) {
|
||||
return {
|
||||
buffer: {
|
||||
active: {
|
||||
cursorX,
|
||||
cursorY: 0,
|
||||
baseY: 0,
|
||||
getLine(line: number) {
|
||||
if (line !== 0) return undefined;
|
||||
return {
|
||||
isWrapped: false,
|
||||
translateToString() {
|
||||
return lineText;
|
||||
},
|
||||
};
|
||||
},
|
||||
},
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
/** Simulates xterm padding empty cells as trailing spaces after wide glyphs. */
|
||||
function createPaddedFakeTerm(content: string, cursorX: number, cols = 80) {
|
||||
const pad = Math.max(0, cols - stringCellWidth(content));
|
||||
const lineText = content + " ".repeat(pad);
|
||||
return createFakeTerm(lineText, cursorX);
|
||||
}
|
||||
|
||||
function createWrappedFakeTerm(rows: string[], cursorY: number, cursorX: number, cols: number) {
|
||||
return {
|
||||
cols,
|
||||
buffer: {
|
||||
active: {
|
||||
cursorX,
|
||||
cursorY,
|
||||
baseY: 0,
|
||||
getLine(line: number) {
|
||||
const lineText = rows[line];
|
||||
if (lineText === undefined) return undefined;
|
||||
return {
|
||||
isWrapped: line > 0,
|
||||
translateToString() {
|
||||
return lineText;
|
||||
},
|
||||
};
|
||||
},
|
||||
},
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
test("keeps raw input when a standard shell prompt echo is still behind", () => {
|
||||
const term = createFakeTerm("$ do", 4);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "doc", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, "$ ");
|
||||
assert.equal(result.prompt.userInput, "do");
|
||||
assert.equal(result.prompt.cursorOffset, 2);
|
||||
assert.equal(result.alignedTyped, null);
|
||||
});
|
||||
|
||||
test("uses reliable typed buffer when prompt echo has not started (IME / high-latency commit)", () => {
|
||||
const term = createFakeTerm("$ ", 2);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "部署", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, "$ ");
|
||||
assert.equal(result.prompt.userInput, "部署");
|
||||
assert.equal(result.prompt.cursorOffset, 2);
|
||||
// Pre-echo input is surfaced for alignment, but empty echo alone must not
|
||||
// authorize history recording or third-party completion providers until echo
|
||||
// validates the line. Local history/fig/snippet popups may still query from
|
||||
// the keystroke buffer (#2830).
|
||||
assert.equal(result.alignedTyped, null);
|
||||
assert.equal(result.allowExternalProviders, false);
|
||||
});
|
||||
|
||||
test("uses typed buffer before echo on padded themed prompts", () => {
|
||||
// robbyrussell-style PS1 pads after the arrow: detectPrompt keeps the
|
||||
// trailing space in userInput, which must still count as visually empty.
|
||||
const term = createFakeTerm("➜ ", 3);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "部署", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.userInput, "部署");
|
||||
assert.equal(result.alignedTyped, null);
|
||||
assert.equal(result.allowExternalProviders, false);
|
||||
});
|
||||
|
||||
test("uses typed buffer before echo on Nerd Font themed terminators", () => {
|
||||
const term = createFakeTerm(" ", 2);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "部署", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, " ");
|
||||
assert.equal(result.prompt.userInput, "部署");
|
||||
assert.equal(result.alignedTyped, null);
|
||||
assert.equal(result.allowExternalProviders, false);
|
||||
});
|
||||
|
||||
test("does not treat empty-echo shell-shaped prompts as validated typed input", () => {
|
||||
const term = createFakeTerm("$ ", 2);
|
||||
const secret = "s3cret-token";
|
||||
|
||||
const result = getAlignedPrompt(term as never, secret, true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
// Keystroke buffer is still visible on the prompt view, but empty echo
|
||||
// must not authorize history recording or external providers.
|
||||
assert.equal(result.prompt.userInput, secret);
|
||||
assert.equal(result.alignedTyped, null);
|
||||
assert.equal(result.allowExternalProviders, false);
|
||||
});
|
||||
test("still trims prompt decorations out of the detected input", () => {
|
||||
const term = createFakeTerm("➜ ~ do", 7);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "do", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, "➜ ~ ");
|
||||
assert.equal(result.prompt.userInput, "do");
|
||||
assert.equal(result.prompt.cursorOffset, 2);
|
||||
assert.equal(result.alignedTyped, "do");
|
||||
});
|
||||
|
||||
test("detects oh-my-posh Nerd Font chevron (U+F105) prompt terminator", () => {
|
||||
// Real-world PS1 captured from oh-my-posh themed bash on a server:
|
||||
// "<U+F31B> root@oracle ~ <U+F105> " then user input
|
||||
const term = createFakeTerm(" root@oracle ~ ls", 21);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "ls", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, " root@oracle ~ ");
|
||||
assert.equal(result.prompt.userInput, "ls");
|
||||
});
|
||||
|
||||
test("detects Powerline right-arrow (U+E0B0) prompt terminator", () => {
|
||||
// oh-my-posh agnoster-style: colored block ending with U+E0B0 + space
|
||||
const term = createFakeTerm(" root ~ git", 16);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "git", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.userInput, "git");
|
||||
assert.ok(result.prompt.promptText.endsWith(" "));
|
||||
});
|
||||
|
||||
test("PUA char without trailing space is not a prompt boundary", () => {
|
||||
// A bare PUA glyph mid-token (e.g. paste artifact) should not trigger detection.
|
||||
const term = createFakeTerm("echo foo", 13);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, false);
|
||||
});
|
||||
|
||||
test("keeps typed command intact when command text contains Powerline glyphs", () => {
|
||||
const typedInput = "echo foo";
|
||||
const lineText = `$ ${typedInput}`;
|
||||
const term = createFakeTerm(lineText, lineText.length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, typedInput, true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, "$ ");
|
||||
assert.equal(result.prompt.userInput, typedInput);
|
||||
assert.equal(result.alignedTyped, typedInput);
|
||||
});
|
||||
|
||||
test("does not treat a mid-line dollar as a prompt boundary", () => {
|
||||
const lineText = "$ echo $HOME";
|
||||
const term = createFakeTerm(lineText, "$ echo $".length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, "$ ");
|
||||
assert.equal(result.prompt.userInput, "echo $");
|
||||
assert.equal(result.prompt.cursorOffset, "echo $".length);
|
||||
});
|
||||
|
||||
test("does not treat a mid-line redirection as a prompt boundary", () => {
|
||||
const lineText = "$ cat >file";
|
||||
const term = createFakeTerm(lineText, "$ cat >".length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, "$ ");
|
||||
assert.equal(result.prompt.userInput, "cat >");
|
||||
assert.equal(result.prompt.cursorOffset, "cat >".length);
|
||||
});
|
||||
|
||||
test("does not treat a spaced redirection as a prompt boundary", () => {
|
||||
const lineText = "$ cat > file";
|
||||
const term = createFakeTerm(lineText, lineText.length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, "$ ");
|
||||
assert.equal(result.prompt.userInput, "cat > file");
|
||||
});
|
||||
|
||||
test("does not treat common interactive program prompts as shell prompts", () => {
|
||||
const cases = [
|
||||
{ lineText: "sftp> get file", typedInput: "get file" },
|
||||
{ lineText: "ftp> ls", typedInput: "ls" },
|
||||
{ lineText: "ghci> :t map", typedInput: ":t map" },
|
||||
{ lineText: "node> .help", typedInput: ".help" },
|
||||
{ lineText: "mongo> db.stats()", typedInput: "db.stats()" },
|
||||
{ lineText: "rs0:PRIMARY> db.stats()", typedInput: "db.stats()" },
|
||||
{ lineText: "rs0 [direct: primary] test> db.stats()", typedInput: "db.stats()" },
|
||||
{ lineText: "rs0 [direct: primary] reporting> db.stats()", typedInput: "db.stats()" },
|
||||
{ lineText: "rs0 [direct: primary] reporting> const x = 1", typedInput: "const x = 1" },
|
||||
{ lineText: "rs0 [direct: primary] reporting> await db.users.findOne()", typedInput: "await db.users.findOne()" },
|
||||
{ lineText: "Atlas a [primary] reporting> db.stats()", typedInput: "db.stats()" },
|
||||
{ lineText: "Atlas a [primary] reporting> await db.users.findOne()", typedInput: "await db.users.findOne()" },
|
||||
{ lineText: "rs0 primary reporting> exit", typedInput: "exit" },
|
||||
{ lineText: "irb(main):001> puts 1", typedInput: "puts 1" },
|
||||
{ lineText: "pry(main)> whereami", typedInput: "whereami" },
|
||||
{ lineText: "[1] pry(main)> whereami", typedInput: "whereami" },
|
||||
{ lineText: "SQL> select 1", typedInput: "select 1" },
|
||||
{ lineText: "cqlsh> select * from users", typedInput: "select * from users" },
|
||||
{ lineText: "hive> select 1", typedInput: "select 1" },
|
||||
{ lineText: "spark-sql> select 1", typedInput: "select 1" },
|
||||
{ lineText: "jshell> /help", typedInput: "/help" },
|
||||
{ lineText: " ...> System.out.println(1)", typedInput: "System.out.println(1)" },
|
||||
{ lineText: "ksql> select 1", typedInput: "select 1" },
|
||||
{ lineText: "trino> select 1", typedInput: "select 1" },
|
||||
{ lineText: "trino:tpch> select 1", typedInput: "select 1" },
|
||||
{ lineText: "presto> show catalogs", typedInput: "show catalogs" },
|
||||
{ lineText: "presto:default> show tables", typedInput: "show tables" },
|
||||
{ lineText: "duckdb> select 1", typedInput: "select 1" },
|
||||
{ lineText: "lftp user@example.com:~> ls", typedInput: "ls" },
|
||||
{ lineText: "cqlsh:cycling> select * from cyclist", typedInput: "select * from cyclist" },
|
||||
{ lineText: "hive (default)> select 1", typedInput: "select 1" },
|
||||
{ lineText: "0: jdbc:hive2://localhost:10000/default> select 1", typedInput: "select 1" },
|
||||
{ lineText: "spark-sql (default)> select 1", typedInput: "select 1" },
|
||||
{ lineText: "test> db.stats()", typedInput: "db.stats()" },
|
||||
{ lineText: "test> const x = 1", typedInput: "const x = 1" },
|
||||
{ lineText: "test> await db.users.findOne()", typedInput: "await db.users.findOne()" },
|
||||
{ lineText: "test> db", typedInput: "db" },
|
||||
{ lineText: "rs0 primary test> db.stats()", typedInput: "db.stats()" },
|
||||
{ lineText: "test> rs.status()", typedInput: "rs.status()" },
|
||||
{ lineText: "test> print(1)", typedInput: "print(1)" },
|
||||
{ lineText: "test> 1 + 1", typedInput: "1 + 1" },
|
||||
{ lineText: "admin@localhost:27017> db.stats()", typedInput: "db.stats()" },
|
||||
];
|
||||
|
||||
for (const { lineText, typedInput } of cases) {
|
||||
const result = getAlignedPrompt(
|
||||
createFakeTerm(lineText, lineText.length) as never,
|
||||
typedInput,
|
||||
true,
|
||||
);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, false, lineText);
|
||||
assert.equal(result.alignedTyped, null, lineText);
|
||||
}
|
||||
});
|
||||
|
||||
test("does not treat sensitive authentication challenges as shell prompts", () => {
|
||||
for (const lineText of [
|
||||
"OTP> 123456",
|
||||
"Verification code> 123456",
|
||||
"Duo passcode: 123456",
|
||||
"验证码> 123456",
|
||||
]) {
|
||||
const typedInput = lineText.slice(lineText.lastIndexOf(" ") + 1);
|
||||
const result = getAlignedPrompt(
|
||||
createFakeTerm(lineText, lineText.length) as never,
|
||||
typedInput,
|
||||
true,
|
||||
);
|
||||
assert.equal(result.prompt.isAtPrompt, false, lineText);
|
||||
assert.equal(result.alignedTyped, null, lineText);
|
||||
}
|
||||
});
|
||||
|
||||
test("does not treat wrapped interactive program prompts as shell prompts", () => {
|
||||
const cases = [
|
||||
{ rows: ["sftp> get very-long-", "remote-file"], typedInput: "get very-long-remote-file" },
|
||||
{ rows: ["node> console.", "log('ok')"], typedInput: "console.log('ok')" },
|
||||
{ rows: ["mongo> db.", "stats()"], typedInput: "db.stats()" },
|
||||
{ rows: ["cqlsh> select *", " from users"], typedInput: "select * from users" },
|
||||
{ rows: ["jshell> System.out.", "println(1)"], typedInput: "System.out.println(1)" },
|
||||
{ rows: [" ...> System.out.", "println(1)"], typedInput: "System.out.println(1)" },
|
||||
{ rows: ["trino> select", " 1"], typedInput: "select 1" },
|
||||
{ rows: ["trino:tpch> select", " 1"], typedInput: "select 1" },
|
||||
{ rows: ["duckdb> select", " 1"], typedInput: "select 1" },
|
||||
{ rows: ["cqlsh:cycling> select *", " from cyclist"], typedInput: "select * from cyclist" },
|
||||
{ rows: ["hive (default)> select", " 1"], typedInput: "select 1" },
|
||||
{ rows: ["0: jdbc:hive2://localhost:10000/default> select", " 1"], typedInput: "select 1" },
|
||||
{ rows: ["test> db.", "stats()"], typedInput: "db.stats()" },
|
||||
{ rows: ["test> d", "b"], typedInput: "db" },
|
||||
{ rows: ["rs0:PRIMARY> db.", "stats()"], typedInput: "db.stats()" },
|
||||
{ rows: ["rs0 [direct: primary] test> db.", "stats()"], typedInput: "db.stats()" },
|
||||
{ rows: ["rs0 [direct: primary]", " test> db.stats()"], typedInput: "db.stats()" },
|
||||
{ rows: ["rs0 [direct: primary]", " reporting> db.stats()"], typedInput: "db.stats()" },
|
||||
{ rows: ["rs0 [direct: primary]", " reporting> const x = 1"], typedInput: "const x = 1" },
|
||||
{ rows: ["Atlas a [primary]", " reporting> db.stats()"], typedInput: "db.stats()" },
|
||||
{ rows: ["rs0 primary test> db.", "stats()"], typedInput: "db.stats()" },
|
||||
{ rows: ["test> print", "(1)"], typedInput: "print(1)" },
|
||||
{ rows: ["test> 1 ", "+ 1"], typedInput: "1 + 1" },
|
||||
{ rows: ["admin@localhost:27017> db.", "stats()"], typedInput: "db.stats()" },
|
||||
];
|
||||
|
||||
for (const { rows, typedInput } of cases) {
|
||||
const result = getAlignedPrompt(
|
||||
createWrappedFakeTerm(rows, 1, rows[1].length, 20) as never,
|
||||
typedInput,
|
||||
true,
|
||||
);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, false, rows[0]);
|
||||
assert.equal(result.alignedTyped, null, rows[0]);
|
||||
}
|
||||
});
|
||||
|
||||
test("keeps non-Mongo-looking default-name greater-than prompts usable", () => {
|
||||
const prompts = ["test> ", "admin> ", "local> ", "config> "];
|
||||
const commands = ["deploy", "exit", "help", "show dbs"];
|
||||
|
||||
for (const prompt of prompts) {
|
||||
for (const typedInput of commands) {
|
||||
const lineText = `${prompt}${typedInput}`;
|
||||
const result = getAlignedPrompt(
|
||||
createFakeTerm(lineText, lineText.length) as never,
|
||||
typedInput,
|
||||
true,
|
||||
);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true, lineText);
|
||||
assert.equal(result.prompt.promptText, prompt, lineText);
|
||||
assert.equal(result.prompt.userInput, typedInput, lineText);
|
||||
assert.equal(result.alignedTyped, typedInput, lineText);
|
||||
assert.equal(
|
||||
getCommandToRecordOnEnter(result.prompt, result.alignedTyped, typedInput, true),
|
||||
typedInput,
|
||||
lineText,
|
||||
);
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
test("keeps wrapped non-Mongo-looking default-name greater-than prompts usable", () => {
|
||||
const cases = [
|
||||
{ rows: ["test> hel", "p"], typedInput: "help", promptText: "test> " },
|
||||
{ rows: ["test> show ", "dbs"], typedInput: "show dbs", promptText: "test> " },
|
||||
{ rows: ["admin> ex", "it"], typedInput: "exit", promptText: "admin> " },
|
||||
{ rows: ["local> dep", "loy"], typedInput: "deploy", promptText: "local> " },
|
||||
];
|
||||
|
||||
for (const { rows, typedInput, promptText } of cases) {
|
||||
const result = getAlignedPrompt(
|
||||
createWrappedFakeTerm(rows, 1, rows[1].length, 20) as never,
|
||||
typedInput,
|
||||
true,
|
||||
);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true, rows[0]);
|
||||
assert.equal(result.prompt.promptText, promptText, rows[0]);
|
||||
assert.equal(result.prompt.userInput, typedInput, rows[0]);
|
||||
assert.equal(result.alignedTyped, typedInput, rows[0]);
|
||||
assert.equal(
|
||||
getCommandToRecordOnEnter(result.prompt, result.alignedTyped, typedInput, true),
|
||||
typedInput,
|
||||
rows[0],
|
||||
);
|
||||
}
|
||||
});
|
||||
|
||||
test("keeps host-style greater-than prompts usable", () => {
|
||||
const prompts = [
|
||||
"prod-web> ",
|
||||
"prod> ",
|
||||
"prod.web> ",
|
||||
"server> ",
|
||||
"staging> ",
|
||||
"webdb> ",
|
||||
"prod.db> ",
|
||||
];
|
||||
const commands = [
|
||||
"deploy",
|
||||
"exit",
|
||||
"show dbs",
|
||||
"use app",
|
||||
"it",
|
||||
"help",
|
||||
"print(1)",
|
||||
"db.stats()",
|
||||
];
|
||||
|
||||
for (const prompt of prompts) {
|
||||
for (const typedInput of commands) {
|
||||
const lineText = `${prompt}${typedInput}`;
|
||||
const result = getAlignedPrompt(
|
||||
createFakeTerm(lineText, lineText.length) as never,
|
||||
typedInput,
|
||||
true,
|
||||
);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true, lineText);
|
||||
assert.equal(result.prompt.promptText, prompt, lineText);
|
||||
assert.equal(result.prompt.userInput, typedInput, lineText);
|
||||
assert.equal(result.alignedTyped, typedInput, lineText);
|
||||
}
|
||||
}
|
||||
});
|
||||
|
||||
test("keeps strong bare Mongo prompt signals out of shell prompts", () => {
|
||||
const cases = [
|
||||
{ lineText: "test> db.stats()", typedInput: "db.stats()" },
|
||||
{ lineText: "test> db", typedInput: "db" },
|
||||
{ lineText: "test> const x = 1", typedInput: "const x = 1" },
|
||||
{ lineText: "test> await db.users.findOne()", typedInput: "await db.users.findOne()" },
|
||||
{ lineText: "test> print(1)", typedInput: "print(1)" },
|
||||
{ lineText: "test> 1 + 1", typedInput: "1 + 1" },
|
||||
];
|
||||
|
||||
for (const { lineText, typedInput } of cases) {
|
||||
const result = getAlignedPrompt(
|
||||
createFakeTerm(lineText, lineText.length) as never,
|
||||
typedInput,
|
||||
true,
|
||||
);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, false, lineText);
|
||||
assert.equal(result.alignedTyped, null, lineText);
|
||||
}
|
||||
});
|
||||
|
||||
test("does not align stale typed input after themed prompt command suffixes", () => {
|
||||
const cases = [
|
||||
"➜ ~ echo sudo",
|
||||
"➜ echo sudo",
|
||||
"➜ make sudo",
|
||||
"➜ docker sudo",
|
||||
"➜ ./script sudo",
|
||||
"➜ ./script sudo",
|
||||
"➜ ~ echo # sudo",
|
||||
];
|
||||
|
||||
for (const lineText of cases) {
|
||||
const result = getAlignedPrompt(createFakeTerm(lineText, lineText.length) as never, "sudo", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true, lineText);
|
||||
assert.equal(result.prompt.promptText, "➜ ", lineText);
|
||||
assert.equal(result.prompt.userInput, lineText.slice("➜ ".length), lineText);
|
||||
assert.equal(result.alignedTyped, null, lineText);
|
||||
}
|
||||
});
|
||||
|
||||
test("aligns themed prompt decorations when command echo lags", () => {
|
||||
const typedInput = "git status";
|
||||
const cases = [
|
||||
{ lineText: "➜ ~ git ", promptText: "➜ ~ " },
|
||||
{ lineText: "➜ ~ git st", promptText: "➜ ~ " },
|
||||
{
|
||||
lineText: "➜ netcatty git:(main) ✗ git ",
|
||||
promptText: "➜ netcatty git:(main) ✗ ",
|
||||
},
|
||||
{
|
||||
lineText: "➜ netcatty git:(main) ✗ git st",
|
||||
promptText: "➜ netcatty git:(main) ✗ ",
|
||||
},
|
||||
];
|
||||
|
||||
for (const { lineText, promptText } of cases) {
|
||||
const result = getAlignedPrompt(
|
||||
createFakeTerm(lineText, lineText.length) as never,
|
||||
typedInput,
|
||||
true,
|
||||
);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true, lineText);
|
||||
assert.equal(result.prompt.promptText, promptText, lineText);
|
||||
assert.equal(result.prompt.userInput, typedInput, lineText);
|
||||
assert.equal(result.alignedTyped, typedInput, lineText);
|
||||
assert.equal(
|
||||
getCommandToRecordOnEnter(result.prompt, result.alignedTyped, typedInput, true),
|
||||
typedInput,
|
||||
lineText,
|
||||
);
|
||||
}
|
||||
});
|
||||
|
||||
test("trims single-space themed prompt decorations out of the detected input", () => {
|
||||
const cases = [
|
||||
{ lineText: "➜ ~/repo do", typedInput: "do", promptText: "➜ ~/repo " },
|
||||
{
|
||||
lineText: "➜ netcatty git:(main) ✗ ls",
|
||||
typedInput: "ls",
|
||||
promptText: "➜ netcatty git:(main) ✗ ",
|
||||
},
|
||||
{
|
||||
lineText: "➜ netcatty git:(main) ✗ + ls",
|
||||
typedInput: "ls",
|
||||
promptText: "➜ netcatty git:(main) ✗ + ",
|
||||
},
|
||||
{ lineText: "➜ netcatty ✗ $ ls", typedInput: "ls", promptText: "➜ netcatty ✗ $ " },
|
||||
{ lineText: "➜ netcatty $ ls", typedInput: "ls", promptText: "➜ netcatty $ " },
|
||||
];
|
||||
|
||||
for (const { lineText, typedInput, promptText } of cases) {
|
||||
const term = createFakeTerm(lineText, lineText.length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, typedInput, true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true, lineText);
|
||||
assert.equal(result.prompt.promptText, promptText, lineText);
|
||||
assert.equal(result.prompt.userInput, typedInput, lineText);
|
||||
assert.equal(result.alignedTyped, typedInput, lineText);
|
||||
}
|
||||
});
|
||||
|
||||
test("does not treat later shell symbols followed by spaces as prompt boundaries", () => {
|
||||
const cases = [
|
||||
"$ echo # comment",
|
||||
"$ printf % value",
|
||||
"$ echo $ value",
|
||||
];
|
||||
|
||||
for (const lineText of cases) {
|
||||
const result = getAlignedPrompt(createFakeTerm(lineText, lineText.length) as never, "", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true, lineText);
|
||||
assert.equal(result.prompt.promptText, "$ ", lineText);
|
||||
assert.equal(result.prompt.userInput, lineText.slice(2), lineText);
|
||||
}
|
||||
});
|
||||
|
||||
test("does not treat command-leading shell symbols as prompt boundaries", () => {
|
||||
const cases = [
|
||||
"$ # comment",
|
||||
"$ > file",
|
||||
"$ % value",
|
||||
"$ $ value",
|
||||
"root@host:~# foo $ value",
|
||||
];
|
||||
|
||||
for (const lineText of cases) {
|
||||
const result = getAlignedPrompt(createFakeTerm(lineText, lineText.length) as never, "", false);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true, lineText);
|
||||
const expectedPrompt = lineText.startsWith("root@host:~#") ? "root@host:~# " : "$ ";
|
||||
assert.equal(result.prompt.promptText, expectedPrompt, lineText);
|
||||
assert.equal(result.prompt.userInput, lineText.slice(expectedPrompt.length), lineText);
|
||||
assert.equal(result.alignedTyped, null, lineText);
|
||||
}
|
||||
});
|
||||
|
||||
test("keeps prompt symbols that are part of the prompt text", () => {
|
||||
const prompts = [
|
||||
"user@host ~/foo#bar $ ",
|
||||
"user@host ~/foo# bar $ ",
|
||||
"user@host:~/foo# bar $ ",
|
||||
"user@host ~/foo% bar $ ",
|
||||
"user@host ~/foo> bar $ ",
|
||||
];
|
||||
const typedInput = "ls";
|
||||
|
||||
for (const prompt of prompts) {
|
||||
const term = createFakeTerm(`${prompt}${typedInput}`, prompt.length + typedInput.length);
|
||||
const result = getAlignedPrompt(term as never, typedInput, true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true, prompt);
|
||||
assert.equal(result.prompt.promptText, prompt, prompt);
|
||||
assert.equal(result.prompt.userInput, typedInput, prompt);
|
||||
assert.equal(result.alignedTyped, typedInput, prompt);
|
||||
}
|
||||
});
|
||||
|
||||
test("keeps prompt symbols in prompt text without typed-buffer alignment", () => {
|
||||
const prompts = [
|
||||
"user@host ~/foo# bar $ ",
|
||||
"user@host ~/foo# git $ ",
|
||||
"user@host ~/foo#git $ ",
|
||||
"root@host ~/foo# bar # ",
|
||||
"root@host ~/foo#bar # ",
|
||||
"fish@host ~/foo# bar % ",
|
||||
"fish@host ~/foo%bar % ",
|
||||
"user@host:~/foo# bar $ ",
|
||||
"user@host ~/repo # $ ",
|
||||
"➜ ~ $ ",
|
||||
"user@host ~/foo% bar $ ",
|
||||
"user@host ~/foo> bar $ ",
|
||||
"user@host ~/foo# bar> ",
|
||||
"user@host ~/foo# bar› ",
|
||||
"user@host ~/foo#bar> ",
|
||||
];
|
||||
|
||||
for (const prompt of prompts) {
|
||||
const lineText = `${prompt}ls`;
|
||||
const result = getAlignedPrompt(createFakeTerm(lineText, lineText.length) as never, "", false);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true, prompt);
|
||||
assert.equal(result.prompt.promptText, prompt, prompt);
|
||||
assert.equal(result.prompt.userInput, "ls", prompt);
|
||||
assert.equal(result.alignedTyped, null, prompt);
|
||||
}
|
||||
});
|
||||
|
||||
test("prefers standard prompt terminator over later Powerline glyphs", () => {
|
||||
const lineText = "$ echo foo";
|
||||
const term = createFakeTerm(lineText, lineText.length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, "$ ");
|
||||
assert.equal(result.prompt.userInput, "echo foo");
|
||||
});
|
||||
|
||||
test("ignores xterm row padding after a no-space root prompt", () => {
|
||||
const prompt = " root@stwo:~#";
|
||||
const term = createFakeTerm(`${prompt} `, prompt.length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, "", true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, prompt);
|
||||
assert.equal(result.prompt.userInput, "");
|
||||
});
|
||||
|
||||
test("aligns typed input after a no-space root prompt", () => {
|
||||
const prompt = " root@stwo:~#";
|
||||
const typedInput = "printf ok";
|
||||
const lineText = `${prompt}${typedInput}`;
|
||||
const term = createFakeTerm(lineText, lineText.length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, typedInput, true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, prompt);
|
||||
assert.equal(result.prompt.userInput, typedInput);
|
||||
assert.equal(result.alignedTyped, typedInput);
|
||||
});
|
||||
|
||||
test("aligns typed input after a no-space root prompt when shell echo lags", () => {
|
||||
const prompt = " root@stwo:~#";
|
||||
const typedInput = "printf ok";
|
||||
const echoedInput = typedInput.slice(0, -1);
|
||||
const lineText = `${prompt}${echoedInput}`;
|
||||
const term = createFakeTerm(lineText, lineText.length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, typedInput, true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, prompt);
|
||||
assert.equal(result.prompt.userInput, typedInput);
|
||||
assert.equal(result.alignedTyped, typedInput);
|
||||
});
|
||||
|
||||
test("aligns typed input after a no-space root prompt when shell echo lags by a word", () => {
|
||||
const prompt = " root@stwo:~#";
|
||||
const typedInput = "printf ok";
|
||||
const echoedInput = "printf ";
|
||||
const lineText = `${prompt}${echoedInput}`;
|
||||
const term = createFakeTerm(lineText, lineText.length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, typedInput, true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, prompt);
|
||||
assert.equal(result.prompt.userInput, typedInput);
|
||||
assert.equal(result.alignedTyped, typedInput);
|
||||
});
|
||||
|
||||
test("aligns typed input after a no-space root prompt when a longer command echo lags by a word", () => {
|
||||
const prompt = "root@host:~#";
|
||||
const typedInput = "git status";
|
||||
const echoedInput = "git ";
|
||||
const lineText = `${prompt}${echoedInput}`;
|
||||
const term = createFakeTerm(lineText, lineText.length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, typedInput, true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, prompt);
|
||||
assert.equal(result.prompt.userInput, typedInput);
|
||||
assert.equal(result.alignedTyped, typedInput);
|
||||
});
|
||||
|
||||
test("aligns typed input after a no-space root prompt when command echo lags mid-word", () => {
|
||||
const prompt = "root@host:~#";
|
||||
const typedInput = "git status";
|
||||
const echoedInput = "git st";
|
||||
const lineText = `${prompt}${echoedInput}`;
|
||||
const term = createFakeTerm(lineText, lineText.length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, typedInput, true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, prompt);
|
||||
assert.equal(result.prompt.userInput, typedInput);
|
||||
assert.equal(result.alignedTyped, typedInput);
|
||||
});
|
||||
|
||||
test("aligns reliable typed input when standard prompt echo lags near completion", () => {
|
||||
const typedInput = "git status";
|
||||
const term = createFakeTerm("$ git statu", "$ git statu".length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, typedInput, true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, "$ ");
|
||||
assert.equal(result.prompt.userInput, typedInput);
|
||||
assert.equal(result.alignedTyped, typedInput);
|
||||
});
|
||||
|
||||
test("aligns reliable typed input when standard prompt echo lags after a word boundary", () => {
|
||||
const typedInput = "git status";
|
||||
const cases = ["$ git ", "$ git st"];
|
||||
|
||||
for (const lineText of cases) {
|
||||
const term = createFakeTerm(lineText, lineText.length);
|
||||
|
||||
const result = getAlignedPrompt(term as never, typedInput, true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true, lineText);
|
||||
assert.equal(result.prompt.promptText, "$ ", lineText);
|
||||
assert.equal(result.prompt.userInput, typedInput, lineText);
|
||||
assert.equal(result.alignedTyped, typedInput, lineText);
|
||||
}
|
||||
});
|
||||
|
||||
test("does not record partial standard prompt input while reliable typed input is still echoing", () => {
|
||||
const typedInput = "sudo";
|
||||
const term = createFakeTerm("$ s", "$ s".length);
|
||||
const result = getAlignedPrompt(term as never, typedInput, true);
|
||||
|
||||
assert.equal(result.prompt.isAtPrompt, true);
|
||||
assert.equal(result.prompt.promptText, "$ ");
|
||||
assert.equal(result.prompt.userInput, "s");
|
||||
assert.equal(result.alignedTyped, null);
|
||||
assert.equal(
|
||||
getCommandToRecordOnEnter(result.prompt, result.alignedTyped, typedInput, true),
|
||||
null,
|
||||
);
|
||||
});
|
||||
|
||||
test("CMD path prompts keep pre-echo Chinese input usable for snippet matching (#2813)", () => {
|
||||
const prompts = [
|
||||
String.raw`C:\Users\foo>`,
|
||||
String.raw`C:\Users\用户>`,
|
||||
String.raw`PS C:\Users\foo> `,
|
||||
String.raw`PS C:\Users\用户> `,
|
||||
];
|
||||
const typedInput = "部署";
|
||||
const snippet = { id: "zh", label: "部署服务", command: "echo deploy" };
|
||||
|
||||
for (const prompt of prompts) {
|
||||
const term = createPaddedFakeTerm(prompt, stringCellWidth(prompt));
|
||||
const raw = detectPrompt(term as never);
|
||||
assert.equal(raw.isAtPrompt, true, prompt);
|
||||
assert.equal(raw.userInput.trim(), "", prompt);
|
||||
|
||||
const aligned = getAlignedPrompt(term as never, typedInput, true);
|
||||
assert.equal(aligned.prompt.isAtPrompt, true, prompt);
|
||||
assert.equal(aligned.prompt.userInput, typedInput, prompt);
|
||||
assert.equal(aligned.alignedTyped, null, prompt);
|
||||
assert.equal(aligned.allowExternalProviders, false, prompt);
|
||||
|
||||
const query = resolveAutocompleteQueryInput(
|
||||
aligned.prompt,
|
||||
typedInput,
|
||||
true,
|
||||
);
|
||||
assert.equal(query, typedInput, prompt);
|
||||
assert.equal(
|
||||
getSnippetSuggestions(query ?? "", [snippet as never], {})[0]?.snippet?.id,
|
||||
"zh",
|
||||
prompt,
|
||||
);
|
||||
}
|
||||
});
|
||||
|
||||
test("CMD path prompts with CJK directories do not absorb padding into echoed Chinese input", () => {
|
||||
const prompt = String.raw`C:\Users\用户>`;
|
||||
const typedInput = "部署";
|
||||
const content = `${prompt}${typedInput}`;
|
||||
const term = createPaddedFakeTerm(content, stringCellWidth(content));
|
||||
|
||||
const raw = detectPrompt(term as never);
|
||||
assert.equal(raw.isAtPrompt, true);
|
||||
assert.equal(raw.promptText, prompt);
|
||||
assert.equal(raw.userInput, typedInput);
|
||||
assert.equal(raw.cursorOffset, typedInput.length);
|
||||
|
||||
// Unreliable typed buffer must still match snippets from the live line
|
||||
// (fresh local CMD sessions often clear keystroke reliability on startup).
|
||||
const query = resolveAutocompleteQueryInput(raw, "", false);
|
||||
assert.equal(query, typedInput);
|
||||
assert.equal(
|
||||
getSnippetSuggestions(query ?? "", [{
|
||||
id: "zh",
|
||||
label: "部署服务",
|
||||
command: "echo deploy",
|
||||
} as never], {})[0]?.snippet?.id,
|
||||
"zh",
|
||||
);
|
||||
});
|
||||
Reference in New Issue
Block a user