feat: add ansi.enble() method

Author: zdm

lib/ansi.js

@@ -15,6 +15,8 @@ const ESC = "\x1B",
     RESET_BACKGROUND = CSI + "49m",
     COLOR_BUFFER = Buffer.alloc( 4 ),
     DEFAULT_STYLES = {
+        "raw": [ null, null ],
+
         "reset": [ 0, null ],
 
         "bold": [ 1, 22 ],
@@ -69,14 +71,29 @@ const ESC = "\x1B",
         "onBrightWhite": [ 107, 49 ],
     };
 
+function parseEnabled ( enabled ) {
+    if ( enabled == null ) {
+        enabled = null;
+    }
+    else if ( enabled instanceof stream.Writable ) {
+        enabled = enabled.isTTY;
+    }
+    else {
+        enabled = Boolean( enabled );
+    }
+
+    return enabled;
+}
+
 const ansiStyle = ansi =>
     class AnsiStyle {
         #ansi = ansi;
         #name;
+        #isEnabled;
         #on;
         #off;
 
-        constructor ( name, on, off ) {
+        constructor ( { name, on, off } ) {
             this.#name = name;
             this.#on = on;
             this.#off = off;
@@ -99,7 +116,25 @@ const ansiStyle = ansi =>
             return this.#off;
         }
 
+        get isEnabled () {
+            return this.#isEnabled ?? this.#ansi.isEnabled;
+        }
+
         // public
+        enable ( enabled ) {
+            if ( this.#name ) {
+                return new this.constructor( {
+                    "on": this.#on,
+                    "off": this.#off,
+                } ).enable( enabled );
+            }
+            else {
+                this.#isEnabled = parseEnabled( enabled );
+
+                return makeCallable( this, "applyStyle" );
+            }
+        }
+
         color ( color ) {
             return this._addStyle( {
                 "on": this.#ansi.createColor( 38, color ),
@@ -119,7 +154,7 @@ const ansiStyle = ansi =>
             // stringify
             string = String( string );
 
-            if ( !this.ansi.isEnabled ) return string;
+            if ( !this.isEnabled ) return string;
 
             if ( !string ) return "";
 
@@ -134,7 +169,13 @@ const ansiStyle = ansi =>
         // protected
         _addStyle ( { on, off } ) {
             if ( this.#name ) {
-                return makeCallable( new this.constructor( null, this.#on + on, off + this.#off ), "applyStyle" );
+                return makeCallable(
+                    new this.constructor( {
+                        "on": this.#on + on,
+                        "off": off + this.#off,
+                    } ),
+                    "applyStyle"
+                );
             }
             else {
                 this.#on = this.#on + on;
@@ -173,7 +214,7 @@ export class Ansi {
     }
 
     static setEnabled ( enabled ) {
-        this.#globalEnabled = Boolean( enabled );
+        this.#globalEnabled = Boolean( parseEnabled( enabled ) );
     }
 
     // properties
@@ -193,17 +234,7 @@ export class Ansi {
     setEnabled ( enabled, callback ) {
         const wasEnabled = this.#isEnabled;
 
-        if ( enabled == null ) {
-            enabled = null;
-        }
-        else if ( enabled instanceof stream.Writable ) {
-            enabled = Boolean( enabled.isTTY );
-        }
-        else {
-            enabled = Boolean( enabled );
-        }
-
-        this.#isEnabled = enabled;
+        this.#isEnabled = parseEnabled( enabled );
 
         if ( callback ) {
             try {
@@ -246,6 +277,10 @@ export class Ansi {
         return this;
     }
 
+    enable ( enabled ) {
+        return this.raw.enable( enabled );
+    }
+
     color ( color ) {
         return this.#createStyle( null, {
             "on": this.createColor( 38, color ),
@@ -313,7 +348,7 @@ export class Ansi {
     }
 
     // private
-    #createStyle ( name, { on, off } ) {
+    #createStyle ( name, { on, off } = {} ) {
         on = on == null
             ? ""
             : String( on );
@@ -322,7 +357,14 @@ export class Ansi {
             ? ""
             : String( off );
 
-        return makeCallable( new this.#AnsiStyle( name, on, off ), "applyStyle" );
+        return makeCallable(
+            new this.#AnsiStyle( {
+                name,
+                on,
+                off,
+            } ),
+            "applyStyle"
+        );
     }
 
     #linkToString ( url, text ) {