Add the rest of the Streams API to installGlobals (#7321)

remix-run · Aug 31, 2023 · 8618338 · 8618338
1 parent 2c5e018
commit 8618338
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/.changeset/five-mugs-leave.md b/.changeset/five-mugs-leave.md
@@ -0,0 +1,5 @@
+---
+"@remix-run/node": patch
+---
+
+Add the rest of the Web Streams API to `installGlobals`
diff --git a/packages/remix-node/globals.ts b/packages/remix-node/globals.ts
@@ -1,6 +1,17 @@
 import {
+  ByteLengthQueuingStrategy as NodeByteLengthQueuingStrategy,
+  CountQueuingStrategy as NodeCountQueuingStrategy,
+  ReadableByteStreamController as NodeReadableByteStreamController,
   ReadableStream as NodeReadableStream,
+  ReadableStreamBYOBReader as NodeReadableStreamBYOBReader,
+  ReadableStreamBYOBRequest as NodeReadableStreamBYOBRequest,
+  ReadableStreamDefaultController as NodeReadableStreamDefaultController,
+  ReadableStreamDefaultReader as NodeReadableStreamDefaultReader,
+  TransformStream as NodeTransformStream,
+  TransformStreamDefaultController as NodeTransformStreamDefaultController,
   WritableStream as NodeWritableStream,
+  WritableStreamDefaultController as NodeWritableStreamDefaultController,
+  WritableStreamDefaultWriter as NodeWritableStreamDefaultWriter,
 } from "@remix-run/web-stream";
 
 import { atob, btoa } from "./base64";
@@ -52,6 +63,19 @@ export function installGlobals() {
   global.fetch = nodeFetch as typeof fetch;
   global.FormData = NodeFormData;
 
+  // Export everything from https://developer.mozilla.org/en-US/docs/Web/API/Streams_API
+  global.ByteLengthQueuingStrategy = NodeByteLengthQueuingStrategy;
+  global.CountQueuingStrategy = NodeCountQueuingStrategy;
+  global.ReadableByteStreamController = NodeReadableByteStreamController;
   global.ReadableStream = NodeReadableStream;
+  global.ReadableStreamBYOBReader = NodeReadableStreamBYOBReader;
+  global.ReadableStreamBYOBRequest = NodeReadableStreamBYOBRequest;
+  global.ReadableStreamDefaultController = NodeReadableStreamDefaultController;
+  global.ReadableStreamDefaultReader = NodeReadableStreamDefaultReader;
+  global.TransformStream = NodeTransformStream;
+  global.TransformStreamDefaultController =
+    NodeTransformStreamDefaultController;
   global.WritableStream = NodeWritableStream;
+  global.WritableStreamDefaultController = NodeWritableStreamDefaultController;
+  global.WritableStreamDefaultWriter = NodeWritableStreamDefaultWriter;
 }