fix: update servers to parse responses better

Snazzah · Sep 16, 2024 · bc17169 · bc17169
1 parent 70de21a
commit bc17169
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 23 deletions.
diff --git a/src/servers/bun.ts b/src/servers/bun.ts
@@ -50,7 +50,7 @@ export class BunServer extends Server {
             );
           } else
             resolve(
-              new Response(JSON.stringify(response.body), {
+              new Response(response.body ? JSON.stringify(response.body) : null, {
                 status: response.status || 200,
                 headers: {
                   ...((response.headers || {}) as Record<string, string>),

diff --git a/src/servers/cfworker.ts b/src/servers/cfworker.ts
@@ -45,7 +45,7 @@ export class CloudflareWorkerServer extends Server {
               );
             } else
               resolve(
-                new Response(JSON.stringify(response.body), {
+                new Response(response.body ? JSON.stringify(response.body) : null, {
                   status: response.status || 200,
                   headers: {
                     ...((response.headers || {}) as Record<string, string>),

diff --git a/src/servers/express.ts b/src/servers/express.ts
@@ -29,26 +29,29 @@ export class ExpressServer extends Server {
 
   /** @private */
   createEndpoint(path: string, handler: ServerRequestHandler) {
-    this.app.post(path, (req: any, res: any) =>
-      handler(
-        {
-          headers: req.headers,
-          body: req.body,
-          request: req,
-          response: res
-        },
-        async (response) => {
-          res.status(response.status || 200);
-          if (response.headers) for (const key in response.headers) res.set(key, response.headers[key]);
-          if (response.files) {
-            const data = new MultipartData();
-            res.set('Content-Type', 'multipart/form-data; boundary=' + data.boundary);
-            for (const i in response.files) data.attach(`files[${i}]`, response.files[i].file, response.files[i].name);
-            data.attach('payload_json', JSON.stringify(response.body));
-            res.send(Buffer.concat(data.finish()));
-          } else res.send(response.body);
-        }
-      )
+    this.app.post(
+      path,
+      (req: any, res: any) =>
+        void handler(
+          {
+            headers: req.headers,
+            body: req.body,
+            request: req,
+            response: res
+          },
+          async (response) => {
+            res.status(response.status || 200);
+            if (response.headers) for (const key in response.headers) res.set(key, response.headers[key]);
+            if (response.files) {
+              const data = new MultipartData();
+              res.set('Content-Type', 'multipart/form-data; boundary=' + data.boundary);
+              for (const i in response.files)
+                data.attach(`files[${i}]`, response.files[i].file, response.files[i].name);
+              data.attach('payload_json', JSON.stringify(response.body));
+              res.send(Buffer.concat(data.finish()));
+            } else res.send(response.body);
+          }
+        )
     );
   }
 

diff --git a/src/servers/lambda.ts b/src/servers/lambda.ts
@@ -57,7 +57,7 @@ export class AWSLambdaServer extends Server {
           callback(null, {
             statusCode: response.status || 200,
             headers: responseHeaders,
-            body: JSON.stringify(response.body)
+            body: response.body ? JSON.stringify(response.body) : undefined
           });
         }
       }