add chalk API proxy

Author: DavidWells

src/index.js

@@ -1,15 +1,16 @@
 const { Command } = require('@oclif/command')
-const chalk = require('chalk')
 const API = require('netlify')
+const merge = require('lodash.merge')
 const { format, inspect } = require('util')
 const getConfigPath = require('./utils/get-config-path')
 const readConfig = require('./utils/read-config')
 const globalConfig = require('./global-config')
 const StateConfig = require('./state')
+const chalkInstance = require('./utils/chalk')
 const openBrowser = require('./utils/open-browser')
 const findRoot = require('./utils/find-root')
 const { track, identify } = require('./utils/telemetry')
-const merge = require('lodash.merge')
+
 const argv = require('minimist')(process.argv.slice(2))
 const { NETLIFY_AUTH_TOKEN } = process.env
 
@@ -127,6 +128,10 @@ class BaseCommand extends Command {
     }, opts))
   }
 
+  get chalk() {
+    // If --json flag disable chalk colors
+    return chalkInstance(argv.json)
+  }
   /**
    * Get user netlify API token
    * @param  {string} - [tokenFromFlag] - value passed in by CLI flag
@@ -212,11 +217,11 @@ class BaseCommand extends Command {
 
     // Log success
     this.log()
-    this.log(`${chalk.greenBright('You are now logged into your Netlify account!')}`)
+    this.log(`${this.chalk.greenBright('You are now logged into your Netlify account!')}`)
     this.log()
-    this.log(`Run ${chalk.cyanBright('netlify status')} for account details`)
+    this.log(`Run ${this.chalk.cyanBright('netlify status')} for account details`)
     this.log()
-    this.log(`To see all available commands run: ${chalk.cyanBright('netlify help')}`)
+    this.log(`To see all available commands run: ${this.chalk.cyanBright('netlify help')}`)
     this.log()
     return accessToken
   }

src/utils/chalk.js

@@ -0,0 +1,48 @@
+const chalk = require('chalk')
+
+/**
+ * Chalk instance for CLI
+ * @param  {boolean} noColors - disable chalk colors
+ * @return {object} - chalk instance or proxy noOp
+ */
+function safeChalk(noColors) {
+  /* if no colors return proxy to chalk API */
+  if (noColors) {
+    return neverNull(chalk)
+  }
+  return chalk
+}
+
+function noop() {}
+
+function neverNull(obj) {
+  function match(some, none = noop) {
+    return obj != null ? some(obj) : none()
+  }
+  return new Proxy((some, none) => {
+    if (some) {
+      // has value return it with no chalk wrapper
+      return some
+    }
+    if (!some && !none) return obj
+    return match(some, none)
+  }, {
+    get: (target, key) => {
+      const obj = target()
+      if (obj !== null && typeof obj === 'object') {
+        return neverNull(obj[key])
+      } else {
+        return neverNull()
+      }
+    },
+    set: (target, key, val) => {
+      const obj = target()
+      if (obj !== null && typeof obj === 'object') {
+        obj[key] = val
+      }
+      return true
+    }
+  })
+}
+
+module.exports = safeChalk