enhance: child type should be token type

huozhi · Jan 25, 2025 · 909d689 · 909d689
1 parent 73a1d6d
commit 909d689
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -659,20 +659,21 @@ function generate(tokens) {
     /** @type {Array<any>} */
     const lineTokens = (
       tokens
-        .map(([type, value]) => (
-          {
+        .map(([type, value]) => {
+          const tokenType = TokenTypes[type]
+          return {
             type: 'element',
             tagName: 'span',
             children: [{
-              type, // token type
+              type: tokenType, // token type
               value: value, // to encode
             }],
             properties: {
-              className: `sh__token--${TokenTypes[type]}`,
-              style: { color: `var(--sh-${TokenTypes[type]})` },
+              className: `sh__token--${tokenType}`,
+              style: { color: `var(--sh-${tokenType})` },
             },
           }
-        ))
+        })
     )
     lines.push(createLine(lineTokens))
   }