toc

quartz/plugins/transformers/description.ts

@@ -15,7 +15,7 @@ export class Description extends QuartzTransformerPlugin {
   name = "Description"
   opts: Options
 
-  constructor(opts?: Options) {
+  constructor(opts?: Partial<Options>) {
     super()
     this.opts = { ...defaultOptions, ...opts }
   }

quartz/plugins/transformers/frontmatter.ts

@@ -17,7 +17,7 @@ export class FrontMatter extends QuartzTransformerPlugin {
   name = "FrontMatter"
   opts: Options
 
-  constructor(opts?: Options) {
+  constructor(opts?: Partial<Options>) {
     super()
     this.opts = { ...defaultOptions, ...opts }
   }

quartz/plugins/transformers/gfm.ts

@@ -19,7 +19,7 @@ export class GitHubFlavoredMarkdown extends QuartzTransformerPlugin {
   name = "GitHubFlavoredMarkdown"
   opts: Options
 
-  constructor(opts?: Options) {
+  constructor(opts?: Partial<Options>) {
     super()
     this.opts = { ...defaultOptions, ...opts }
   }

quartz/plugins/transformers/index.ts

@@ -6,3 +6,4 @@ export { Description } from './description'
 export { ResolveLinks } from './links'
 export { ObsidianFlavoredMarkdown } from './ofm'
 export { SyntaxHighlighting } from './syntax'
+export { TableOfContents } from './toc'

quartz/plugins/transformers/lastmod.ts

@@ -16,7 +16,7 @@ export class CreatedModifiedDate extends QuartzTransformerPlugin {
   name = "CreatedModifiedDate"
   opts: Options
 
-  constructor(opts?: Options) {
+  constructor(opts?: Partial<Options>) {
     super()
     this.opts = {
       ...defaultOptions,

quartz/plugins/transformers/links.ts

@@ -21,7 +21,7 @@ export class ResolveLinks extends QuartzTransformerPlugin {
   name = "LinkProcessing"
   opts: Options
 
-  constructor(opts?: Options) {
+  constructor(opts?: Partial<Options>) {
     super()
     this.opts = { ...defaultOptions, ...opts }
   }

quartz/plugins/transformers/ofm.ts

@@ -93,7 +93,7 @@ export class ObsidianFlavoredMarkdown extends QuartzTransformerPlugin {
   name = "ObsidianFlavoredMarkdown"
   opts: Options
 
-  constructor(opts?: Options) {
+  constructor(opts?: Partial<Options>) {
     super()
     this.opts = { ...defaultOptions, ...opts }
   }

quartz/plugins/transformers/toc.ts

@@ -0,0 +1,72 @@
+import { PluggableList } from "unified"
+import { QuartzTransformerPlugin } from "../types"
+import { Root } from "mdast"
+import { visit } from "unist-util-visit"
+import { toString } from "mdast-util-to-string"
+import { slugAnchor } from "../../path"
+
+export interface Options {
+  maxDepth: 1 | 2 | 3 | 4 | 5 | 6,
+  minEntries: 1,
+  showByDefault: boolean
+}
+
+const defaultOptions: Options = {
+  maxDepth: 3,
+  minEntries: 1,
+  showByDefault: true,
+}
+
+interface TocEntry {
+  depth: number,
+  text: string,
+  slug: string
+}
+
+export class TableOfContents extends QuartzTransformerPlugin {
+  name = "TableOfContents"
+  opts: Options
+
+  constructor(opts?: Partial<Options>) {
+    super()
+    this.opts = { ...defaultOptions, ...opts }
+  }
+
+  markdownPlugins(): PluggableList {
+    return [() => {
+      return async (tree: Root, file) => {
+        const display = file.data.frontmatter?.enableToc ?? this.opts.showByDefault
+        if (display) {
+          const toc: TocEntry[] = []
+          let highestDepth: number = this.opts.maxDepth
+          visit(tree, 'heading', (node) => {
+            if (node.depth <= this.opts.maxDepth) {
+              const text = toString(node)
+              highestDepth = Math.min(highestDepth, node.depth)
+              toc.push({
+                depth: node.depth,
+                text,
+                slug: slugAnchor.slug(text)
+              })
+            }
+          })
+
+          if (toc.length > this.opts.minEntries) {
+            file.data.toc = toc.map(entry => ({ ...entry, depth: entry.depth - highestDepth }))
+          }
+        }
+      }
+    }]
+  }
+
+  htmlPlugins(): PluggableList {
+    return []
+  }
+}
+
+declare module 'vfile' {
+  interface DataMap {
+    toc: TocEntry[]
+  }
+}
+