feat: add modifyDefaultTemplate for plugin

wangxue16292016 · Jun 28, 2018 · 8009fc6 · 8009fc6
1 parent 0ad6e0b
commit 8009fc6
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/packages/umi-build-dev/src/html/getHTMLContent.js b/packages/umi-build-dev/src/html/getHTMLContent.js
@@ -22,10 +22,17 @@ export default function(path, service, chunksMap, minifyHTML, isProduction) {
   const customizedDocPath = document
     ? join(paths.cwd, document)
     : paths.absPageDocumentPath;
-  const docPath = existsSync(customizedDocPath)
+  const existsCustomTpl = existsSync(customizedDocPath);
+  const docPath = existsCustomTpl
     ? customizedDocPath
     : paths.defaultDocumentPath;
-  const tpl = readFileSync(docPath, 'utf-8');
+  let tpl = readFileSync(docPath, 'utf-8');
+
+  if (!existsCustomTpl) {
+    tpl = service.applyPlugins('modifyDefaultTemplate', {
+      initialValue: tpl,
+    });
+  }
 
   if (config.exportStatic && !config.exportStatic.htmlSuffix) {
     path = makeSureLastSlash(path);