Spaces:
Running
Running
; | |
const stringWidth = require('string-width'); | |
const stripAnsi = require('strip-ansi'); | |
const ansiStyles = require('ansi-styles'); | |
const ESCAPES = new Set([ | |
'\u001B', | |
'\u009B' | |
]); | |
const END_CODE = 39; | |
const wrapAnsi = code => `${ESCAPES.values().next().value}[${code}m`; | |
// Calculate the length of words split on ' ', ignoring | |
// the extra characters added by ansi escape codes | |
const wordLengths = string => string.split(' ').map(character => stringWidth(character)); | |
// Wrap a long word across multiple rows | |
// Ansi escape codes do not count towards length | |
const wrapWord = (rows, word, columns) => { | |
const characters = [...word]; | |
let isInsideEscape = false; | |
let visible = stringWidth(stripAnsi(rows[rows.length - 1])); | |
for (const [index, character] of characters.entries()) { | |
const characterLength = stringWidth(character); | |
if (visible + characterLength <= columns) { | |
rows[rows.length - 1] += character; | |
} else { | |
rows.push(character); | |
visible = 0; | |
} | |
if (ESCAPES.has(character)) { | |
isInsideEscape = true; | |
} else if (isInsideEscape && character === 'm') { | |
isInsideEscape = false; | |
continue; | |
} | |
if (isInsideEscape) { | |
continue; | |
} | |
visible += characterLength; | |
if (visible === columns && index < characters.length - 1) { | |
rows.push(''); | |
visible = 0; | |
} | |
} | |
// It's possible that the last row we copy over is only | |
// ansi escape characters, handle this edge-case | |
if (!visible && rows[rows.length - 1].length > 0 && rows.length > 1) { | |
rows[rows.length - 2] += rows.pop(); | |
} | |
}; | |
// Trims spaces from a string ignoring invisible sequences | |
const stringVisibleTrimSpacesRight = str => { | |
const words = str.split(' '); | |
let last = words.length; | |
while (last > 0) { | |
if (stringWidth(words[last - 1]) > 0) { | |
break; | |
} | |
last--; | |
} | |
if (last === words.length) { | |
return str; | |
} | |
return words.slice(0, last).join(' ') + words.slice(last).join(''); | |
}; | |
// The wrap-ansi module can be invoked in either 'hard' or 'soft' wrap mode | |
// | |
// 'hard' will never allow a string to take up more than columns characters | |
// | |
// 'soft' allows long words to expand past the column length | |
const exec = (string, columns, options = {}) => { | |
if (options.trim !== false && string.trim() === '') { | |
return ''; | |
} | |
let pre = ''; | |
let ret = ''; | |
let escapeCode; | |
const lengths = wordLengths(string); | |
let rows = ['']; | |
for (const [index, word] of string.split(' ').entries()) { | |
if (options.trim !== false) { | |
rows[rows.length - 1] = rows[rows.length - 1].trimLeft(); | |
} | |
let rowLength = stringWidth(rows[rows.length - 1]); | |
if (index !== 0) { | |
if (rowLength >= columns && (options.wordWrap === false || options.trim === false)) { | |
// If we start with a new word but the current row length equals the length of the columns, add a new row | |
rows.push(''); | |
rowLength = 0; | |
} | |
if (rowLength > 0 || options.trim === false) { | |
rows[rows.length - 1] += ' '; | |
rowLength++; | |
} | |
} | |
// In 'hard' wrap mode, the length of a line is never allowed to extend past 'columns' | |
if (options.hard && lengths[index] > columns) { | |
const remainingColumns = (columns - rowLength); | |
const breaksStartingThisLine = 1 + Math.floor((lengths[index] - remainingColumns - 1) / columns); | |
const breaksStartingNextLine = Math.floor((lengths[index] - 1) / columns); | |
if (breaksStartingNextLine < breaksStartingThisLine) { | |
rows.push(''); | |
} | |
wrapWord(rows, word, columns); | |
continue; | |
} | |
if (rowLength + lengths[index] > columns && rowLength > 0 && lengths[index] > 0) { | |
if (options.wordWrap === false && rowLength < columns) { | |
wrapWord(rows, word, columns); | |
continue; | |
} | |
rows.push(''); | |
} | |
if (rowLength + lengths[index] > columns && options.wordWrap === false) { | |
wrapWord(rows, word, columns); | |
continue; | |
} | |
rows[rows.length - 1] += word; | |
} | |
if (options.trim !== false) { | |
rows = rows.map(stringVisibleTrimSpacesRight); | |
} | |
pre = rows.join('\n'); | |
for (const [index, character] of [...pre].entries()) { | |
ret += character; | |
if (ESCAPES.has(character)) { | |
const code = parseFloat(/\d[^m]*/.exec(pre.slice(index, index + 4))); | |
escapeCode = code === END_CODE ? null : code; | |
} | |
const code = ansiStyles.codes.get(Number(escapeCode)); | |
if (escapeCode && code) { | |
if (pre[index + 1] === '\n') { | |
ret += wrapAnsi(code); | |
} else if (character === '\n') { | |
ret += wrapAnsi(escapeCode); | |
} | |
} | |
} | |
return ret; | |
}; | |
// For each newline, invoke the method separately | |
module.exports = (string, columns, options) => { | |
return String(string) | |
.normalize() | |
.replace(/\r\n/g, '\n') | |
.split('\n') | |
.map(line => exec(line, columns, options)) | |
.join('\n'); | |
}; | |