b0523a4438
- Precomputed style function
- Skip arguments to array + join if there's only one argument (the
common case)
- Merge multiple return statements to one
To calculate the performance benefit:
```javascript
var chalk = require('./index.js');
console.time('100000 iterations');
for (var i = 0; i < 100000; i++) {
chalk.red('A string that is about 80 characters long (normal use I think?)');
}
console.timeEnd('100000 iterations');
```
Running this before this commit:
```shell
for i in {1..5}; do node time.js; done
100000 iterations: 19485ms
100000 iterations: 18933ms
100000 iterations: 19365ms
100000 iterations: 19332ms
100000 iterations: 18660ms
```
After:
```shell
100000 iterations: 268ms
100000 iterations: 261ms
100000 iterations: 264ms
100000 iterations: 259ms
100000 iterations: 254ms
```
Performance gain, taking the middle result of both:
```shell
19332 / 261 = 74.~
```
Closes #16
70 lines
1.7 KiB
JavaScript
70 lines
1.7 KiB
JavaScript
'use strict';
|
|
var escapeStringRegexp = require('escape-string-regexp');
|
|
var ansiStyles = require('ansi-styles');
|
|
var stripAnsi = require('strip-ansi');
|
|
var supportsColor = require('supports-color');
|
|
var defineProps = Object.defineProperties;
|
|
var chalk = module.exports;
|
|
|
|
var styles = (function () {
|
|
var ret = {};
|
|
|
|
ansiStyles.grey = ansiStyles.gray;
|
|
|
|
Object.keys(ansiStyles).forEach(function (key) {
|
|
ansiStyles[key].closeRe = new RegExp(escapeStringRegexp(ansiStyles[key].close), 'g');
|
|
|
|
ret[key] = {
|
|
get: function () {
|
|
this._styles.push(key);
|
|
return this;
|
|
}
|
|
};
|
|
});
|
|
|
|
return ret;
|
|
})();
|
|
|
|
function applyStyle() {
|
|
// support varags, but simply cast to string in case there's only one arg
|
|
var str = arguments.length === 1 ? arguments[0] + '' : [].slice.call(arguments).join(' ');
|
|
|
|
if (!chalk.enabled || !str) {
|
|
return str;
|
|
}
|
|
|
|
return applyStyle._styles.reduce(function (str, name) {
|
|
var code = ansiStyles[name];
|
|
// Replace any instances already present with a re-opening code
|
|
// otherwise only the part of the string until said closing code
|
|
// will be colored, and the rest will simply be 'plain'.
|
|
return code.open + str.replace(code.closeRe, code.open) + code.close;
|
|
}, str) ;
|
|
}
|
|
|
|
function init() {
|
|
var ret = {};
|
|
|
|
Object.keys(styles).forEach(function (name) {
|
|
var style = defineProps(applyStyle, styles);
|
|
ret[name] = {
|
|
get: function () {
|
|
style._styles = [];
|
|
return style[name];
|
|
}
|
|
};
|
|
});
|
|
|
|
return ret;
|
|
}
|
|
|
|
defineProps(chalk, init());
|
|
|
|
chalk.styles = ansiStyles;
|
|
chalk.stripColor = stripAnsi;
|
|
chalk.supportsColor = supportsColor;
|
|
|
|
// detect mode if not set manually
|
|
if (chalk.enabled === undefined) {
|
|
chalk.enabled = chalk.supportsColor;
|
|
}
|