feat(core): pass raw token to transformer.span api (#782)

zzuu666 · web-flow · commit bba452c41a7b · 2024-09-26T09:35:42.000+08:00
diff --git a/packages/core/src/highlight/code-to-hast.ts b/packages/core/src/highlight/code-to-hast.ts
@@ -180,7 +180,7 @@ export function tokensToHast(
         tokenNode.properties.style = style
 
       for (const transformer of transformers)
-        tokenNode = transformer?.span?.call(context, tokenNode, idx + 1, col, lineNode) || tokenNode
+        tokenNode = transformer?.span?.call(context, tokenNode, idx + 1, col, lineNode, token) || tokenNode
 
       if (structure === 'inline')
         root.children.push(tokenNode)
diff --git a/packages/types/src/transformers.ts b/packages/types/src/transformers.ts
@@ -81,7 +81,7 @@ export interface ShikiTransformer {
   /**
    * Transform each token `<span>` element.
    */
-  span?: (this: ShikiTransformerContext, hast: Element, line: number, col: number, lineElement: Element) => Element | void
+  span?: (this: ShikiTransformerContext, hast: Element, line: number, col: number, lineElement: Element, token: ThemedToken) => Element | void
   /**
    * Transform the generated HTML string before returning.
    * This hook will only be called with `codeToHtml`.
