@belmonddev/catch-request-express-middleware 3.1.2 → 3.2.0

package/deleteme.js

@@ -0,0 +1,175 @@
+import http from 'http';
+import https from 'https';
+import zlib from 'zlib';
+
+const override = (module, attr, fn) => {
+  let original = module[attr];
+
+  function wrapper(...args) {
+    const result = original.apply(this, args);
+    fn.apply(this, [result, ...args]);
+    return result;
+  }
+
+  module[attr] = wrapper;
+};
+
+export const getHeaderValue = (headers, key) =>
+  Object.entries(headers).find(([k]) => k.toLowerCase() === key)?.[1];
+
+const catchHttpResponse = (reqObject, responseCb) => {
+  override(reqObject, 'emit', (result, eventName, response) => {
+    const chunks = [];
+    switch (eventName) {
+      case 'response': {
+        response.on('data', (d) => {
+          chunks.push(d);
+        });
+
+        response.on('end', () => {
+          const responseObj = {
+            headers: response.headers,
+            statusCode: response.statusCode,
+            statusMessage: response.statusMessage,
+          };
+          const contentEncoding = getHeaderValue(
+            response.headers,
+            'content-encoding'
+          );
+          if (['gzip', 'deflate'].indexOf(contentEncoding) !== -1) {
+            const buffer = Buffer.concat(chunks);
+            const zlibFunc =
+              contentEncoding === 'gzip' ? zlib.gunzip : zlib.inflate;
+            zlibFunc(buffer, (_, decoded) => {
+              const stringBody = decoded.toString('utf8');
+              const request = { ...responseObj, body: stringBody };
+              responseCb(request);
+            });
+          } else {
+            const stringBody = chunks.map((c) => c.toString('utf8')).join('');
+            const request = { ...responseObj, body: stringBody };
+            responseCb(request);
+          }
+        });
+      }
+    }
+  });
+};
+
+const overrideHttpModule = (httpModule, requestSentCb) => {
+  override(httpModule, 'request', (reqObject, request) => {
+    const contentLength = getHeaderValue(
+      request.headers || {},
+      'content-length'
+    );
+    const requestHasBody = !!contentLength;
+    const params = Object.fromEntries(
+      new URLSearchParams(request.query).entries()
+    );
+
+    if (requestHasBody) {
+      override(reqObject, 'write', (result, data) => {
+        const body = data.toString();
+        const responseCb = requestSentCb({
+          ...request,
+          params,
+          body,
+        });
+        if (responseCb) {
+          catchHttpResponse(reqObject, responseCb);
+        }
+      });
+    } else {
+      const responseCb = requestSentCb({ ...request, params, body: null });
+      if (responseCb) {
+        catchHttpResponse(reqObject, responseCb);
+      }
+    }
+  });
+};
+
+const callResponseCb = (responseCb, res) => {
+  const response = {
+    headers: res.getHeaders(),
+    statusCode: res.statusCode,
+    statusMessage: res.statusMessage,
+    body: res.bodyWritten,
+  };
+  responseCb(response);
+};
+
+const callReceivedCbs = (req, res, requestReceivedCb, body) => {
+  const responseCb = requestReceivedCb({
+    originalRequest: req,
+    headers: req.headers,
+    method: req.method,
+    host: req.headers.host,
+    pathname: req._parsedUrl.pathname,
+    params: req.params,
+    query: req.query,
+    body,
+  });
+
+  if (responseCb) {
+    res.setMaxListeners(100);
+    if (res.writableFinished) {
+      callResponseCb(responseCb, res);
+    } else {
+      res.on('finish', () => {
+        callResponseCb(responseCb, res);
+      });
+    }
+  }
+};
+
+const catchExpressReceivedRequest = (req, res, requestReceivedCb) => {
+  // we override write method ASAP cause otherwise if the request finishes too fast the response callback is not called
+  let writeCount = 0;
+  let firstData = '';
+  override(req.socket, 'write', (result, data) => {
+    writeCount++;
+    if (writeCount === 1) {
+      firstData = data;
+    } else if (writeCount === 2) {
+      res.bodyWritten = data.toString();
+      if (!res.bodyWritten.length) {
+        res.bodyWritten = firstData.split('\r\n\r\n')?.[1];
+      }
+    }
+  });
+
+  if (req.body !== undefined) {
+    callReceivedCbs(req, res, requestReceivedCb, req.body);
+  } else {
+    const contentLength = getHeaderValue(req.headers, 'content-length');
+    if (contentLength) {
+      let body = '';
+      req.on('data', function (data) {
+        body += data;
+      });
+      req.on('end', function () {
+        callReceivedCbs(req, res, requestReceivedCb, body);
+      });
+    } else {
+      callReceivedCbs(req, res, requestReceivedCb, '');
+    }
+  }
+};
+
+const catchRequestExpressMiddleware = (
+  requestReceivedCb = null,
+  requestSentCb = null
+) => {
+  if (requestSentCb) {
+    overrideHttpModule(http, requestSentCb);
+    overrideHttpModule(https, requestSentCb);
+  }
+  return (req, res, next) => {
+    if (requestReceivedCb) {
+      catchExpressReceivedRequest(req, res, requestReceivedCb);
+    }
+    next();
+  };
+};
+
+export default catchRequestExpressMiddleware;

package/index.js

@@ -67,10 +67,11 @@ const overrideHttpModule = (httpModule, requestSentCb) => {
       new URLSearchParams(request.query).entries()
     );
 
+    let responseCb;
     if (requestHasBody) {
       override(reqObject, 'write', (result, data) => {
         const body = data.toString();
-        const responseCb = requestSentCb({
+        responseCb = requestSentCb({
           ...request,
           params,
           body,
@@ -80,11 +81,23 @@ const overrideHttpModule = (httpModule, requestSentCb) => {
         }
       });
     } else {
-      const responseCb = requestSentCb({ ...request, params, body: null });
+      responseCb = requestSentCb({ ...request, params, body: null });
       if (responseCb) {
         catchHttpResponse(reqObject, responseCb);
       }
     }
+
+    reqObject.on('error', (err) => {
+      if (responseCb) {
+        const response = {
+          headers: {},
+          statusCode: null,
+          statusMessage: null,
+          body: err.toString(),
+        };
+        responseCb(response);
+      }
+    });
   });
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@belmonddev/catch-request-express-middleware",
-  "version": "3.1.2",
+  "version": "3.2.0",
   "main": "index.js",
   "scripts": {
     "start": "nodemon -L --experimental-specifier-resolution=node ./src/index.js"