fix(plugin-legacy): move polyfills in plugin post

vitejs · May 3, 2021 · babd45a · babd45a
1 parent 38dd448
commit babd45a
Showing 1 changed file with 6 additions and 10 deletions.
diff --git a/packages/plugin-legacy/index.js b/packages/plugin-legacy/index.js
@@ -533,17 +533,13 @@ function isLegacyOutput(options) {
 function recordAndRemovePolyfillBabelPlugin(polyfills) {
   return ({ types: t }) => ({
     name: 'vite-remove-polyfill-import',
-    visitor: {
-      Program: {
-        exit(path) {
-          path.get('body').forEach((p) => {
-            if (t.isImportDeclaration(p)) {
-              polyfills.add(p.node.source.value)
-              p.remove()
-            }
-          })
+    post({ path }) {
+      path.get('body').forEach((p) => {
+        if (t.isImportDeclaration(p)) {
+          polyfills.add(p.node.source.value)
+          p.remove()
         }
-      }
+      })
     }
   })
 }