Add opts.colors

Author: ehmicky

src/colors.js

@@ -0,0 +1,15 @@
+'use strict'
+
+const chalk = require('chalk')
+
+const getColors = function({ opts: { colors } }) {
+  // Can disable colors with `opts.colors: false`.
+  // chalk will automatically disable colors if output does not support it.
+  const enabled = Boolean(colors)
+
+  return chalk.constructor({ enabled })
+}
+
+module.exports = {
+  getColors,
+}

src/default.js

@@ -10,6 +10,7 @@ const DEFAULT_OPTS = {
   getMessage: defaultGetMessage,
   log: defaultLog,
   exitOnExceptions: true,
+  colors: true,
 }
 
 module.exports = {

src/handle.js

@@ -3,6 +3,7 @@
 const { exit } = require('process')
 
 const { getInfo } = require('./info')
+const { getColors } = require('./colors')
 const { getLevel } = require('./level')
 const { getMessage } = require('./message')
 
@@ -29,8 +30,9 @@ const handleEvent = async function({
     return
   }
 
+  const colors = getColors({ opts })
   const level = getLevel({ opts, info })
-  const message = getMessage({ opts, info, level })
+  const message = getMessage({ opts, info, level, colors })
 
   opts.log(message, level, info)
 

src/level.js

@@ -2,8 +2,6 @@
 
 const { platform } = require('process')
 
-const { red, yellow } = require('chalk')
-
 // Retrieve error's `level`
 const getLevel = function({ opts, info }) {
   const level = opts.getLevel(info)
@@ -21,11 +19,11 @@ const isWindows = platform === 'win32'
 // Each level is printed in a different way
 const LEVELS = {
   warn: {
-    COLOR: yellow,
+    COLOR: 'yellow',
     SIGN: isWindows ? '\u203C' : '\u26A0',
   },
   error: {
-    COLOR: red,
+    COLOR: 'red',
     SIGN: isWindows ? '\u00D7' : '\u2718',
   },
 }

src/message.js

@@ -6,8 +6,8 @@ const { printMultiline, print, prettify } = require('./serialize')
 
 // Retrieve `message` which sums up all information that can be gathered about
 // the event.
-const getMessage = function({ opts, info, level }) {
-  const message = opts.getMessage({ ...info, level })
+const getMessage = function({ opts, info, level, colors }) {
+  const message = opts.getMessage({ ...info, level, colors })
   // Ensure this is a string
   const messageA = typeof message === 'string' ? message : inspect(message)
   return messageA
@@ -22,6 +22,7 @@ const defaultGetMessage = function({
   secondPromiseValue,
   error,
   level,
+  colors,
 }) {
   const message = MESSAGES[eventName]({
     promiseState,
@@ -31,7 +32,7 @@ const defaultGetMessage = function({
     error,
   })
 
-  const messageA = prettify({ message, eventName, level })
+  const messageA = prettify({ message, eventName, level, colors })
   return messageA
 }
 

src/serialize.js

@@ -2,8 +2,6 @@
 
 const { inspect } = require('util')
 
-const { bold, dim, inverse } = require('chalk')
-
 const { LEVELS } = require('./level')
 
 const printMultiline = function(value) {
@@ -21,12 +19,20 @@ const print = function(value) {
   return inspect(value)
 }
 
-const prettify = function({ message, eventName, level }) {
+const prettify = function({
+  message,
+  eventName,
+  level,
+  colors,
+  colors: { bold, dim, inverse },
+}) {
   const [header, ...lines] = message.split('\n')
 
   // Add color, icon and `eventName` to first message line.
   const { COLOR, SIGN } = LEVELS[level]
-  const headerA = COLOR(`${bold(inverse(` ${SIGN}  ${eventName} `))} ${header}`)
+  const headerA = colors[COLOR](
+    `${bold(inverse(` ${SIGN}  ${eventName} `))} ${header}`,
+  )
   // Add gray color and indentation to other lines.
   const linesA = lines.map(line => dim(`\t${VERTICAL_BAR} ${line}`))