From 5dbc73f76830f2e65088aab5cbff10c1b01c8d3d Mon Sep 17 00:00:00 2001
From: Boopathi Rajaa <boopathi@users.noreply.github.com>
Date: Thu, 17 May 2018 18:58:10 +0300
Subject: [PATCH] refactor(simplify): hoist path.get calls (#858)

---
 .../babel-plugin-minify-simplify/src/index.js | 47 +++++++++++--------
 1 file changed, 27 insertions(+), 20 deletions(-)

diff --git a/packages/babel-plugin-minify-simplify/src/index.js b/packages/babel-plugin-minify-simplify/src/index.js
index 7a1de9637..1845e1034 100644
--- a/packages/babel-plugin-minify-simplify/src/index.js
+++ b/packages/babel-plugin-minify-simplify/src/index.js
@@ -946,47 +946,54 @@ module.exports = ({ types: t }) => {
   }
 
   function earlyReturnTransform(path) {
-    const { node } = path;
+    const block = path.get("body");
 
-    if (!t.isBlockStatement(node.body)) {
+    if (!block.isBlockStatement()) {
       return;
     }
 
-    for (let i = node.body.body.length; i >= 0; i--) {
-      const statement = node.body.body[i];
+    const body = block.get("body");
+
+    for (let i = body.length - 1; i >= 0; i--) {
+      const statement = body[i];
       if (
-        t.isIfStatement(statement) &&
-        !statement.alternate &&
-        t.isReturnStatement(statement.consequent) &&
-        !statement.consequent.argument
+        t.isIfStatement(statement.node) &&
+        !statement.node.alternate &&
+        t.isReturnStatement(statement.node.consequent) &&
+        !statement.node.consequent.argument
       ) {
-        genericEarlyExitTransform(path.get("body").get("body")[i]);
+        genericEarlyExitTransform(statement);
       }
     }
   }
 
   function earlyContinueTransform(path) {
-    const { node } = path;
+    const block = path.get("body");
 
-    if (!t.isBlockStatement(node.body)) {
+    if (!block.isBlockStatement()) {
       return;
     }
 
-    for (let i = node.body.body.length; i >= 0; i--) {
-      const statement = node.body.body[i];
+    let body = block.get("body");
+
+    for (let i = body.length - 1; i >= 0; i--) {
+      const statement = body[i];
       if (
-        t.isIfStatement(statement) &&
-        !statement.alternate &&
-        t.isContinueStatement(statement.consequent) &&
-        !statement.consequent.label
+        t.isIfStatement(statement.node) &&
+        !statement.node.alternate &&
+        t.isContinueStatement(statement.node.consequent) &&
+        !statement.node.consequent.label
       ) {
-        genericEarlyExitTransform(path.get("body").get("body")[i]);
+        genericEarlyExitTransform(statement);
       }
     }
 
+    // because we might have folded or removed statements
+    body = block.get("body");
+
     // We may have reduced the body to a single statement.
-    if (node.body.body.length === 1 && !needsBlock(node.body, node)) {
-      path.get("body").replaceWith(node.body.body[0]);
+    if (body.length === 1 && !needsBlock(block.node, path.node)) {
+      block.replaceWith(body[0].node);
     }
   }