run DCE on program exit (fix babel#289)

vigneshshanmugam · Nov 22, 2016 · 67611fd · kangax · Nov 23, 2016 · 67611fd
1 parent eb7b3c8
commit 67611fd
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 16 deletions.
diff --git a/packages/babel-plugin-minify-dead-code-elimination/src/index.js b/packages/babel-plugin-minify-dead-code-elimination/src/index.js
@@ -690,21 +690,23 @@ module.exports = ({ types: t, traverse }) => {
           path.remove();
         }
       },
-      Program(path, {
-        opts: {
-          // set defaults
-          optimizeRawSize = false,
-          keepFnName = false,
-          keepFnArgs = false,
-        } = {}
-      } = {}) {
-        // We need to run this plugin in isolation.
-        path.traverse(main, {
-          functionToBindings: new Map(),
-          optimizeRawSize,
-          keepFnName,
-          keepFnArgs,
-        });
+      Program: {
+        exit(path, {
+          opts: {
+            // set defaults
+            optimizeRawSize = false,
+            keepFnName = false,
+            keepFnArgs = false,
+          } = {}
+        } = {}) {
+          // We need to run this plugin in isolation.
+          path.traverse(main, {
+            functionToBindings: new Map(),
+            optimizeRawSize,
+            keepFnName,
+            keepFnArgs,
+          });
+        }
       },
     },
   };

diff --git a/packages/babel-preset-babili/__tests__/preset-tests.js b/packages/babel-preset-babili/__tests__/preset-tests.js
@@ -34,7 +34,7 @@ describe("preset", () => {
     const expected = unpad(`
       function foo() {
         var d, e, f;
-        d ? e && f : !b && f;
+        d ? e && f : e || f;
       }
     `);
     expect(transform(source)).toBe(expected);