cool

trpc · Oct 26, 2023 · 4b0fe67 · 4b0fe67
1 parent 94751a1
commit 4b0fe67
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/src/async/iterableUtils.ts b/src/async/iterableUtils.ts
@@ -1,10 +1,16 @@
 import { assert } from "../internals/assert.js";
-import { WebReadableStreamEsque } from "../internals/esque.js";
+import {
+	NodeJSReadableStreamEsque,
+	WebReadableStreamEsque,
+} from "../internals/esque.js";
 
 export async function* readableStreamToAsyncIterable<T>(
-	stream: NodeJS.ReadableStream | ReadableStream<T> | WebReadableStreamEsque,
+	stream:
+		| NodeJSReadableStreamEsque
+		| ReadableStream<T>
+		| WebReadableStreamEsque,
 ): AsyncIterable<T> {
-	if (!("getReader" in stream)) {
+	if (Symbol.asyncIterator in stream) {
 		// NodeJS.ReadableStream
 		for await (const chunk of stream) {
 			yield chunk as T;

diff --git a/src/internals/esque.ts b/src/internals/esque.ts
@@ -13,3 +13,10 @@ export interface TextDecoderEsque {
 export interface WebReadableStreamEsque {
 	getReader: () => ReadableStreamDefaultReader<Uint8Array>;
 }
+
+/**
+ * @see NodeJS.ReadableStream from @types/node
+ */
+export interface NodeJSReadableStreamEsque {
+	[Symbol.asyncIterator]: () => AsyncIterableIterator<Buffer | string>;
+}