Exports helpers to enhance request or response

Author: ctrlplusb

package.json

@@ -37,20 +37,20 @@
   },
   "devDependencies": {
     "@types/content-type": "^1.1.3",
-    "@types/cookie": "^0.3.3",
-    "@types/jest": "^25.2.1",
+    "@types/cookie": "^0.4.0",
+    "@types/jest": "^25.2.3",
     "@types/micro": "^7.3.3",
     "@types/test-listen": "^1.1.0",
     "axios": "^0.19.2",
-    "codecov": "^3.6.5",
+    "codecov": "^3.7.0",
     "form-data": "^3.0.0",
     "husky": "^4.2.5",
-    "jest": "^25.5.4",
+    "jest": "^26.0.1",
     "test-listen": "^1.1.0",
-    "ts-jest": "^25.5.1",
+    "ts-jest": "^26.0.0",
     "tsdx": "^0.13.2",
-    "tslib": "^1.11.2",
-    "typescript": "^3.8.3"
+    "tslib": "^1.13.0",
+    "typescript": "^3.8.4"
   },
   "dependencies": {
     "@vercel/node": "^1.6.1",

src/index.ts

@@ -72,37 +72,48 @@ function parseCookie(req: IncomingMessage): NowRequestCookies {
   return parse(Array.isArray(header) ? header.join(';') : header);
 }
 
+export const enhanceRequest = async (
+  req: IncomingMessage
+): Promise<NowRequest> => {
+  const bufferOrString = await buffer(req);
+  return Object.assign(req, {
+    body:
+      typeof bufferOrString === 'string'
+        ? bufferOrString
+        : parseBody(req, bufferOrString),
+    cookies: parseCookie(req),
+    query: parseQuery(req),
+  });
+};
+
+export const enhanceResponse = (res: ServerResponse): NowResponse => {
+  let _status: number;
+  const nowRes = Object.assign(res, {
+    status: (status: number) => {
+      _status = status;
+      return nowRes;
+    },
+    json: (jsonBody: any) => {
+      send(nowRes, _status || 200, jsonBody);
+      return nowRes;
+    },
+    send: (body: string | object | Buffer) => {
+      send(nowRes, _status || 200, body);
+      return nowRes;
+    },
+    text: (body: string) => {
+      send(nowRes, _status || 200, body);
+      return nowRes;
+    },
+  });
+  return nowRes;
+};
+
 export const createServer = (
   route: (req: NowRequest, res: NowResponse) => any | Promise<any>
 ) =>
   micro(async (req: IncomingMessage, res: ServerResponse) => {
-    const bufferOrString = await buffer(req);
-    const nowReq = Object.assign(req, {
-      body:
-        typeof bufferOrString === 'string'
-          ? bufferOrString
-          : parseBody(req, bufferOrString),
-      cookies: parseCookie(req),
-      query: parseQuery(req),
-    });
-    let _status: number;
-    const nowRes = Object.assign(res, {
-      status: (status: number) => {
-        _status = status;
-        return nowRes;
-      },
-      json: (jsonBody: any) => {
-        send(nowRes, _status || 200, jsonBody);
-        return nowRes;
-      },
-      send: (body: string | object | Buffer) => {
-        send(nowRes, _status || 200, body);
-        return nowRes;
-      },
-      text: (body: string) => {
-        send(nowRes, _status || 200, body);
-        return nowRes;
-      },
-    });
+    const nowReq = await enhanceRequest(req);
+    const nowRes = enhanceResponse(res);
     return await route(nowReq, nowRes);
   });