From 4bc4331e55f87f94c06b4acc4ff1dc088561d328 Mon Sep 17 00:00:00 2001
From: ULIVZ <472590061@qq.com>
Date: Sat, 22 Sep 2018 23:40:16 +0800
Subject: [PATCH] feat($core): use markdown-it-chain

---
 packages/@vuepress/core/lib/markdown/index.js | 117 +++++++++++++-----
 .../@vuepress/core/lib/prepare/AppContext.js  |   2 +-
 2 files changed, 85 insertions(+), 34 deletions(-)

diff --git a/packages/@vuepress/core/lib/markdown/index.js b/packages/@vuepress/core/lib/markdown/index.js
index b398e611..01e57983 100644
--- a/packages/@vuepress/core/lib/markdown/index.js
+++ b/packages/@vuepress/core/lib/markdown/index.js
@@ -1,3 +1,10 @@
+'use strict'
+
+/**
+ * Module dependencies.
+ */
+
+const Config = require('markdown-it-chain')
 const highlight = require('./highlight')
 const highlightLines = require('./highlightLines')
 const preWrapper = require('./preWrapper')
@@ -13,49 +20,93 @@ const toc = require('markdown-it-table-of-contents')
 const _slugify = require('./slugify')
 const { parseHeaders } = require('@vuepress/shared-utils')
 
-module.exports = ({ markdown = {}} = {}) => {
+/**
+ * Create markdown by config.
+ */
+
+module.exports = ({
+  markdown = {}
+} = {}) => {
   // allow user config slugify
   const slugify = markdown.slugify || _slugify
+  // using chainedAPI
+  const config = new Config()
 
-  const md = require('markdown-it')({
-    html: true,
-    highlight
-  })
-    // custom plugins
-    .use(component)
-    .use(highlightLines)
-    .use(preWrapper)
-    .use(snippet)
-    .use(convertRouterLink, Object.assign({
-      target: '_blank',
-      rel: 'noopener noreferrer'
-    }, markdown.externalLinks))
-    .use(hoistScriptStyle)
-    .use(containers)
+  config
+    .options
+      .html(true)
+      .highlight(highlight)
+      .end()
 
-    // 3rd party plugins
-    .use(emoji)
-    .use(anchor, Object.assign({
-      slugify,
-      permalink: true,
-      permalinkBefore: true,
-      permalinkSymbol: '#'
-    }, markdown.anchor))
-    .use(toc, Object.assign({
-      slugify,
-      includeLevel: [2, 3],
-      format: parseHeaders
-    }, markdown.toc))
+    .plugin('component')
+      .use(component)
+      .end()
+
+    .plugin('highlight-lines')
+      .use(highlightLines)
+      .end()
+
+    .plugin('pre-wrapper')
+      .use(preWrapper)
+      .end()
+
+    .plugin('snippet')
+      .use(snippet)
+      .end()
+
+    .plugin('convert-router-link')
+      .use(convertRouterLink, [Object.assign({
+        target: '_blank',
+        rel: 'noopener noreferrer'
+      }, markdown.externalLinks)])
+      .end()
+
+    .plugin('hoist-script-style')
+      .use(hoistScriptStyle)
+      .end()
+
+    .plugin('containers')
+      .use(containers)
+      .end()
+
+    .plugin('emoji')
+      .use(emoji)
+      .end()
+
+    .plugin('anchor')
+      .use(anchor, [Object.assign({
+        slugify,
+        permalink: true,
+        permalinkBefore: true,
+        permalinkSymbol: '#'
+      }, markdown.anchor)])
+      .end()
+
+    .plugin('toc')
+      .use(toc, [Object.assign({
+        slugify,
+        includeLevel: [2, 3],
+        format: parseHeaders
+      }, markdown.toc)])
+      .end()
+
+  if (markdown.lineNumbers) {
+    config
+      .plugin('line-numbers')
+        .use(lineNumbers)
+  }
+
+  if (markdown.chainMarkdown) {
+    markdown.chainMarkdown(config)
+  }
+
+  const md = config.toMd()
 
   // apply user config
   if (markdown.config) {
     markdown.config(md)
   }
 
-  if (markdown.lineNumbers) {
-    md.use(lineNumbers)
-  }
-
   module.exports.dataReturnable(md)
 
   // expose slugify
diff --git a/packages/@vuepress/core/lib/prepare/AppContext.js b/packages/@vuepress/core/lib/prepare/AppContext.js
index 4420296a..89d17a00 100644
--- a/packages/@vuepress/core/lib/prepare/AppContext.js
+++ b/packages/@vuepress/core/lib/prepare/AppContext.js
@@ -40,7 +40,6 @@ module.exports = class AppContext {
       ? path.resolve(this.siteConfig.dest)
       : path.resolve(sourceDir, '.vuepress/dist')
 
-    this.markdown = createMarkdown(this.siteConfig)
     this.pluginAPI = new PluginAPI(this)
     this.pages = [] // Array<Page>
     this.I18nConstructor = I18n(null)
@@ -52,6 +51,7 @@ module.exports = class AppContext {
    */
   async process () {
     this.normalizeHeadTagUrls()
+    this.markdown = createMarkdown(this.siteConfig)
     this.resolveTemplates()
     await this.resolveTheme()
     this.resolvePlugins()