62 lines
1.6 KiB
JavaScript
62 lines
1.6 KiB
JavaScript
import emojiRegex from "emoji-regex";
|
|
import {
|
|
// @ts-expect-error -- Private
|
|
_isFullWidth as isFullWidth,
|
|
// @ts-expect-error -- Private
|
|
_isWide as isWide,
|
|
} from "get-east-asian-width";
|
|
import { isNarrowEmojiCharacter } from "narrow-emojis";
|
|
|
|
const notAsciiRegex = /[^\x20-\x7F]/;
|
|
|
|
// Similar to https://github.com/sindresorhus/string-width
|
|
// We don't strip ansi, always treat ambiguous width characters as having narrow width.
|
|
/**
|
|
* @param {string} text
|
|
* @returns {number}
|
|
*/
|
|
function getStringWidth(text) {
|
|
if (!text) {
|
|
return 0;
|
|
}
|
|
|
|
// shortcut to avoid needless string `RegExp`s, replacements, and allocations
|
|
if (!notAsciiRegex.test(text)) {
|
|
return text.length;
|
|
}
|
|
|
|
let width = 0;
|
|
text = text.replace(emojiRegex(), (character) => {
|
|
width += isNarrowEmojiCharacter(character) ? 1 : 2;
|
|
return "";
|
|
});
|
|
|
|
// Use `Intl.Segmenter` when we drop support for Node.js v14
|
|
// https://github.com/prettier/prettier/pull/14793#discussion_r1185840038
|
|
// https://github.com/sindresorhus/string-width/pull/47
|
|
for (const character of text) {
|
|
const codePoint = character.codePointAt(0);
|
|
|
|
// Ignore control characters
|
|
if (codePoint <= 0x1f || (codePoint >= 0x7f && codePoint <= 0x9f)) {
|
|
continue;
|
|
}
|
|
|
|
// Ignore combining characters
|
|
if (codePoint >= 0x300 && codePoint <= 0x36f) {
|
|
continue;
|
|
}
|
|
|
|
// Ignore Variation Selectors
|
|
if (codePoint >= 0xfe00 && codePoint <= 0xfe0f) {
|
|
continue;
|
|
}
|
|
|
|
width += isFullWidth(codePoint) || isWide(codePoint) ? 2 : 1;
|
|
}
|
|
|
|
return width;
|
|
}
|
|
|
|
export default getStringWidth;
|