Add flow checks for chalk

Author: jfmengels

lib/chalk.js

@@ -1,23 +1,21 @@
 'use strict';
+// @flow
 const supportsColor = require('./supports-color');
 
 // Find more colors/styles in
 // https://github.com/chalk/ansi-styles/blob/main/index.js
-const colors = {
-  blue: { open: '\x1B[34m', close: '\x1B[39m' },
-  red: { open: '\x1B[31m', close: '\x1B[39m' },
-};
-
-const chalk = { supportsColor };
 
-for (const [styleName, style] of Object.entries(colors)) {
-  chalk[styleName] = (string) => {
-    if (!supportsColor || !string) {
-      return string;
-    }
+function red(string /*: string */) /*: string */ {
+  return supportsColor ? `\x1B[31m${string}\x1B[39m` : string;
+}
 
-    return style.open + string + style.close;
-  };
+function blue(string /*: string */) /*: string */ {
+  return supportsColor ? `\x1B[34m${string}\x1B[39m` : string;
 }
 
-module.exports = chalk;
+module.exports = {
+  supportsColor,
+
+  red,
+  blue,
+};