perf: scan scopeStack for once

babel · Jun 9, 2021 · c9155bd · c9155bd
1 parent e92666f
commit c9155bd
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/packages/babel-parser/src/parser/expression.js b/packages/babel-parser/src/parser/expression.js
@@ -2415,7 +2415,7 @@ export default class ExpressionParser extends LValParser {
       if (this.prodParam.hasAwait) {
         this.raise(startLoc, Errors.AwaitBindingIdentifier);
         return;
-      } else if (this.scope.inStaticBlock && !this.scope.inFunction) {
+      } else if (this.scope.inStaticBlock) {
         this.raise(startLoc, Errors.AwaitBindingIdentifierInStaticBlock);
         return;
       } else {

diff --git a/packages/babel-parser/src/util/scope.js b/packages/babel-parser/src/util/scope.js
@@ -65,7 +65,20 @@ export default class ScopeHandler<IScope: Scope = Scope> {
     return (flags & SCOPE_CLASS) > 0 && (flags & SCOPE_FUNCTION) === 0;
   }
   get inStaticBlock() {
-    return (this.currentThisScopeFlags() & SCOPE_STATIC_BLOCK) > 0;
+    for (let i = this.scopeStack.length - 1; ; i--) {
+      const { flags } = this.scopeStack[i];
+      if (flags & SCOPE_VAR) {
+        // function body, module body
+        return false;
+      }
+      if (flags & SCOPE_STATIC_BLOCK) {
+        return true;
+      }
+      if (flags & SCOPE_CLASS) {
+        // class property initializers
+        return false;
+      }
+    }
   }
   get inNonArrowFunction() {
     return (this.currentThisScopeFlags() & SCOPE_FUNCTION) > 0;