fix transform (#154)

rollup · Apr 11, 2020 · 1b6e0e7 · 1b6e0e7
1 parent 64bc64b
commit 1b6e0e7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/packages/rollup-plugin-babel/src/helperPlugin.js b/packages/rollup-plugin-babel/src/helperPlugin.js
@@ -1,8 +1,8 @@
 export default function importHelperPlugin () {
 	return {
 		visitor: {
-			Program (path, state) {
-				state.file.addHelper('classCallCheck');
+			ClassDeclaration (path, state) {
+				path.replaceWith(state.file.addHelper('classCallCheck'));
 			}
 		}
 	};

diff --git a/packages/rollup-plugin-babel/test/helperPlugin.js b/packages/rollup-plugin-babel/test/helperPlugin.js
@@ -1,8 +1,8 @@
 module.exports = function importHelperPlugin () {
   return {
     visitor: {
-      Program (path, state) {
-        state.file.addHelper('classCallCheck');
+      ClassDeclaration (path, state) {
+        path.replaceWith(state.file.addHelper('classCallCheck'));
       }
     }
   };