alexion
6081e3e325
Imported as a linear vendor commit so forge rebases do not replay the upstream root tree. git-subtree-dir: third_party/pi git-subtree-split: 71023aec1546d6d677a65669c5665c0927ce0dd0
83 lines
3.5 KiB
TypeScript
83 lines
3.5 KiB
TypeScript
/*
|
|
* Portions of this file are derived from:
|
|
* - ansi-regex (https://github.com/chalk/ansi-regex)
|
|
* - strip-ansi (https://github.com/chalk/strip-ansi)
|
|
*
|
|
* MIT License
|
|
*
|
|
* Copyright (c) Sindre Sorhus <sindresorhus@gmail.com> (https://sindresorhus.com)
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in all
|
|
* copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
* SOFTWARE.
|
|
*/
|
|
|
|
// Valid string terminator sequences are BEL, ESC\, and 0x9c
|
|
const ST = "(?:\\u0007|\\u001B\\u005C|\\u009C)";
|
|
|
|
// OSC sequences: ESC ] ... ST
|
|
const OSC_START = "\\u001B\\]";
|
|
|
|
// CSI and related: ESC/C1, optional intermediates, optional params (supports ; and :), then final byte
|
|
const CSI_START = "[\\u001B\\u009B][[\\]()#;?]*(?:\\d{1,4}(?:[;:]\\d{0,4})*)?";
|
|
const CSI_FINAL = "[\\dA-PR-TZcf-nq-uy=><~]";
|
|
|
|
// Complete sequences. OSC is non-greedy until the first ST.
|
|
const ansiRegex = new RegExp(`(?:${OSC_START}[\\s\\S]*?${ST})|${CSI_START}${CSI_FINAL}`, "g");
|
|
|
|
// Unfinished sequence at the end of the text: OSC without its ST (a trailing ESC may start ESC\),
|
|
// or CSI without its final byte.
|
|
const unfinishedAnsiAtEndRegex = new RegExp(
|
|
`(?:${OSC_START}(?:[^\\u0007\\u009C\\u001B]|\\u001B(?!\\\\))*|${CSI_START})$`,
|
|
);
|
|
|
|
// Longest unfinished sequence held back while streaming. Longer ones are processed as-is.
|
|
const MAX_PENDING_ANSI_LENGTH = 256;
|
|
|
|
/**
|
|
* Split streamed text into a part that is safe to pass to stripAnsi now and a trailing
|
|
* unfinished escape sequence that should be prepended to the next chunk.
|
|
*/
|
|
export function splitIncompleteAnsiSuffix(value: string): { complete: string; pending: string } {
|
|
if (!value.includes("\u001B") && !value.includes("\u009B")) {
|
|
return { complete: value, pending: "" };
|
|
}
|
|
const windowStart = Math.max(0, value.length - MAX_PENDING_ANSI_LENGTH);
|
|
const match = unfinishedAnsiAtEndRegex.exec(value.slice(windowStart));
|
|
if (!match) {
|
|
return { complete: value, pending: "" };
|
|
}
|
|
const splitAt = windowStart + match.index;
|
|
return { complete: value.slice(0, splitAt), pending: value.slice(splitAt) };
|
|
}
|
|
|
|
export function stripAnsi(value: string): string {
|
|
if (typeof value !== "string") {
|
|
throw new TypeError(`Expected a \`string\`, got \`${typeof value}\``);
|
|
}
|
|
|
|
// Fast path: ANSI codes require ESC (7-bit) or CSI (8-bit) introducer
|
|
if (!value.includes("\u001B") && !value.includes("\u009B")) {
|
|
return value;
|
|
}
|
|
|
|
// Even though the regex is global, we don't need to reset the `.lastIndex`
|
|
// because unlike `.exec()` and `.test()`, `.replace()` does it automatically
|
|
// and doing it manually has a performance penalty.
|
|
return value.replace(ansiRegex, "");
|
|
}
|