feat($core): config the dev and ssr template. (close: #733)

Author: ulivz

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

@@ -25,6 +25,7 @@ module.exports = async function build (sourceDir, cliOptions = {}) {
     return console.error(logger.error(chalk.red('Unexpected option: outDir cannot be set to the current working directory.\n'), false))
   }
   await fs.remove(outDir)
+  logger.debug('Dist directory: ' + chalk.gray(path.resolve(outDir)))
 
   let clientConfig = createClientConfig(options, cliOptions).toConfig()
   let serverConfig = createServerConfig(options, cliOptions).toConfig()
@@ -56,7 +57,7 @@ module.exports = async function build (sourceDir, cliOptions = {}) {
     runInNewContext: false,
     inject: false,
     shouldPrefetch: options.siteConfig.shouldPrefetch || (() => true),
-    template: await fs.readFile(path.resolve(__dirname, 'app/index.ssr.html'), 'utf-8')
+    template: await fs.readFile(options.ssrTemplate, 'utf-8')
   })
 
   // pre-render head tags from user config

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

@@ -65,7 +65,7 @@ module.exports = async function dev (sourceDir, cliOptions = {}) {
     // using a fork of html-webpack-plugin to avoid it requiring webpack
     // internals from an incompatible version.
     .use(require('vuepress-html-webpack-plugin'), [{
-      template: path.resolve(__dirname, 'app/index.dev.html')
+      template: options.devTemplate
     }])
 
   config

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

@@ -48,6 +48,7 @@ module.exports = class AppContext {
    */
   async process () {
     this.normalizeHeadTagUrls()
+    this.resolveTemplates()
     await this.resolveTheme()
     this.resolvePlugins()
 
@@ -124,6 +125,30 @@ module.exports = class AppContext {
     }
   }
 
+  /**
+   * Make template configurable
+   */
+  resolveTemplates () {
+    let { ssrTemplate, devTemplate } = this.siteConfig
+    const templateDir = path.resolve(this.vuepressDir, 'templates')
+    if (!devTemplate) {
+      devTemplate = path.resolve(templateDir, 'dev.html')
+      if (!fs.existsSync(devTemplate)) {
+        devTemplate = path.resolve(__dirname, '../app/index.dev.html')
+      }
+    }
+    if (!ssrTemplate) {
+      ssrTemplate = path.resolve(templateDir, 'ssr.html')
+      if (!fs.existsSync(ssrTemplate)) {
+        ssrTemplate = path.resolve(__dirname, '../app/index.ssr.html')
+      }
+    }
+    logger.debug('SSR Template File: ' + chalk.gray(ssrTemplate))
+    logger.debug('DEV Template File: ' + chalk.gray(devTemplate))
+    this.devTemplate = devTemplate
+    this.ssrTemplate = ssrTemplate
+  }
+
   /**
    * Find all page source files located in sourceDir
    * @returns {Promise<void>}