refactor: refactor ansi

Author: zdm

lib/ansi.js

@@ -15,8 +15,6 @@ const ESC = "\x1B",
     RESET_BACKGROUND = CSI + "49m",
     COLOR_BUFFER = Buffer.alloc( 4 ),
     DEFAULT_STYLES = {
-        "raw": [ null, null ],
-
         "reset": [ 0, null ],
 
         "bold": [ 1, 22 ],
@@ -93,10 +91,11 @@ const ansiStyle = ansi =>
         #on;
         #off;
 
-        constructor ( { name, on, off } ) {
+        constructor ( { name, on, off, enabled } ) {
             this.#name = name;
             this.#on = on;
             this.#off = off;
+            this.#isEnabled = parseEnabled( enabled );
         }
 
         // properties
@@ -122,12 +121,17 @@ const ansiStyle = ansi =>
 
         // public
         enable ( enabled ) {
+
+            // clone named style
             if ( this.#name ) {
-                return new this.constructor( {
-                    "on": this.#on,
-                    "off": this.#off,
-                } ).enable( enabled );
+                return this._addStyle( {
+                    "on": "",
+                    "off": "",
+                    enabled,
+                } );
             }
+
+            // update style
             else {
                 this.#isEnabled = parseEnabled( enabled );
 
@@ -167,16 +171,21 @@ const ansiStyle = ansi =>
         }
 
         // protected
-        _addStyle ( { on, off } ) {
+        _addStyle ( { on, off, enabled } ) {
+
+            // clone named style
             if ( this.#name ) {
                 return makeCallable(
                     new this.constructor( {
                         "on": this.#on + on,
                         "off": off + this.#off,
+                        enabled,
                     } ),
                     "applyStyle"
                 );
             }
+
+            // update style
             else {
                 this.#on = this.#on + on;
                 this.#off = off + this.#off;
@@ -198,6 +207,7 @@ export class Ansi {
         // define default styles
         for ( const [ name, [ on, off ] ] of Object.entries( DEFAULT_STYLES ) ) {
             this.defineStyle( name, {
+                "configurable": false,
                 "on": on
                     ? CSI + on + "m"
                     : null,
@@ -213,8 +223,23 @@ export class Ansi {
         return this.#globalEnabled;
     }
 
-    static setEnabled ( enabled ) {
+    static setEnabled ( enabled, callback ) {
+        const wasEnabled = this.#globalEnabled;
+
         this.#globalEnabled = Boolean( parseEnabled( enabled ) );
+
+        if ( callback ) {
+            try {
+                callback();
+
+                this.#globalEnabled = wasEnabled;
+            }
+            catch ( e ) {
+                this.#globalEnabled = wasEnabled;
+
+                throw e;
+            }
+        }
     }
 
     // properties
@@ -250,17 +275,21 @@ export class Ansi {
         }
     }
 
-    defineStyle ( name, style ) {
-        style = this.#createStyle( name, style );
+    defineStyle ( name, { configurable, on, off } ) {
+        configurable = configurable == null
+            ? true
+            : Boolean( configurable );
+
+        const style = this.#createStyle( name, { on, off } );
 
         Object.defineProperty( this, name, {
-            "configurable": true,
+            configurable,
             "writable": false,
             "value": style,
         } );
 
         Object.defineProperty( this.#AnsiStyle.prototype, name, {
-            "configurable": true,
+            configurable,
             get () {
                 return this._addStyle( style );
             },
@@ -278,7 +307,7 @@ export class Ansi {
     }
 
     enable ( enabled ) {
-        return this.raw.enable( enabled );
+        return this.#createStyle( null, { enabled } );
     }
 
     color ( color ) {
@@ -348,7 +377,7 @@ export class Ansi {
     }
 
     // private
-    #createStyle ( name, { on, off } = {} ) {
+    #createStyle ( name, { on, off, enabled } = {} ) {
         on = on == null
             ? ""
             : String( on );
@@ -362,6 +391,7 @@ export class Ansi {
                 name,
                 on,
                 off,
+                enabled,
             } ),
             "applyStyle"
         );
@@ -383,7 +413,6 @@ export default ansi;
 
 ansi.defineStyles( {
     "hl": ansi.brightWhite,
-    "dim": ansi.gray,
     "ok": ansi.bold.brightWhite.onColor( 0x64_00 ),
     "warn": ansi.color( 0x0 ).onColor( 0xCC_CC_00 ),
     "error": ansi.bold.brightWhite.onRed,