refactor(traverse): simpler code for entering/exiting unconditional s…

…copes
oxc-project · Aug 6, 2024 · 23f6efd · 23f6efd
1 parent 796d58d
commit 23f6efd
Show file tree

Hide file tree

Showing 2 changed files with 139 additions and 182 deletions.
diff --git a/crates/oxc_traverse/scripts/lib/walk.mjs b/crates/oxc_traverse/scripts/lib/walk.mjs
@@ -78,19 +78,28 @@ function generateWalkForStruct(type, types) {
         // but we don't take that into account.
         // Visitor should not do that though, so maybe it's OK.
         // In final version, we should not make `scope_id` fields `Cell`s to prevent this.
-        enterScopeCode = `
-            let mut previous_scope_id = None;
-            if let Some(scope_id) = (*(${makeFieldCode(scopeIdField)})).get() {
-                previous_scope_id = Some(ctx.current_scope_id());
-                ctx.set_current_scope_id(scope_id);
-            }
-        `;
+        if (scopeArgs.if) {
+            enterScopeCode = `
+                let mut previous_scope_id = None;
+                if let Some(scope_id) = (*(${makeFieldCode(scopeIdField)})).get() {
+                    previous_scope_id = Some(ctx.current_scope_id());
+                    ctx.set_current_scope_id(scope_id);
+                }
+            `;
 
-        exitScopeCode = `
-            if let Some(previous_scope_id) = previous_scope_id {
-                ctx.set_current_scope_id(previous_scope_id);
-            }
-        `;
+            exitScopeCode = `
+                if let Some(previous_scope_id) = previous_scope_id {
+                    ctx.set_current_scope_id(previous_scope_id);
+                }
+            `;
+        } else {
+            enterScopeCode = `
+                let previous_scope_id = ctx.current_scope_id();
+                ctx.set_current_scope_id((*(${makeFieldCode(scopeIdField)})).get().unwrap());
+            `;
+
+            exitScopeCode = `ctx.set_current_scope_id(previous_scope_id);`;
+        }
     }
 
     const fieldsCodes = visitedFields.map((field, index) => {