123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168 |
- 'use strict';
- var stringWidth = require('string-width');
- var stripAnsi = require('strip-ansi');
- var ESCAPES = [
- '\u001b',
- '\u009b'
- ];
- var END_CODE = 39;
- var ESCAPE_CODES = {
- 0: 0,
- 1: 22,
- 2: 22,
- 3: 23,
- 4: 24,
- 7: 27,
- 8: 28,
- 9: 29,
- 30: 39,
- 31: 39,
- 32: 39,
- 33: 39,
- 34: 39,
- 35: 39,
- 36: 39,
- 37: 39,
- 90: 39,
- 40: 49,
- 41: 49,
- 42: 49,
- 43: 49,
- 44: 49,
- 45: 49,
- 46: 49,
- 47: 49
- };
- function wrapAnsi(code) {
- return ESCAPES[0] + '[' + code + 'm';
- }
- function wordLengths(str) {
- return str.split(' ').map(function (s) {
- return stringWidth(s);
- });
- }
- function wrapWord(rows, word, cols) {
- var insideEscape = false;
- var visible = stripAnsi(rows[rows.length - 1]).length;
- for (var i = 0; i < word.length; i++) {
- var x = word[i];
- rows[rows.length - 1] += x;
- if (ESCAPES.indexOf(x) !== -1) {
- insideEscape = true;
- } else if (insideEscape && x === 'm') {
- insideEscape = false;
- continue;
- }
- if (insideEscape) {
- continue;
- }
- visible++;
- if (visible >= cols && i < word.length - 1) {
- rows.push('');
- visible = 0;
- }
- }
-
-
- if (!visible && rows[rows.length - 1].length > 0 && rows.length > 1) {
- rows[rows.length - 2] += rows.pop();
- }
- }
- function exec(str, cols, opts) {
- var options = opts || {};
- var pre = '';
- var ret = '';
- var escapeCode;
- var lengths = wordLengths(str);
- var words = str.split(' ');
- var rows = [''];
- for (var i = 0, word; (word = words[i]) !== undefined; i++) {
- var rowLength = stringWidth(rows[rows.length - 1]);
- if (rowLength) {
- rows[rows.length - 1] += ' ';
- rowLength++;
- }
-
-
- if (lengths[i] > cols && options.hard) {
- if (rowLength) {
- rows.push('');
- }
- wrapWord(rows, word, cols);
- continue;
- }
- if (rowLength + lengths[i] > cols && rowLength > 0) {
- if (options.wordWrap === false && rowLength < cols) {
- wrapWord(rows, word, cols);
- continue;
- }
- rows.push('');
- }
- rows[rows.length - 1] += word;
- }
- pre = rows.map(function (r) {
- return r.trim();
- }).join('\n');
- for (var j = 0; j < pre.length; j++) {
- var y = pre[j];
- ret += y;
- if (ESCAPES.indexOf(y) !== -1) {
- var code = parseFloat(/[0-9][^m]*/.exec(pre.slice(j, j + 4)));
- escapeCode = code === END_CODE ? null : code;
- }
- if (escapeCode && ESCAPE_CODES[escapeCode]) {
- if (pre[j + 1] === '\n') {
- ret += wrapAnsi(ESCAPE_CODES[escapeCode]);
- } else if (y === '\n') {
- ret += wrapAnsi(escapeCode);
- }
- }
- }
- return ret;
- }
- module.exports = function (str, cols, opts) {
- return String(str).split('\n').map(function (substr) {
- return exec(substr, cols, opts);
- }).join('\n');
- };
|