1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859 |
- '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;
- };
|