internal: factor colors into style object

Author: rwe

lib/index.js

@@ -8,6 +8,17 @@
  * @property {boolean | null | undefined} [showPositions=true]
  *   Whether to include positional information (default: `true`).
  *
+ * @typedef Style
+ *   Styling functions.
+ * @property {(_: string) => string} bold
+ *   Style a node type.
+ * @property {(_: string) => string} dim
+ *   Style structural punctuation.
+ * @property {(_: string) => string} yellow
+ *   Style the numeric count of node children.
+ * @property {(_: string) => string} green
+ *   Style a non-tree value.
+ *
  * @typedef State
  *   Info passed around.
  * @property {boolean} showPositions
@@ -31,10 +42,15 @@ export const inspect = color ? inspectColor : inspectNoColor
 
 const own = {}.hasOwnProperty
 
-const bold = ansiColor(1, 22)
-const dim = ansiColor(2, 22)
-const yellow = ansiColor(33, 39)
-const green = ansiColor(32, 39)
+/**
+ * @type Style
+ */
+const style = {
+  bold: ansiColor(1, 22),
+  dim: ansiColor(2, 22),
+  yellow: ansiColor(33, 39),
+  green: ansiColor(32, 39)
+}
 
 // ANSI color regex.
 /* eslint-disable no-control-regex */
@@ -132,15 +148,16 @@ function inspectNodes(nodes, state) {
 
   while (++index < nodes.length) {
     result.push(
-      dim(
+      style.dim(
         (index < nodes.length - 1 ? '├' : '└') +
           '─' +
           String(index).padEnd(size)
       ) +
         ' ' +
         indent(
           inspectValue(nodes[index], state),
-          (index < nodes.length - 1 ? dim('│') : ' ') + ' '.repeat(size + 2),
+          (index < nodes.length - 1 ? style.dim('│') : ' ') +
+            ' '.repeat(size + 2),
           true
         )
     )
@@ -205,14 +222,17 @@ function inspectFields(object, state) {
     }
 
     result.push(
-      key + dim(':') + (/\s/.test(formatted.charAt(0)) ? '' : ' ') + formatted
+      key +
+        style.dim(':') +
+        (/\s/.test(formatted.charAt(0)) ? '' : ' ') +
+        formatted
     )
   }
 
   return indent(
     result.join('\n'),
     (isArrayUnknown(object.children) && object.children.length > 0
-      ? dim('│')
+      ? style.dim('│')
       : ' ') + ' '
   )
 }
@@ -253,7 +273,7 @@ function inspectTree(node, state) {
  *   Formatted node.
  */
 function formatNode(node, state) {
-  const result = [bold(node.type)]
+  const result = [style.bold(node.type)]
   // Cast as record to allow indexing.
   const map = /** @type {Record<string, unknown>} */ (
     /** @type {unknown} */ (node)
@@ -266,13 +286,17 @@ function formatNode(node, state) {
   }
 
   if (isArrayUnknown(map.children)) {
-    result.push(dim('['), yellow(String(map.children.length)), dim(']'))
+    result.push(
+      style.dim('['),
+      style.yellow(String(map.children.length)),
+      style.dim(']')
+    )
   } else if (typeof map.value === 'string') {
-    result.push(' ', green(inspectNonTree(map.value)))
+    result.push(' ', style.green(inspectNonTree(map.value)))
   }
 
   if (position) {
-    result.push(' ', dim('('), position, dim(')'))
+    result.push(' ', style.dim('('), position, style.dim(')'))
   }
 
   return result.join('')