Keep function prototype methods (#434)

Richienb · web-flow · commit 0fba91b0373c · 2021-04-20T13:48:30.000+07:00
diff --git a/source/index.js b/source/index.js
@@ -54,6 +54,8 @@ function createChalk(options) {
 	return chalkFactory(options);
 }
 
+Object.setPrototypeOf(createChalk.prototype, Function.prototype);
+
 for (const [styleName, style] of Object.entries(ansiStyles)) {
 	styles[styleName] = {
 		get() {
diff --git a/test/chalk.js b/test/chalk.js
@@ -117,3 +117,9 @@ test('sets correct level for chalkStderr and respects it', t => {
 	t.is(chalkStderr.level, 3);
 	t.is(chalkStderr.red.bold('foo'), '\u001B[31m\u001B[1mfoo\u001B[22m\u001B[39m');
 });
+
+test('keeps function prototype methods', t => {
+	t.is(chalk.apply(chalk, ['foo']), 'foo');
+	t.is(chalk.bind(chalk, 'foo')(), 'foo');
+	t.is(chalk.call(chalk, 'foo'), 'foo');
+});

Original file line number	Diff line number	Diff line change
`@@ -54,6 +54,8 @@ function createChalk(options) {`
`54`	`54`	`return chalkFactory(options);`
`55`	`55`	`}`
`56`	`56`
	`57`	`+Object.setPrototypeOf(createChalk.prototype, Function.prototype);`
	`58`	`+`
`57`	`59`	`for (const [styleName, style] of Object.entries(ansiStyles)) {`
`58`	`60`	`styles[styleName] = {`
`59`	`61`	`get() {`