Fix codegen bug with iifes

Author: GearsDatapacks

CHANGELOG.md

@@ -395,3 +395,7 @@
 - Fixed a bug where the compiler would not correctly check the size of bit
   arrays when doing bit array pattern matching on the JavaScript target.
   ([Giacomo Cavalieri](https://github.com/giacomocavalieri))
+
+- Fixed a bug where using the pipe operator inside a record update would cause
+  the compiler to generate invalid code on the JavaScript target.
+  ([Surya Rose](https://github.com/GearsDatapacks))

compiler-core/src/javascript/expression.rs

@@ -225,9 +225,7 @@ impl<'module, 'a> Generator<'module, 'a> {
         } else {
             let mut statements = std::mem::take(&mut self.statement_level);
             statements.push(expression);
-            Itertools::intersperse(statements.into_iter(), line())
-                .collect_vec()
-                .to_doc()
+            join(statements.into_iter(), line())
         }
     }
 
@@ -582,7 +580,8 @@ impl<'module, 'a> Generator<'module, 'a> {
         self.scope_position = scope_position;
 
         // Wrap in iife document
-        let doc = immediately_invoked_function_expression_document(result?);
+        let doc =
+            immediately_invoked_function_expression_document(self.add_statement_level(result?));
         Ok(self.wrap_return(doc))
     }
 

compiler-core/src/javascript/tests/functions.rs

@@ -541,3 +541,22 @@ pub fn main() {
 "
     );
 }
+
+// https://github.com/gleam-lang/gleam/issues/4533
+#[test]
+fn immediately_invoked_function_expressions_include_statement_level() {
+    assert_js!(
+        "
+fn identity(x) { x }
+
+pub type Wibble {
+  Wibble(a: Int, b: Int)
+}
+
+pub fn main() {
+  let w = Wibble(1, 2)
+  identity(Wibble(..w |> identity, b: 4)) |> identity
+}
+"
+    );
+}

compiler-core/src/javascript/tests/snapshots/gleam_core__javascript__tests__functions__immediately_invoked_function_expressions_include_statement_level.snap

@@ -0,0 +1,42 @@
+---
+source: compiler-core/src/javascript/tests/functions.rs
+expression: "\nfn identity(x) { x }\n\npub type Wibble {\n  Wibble(a: Int, b: Int)\n}\n\npub fn main() {\n  let w = Wibble(1, 2)\n  identity(Wibble(..w |> identity, b: 4)) |> identity\n}\n"
+---
+----- SOURCE CODE
+
+fn identity(x) { x }
+
+pub type Wibble {
+  Wibble(a: Int, b: Int)
+}
+
+pub fn main() {
+  let w = Wibble(1, 2)
+  identity(Wibble(..w |> identity, b: 4)) |> identity
+}
+
+
+----- COMPILED JAVASCRIPT
+import { CustomType as $CustomType } from "../gleam.mjs";
+
+export class Wibble extends $CustomType {
+  constructor(a, b) {
+    super();
+    this.a = a;
+    this.b = b;
+  }
+}
+
+function identity(x) {
+  return x;
+}
+
+export function main() {
+  let w = new Wibble(1, 2);
+  let _pipe = identity((() => { let _block;
+      let _pipe = w;
+      _block = identity(_pipe);
+      let _record = _block;
+      return new Wibble(_record.a, 4); })());
+  return identity(_pipe);
+}