1 | 'use strict';
|
2 | const stringWidth = require('string-width');
|
3 | const stripAnsi = require('strip-ansi');
|
4 | const ansiStyles = require('ansi-styles');
|
5 |
|
6 | const ESCAPES = new Set([
|
7 | '\u001B',
|
8 | '\u009B'
|
9 | ]);
|
10 |
|
11 | const END_CODE = 39;
|
12 |
|
13 | const ANSI_ESCAPE_BELL = '\u0007';
|
14 | const ANSI_CSI = '[';
|
15 | const ANSI_OSC = ']';
|
16 | const ANSI_SGR_TERMINATOR = 'm';
|
17 | const ANSI_ESCAPE_LINK = `${ANSI_OSC}8;;`;
|
18 |
|
19 | const wrapAnsi = code => `${ESCAPES.values().next().value}${ANSI_CSI}${code}${ANSI_SGR_TERMINATOR}`;
|
20 | const wrapAnsiHyperlink = uri => `${ESCAPES.values().next().value}${ANSI_ESCAPE_LINK}${uri}${ANSI_ESCAPE_BELL}`;
|
21 |
|
22 |
|
23 |
|
24 | const wordLengths = string => string.split(' ').map(character => stringWidth(character));
|
25 |
|
26 |
|
27 |
|
28 | const wrapWord = (rows, word, columns) => {
|
29 | const characters = [...word];
|
30 |
|
31 | let isInsideEscape = false;
|
32 | let isInsideLinkEscape = false;
|
33 | let visible = stringWidth(stripAnsi(rows[rows.length - 1]));
|
34 |
|
35 | for (const [index, character] of characters.entries()) {
|
36 | const characterLength = stringWidth(character);
|
37 |
|
38 | if (visible + characterLength <= columns) {
|
39 | rows[rows.length - 1] += character;
|
40 | } else {
|
41 | rows.push(character);
|
42 | visible = 0;
|
43 | }
|
44 |
|
45 | if (ESCAPES.has(character)) {
|
46 | isInsideEscape = true;
|
47 | isInsideLinkEscape = characters.slice(index + 1).join('').startsWith(ANSI_ESCAPE_LINK);
|
48 | }
|
49 |
|
50 | if (isInsideEscape) {
|
51 | if (isInsideLinkEscape) {
|
52 | if (character === ANSI_ESCAPE_BELL) {
|
53 | isInsideEscape = false;
|
54 | isInsideLinkEscape = false;
|
55 | }
|
56 | } else if (character === ANSI_SGR_TERMINATOR) {
|
57 | isInsideEscape = false;
|
58 | }
|
59 |
|
60 | continue;
|
61 | }
|
62 |
|
63 | visible += characterLength;
|
64 |
|
65 | if (visible === columns && index < characters.length - 1) {
|
66 | rows.push('');
|
67 | visible = 0;
|
68 | }
|
69 | }
|
70 |
|
71 |
|
72 |
|
73 | if (!visible && rows[rows.length - 1].length > 0 && rows.length > 1) {
|
74 | rows[rows.length - 2] += rows.pop();
|
75 | }
|
76 | };
|
77 |
|
78 |
|
79 | const stringVisibleTrimSpacesRight = string => {
|
80 | const words = string.split(' ');
|
81 | let last = words.length;
|
82 |
|
83 | while (last > 0) {
|
84 | if (stringWidth(words[last - 1]) > 0) {
|
85 | break;
|
86 | }
|
87 |
|
88 | last--;
|
89 | }
|
90 |
|
91 | if (last === words.length) {
|
92 | return string;
|
93 | }
|
94 |
|
95 | return words.slice(0, last).join(' ') + words.slice(last).join('');
|
96 | };
|
97 |
|
98 |
|
99 |
|
100 |
|
101 |
|
102 |
|
103 | const exec = (string, columns, options = {}) => {
|
104 | if (options.trim !== false && string.trim() === '') {
|
105 | return '';
|
106 | }
|
107 |
|
108 | let returnValue = '';
|
109 | let escapeCode;
|
110 | let escapeUrl;
|
111 |
|
112 | const lengths = wordLengths(string);
|
113 | let rows = [''];
|
114 |
|
115 | for (const [index, word] of string.split(' ').entries()) {
|
116 | if (options.trim !== false) {
|
117 | rows[rows.length - 1] = rows[rows.length - 1].trimStart();
|
118 | }
|
119 |
|
120 | let rowLength = stringWidth(rows[rows.length - 1]);
|
121 |
|
122 | if (index !== 0) {
|
123 | if (rowLength >= columns && (options.wordWrap === false || options.trim === false)) {
|
124 |
|
125 | rows.push('');
|
126 | rowLength = 0;
|
127 | }
|
128 |
|
129 | if (rowLength > 0 || options.trim === false) {
|
130 | rows[rows.length - 1] += ' ';
|
131 | rowLength++;
|
132 | }
|
133 | }
|
134 |
|
135 |
|
136 | if (options.hard && lengths[index] > columns) {
|
137 | const remainingColumns = (columns - rowLength);
|
138 | const breaksStartingThisLine = 1 + Math.floor((lengths[index] - remainingColumns - 1) / columns);
|
139 | const breaksStartingNextLine = Math.floor((lengths[index] - 1) / columns);
|
140 | if (breaksStartingNextLine < breaksStartingThisLine) {
|
141 | rows.push('');
|
142 | }
|
143 |
|
144 | wrapWord(rows, word, columns);
|
145 | continue;
|
146 | }
|
147 |
|
148 | if (rowLength + lengths[index] > columns && rowLength > 0 && lengths[index] > 0) {
|
149 | if (options.wordWrap === false && rowLength < columns) {
|
150 | wrapWord(rows, word, columns);
|
151 | continue;
|
152 | }
|
153 |
|
154 | rows.push('');
|
155 | }
|
156 |
|
157 | if (rowLength + lengths[index] > columns && options.wordWrap === false) {
|
158 | wrapWord(rows, word, columns);
|
159 | continue;
|
160 | }
|
161 |
|
162 | rows[rows.length - 1] += word;
|
163 | }
|
164 |
|
165 | if (options.trim !== false) {
|
166 | rows = rows.map(stringVisibleTrimSpacesRight);
|
167 | }
|
168 |
|
169 | const pre = [...rows.join('\n')];
|
170 |
|
171 | for (const [index, character] of pre.entries()) {
|
172 | returnValue += character;
|
173 |
|
174 | if (ESCAPES.has(character)) {
|
175 | const {groups} = new RegExp(`(?:\\${ANSI_CSI}(?<code>\\d+)m|\\${ANSI_ESCAPE_LINK}(?<uri>.*)${ANSI_ESCAPE_BELL})`).exec(pre.slice(index).join('')) || {groups: {}};
|
176 | if (groups.code !== undefined) {
|
177 | const code = Number.parseFloat(groups.code);
|
178 | escapeCode = code === END_CODE ? undefined : code;
|
179 | } else if (groups.uri !== undefined) {
|
180 | escapeUrl = groups.uri.length === 0 ? undefined : groups.uri;
|
181 | }
|
182 | }
|
183 |
|
184 | const code = ansiStyles.codes.get(Number(escapeCode));
|
185 |
|
186 | if (pre[index + 1] === '\n') {
|
187 | if (escapeUrl) {
|
188 | returnValue += wrapAnsiHyperlink('');
|
189 | }
|
190 |
|
191 | if (escapeCode && code) {
|
192 | returnValue += wrapAnsi(code);
|
193 | }
|
194 | } else if (character === '\n') {
|
195 | if (escapeCode && code) {
|
196 | returnValue += wrapAnsi(escapeCode);
|
197 | }
|
198 |
|
199 | if (escapeUrl) {
|
200 | returnValue += wrapAnsiHyperlink(escapeUrl);
|
201 | }
|
202 | }
|
203 | }
|
204 |
|
205 | return returnValue;
|
206 | };
|
207 |
|
208 |
|
209 | module.exports = (string, columns, options) => {
|
210 | return String(string)
|
211 | .normalize()
|
212 | .replace(/\r\n/g, '\n')
|
213 | .split('\n')
|
214 | .map(line => exec(line, columns, options))
|
215 | .join('\n');
|
216 | };
|