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