Keep function prototype methods (#434)

chalk · Apr 20, 2021 · 0fba91b · 0fba91b
1 parent d798222
commit 0fba91b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
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');
+});