Add --color and --no-color support (#77)

index.js

@@ -1,9 +1,10 @@
 import * as tty from "tty"
 
 const env = process.env || {}
+const argv = process.argv || []
 
-const isDisabled = "NO_COLOR" in env
-const isForced = "FORCE_COLOR" in env
+const isDisabled = "NO_COLOR" in env || argv.includes("--no-color")
+const isForced = "FORCE_COLOR" in env || argv.includes("--color")
 const isWindows = process.platform === "win32"
 
 const isCompatibleTerminal =

tests/scripts/--color.NO_COLOR.sh

@@ -0,0 +1,3 @@
+#!/bin/sh
+
+[ `NO_COLOR= node --eval 'import("./index.js").then(({ blue }) => console.log(blue("foobar")))' -- --color` = foobar ]

tests/scripts/--color.sh

@@ -0,0 +1,5 @@
+#!/bin/sh
+
+[ `node --eval 'import("./index.js").then(({ blue }) => console.log(blue("foobar")))' -- --color` = `printf '\e[34mfoobar\e[39m'` ]
+
+

tests/scripts/--no-color.sh

@@ -0,0 +1,4 @@
+#!/bin/sh
+
+[ `node --eval 'import("./index.js").then(({ blue }) => console.log(blue("foobar")))' -- --no-color` = 'foobar' ]
+

tests/scripts/NO_COLOR.sh

@@ -6,3 +6,4 @@
       console.log(createColors().blue("foobar")))
 
   '` = "foobar" ]
+

tests/variables.test.js

@@ -17,5 +17,14 @@ export default [
     t("`NO_COLOR` disables color", [
       sh(`${SCRIPTS_PATH}/NO_COLOR.sh`, exitStatusSuccess),
     ]),
+    t("`--no-color` disables color", [
+      sh(`${SCRIPTS_PATH}/--no-color.sh`, exitStatusSuccess),
+    ]),
+    t("`--color` enables color", [
+      sh(`${SCRIPTS_PATH}/--color.sh`, exitStatusSuccess),
+    ]),
+    t("`NO_COLOR beats --color` enables color", [
+      sh(`${SCRIPTS_PATH}/--color.NO_COLOR.sh`, exitStatusSuccess),
+    ]),
   ]),
 ]