Fix to align large indices

Closes GH-18.
syntax-tree · Jul 23, 2022 · 42b5370 · 42b5370
1 parent 56f6980
commit 42b5370
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -86,17 +86,22 @@ export function inspectColor(tree, options = {}) {
    * @returns {string}
    */
   function inspectNodes(nodes) {
+    const size = String(nodes.length - 1).length
     /** @type {Array<string>} */
     const result = []
     let index = -1
 
     while (++index < nodes.length) {
       result.push(
-        dim((index < nodes.length - 1 ? '├' : '└') + '─' + index) +
+        dim(
+          (index < nodes.length - 1 ? '├' : '└') +
+            '─' +
+            String(index).padEnd(size)
+        ) +
           ' ' +
           indent(
             inspectValue(nodes[index]),
-            (index < nodes.length - 1 ? dim('│') : ' ') + '   ',
+            (index < nodes.length - 1 ? dim('│') : ' ') + ' '.repeat(size + 2),
             true
           )
       )

diff --git a/test.js b/test.js
@@ -64,6 +64,43 @@ test('inspect()', (t) => {
     st.end()
   })
 
+  t.equal(
+    strip(
+      inspect(
+        Array.from({length: 11}).map((/** @type {undefined} */ d, i) => ({
+          type: 'text',
+          value: String(i),
+          data: {id: String.fromCodePoint(97 + i)}
+        }))
+      )
+    ),
+    [
+      '├─0  text "0"',
+      '│      data: {"id":"a"}',
+      '├─1  text "1"',
+      '│      data: {"id":"b"}',
+      '├─2  text "2"',
+      '│      data: {"id":"c"}',
+      '├─3  text "3"',
+      '│      data: {"id":"d"}',
+      '├─4  text "4"',
+      '│      data: {"id":"e"}',
+      '├─5  text "5"',
+      '│      data: {"id":"f"}',
+      '├─6  text "6"',
+      '│      data: {"id":"g"}',
+      '├─7  text "7"',
+      '│      data: {"id":"h"}',
+      '├─8  text "8"',
+      '│      data: {"id":"i"}',
+      '├─9  text "9"',
+      '│      data: {"id":"j"}',
+      '└─10 text "10"',
+      '       data: {"id":"k"}'
+    ].join('\n'),
+    'should align and indent large numbers'
+  )
+
   t.equal(
     strip(
       inspect({