fix: Allow toplevel await when option true and correctly mark await keyword as unexpected (#9371)

packages/babel-parser/src/parser/statement.js

@@ -453,16 +453,19 @@ export default class StatementParser extends ExpressionParser {
     this.next();
     this.state.labels.push(loopLabel);
 
-    let forAwait = false;
+    let awaitAt = -1;
-    if (this.state.inAsync && this.isContextual("await")) {
+    if (
-      forAwait = true;
+      (this.state.inAsync ||
-      this.next();
+        (!this.state.inFunction && this.options.allowAwaitOutsideFunction)) &&
+      this.eatContextual("await")
+    ) {
+      awaitAt = this.state.lastTokStart;
     }
     this.expect(tt.parenL);
 
     if (this.match(tt.semi)) {
-      if (forAwait) {
+      if (awaitAt > -1) {
-        this.unexpected();
+        this.unexpected(awaitAt);
       }
       return this.parseFor(node, null);
     }
@@ -487,12 +490,12 @@ export default class StatementParser extends ExpressionParser {
           if (this.state.strict && isForInInitializer) {
             this.raise(this.state.start, "for-in initializer in strict mode");
           } else if (isForInInitializer || !declaration.init) {
-            return this.parseForIn(node, init, forAwait);
+            return this.parseForIn(node, init, awaitAt);
           }
         }
       }
-      if (forAwait) {
+      if (awaitAt > -1) {
-        this.unexpected();
+        this.unexpected(awaitAt);
       }
       return this.parseFor(node, init);
     }
@@ -505,12 +508,12 @@ export default class StatementParser extends ExpressionParser {
         : "for-in statement";
       this.toAssignable(init, undefined, description);
       this.checkLVal(init, undefined, undefined, description);
-      return this.parseForIn(node, init, forAwait);
+      return this.parseForIn(node, init, awaitAt);
     } else if (refShorthandDefaultPos.start) {
       this.unexpected(refShorthandDefaultPos.start);
     }
-    if (forAwait) {
+    if (awaitAt > -1) {
-      this.unexpected();
+      this.unexpected(awaitAt);
     }
     return this.parseFor(node, init);
   }
@@ -872,16 +875,16 @@ export default class StatementParser extends ExpressionParser {
   parseForIn(
     node: N.ForInOf,
     init: N.VariableDeclaration,
-    forAwait: boolean,
+    awaitAt: number,
   ): N.ForInOf {
     const type = this.match(tt._in) ? "ForInStatement" : "ForOfStatement";
-    if (forAwait) {
+    if (awaitAt > -1) {
       this.eatContextual("of");
     } else {
       this.next();
     }
     if (type === "ForOfStatement") {
-      node.await = !!forAwait;
+      node.await = awaitAt > -1;
     }
     node.left = init;
     node.right = this.parseExpression();

packages/babel-parser/test/fixtures/core/opts/allowAwaitOutsideFunction-false/input.js

@@ -0,0 +1 @@
+for await (const i of imports) {}

packages/babel-parser/test/fixtures/core/opts/allowAwaitOutsideFunction-false/options.json

@@ -0,0 +1,3 @@
+{
+  "throws": "Unexpected token, expected \"(\" (1:4)"
+}

packages/babel-parser/test/fixtures/core/opts/allowAwaitOutsideFunction-true/input.js

@@ -0,0 +1 @@
+for await (const i of imports) {}

packages/babel-parser/test/fixtures/core/opts/allowAwaitOutsideFunction-true/options.json

@@ -0,0 +1,3 @@
+{
+  "allowAwaitOutsideFunction": true
+}

packages/babel-parser/test/fixtures/core/opts/allowAwaitOutsideFunction-true/output.json

@@ -0,0 +1,136 @@
+{
+  "type": "File",
+  "start": 0,
+  "end": 33,
+  "loc": {
+    "start": {
+      "line": 1,
+      "column": 0
+    },
+    "end": {
+      "line": 1,
+      "column": 33
+    }
+  },
+  "program": {
+    "type": "Program",
+    "start": 0,
+    "end": 33,
+    "loc": {
+      "start": {
+        "line": 1,
+        "column": 0
+      },
+      "end": {
+        "line": 1,
+        "column": 33
+      }
+    },
+    "sourceType": "script",
+    "interpreter": null,
+    "body": [
+      {
+        "type": "ForOfStatement",
+        "start": 0,
+        "end": 33,
+        "loc": {
+          "start": {
+            "line": 1,
+            "column": 0
+          },
+          "end": {
+            "line": 1,
+            "column": 33
+          }
+        },
+        "await": true,
+        "left": {
+          "type": "VariableDeclaration",
+          "start": 11,
+          "end": 18,
+          "loc": {
+            "start": {
+              "line": 1,
+              "column": 11
+            },
+            "end": {
+              "line": 1,
+              "column": 18
+            }
+          },
+          "declarations": [
+            {
+              "type": "VariableDeclarator",
+              "start": 17,
+              "end": 18,
+              "loc": {
+                "start": {
+                  "line": 1,
+                  "column": 17
+                },
+                "end": {
+                  "line": 1,
+                  "column": 18
+                }
+              },
+              "id": {
+                "type": "Identifier",
+                "start": 17,
+                "end": 18,
+                "loc": {
+                  "start": {
+                    "line": 1,
+                    "column": 17
+                  },
+                  "end": {
+                    "line": 1,
+                    "column": 18
+                  },
+                  "identifierName": "i"
+                },
+                "name": "i"
+              },
+              "init": null
+            }
+          ],
+          "kind": "const"
+        },
+        "right": {
+          "type": "Identifier",
+          "start": 22,
+          "end": 29,
+          "loc": {
+            "start": {
+              "line": 1,
+              "column": 22
+            },
+            "end": {
+              "line": 1,
+              "column": 29
+            },
+            "identifierName": "imports"
+          },
+          "name": "imports"
+        },
+        "body": {
+          "type": "BlockStatement",
+          "start": 31,
+          "end": 33,
+          "loc": {
+            "start": {
+              "line": 1,
+              "column": 31
+            },
+            "end": {
+              "line": 1,
+              "column": 33
+            }
+          },
+          "body": [],
+          "directives": []
+        }
+      }
+    ],
+    "directives": []
+  }
+}

packages/babel-parser/test/fixtures/es2018/async-generators/for-await-no-semi-1/options.json

@@ -1,3 +1,3 @@
 {
-  "throws": "Unexpected token (2:13)"
+  "throws": "Unexpected token (2:6)"
 }

packages/babel-parser/test/fixtures/es2018/async-generators/for-await-no-semi-2/options.json

@@ -1,3 +1,3 @@
 {
-  "throws": "Unexpected token (2:22)"
+  "throws": "Unexpected token (2:6)"
 }

packages/babel-parser/test/fixtures/es2018/async-generators/for-await-no-semi-3/options.json

@@ -1,3 +1,3 @@
 {
-  "throws": "Unexpected token (2:18)"
+  "throws": "Unexpected token (2:6)"
 }