fix(coding-agent): align stripAnsi with strip-ansi
This commit is contained in:
@@ -1,7 +1,95 @@
|
||||
import { describe, expect, it } from "vitest";
|
||||
import { stripAnsi } from "../src/utils/ansi.js";
|
||||
|
||||
function referenceAnsiRegex(): RegExp {
|
||||
const ST = "(?:\\u0007|\\u001B\\u005C|\\u009C)";
|
||||
const osc = `(?:\\u001B\\][\\s\\S]*?${ST})`;
|
||||
const csi = "[\\u001B\\u009B][[\\]()#;?]*(?:\\d{1,4}(?:[;:]\\d{0,4})*)?[\\dA-PR-TZcf-nq-uy=><~]";
|
||||
return new RegExp(`${osc}|${csi}`, "g");
|
||||
}
|
||||
|
||||
const referenceRegex = referenceAnsiRegex();
|
||||
|
||||
function referenceStripAnsi(value: string): string {
|
||||
if (!value.includes("\u001B") && !value.includes("\u009B")) {
|
||||
return value;
|
||||
}
|
||||
return value.replace(referenceRegex, "");
|
||||
}
|
||||
|
||||
function getCompatibilityInputs(): string[] {
|
||||
const inputs = [
|
||||
"plain",
|
||||
"a\x1b[31mred\x1b[0mz",
|
||||
"a\x1b]8;;https://example.com\x07link\x1b]8;;\x07z",
|
||||
"a\x1b]unterminated",
|
||||
"a\x1b]funterminated",
|
||||
"a\x1bPabc\x1b\\z",
|
||||
"a\x1b^abc\x07z",
|
||||
"a\x1b_abc\x9cz",
|
||||
"a\x90abc\x9cz",
|
||||
"a\x9dabc\x9cz",
|
||||
"a\x9b31mred",
|
||||
"a\x1b(0x",
|
||||
"a\x1b*0x",
|
||||
"a\x1b+c",
|
||||
"a\x1b/0x",
|
||||
"a\x1bcok",
|
||||
"a\x1b\\ok",
|
||||
];
|
||||
const chars = [
|
||||
"a",
|
||||
"f",
|
||||
"0",
|
||||
"1",
|
||||
";",
|
||||
":",
|
||||
"[",
|
||||
"]",
|
||||
"(",
|
||||
")",
|
||||
"#",
|
||||
"?",
|
||||
"m",
|
||||
"P",
|
||||
"_",
|
||||
"\\",
|
||||
"\x07",
|
||||
"\x1b",
|
||||
"\x9b",
|
||||
"\x9c",
|
||||
"\x90",
|
||||
"\x9d",
|
||||
];
|
||||
|
||||
for (const char of chars) {
|
||||
inputs.push(`x\x1b${char}y`);
|
||||
inputs.push(`x\x9b${char}y`);
|
||||
for (let index = 0; index < chars.length; index += 3) {
|
||||
inputs.push(`x\x1b${char}${chars[index]}y`);
|
||||
}
|
||||
}
|
||||
|
||||
return inputs;
|
||||
}
|
||||
|
||||
describe("stripAnsi", () => {
|
||||
it("matches chalk strip-ansi for generated compatibility inputs", () => {
|
||||
for (const input of getCompatibilityInputs()) {
|
||||
expect(stripAnsi(input)).toBe(referenceStripAnsi(input));
|
||||
}
|
||||
});
|
||||
|
||||
it("throws the same TypeError as chalk strip-ansi for non-string values", () => {
|
||||
const stripAnsiUnknown = stripAnsi as (value: unknown) => string;
|
||||
|
||||
for (const value of [undefined, null, 123, {}, Object("x")]) {
|
||||
const message = `Expected a \`string\`, got \`${typeof value}\``;
|
||||
expect(() => stripAnsiUnknown(value)).toThrow(TypeError);
|
||||
expect(() => stripAnsiUnknown(value)).toThrow(message);
|
||||
}
|
||||
});
|
||||
|
||||
it("strips RIS without leaking the final byte", () => {
|
||||
expect(stripAnsi("\x1bcdone")).toBe("done");
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user