feat: support response objects in InputFile (#434)

grammyjs · Jun 25, 2023 · 29d6a88 · 29d6a88
1 parent ecf1abe
commit 29d6a88
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/src/types.deno.ts b/src/types.deno.ts
@@ -60,6 +60,7 @@ export class InputFile {
             | string
             | Blob
             | Deno.FsFile
+            | Response
             | URL
             | URLLike
             | Uint8Array
@@ -118,6 +119,11 @@ export class InputFile {
         }
         if (data instanceof Blob) return data.stream();
         if (isDenoFile(data)) return iterateReader(data);
+        // Handle Response objects
+        if (data instanceof Response) {
+            if (data.body === null) throw new Error(`No response body!`);
+            return data.body;
+        }
         // Handle URL and URLLike objects
         if (data instanceof URL) return fetchFile(data);
         if ("url" in data) return fetchFile(data.url);

diff --git a/test/types.test.ts b/test/types.test.ts
@@ -152,6 +152,18 @@ Deno.test({
     },
 });
 
+Deno.test({
+    name: "convert Response to raw",
+    async fn() {
+        const bytes = new Uint8Array([65, 66, 67]);
+        const file0 = new InputFile(new Response(bytes));
+        const data0 = await file0.toRaw();
+        if (data0 instanceof Uint8Array) throw new Error("no itr");
+        const values0 = await readAll(readerFromIterable(data0));
+        assertEquals(values0, bytes);
+    },
+});
+
 Deno.test({
     name: "convert supplier function to raw",
     async fn() {