test for custom compiler modules

Author: yyx990803

lib/loader.js

@@ -80,7 +80,11 @@ module.exports = function (content) {
     id: moduleId,
     transformToRequire: options.transformToRequire,
     preserveWhitespace: options.preserveWhitespace,
-    buble: options.buble
+    buble: options.buble,
+    // only pass compilerModules if it's a path string
+    compilerModules: typeof options.compilerModules === 'string'
+      ? options.compilerModules
+      : undefined
   })
 
   var defaultLoaders = {

lib/template-transform-require.js renamed to lib/template-compiler-modules/transform-require.js

@@ -4,22 +4,30 @@ var compiler = require('vue-template-compiler')
 var beautify = require('js-beautify').js_beautify
 var transpile = require('vue-template-es2015-compiler')
 var hotReloadAPIPath = normalize.dep('vue-hot-reload-api')
-var transformRequire = require('./template-transform-require')
+var transformRequire = require('./template-compiler-modules/transform-require')
 
-var defaultCompileModules = [transformRequire]
+var defaultCompilerModules = [transformRequire]
 
 module.exports = function (html) {
   this.cacheable()
   var isServer = this.target === 'node'
   var isProduction = this.minimize || process.env.NODE_ENV === 'production'
+  var vueOptions = this.options.__vueOptions__ || {}
   var options = loaderUtils.getOptions(this) || {}
 
   // set user transformToRequire options on module
   transformRequire._userOptions = options.transformToRequire
 
+  // user compiler modules
+  var userCompilerModules = options.compilerModules || vueOptions.compilerModules
+  // for HappyPack cross-process use cases
+  if (typeof userCompilerModules === 'string') {
+    userCompilerModules = require(userCompilerModules)
+  }
+
   var compiled = compiler.compile(html, {
     preserveWhitespace: options.preserveWhitespace,
-    modules: defaultCompileModules.concat(options.modules || [])
+    modules: defaultCompilerModules.concat(userCompilerModules || [])
   })
 
   // tips

lib/template-compiler.js

@@ -0,0 +1,9 @@
+module.exports = [
+  {
+    postTransformNode: el => {
+      if (el.staticClass) {
+        el.staticClass = '"red blue"'
+      }
+    }
+  }
+]

test/fixtures/custom-compiler-module.js

@@ -700,4 +700,42 @@ describe('vue-loader', function () {
       done()
     })
   })
+
+  it('custom compiler modules', done => {
+    test({
+      entry: './test/fixtures/basic.vue',
+      vue: {
+        compilerModules: [
+          {
+            postTransformNode: el => {
+              if (el.staticClass) {
+                el.staticClass = '"red blue"'
+              }
+            }
+          }
+        ]
+      }
+    }, (window, module) => {
+      var vnode = mockRender(module, {
+        msg: 'hi'
+      })
+      expect(vnode.data.staticClass).to.equal('red blue')
+      done()
+    })
+  })
+
+  it('custom compiler modules (string)', done => {
+    test({
+      entry: './test/fixtures/basic.vue',
+      vue: {
+        compilerModules: require.resolve('./fixtures/custom-compiler-module')
+      }
+    }, (window, module) => {
+      var vnode = mockRender(module, {
+        msg: 'hi'
+      })
+      expect(vnode.data.staticClass).to.equal('red blue')
+      done()
+    })
+  })
 })