feat: prepare transform_insideRendering

package.json

@@ -83,7 +83,7 @@
     "@mdit-vue/types": "^2.1.0",
     "@types/markdown-it": "^14.1.2",
     "markdown-it": "^14.1.0",
-    "markdown-it-async": "^2.0.0",
+    "markdown-it-async": "^2.1.0",
     "unplugin": "^2.2.0",
     "unplugin-utils": "^0.2.4"
   },

src/core/markdown.ts

@@ -114,7 +114,13 @@ export function createMarkdown(options: ResolvedOptions) {
     raw = await transforms.before?.(raw, id) ?? raw
 
     const env: MarkdownEnv = { id }
-    let html = await md.renderAsync(raw, env)
+
+    let tokens = md.parse(raw, env)
+
+    tokens = await transforms.tokens?.(tokens, options.markdownItOptions, env, md) ?? tokens
+
+    let html = await md.rendererRenderAsync(tokens, md.options as any, env)
+
     const { excerpt = '', frontmatter: data = null } = env
 
     if (wrapperDiv) {

src/index.ts

@@ -1,7 +1,7 @@
 import type { UnpluginFactory } from 'unplugin'
 import type { Options } from './types'
-import { createFilter } from 'unplugin-utils'
 import { createUnplugin } from 'unplugin'
+import { createFilter } from 'unplugin-utils'
 import { createMarkdown } from './core/markdown'
 import { resolveOptions } from './core/options'
 

src/types.ts

@@ -6,6 +6,7 @@ import type {
   MarkdownItAsyncOptions,
   PluginSimple as MarkdownItPluginSimple,
   PluginWithOptions as MarkdownItPluginWithOptions,
+  Token,
 } from 'markdown-it-async'
 import type { FilterPattern } from 'unplugin-utils'
 import type { preprocessHead } from './core/head'
@@ -196,12 +197,22 @@ export interface Options {
    * Custom tranformations apply before and after the markdown transformation
    */
   transforms?: {
-    before?: (code: string, id: string) => string | Promise<string>
-    after?: (code: string, id: string) => string | Promise<string>
+    /**
+     * Pre-transformations
+     */
+    before?: (code: string, id: string) => Awaitable<string | void>
+    /**
+     * Transform internal MarkdownIt tokens
+     */
+    tokens?: (tokens: Token[], options: MarkdownItAsyncOptions, env: any, self: MarkdownItAsync) => Awaitable<Token[] | void>
+    /**
+     * Post-transformations
+     */
+    after?: (code: string, id: string) => Awaitable<string | void>
     /**
      * Return extra code to be injected into the `<script>` tag
      */
-    extraScripts?: (frontmatter: Record<string, any>, id: string) => string[] | Promise<string[]>
+    extraScripts?: (frontmatter: Record<string, any>, id: string) => Awaitable<string[] | void>
   }
 
   include?: FilterPattern
@@ -213,3 +224,5 @@ export interface ResolvedOptions extends Required<Options> { }
 export interface MarkdownEnv extends MarkdownItEnv {
   id: string
 }
+
+export type Awaitable<T> = T | Promise<T>