fix: handle NodeJS.ReadableStream (#81)

Author: KATT

src/async/iterableUtils.ts

@@ -1,8 +1,24 @@
 import { assert } from "../internals/assert.js";
+import {
+	NodeJSReadableStreamEsque,
+	WebReadableStreamEsque,
+} from "../internals/esque.js";
 
 export async function* readableStreamToAsyncIterable<T>(
-	stream: ReadableStream<T>,
+	stream:
+		| NodeJSReadableStreamEsque
+		| ReadableStream<T>
+		| WebReadableStreamEsque,
 ): AsyncIterable<T> {
+	if (Symbol.asyncIterator in stream) {
+		// NodeJS.ReadableStream
+		for await (const chunk of stream) {
+			yield chunk as T;
+		}
+
+		return;
+	}
+
 	// Get a lock on the stream
 	const reader = stream.getReader();
 
@@ -18,7 +34,7 @@ export async function* readableStreamToAsyncIterable<T>(
 			}
 
 			// Else yield the chunk
-			yield result.value;
+			yield result.value as T;
 		}
 	} finally {
 		reader.releaseLock();

src/internals/esque.ts

@@ -4,3 +4,19 @@
 export interface TextDecoderEsque {
 	decode(chunk: Uint8Array): string;
 }
+
+/**
+ * A subset of the standard ReadableStream properties needed by tRPC internally.
+ * @see ReadableStream from lib.dom.d.ts
+ * @internal
+ */
+export interface WebReadableStreamEsque {
+	getReader: () => ReadableStreamDefaultReader<Uint8Array>;
+}
+
+/**
+ * @see NodeJS.ReadableStream from @types/node
+ */
+export interface NodeJSReadableStreamEsque {
+	[Symbol.asyncIterator]: () => AsyncIterableIterator<unknown>;
+}