Add option "allowedTokenTypes"

nagaozen · Aug 26, 2021 · d74b36f · d74b36f
1 parent 0a9fe6d
commit d74b36f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/index.js b/index.js
@@ -35,7 +35,8 @@ function tocPlugin (md, options) {
     level: 1,
     listType: 'ol',
     format: undefined,
-    callback: undefined/* function(html, ast) {} */
+    callback: undefined/* function(html, ast) {} */,
+    allowedTokenTypes: ["text", "code_inline"]
   }, options)
 
   let ast
@@ -156,7 +157,7 @@ function tocPlugin (md, options) {
         const key = (
           tokens[i + 1]
             .children
-            .filter(function (token) { return token.type === 'text' || token.type === 'code_inline' })
+            .filter(function (token) { return options.allowedTokenTypes.indexOf(token.type) !== -1 })
             .reduce(function (s, t) { return s + t.content }, '')
         )
 

diff --git a/types/index.d.ts b/types/index.d.ts
@@ -13,7 +13,8 @@ declare module 'markdown-it-toc-done-right' {
         level: number | number[]
         listType: 'ol' | 'ul'
         format: (s: string) => string
-        callback: (tocCode: string, ast: TocAst) => void
+        callback: (tocCode: string, ast: TocAst) => void,
+        allowedTokenTypes: string[]
     }
 
     export interface TocAst {