Use FORCE_COLOR environmental variable to force colorized output (#7033)

* Use FORCE_COLOR environmental variable to force colorized output

* Update CHANGELOG.md

Author: mediremi

CHANGELOG.md

@@ -12,6 +12,10 @@
 
 # 12.0.0-alpha.4 (Unreleased)
 
+#### :rocket: New Feature
+
+- Use FORCE_COLOR environmental variable to force colorized output https://github.com/rescript-lang/rescript-compiler/pull/7033
+
 #### :bug: Bug fix
 
 - Fix tuple coercion. https://github.com/rescript-lang/rescript-compiler/pull/7024

cli/rescript_bsb.js

@@ -110,8 +110,8 @@ function clean(args) {
   delegate(["clean", ...args]);
 }
 
-const isTtyError = process.stderr.isTTY;
-const isTtyStd = process.stdout.isTTY;
+const shouldColorizeError = process.stderr.isTTY || process.env.FORCE_COLOR == "1";
+const shouldColorize = process.stdout.isTTY || process.env.FORCE_COLOR == "1";
 
 /**
  * @type {[number,number]}
@@ -133,7 +133,7 @@ function logFinishCompiling(code) {
   if (code) {
     log = log + " (exit: " + code + ")";
   }
-  if (isTtyStd) {
+  if (shouldColorize) {
     log = "\x1b[36m" + log + "\x1b[0m";
   }
   if (code) {
@@ -146,7 +146,7 @@ function logFinishCompiling(code) {
 function logStartCompiling() {
   updateStartTime();
   let log = `>>>> Start compiling`;
-  if (isTtyStd) {
+  if (shouldColorize) {
     log = "\x1b[36m" + log + "\x1b[0m";
   }
   console.log(log);
@@ -265,7 +265,7 @@ function watch(args) {
       .on("error", function (err) {
         // @ts-ignore
         if (err !== undefined && err.code === "EADDRINUSE") {
-          var error = isTtyStd ? `\x1b[1;31mERROR:\x1b[0m` : `ERROR:`;
+          var error = shouldColorize ? `\x1b[1;31mERROR:\x1b[0m` : `ERROR:`;
           console.error(`${error} The websocket port number ${webSocketPort} is in use.
 Please pick a different one using the \`-ws [host:]port\` flag from bsb.`);
         } else {
@@ -358,7 +358,7 @@ Please pick a different one using the \`-ws [host:]port\` flag from bsb.`);
    * @param highlight {string}
    */
   function outputError(error, highlight) {
-    if (isTtyError && highlight) {
+    if (shouldColorizeError && highlight) {
       process.stderr.write(
         error.replace(highlight, "\x1b[1;31m" + highlight + "\x1b[0m"),
       );

rescript

@@ -14,13 +14,17 @@ var bsb = require("./cli/rescript_bsb.js");
 var cwd = process.cwd();
 process.env.BSB_PROJECT_ROOT = cwd;
 
-if (process.env.NINJA_ANSI_FORCED === undefined) {
+if (process.env.FORCE_COLOR === undefined) {
   if (require("tty").isatty(1)) {
+    process.env.FORCE_COLOR = "1";
     process.env.NINJA_ANSI_FORCED = "1";
   }
 } else {
+  if (process.env.FORCE_COLOR === "1" && process.env.NINJA_ANSI_FORCED === undefined) {
+    process.env.NINJA_ANSI_FORCED = "1";
+  }
   if (process.argv.includes("-verbose")) {
-    console.log(`NINJA_ANSI_FORCED: "${process.env.NINJA_ANSI_FORCED}"`);
+    console.log(`FORCE_COLOR: "${process.env.FORCE_COLOR}"`);
   }
 }