perf: use template argument property instead of babel-template

59naga · May 4, 2016 · a07c274 · a07c274 · jdalton · May 6, 2016
1 parent d1922e6
commit a07c274
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 10 deletions.
diff --git a/package.json b/package.json
@@ -38,9 +38,6 @@
       "watch": false
     }
   },
-  "dependencies": {
-    "babel-template": "^6.5.0"
-  },
   "devDependencies": {
     "abigail": "^1.6.1",
     "babel-cli": "^6.5.1",

diff --git a/src/index.js b/src/index.js
@@ -1,6 +1,4 @@
-import babelTemplate from 'babel-template'
-
-module.exports = {
+module.exports = ({template}) => ({
   visitor: {
     Program: {
       exit (path) {
@@ -30,7 +28,7 @@ module.exports = {
 
         if (hasExportDefault && !hasExportNamed) {
           const topNodes = []
-          topNodes.push(babelTemplate("module.exports = exports['default']")())
+          topNodes.push(template("module.exports = exports['default']")())
 
           path.pushContainer('body', topNodes)
         }
@@ -39,4 +37,4 @@ module.exports = {
       }
     }
   }
-}
+})
diff --git a/test/index.js b/test/index.js
@@ -13,8 +13,7 @@ describe('babel-plugin-add-module-exports', () => {
 
   it('plugin should export to module.exports(#31)', () => {
     const plugin = require('../lib')
-    assert(typeof plugin === 'object')
-    assert(typeof plugin.visitor === 'object')
+    assert(typeof plugin === 'function')
   })
 
   it('should handle duplicated plugin references (#1)', () =>