[GR-59077] Backport to 24.1: Fix of top-level for-await-of in a module.

PullRequest: js/3338

Author: hachim-hassani

graal-js/src/com.oracle.js.parser/src/com/oracle/js/parser/Parser.java

@@ -2957,6 +2957,13 @@ private void forStatement(boolean yield, boolean await) {
                     if (!await) {
                         throw error(AbstractParser.message(MSG_INVALID_FOR_AWAIT_OF), token);
                     }
+                    if (isModule) {
+                        ParserContextFunctionNode currentFunction = lc.getCurrentFunction();
+                        if (currentFunction.isModule()) {
+                            // Top-level for-await: mark the module function as async.
+                            currentFunction.setFlag(FunctionNode.IS_ASYNC);
+                        }
+                    }
                     isForAwaitOf = true;
                     next();
                 }

graal-js/src/com.oracle.js.parser/src/com/oracle/js/parser/ParserContextFunctionNode.java

@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2014, 2023, Oracle and/or its affiliates. All rights reserved.
+ * Copyright (c) 2014, 2024, Oracle and/or its affiliates. All rights reserved.
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  *
  * The Universal Permissive License (UPL), Version 1.0
@@ -167,7 +167,7 @@ public boolean isStrict() {
     }
 
     /**
-     * @return if function in strict mode
+     * @return true if the function is a module.
      */
     public boolean isModule() {
         return getFlag(FunctionNode.IS_MODULE) != 0;

graal-js/src/com.oracle.truffle.js.test/js/module/GR-59025.mjs

@@ -0,0 +1,16 @@
+/*
+ * Copyright (c) 2024, 2024, Oracle and/or its affiliates. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * Licensed under the Universal Permissive License v 1.0 as shown at http://oss.oracle.com/licenses/upl.
+ */
+
+/**
+ * Regression test of the usage of for-await-of in the body of a module.
+ */
+
+let sum = 0;
+for await (let x of [42, 211]) sum += x;
+if (sum !== 42 + 211) {
+   throw new Error("Sum of 24 and 211 is not " + sum);
+}