import injector again

nicholaslee119 · Oct 17, 2017 · b1d808b · b1d808b
1 parent 294588c
commit b1d808b
Show file tree

Hide file tree

Showing 8 changed files with 3,363 additions and 505 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,4 +1,6 @@
 node_modules
 test/assets
 test/assetsCoreTest
-test/coverage
+test/coverage
+.idea
+jest_0
diff --git a/index.js b/index.js
@@ -32,6 +32,7 @@ module.exports = function loaderEntry(source) {
   checkPath(options.srcPath, options.builtTemplatePath, this.resourcePath);
   checkFunction(options.extractor, options.addScopeAttr);
   checkFunction(options.addScopeAttr);
+  checkFunction(options.injector);
   checkSource(source);
 
   const buildOption = {
@@ -40,9 +41,11 @@ module.exports = function loaderEntry(source) {
     builtTemplatePath: options.builtTemplatePath,
     isCodeSplit: options.isCodeSplit,
     extractor: options.extractor,
+    injector: options.injector,
     addScopeAttr: options.addScopeAttr,
     currentPagePath: this.resourcePath,
-  }
+    publicPath: this.options.output.publicPath || '',
+  };
 
   return core(source, buildOption);
 };
diff --git a/lib/rewriteTemplate.js b/lib/rewriteTemplate.js
@@ -8,7 +8,8 @@ module.exports = async function rewriteTemplate(component) {
   if (fsx.existsSync(templatePath)) {
     try {
       const template = fsx.readFileSync(templatePath, 'utf8');
-      const res = await buildOption.addScopeAttr(template, component);
+      let res = await buildOption.addScopeAttr(template, component);
+      if (component.pageFlag) res = buildOption.injector(res, component, buildOption);
       fsx.ensureDirSync(path.resolve(buildOption.builtTemplatePath, component.dir));
       fsx.writeFileSync(path.resolve(buildOption.builtTemplatePath, component.dir, `${component.base}`), res);
     } catch (e) {