orbit-oss/chalk
5
4
Fork 1
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

export classy chalk object to create isolated ctx

Browse source
This commit is contained in:
David Keijser 2015-01-23 15:45:40 +01:00
parent aa7e67741f
commit eeadb9f40d
2 changed files with 25 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

30
index.js
View file

@ -5,7 +5,13 @@ var stripAnsi = require('strip-ansi');
var hasAnsi = require('has-ansi');
var supportsColor = require('supports-color');
var defineProps = Object.defineProperties;
var chalk = module.exports;
function Chalk(enabled) {
// detect mode if not set manually
if (enabled === undefined) {
this.enabled = this.supportsColor;
}
}
// use bright blue on Windows as the normal blue color is illegible
if (process.platform === 'win32') {
@ -17,6 +23,7 @@ function build(_styles) {
return applyStyle.apply(builder, arguments);
};
builder._styles = _styles;
builder.enabled = this.enabled;
// __proto__ is used because we must return a function, but there is
// no way to create a function with a different prototype.
builder.__proto__ = proto;
@ -31,7 +38,7 @@ var styles = (function () {
ret[key] = {
get: function () {
return build(this._styles.concat(key));
return build.call(this, this._styles.concat(key));
}
};
});
@ -53,7 +60,7 @@ function applyStyle() {
}
}
if (!chalk.enabled || !str) {
if (!this.enabled || !str) {
return str;
}
@ -78,7 +85,7 @@ function init() {
Object.keys(styles).forEach(function (name) {
ret[name] = {
get: function () {
return build([name]);
return build.call(this, [name]);
}
};
});
@ -86,14 +93,11 @@ function init() {
return ret;
}
defineProps(chalk, init());
defineProps(Chalk.prototype, init());
chalk.styles = ansiStyles;
chalk.hasColor = hasAnsi;
chalk.stripColor = stripAnsi;
chalk.supportsColor = supportsColor;
Chalk.prototype.styles = ansiStyles;
Chalk.prototype.hasColor = hasAnsi;
Chalk.prototype.stripColor = stripAnsi;
Chalk.prototype.supportsColor = supportsColor;
// detect mode if not set manually
if (chalk.enabled === undefined) {
chalk.enabled = chalk.supportsColor;
}
module.exports = new Chalk();

8
test.js
View file

@ -68,6 +68,14 @@ describe('chalk.enabled', function () {
});
});
describe('chalk.constructor', function () {
it('should create a isolated context where colors can be disabled', function () {
var ctx = new chalk.constructor(false);
assert.equal(ctx.red('foo'), 'foo');
assert.equal(chalk.red('foo'), '\u001b[31mfoo\u001b[39m');
});
});
describe('chalk.styles', function () {
it('should expose the styles as ANSI escape codes', function () {
assert.equal(chalk.styles.red.open, '\u001b[31m');