Define Headers to Parse during Page Prep - resolve #984

packages/@vuepress/core/lib/index.js

@@ -3,3 +3,4 @@
 exports.dev = require('./dev')
 exports.build = require('./build')
 exports.eject = require('./eject')
+exports.loadConfig = require('./prepare/loadConfig')

packages/@vuepress/core/lib/prepare/AppContext.js

@@ -290,6 +290,7 @@ module.exports = class AppContext {
 
   async addPage (options) {
     options.permalinkPattern = this.siteConfig.permalink
+    options.siteConfig = this.siteConfig
     const page = new Page(options, this)
     await page.process({
       markdown: this.markdown,

packages/@vuepress/core/lib/prepare/Page.js

@@ -45,7 +45,8 @@ module.exports = class Page {
     relative,
     permalink,
     frontmatter = {},
-    permalinkPattern
+    permalinkPattern,
+    siteConfig
   }, context) {
     this.title = title
     this._meta = meta
@@ -54,6 +55,7 @@ module.exports = class Page {
     this._permalink = permalink
     this.frontmatter = frontmatter
     this._permalinkPattern = permalinkPattern
+    this._siteConfig = siteConfig
     this._context = context
 
     if (relative) {
@@ -108,14 +110,18 @@ module.exports = class Page {
         }
 
         // headers
-        const headers = extractHeaders(
-          this._strippedContent,
-          ['h2', 'h3'],
-          markdown
-        )
-        if (headers.length) {
-          this.headers = headers
-        }
+      this.headersToExtract = ['h2', 'h3']
+      if (this._siteConfig.markdown && this._siteConfig.markdown.extractHeaders) {
+        this.headersToExtract = this._siteConfig.markdown.extractHeaders
+      }
+      const headers = extractHeaders(
+        this._strippedContent,
+        this.headersToExtract,
+        markdown
+      )
+      if (headers.length) {
+        this.headers = headers
+      }
 
         if (excerpt) {
           const { html } = markdown.render(excerpt)

packages/@vuepress/markdown-loader/index.js

@@ -6,7 +6,11 @@
 
 const { EventEmitter } = require('events')
 const { getOptions } = require('loader-utils')
-const { fs, path, hash, parseFrontmatter, inferTitle, extractHeaders } = require('@vuepress/shared-utils')
+const { loadConfig } = require('@vuepress/core')
+const {
+  fs, path, hash, parseFrontmatter, inferTitle, extractHeaders,
+  datatypes: { isFunction }
+} = require('@vuepress/shared-utils')
 const LRU = require('lru-cache')
 const md = require('@vuepress/markdown')
 
@@ -26,7 +30,10 @@ module.exports = function (src) {
   if (!markdown) {
     markdown = md()
   }
-
+  let config = loadConfig(path.resolve(sourceDir, '.vuepress'), false)
+  if (isFunction(config)) {
+    config = config(this)
+  }
   // we implement a manual cache here because this loader is chained before
   // vue-loader, and will be applied on the same file multiple times when
   // selecting the individual blocks.
@@ -42,7 +49,11 @@ module.exports = function (src) {
 
   if (!isProd && !isServer) {
     const inferredTitle = inferTitle(frontmatter.data, frontmatter.content)
-    const headers = extractHeaders(content, ['h2', 'h3'], markdown)
+    let headersToExtract = ['h2', 'h3']
+    if (config.markdown && config.markdown.extractHeaders) {
+      headersToExtract = this._siteConfig.markdown.extractHeaders
+    }
+    const headers = extractHeaders(content, headersToExtract, markdown)
     delete frontmatter.content
 
     // diff frontmatter and title, since they are not going to be part of the

packages/@vuepress/markdown-loader/package.json

@@ -18,6 +18,7 @@
   ],
   "dependencies": {
     "@vuepress/markdown": "^1.0.0-alpha.27",
+    "@vuepress/core": "^1.0.0-alpha.27",
     "loader-utils": "^1.1.0"
   },
   "author": "Evan You",

packages/docs/docs/theme/default-theme-config.md

@@ -152,6 +152,20 @@ module.exports = {
 }
 ```
 
+### Extract Headers
+
+While preparing the page, headers are extracted from the markdown file and stored in `this.$page.headers`. By default, VuePress will extract `h2` and `h3` elements for you.
+
+You can override the headers it pulls out in your `markdown` options.
+
+``` js
+module.exports = {
+  markdown: {
+    extractHeaders: [ 'h2', 'h3', 'h4' ]
+  }
+}
+```
+
 ### Active Header Links
 
 By default, the nested header links and the hash in the URL are updated as the user scrolls to view the different sections of the page. This behavior can be disabled with the following theme config:
@@ -329,7 +343,7 @@ module.exports = {
 ```
 
 ::: warning Note
-Unlike the [built-in search](#built-in-search) engine which works out of the box, [Algolia DocSearch](https://community.algolia.com/docsearch/) requires you to submit your site to them for indexing before it starts working. 
+Unlike the [built-in search](#built-in-search) engine which works out of the box, [Algolia DocSearch](https://community.algolia.com/docsearch/) requires you to submit your site to them for indexing before it starts working.
 :::
 
 For more options, refer to [Algolia DocSearch's documentation](https://github.com/algolia/docsearch#docsearch-options).
@@ -373,10 +387,10 @@ module.exports = {
   themeConfig: {
     serviceWorker: {
       updatePopup: true // Boolean | Object, default to undefined.
-      // If set to true, the default text config will be: 
-      // updatePopup: { 
-      //    message: "New content is available.", 
-      //    buttonText: "Refresh" 
+      // If set to true, the default text config will be:
+      // updatePopup: {
+      //    message: "New content is available.",
+      //    buttonText: "Refresh"
       // }
     }
   }