Refactor code

src/app/api/image/[...key]/route.ts

@@ -1,10 +1,27 @@
 import { s3 } from "@/lib/s3";
+import type { S3Body } from "@/types/s3";
 import { GetObjectCommand } from "@aws-sdk/client-s3";
 import type { NextRequest } from "next/server";
+import { Readable } from "stream";
 
+function isWebReadableStream(value: unknown): value is ReadableStream<Uint8Array> {
+  return !!value && typeof (value as ReadableStream<Uint8Array>).getReader === "function";
+}
+
+function toBodyInit(body: S3Body): BodyInit {
+  if (body instanceof Readable) {
+    return Readable.toWeb(body) as ReadableStream<Uint8Array>;
+  }
+  if (isWebReadableStream(body)) {
+    return body;
+  }
+  return body as BodyInit;
+}
+
+// Streams images from S3 for the admin app.
 export async function GET(_req: NextRequest, context: { params: Promise<{ key: string[] }> }) {
   const { key } = await context.params;
-  const s3Key = key.join("/"); 
+  const s3Key = key.join("/");
 
   try {
     const command = new GetObjectCommand({
@@ -20,7 +37,7 @@ export async function GET(_req: NextRequest, context: { params: Promise<{ key: s
 
     const contentType = response.ContentType ?? "application/octet-stream";
 
-    return new Response(response.Body as ReadableStream, {
+    return new Response(toBodyInit(response.Body as S3Body), {
       headers: {
         "Content-Type": contentType,
         "Cache-Control": "public, max-age=3600",
@@ -28,7 +45,7 @@ export async function GET(_req: NextRequest, context: { params: Promise<{ key: s
       },
     });
   } catch (err) {
-    console.log(err)
+    console.error(err);
     return new Response("Image not found", { status: 404 });
   }
-}
+}