ansi-styles and support-color correction
This commit is contained in:
parent
b17d862944
commit
68030d79be
1 changed files with 65 additions and 42 deletions
107
source/index.js
107
source/index.js
|
|
@ -1,29 +1,32 @@
|
||||||
import ansiStyles from '#ansi-styles';
|
import ansiStyles from "ansi-styles";
|
||||||
import supportsColor from '#supports-color';
|
import supportsColor from "supports-color";
|
||||||
import { // eslint-disable-line import/order
|
import {
|
||||||
|
// eslint-disable-line import/order
|
||||||
stringReplaceAll,
|
stringReplaceAll,
|
||||||
stringEncaseCRLFWithFirstIndex,
|
stringEncaseCRLFWithFirstIndex,
|
||||||
} from './utilities.js';
|
} from "./utilities.js";
|
||||||
|
|
||||||
const {stdout: stdoutColor, stderr: stderrColor} = supportsColor;
|
const { stdout: stdoutColor, stderr: stderrColor } = supportsColor;
|
||||||
|
|
||||||
const GENERATOR = Symbol('GENERATOR');
|
const GENERATOR = Symbol("GENERATOR");
|
||||||
const STYLER = Symbol('STYLER');
|
const STYLER = Symbol("STYLER");
|
||||||
const IS_EMPTY = Symbol('IS_EMPTY');
|
const IS_EMPTY = Symbol("IS_EMPTY");
|
||||||
|
|
||||||
// `supportsColor.level` → `ansiStyles.color[name]` mapping
|
// `supportsColor.level` → `ansiStyles.color[name]` mapping
|
||||||
const levelMapping = [
|
const levelMapping = ["ansi", "ansi", "ansi256", "ansi16m"];
|
||||||
'ansi',
|
|
||||||
'ansi',
|
|
||||||
'ansi256',
|
|
||||||
'ansi16m',
|
|
||||||
];
|
|
||||||
|
|
||||||
const styles = Object.create(null);
|
const styles = Object.create(null);
|
||||||
|
|
||||||
const applyOptions = (object, options = {}) => {
|
const applyOptions = (object, options = {}) => {
|
||||||
if (options.level && !(Number.isInteger(options.level) && options.level >= 0 && options.level <= 3)) {
|
if (
|
||||||
throw new Error('The `level` option should be an integer from 0 to 3');
|
options.level &&
|
||||||
|
!(
|
||||||
|
Number.isInteger(options.level) &&
|
||||||
|
options.level >= 0 &&
|
||||||
|
options.level <= 3
|
||||||
|
)
|
||||||
|
) {
|
||||||
|
throw new Error("The `level` option should be an integer from 0 to 3");
|
||||||
}
|
}
|
||||||
|
|
||||||
// Detect level if not set manually
|
// Detect level if not set manually
|
||||||
|
|
@ -38,8 +41,8 @@ export class Chalk {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const chalkFactory = options => {
|
const chalkFactory = (options) => {
|
||||||
const chalk = (...strings) => strings.join(' ');
|
const chalk = (...strings) => strings.join(" ");
|
||||||
applyOptions(chalk, options);
|
applyOptions(chalk, options);
|
||||||
|
|
||||||
Object.setPrototypeOf(chalk, createChalk.prototype);
|
Object.setPrototypeOf(chalk, createChalk.prototype);
|
||||||
|
|
@ -56,8 +59,12 @@ Object.setPrototypeOf(createChalk.prototype, Function.prototype);
|
||||||
for (const [styleName, style] of Object.entries(ansiStyles)) {
|
for (const [styleName, style] of Object.entries(ansiStyles)) {
|
||||||
styles[styleName] = {
|
styles[styleName] = {
|
||||||
get() {
|
get() {
|
||||||
const builder = createBuilder(this, createStyler(style.open, style.close, this[STYLER]), this[IS_EMPTY]);
|
const builder = createBuilder(
|
||||||
Object.defineProperty(this, styleName, {value: builder});
|
this,
|
||||||
|
createStyler(style.open, style.close, this[STYLER]),
|
||||||
|
this[IS_EMPTY]
|
||||||
|
);
|
||||||
|
Object.defineProperty(this, styleName, { value: builder });
|
||||||
return builder;
|
return builder;
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
@ -66,50 +73,63 @@ for (const [styleName, style] of Object.entries(ansiStyles)) {
|
||||||
styles.visible = {
|
styles.visible = {
|
||||||
get() {
|
get() {
|
||||||
const builder = createBuilder(this, this[STYLER], true);
|
const builder = createBuilder(this, this[STYLER], true);
|
||||||
Object.defineProperty(this, 'visible', {value: builder});
|
Object.defineProperty(this, "visible", { value: builder });
|
||||||
return builder;
|
return builder;
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
const getModelAnsi = (model, level, type, ...arguments_) => {
|
const getModelAnsi = (model, level, type, ...arguments_) => {
|
||||||
if (model === 'rgb') {
|
if (model === "rgb") {
|
||||||
if (level === 'ansi16m') {
|
if (level === "ansi16m") {
|
||||||
return ansiStyles[type].ansi16m(...arguments_);
|
return ansiStyles[type].ansi16m(...arguments_);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (level === 'ansi256') {
|
if (level === "ansi256") {
|
||||||
return ansiStyles[type].ansi256(ansiStyles.rgbToAnsi256(...arguments_));
|
return ansiStyles[type].ansi256(ansiStyles.rgbToAnsi256(...arguments_));
|
||||||
}
|
}
|
||||||
|
|
||||||
return ansiStyles[type].ansi(ansiStyles.rgbToAnsi(...arguments_));
|
return ansiStyles[type].ansi(ansiStyles.rgbToAnsi(...arguments_));
|
||||||
}
|
}
|
||||||
|
|
||||||
if (model === 'hex') {
|
if (model === "hex") {
|
||||||
return getModelAnsi('rgb', level, type, ...ansiStyles.hexToRgb(...arguments_));
|
return getModelAnsi(
|
||||||
|
"rgb",
|
||||||
|
level,
|
||||||
|
type,
|
||||||
|
...ansiStyles.hexToRgb(...arguments_)
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
return ansiStyles[type][model](...arguments_);
|
return ansiStyles[type][model](...arguments_);
|
||||||
};
|
};
|
||||||
|
|
||||||
const usedModels = ['rgb', 'hex', 'ansi256'];
|
const usedModels = ["rgb", "hex", "ansi256"];
|
||||||
|
|
||||||
for (const model of usedModels) {
|
for (const model of usedModels) {
|
||||||
styles[model] = {
|
styles[model] = {
|
||||||
get() {
|
get() {
|
||||||
const {level} = this;
|
const { level } = this;
|
||||||
return function (...arguments_) {
|
return function (...arguments_) {
|
||||||
const styler = createStyler(getModelAnsi(model, levelMapping[level], 'color', ...arguments_), ansiStyles.color.close, this[STYLER]);
|
const styler = createStyler(
|
||||||
|
getModelAnsi(model, levelMapping[level], "color", ...arguments_),
|
||||||
|
ansiStyles.color.close,
|
||||||
|
this[STYLER]
|
||||||
|
);
|
||||||
return createBuilder(this, styler, this[IS_EMPTY]);
|
return createBuilder(this, styler, this[IS_EMPTY]);
|
||||||
};
|
};
|
||||||
},
|
},
|
||||||
};
|
};
|
||||||
|
|
||||||
const bgModel = 'bg' + model[0].toUpperCase() + model.slice(1);
|
const bgModel = "bg" + model[0].toUpperCase() + model.slice(1);
|
||||||
styles[bgModel] = {
|
styles[bgModel] = {
|
||||||
get() {
|
get() {
|
||||||
const {level} = this;
|
const { level } = this;
|
||||||
return function (...arguments_) {
|
return function (...arguments_) {
|
||||||
const styler = createStyler(getModelAnsi(model, levelMapping[level], 'bgColor', ...arguments_), ansiStyles.bgColor.close, this[STYLER]);
|
const styler = createStyler(
|
||||||
|
getModelAnsi(model, levelMapping[level], "bgColor", ...arguments_),
|
||||||
|
ansiStyles.bgColor.close,
|
||||||
|
this[STYLER]
|
||||||
|
);
|
||||||
return createBuilder(this, styler, this[IS_EMPTY]);
|
return createBuilder(this, styler, this[IS_EMPTY]);
|
||||||
};
|
};
|
||||||
},
|
},
|
||||||
|
|
@ -152,7 +172,11 @@ const createStyler = (open, close, parent) => {
|
||||||
const createBuilder = (self, _styler, _isEmpty) => {
|
const createBuilder = (self, _styler, _isEmpty) => {
|
||||||
// Single argument is hot path, implicit coercion is faster than anything
|
// Single argument is hot path, implicit coercion is faster than anything
|
||||||
// eslint-disable-next-line no-implicit-coercion
|
// eslint-disable-next-line no-implicit-coercion
|
||||||
const builder = (...arguments_) => applyStyle(builder, (arguments_.length === 1) ? ('' + arguments_[0]) : arguments_.join(' '));
|
const builder = (...arguments_) =>
|
||||||
|
applyStyle(
|
||||||
|
builder,
|
||||||
|
arguments_.length === 1 ? "" + arguments_[0] : arguments_.join(" ")
|
||||||
|
);
|
||||||
|
|
||||||
// We alter the prototype because we must return a function, but there is
|
// We alter the prototype because we must return a function, but there is
|
||||||
// no way to create a function with a different prototype
|
// no way to create a function with a different prototype
|
||||||
|
|
@ -167,7 +191,7 @@ const createBuilder = (self, _styler, _isEmpty) => {
|
||||||
|
|
||||||
const applyStyle = (self, string) => {
|
const applyStyle = (self, string) => {
|
||||||
if (self.level <= 0 || !string) {
|
if (self.level <= 0 || !string) {
|
||||||
return self[IS_EMPTY] ? '' : string;
|
return self[IS_EMPTY] ? "" : string;
|
||||||
}
|
}
|
||||||
|
|
||||||
let styler = self[STYLER];
|
let styler = self[STYLER];
|
||||||
|
|
@ -176,8 +200,8 @@ const applyStyle = (self, string) => {
|
||||||
return string;
|
return string;
|
||||||
}
|
}
|
||||||
|
|
||||||
const {openAll, closeAll} = styler;
|
const { openAll, closeAll } = styler;
|
||||||
if (string.includes('\u001B')) {
|
if (string.includes("\u001B")) {
|
||||||
while (styler !== undefined) {
|
while (styler !== undefined) {
|
||||||
// Replace any instances already present with a re-opening code
|
// Replace any instances already present with a re-opening code
|
||||||
// otherwise only the part of the string until said closing code
|
// otherwise only the part of the string until said closing code
|
||||||
|
|
@ -191,7 +215,7 @@ const applyStyle = (self, string) => {
|
||||||
// We can move both next actions out of loop, because remaining actions in loop won't have
|
// We can move both next actions out of loop, because remaining actions in loop won't have
|
||||||
// any/visible effect on parts we add here. Close the styling before a linebreak and reopen
|
// any/visible effect on parts we add here. Close the styling before a linebreak and reopen
|
||||||
// after next line to fix a bleed issue on macOS: https://github.com/chalk/chalk/pull/92
|
// after next line to fix a bleed issue on macOS: https://github.com/chalk/chalk/pull/92
|
||||||
const lfIndex = string.indexOf('\n');
|
const lfIndex = string.indexOf("\n");
|
||||||
if (lfIndex !== -1) {
|
if (lfIndex !== -1) {
|
||||||
string = stringEncaseCRLFWithFirstIndex(string, closeAll, openAll, lfIndex);
|
string = stringEncaseCRLFWithFirstIndex(string, closeAll, openAll, lfIndex);
|
||||||
}
|
}
|
||||||
|
|
@ -202,11 +226,10 @@ const applyStyle = (self, string) => {
|
||||||
Object.defineProperties(createChalk.prototype, styles);
|
Object.defineProperties(createChalk.prototype, styles);
|
||||||
|
|
||||||
const chalk = createChalk();
|
const chalk = createChalk();
|
||||||
export const chalkStderr = createChalk({level: stderrColor ? stderrColor.level : 0});
|
export const chalkStderr = createChalk({
|
||||||
|
level: stderrColor ? stderrColor.level : 0,
|
||||||
|
});
|
||||||
|
|
||||||
export {
|
export { stdoutColor as supportsColor, stderrColor as supportsColorStderr };
|
||||||
stdoutColor as supportsColor,
|
|
||||||
stderrColor as supportsColorStderr,
|
|
||||||
};
|
|
||||||
|
|
||||||
export default chalk;
|
export default chalk;
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue