Add level

Author: ehmicky

src/default.js

@@ -1,13 +1,15 @@
 'use strict'
 
+const { defaultGetLevel } = require('./level')
+
 // Default `opts.handlerFunc`
-const defaultHandler = function({ eventName, message }) {
-  const level = eventName === 'warning' ? 'warn' : 'error'
+const defaultHandler = function(message, level) {
   // eslint-disable-next-line no-restricted-globals, no-console
   console[level](message)
 }
 
 const DEFAULT_OPTS = {
+  getLevel: defaultGetLevel,
   handlerFunc: defaultHandler,
   exitOnExceptions: true,
 }

src/handle.js

@@ -3,10 +3,12 @@
 const { exit } = require('process')
 
 const { getInfo } = require('./info')
+const { getLevel } = require('./level')
 const { getMessage } = require('./message')
 
 // Generic event handler for all events.
 const handleEvent = async function({
+  opts,
   opts: { handlerFunc, exitOnExceptions },
   eventName,
   error,
@@ -24,9 +26,10 @@ const handleEvent = async function({
     secondPromiseValue,
   })
 
-  const message = getMessage(info)
+  const { level, levelInfo } = getLevel({ opts, info })
+  const message = getMessage({ ...info, levelInfo })
 
-  handlerFunc({ ...info, message })
+  handlerFunc(message, level, info)
 
   exitProcess({ eventName, exitOnExceptions })
 }

src/level.js

@@ -0,0 +1,44 @@
+'use strict'
+
+const { platform } = require('process')
+
+const { red, yellow } = require('chalk')
+
+const getLevel = function({ opts, info }) {
+  const level = opts.getLevel(info)
+  const levelInfo = LEVELS[level]
+
+  if (levelInfo === undefined) {
+    const levels = Object.keys(LEVELS).join(', ')
+    // TODO: infinite recursion?
+    throw new Error(`Level ${level} is invalid. Must be one of: ${levels}`)
+  }
+
+  return { level, levelInfo }
+}
+
+const isWindows = platform === 'win32'
+// Each level is printed in a different way
+const LEVELS = {
+  warn: {
+    COLOR: yellow,
+    SIGN: isWindows ? '\u203C' : '\u26A0',
+  },
+  error: {
+    COLOR: red,
+    SIGN: isWindows ? '\u00D7' : '\u2718',
+  },
+}
+
+const defaultGetLevel = function({ eventName }) {
+  if (eventName === 'warning') {
+    return 'warn'
+  }
+
+  return 'error'
+}
+
+module.exports = {
+  getLevel,
+  defaultGetLevel,
+}

src/message.js

@@ -1,9 +1,8 @@
 'use strict'
 
-const { platform } = require('process')
 const { inspect } = require('util')
 
-const { red, yellow, bold, dim, inverse } = require('chalk')
+const { bold, dim, inverse } = require('chalk')
 
 // Retrieve `message` which sums up all information that can be gathered about
 // the event.
@@ -14,6 +13,7 @@ const getMessage = function({
   secondPromiseState,
   secondPromiseValue,
   error,
+  levelInfo,
 }) {
   const message = MESSAGES[eventName]({
     promiseState,
@@ -23,8 +23,7 @@ const getMessage = function({
     error,
   })
 
-  const level = eventName === 'warning' ? 'warn' : 'error'
-  const messageA = prettify({ message, level, eventName })
+  const messageA = prettify({ message, eventName, levelInfo })
   return messageA
 }
 
@@ -86,11 +85,10 @@ const printValue = function(value) {
   return inspect(value)
 }
 
-const prettify = function({ message, level, eventName }) {
+const prettify = function({ message, eventName, levelInfo: { COLOR, SIGN } }) {
   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}`)
   // Add gray color and indentation to other lines.
   const linesA = lines.map(line => dim(`\t${VERTICAL_BAR} ${line}`))
@@ -99,20 +97,6 @@ const prettify = function({ message, level, eventName }) {
   return messageA
 }
 
-// Start the message with an icon followed by `Error` or `Warning`
-// Also add colors
-const isWindows = platform === 'win32'
-const LEVELS = {
-  warn: {
-    COLOR: yellow,
-    SIGN: isWindows ? '\u203C' : '\u26A0',
-  },
-  error: {
-    COLOR: red,
-    SIGN: isWindows ? '\u00D7' : '\u2718',
-  },
-}
-
 const VERTICAL_BAR = '\u2016'
 
 module.exports = {