123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960 |
- 'use strict';
- const isFullwidthCodePoint = require('is-fullwidth-code-point');
- const astralRegex = require('astral-regex');
- const ansiStyles = require('ansi-styles');
- const ESCAPES = [
- '\u001B',
- '\u009B'
- ];
- const END_CODE = 39;
- const wrapAnsi = code => `${ESCAPES[0]}[${code}m`;
- module.exports = (str, begin, end) => {
- const arr = [...str.normalize()];
- end = typeof end === 'number' ? end : arr.length;
- let insideEscape = false;
- let escapeCode = null;
- let visible = 0;
- let output = '';
- for (const [i, x] of arr.entries()) {
- let leftEscape = false;
- if (ESCAPES.includes(x)) {
- insideEscape = true;
- const code = /\d[^m]*/.exec(str.slice(i, i + 18));
- escapeCode = code === END_CODE ? null : code;
- } else if (insideEscape && x === 'm') {
- insideEscape = false;
- leftEscape = true;
- }
- if (!insideEscape && !leftEscape) {
- ++visible;
- }
- if (!astralRegex({exact: true}).test(x) && isFullwidthCodePoint(x.codePointAt())) {
- ++visible;
- }
- if (visible > begin && visible <= end) {
- output += x;
- } else if (visible === begin && !insideEscape && escapeCode !== null && escapeCode !== END_CODE) {
- output += wrapAnsi(escapeCode);
- } else if (visible >= end) {
- if (escapeCode !== null) {
- output += wrapAnsi(ansiStyles.codes.get(parseInt(escapeCode, 10)) || END_CODE);
- }
- break;
- }
- }
- return output;
- };
|